| OLD | NEW |
|---|
| (Empty) |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 import StringIO | |
| 5 import csv | |
| 6 import os | |
| 7 import unittest | |
| 8 | |
| 9 from telemetry.page import csv_page_benchmark_results | |
| 10 from telemetry.page import page_set | |
| 11 | |
| 12 def _MakePageSet(): | |
| 13 return page_set.PageSet.FromDict({ | |
| 14 "description": "hello", | |
| 15 "pages": [ | |
| 16 {"url": "http://www.foo.com/"}, | |
| 17 {"url": "http://www.bar.com/"} | |
| 18 ] | |
| 19 }, os.path.dirname(__file__)) | |
| 20 | |
| 21 class NonPrintingCsvPageBenchmarkResults( | |
| 22 csv_page_benchmark_results.CsvPageBenchmarkResults): | |
| 23 def __init__(self, *args): | |
| 24 super(NonPrintingCsvPageBenchmarkResults, self).__init__(*args) | |
| 25 | |
| 26 def _PrintPerfResult(self, *args): | |
| 27 pass | |
| 28 | |
| 29 class CsvPageBenchmarkResultsTest(unittest.TestCase): | |
| 30 def setUp(self): | |
| 31 self._output = StringIO.StringIO() | |
| 32 self._page_set = _MakePageSet() | |
| 33 | |
| 34 @property | |
| 35 def lines(self): | |
| 36 lines = StringIO.StringIO(self._output.getvalue()).readlines() | |
| 37 return lines | |
| 38 | |
| 39 @property | |
| 40 def output_header_row(self): | |
| 41 rows = list(csv.reader(self.lines)) | |
| 42 return rows[0] | |
| 43 | |
| 44 @property | |
| 45 def output_data_rows(self): | |
| 46 rows = list(csv.reader(self.lines)) | |
| 47 return rows[1:] | |
| 48 | |
| 49 def test_with_output_after_every_page(self): | |
| 50 results = NonPrintingCsvPageBenchmarkResults(csv.writer(self._output), True) | |
| 51 results.WillMeasurePage(self._page_set[0]) | |
| 52 results.Add('foo', 'seconds', 3) | |
| 53 results.DidMeasurePage() | |
| 54 self.assertEquals( | |
| 55 self.output_header_row, | |
| 56 ['url', 'foo (seconds)']) | |
| 57 self.assertEquals( | |
| 58 self.output_data_rows[0], | |
| 59 [self._page_set[0].url, '3']) | |
| 60 | |
| 61 results.WillMeasurePage(self._page_set[1]) | |
| 62 results.Add('foo', 'seconds', 4) | |
| 63 results.DidMeasurePage() | |
| 64 self.assertEquals( | |
| 65 len(self.output_data_rows), | |
| 66 2) | |
| 67 self.assertEquals( | |
| 68 self.output_data_rows[1], | |
| 69 [self._page_set[1].url, '4']) | |
| 70 | |
| 71 def test_with_output_after_every_page_and_inconsistency(self): | |
| 72 results = NonPrintingCsvPageBenchmarkResults(csv.writer(self._output), True) | |
| 73 results.WillMeasurePage(self._page_set[0]) | |
| 74 results.Add('foo', 'seconds', 3) | |
| 75 results.DidMeasurePage() | |
| 76 | |
| 77 # We printed foo, now change to bar | |
| 78 results.WillMeasurePage(self._page_set[1]) | |
| 79 results.Add('bar', 'seconds', 4) | |
| 80 | |
| 81 self.assertRaises( | |
| 82 Exception, | |
| 83 lambda: results.DidMeasurePage()) # pylint: disable=W0108 | |
| 84 | |
| 85 def test_with_output_at_print_summary_time(self): | |
| 86 results = NonPrintingCsvPageBenchmarkResults(csv.writer(self._output), | |
| 87 False) | |
| 88 results.WillMeasurePage(self._page_set[0]) | |
| 89 results.Add('foo', 'seconds', 3) | |
| 90 results.DidMeasurePage() | |
| 91 | |
| 92 results.WillMeasurePage(self._page_set[1]) | |
| 93 results.Add('bar', 'seconds', 4) | |
| 94 results.DidMeasurePage() | |
| 95 | |
| 96 results.PrintSummary('tag') | |
| 97 | |
| 98 self.assertEquals( | |
| 99 self.output_header_row, | |
| 100 ['url', 'bar (seconds)', 'foo (seconds)']) | |
| 101 self.assertEquals( | |
| 102 self.output_data_rows, | |
| 103 [[self._page_set[0].url, '-', '3'], | |
| 104 [self._page_set[1].url, '4', '-']]) | |
| 105 | |
| 106 def test_histogram(self): | |
| 107 results = NonPrintingCsvPageBenchmarkResults(csv.writer(self._output), | |
| 108 False) | |
| 109 results.WillMeasurePage(self._page_set[0]) | |
| 110 results.Add('a', '', | |
| 111 '{"buckets": [{"low": 1, "high": 2, "count": 1}]}', | |
| 112 data_type='histogram') | |
| 113 results.DidMeasurePage() | |
| 114 | |
| 115 results.WillMeasurePage(self._page_set[1]) | |
| 116 results.Add('a', '', | |
| 117 '{"buckets": [{"low": 2, "high": 3, "count": 1}]}', | |
| 118 data_type='histogram') | |
| 119 results.DidMeasurePage() | |
| 120 | |
| 121 results.PrintSummary('tag') | |
| 122 | |
| 123 self.assertEquals( | |
| 124 self.output_header_row, | |
| 125 ['url', 'a ()']) | |
| 126 self.assertEquals( | |
| 127 self.output_data_rows, | |
| 128 [[self._page_set[0].url, '1.5'], | |
| 129 [self._page_set[1].url, '2.5']]) | |
| OLD | NEW |
|---|
Chromium Code Reviews
