| OLD | NEW |
| 1 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 from optparse import OptionParser | 5 from optparse import OptionParser |
| 6 import unittest | 6 import unittest |
| 7 | 7 |
| 8 from telemetry.page import shared_page_state | 8 from telemetry.page import shared_page_state |
| 9 | 9 |
| 10 from benchmarks import rasterize_and_record_micro | 10 from contrib.cluster_telemetry import rasterize_and_record_micro_ct |
| 11 from benchmarks import repaint | 11 from contrib.cluster_telemetry import repaint |
| 12 from benchmarks import skpicture_printer | 12 from contrib.cluster_telemetry import skpicture_printer |
| 13 | 13 |
| 14 | 14 |
| 15 class MockErrorParser(object): | 15 class MockErrorParser(object): |
| 16 | 16 |
| 17 def __init__(self): | 17 def __init__(self): |
| 18 self.err_msg = None | 18 self.err_msg = None |
| 19 | 19 |
| 20 def error(self, err_msg): | 20 def error(self, err_msg): |
| 21 self.err_msg = err_msg | 21 self.err_msg = err_msg |
| 22 | 22 |
| 23 | 23 |
| 24 class CTBenchmarks(unittest.TestCase): | 24 class CTBenchmarks(unittest.TestCase): |
| 25 | 25 |
| 26 def setUp(self): | 26 def setUp(self): |
| 27 self.ct_benchmarks = [ | 27 self.ct_benchmarks = [ |
| 28 rasterize_and_record_micro.RasterizeAndRecordMicroCT(), | 28 rasterize_and_record_micro_ct.RasterizeAndRecordMicroCT(), |
| 29 repaint.RepaintCT(), | 29 repaint.RepaintCT(), |
| 30 skpicture_printer.SkpicturePrinterCT(), | 30 skpicture_printer.SkpicturePrinterCT(), |
| 31 ] | 31 ] |
| 32 self.shared_page_state_class = shared_page_state.SharedMobilePageState | 32 self.shared_page_state_class = shared_page_state.SharedMobilePageState |
| 33 self.archive_data_file = '/b/test' | 33 self.archive_data_file = '/b/test' |
| 34 self.urls_list = 'http://test1.com,http://test2.com,http://test3.net' | 34 self.urls_list = 'http://test1.com,http://test2.com,http://test3.net' |
| 35 self.mock_parser = MockErrorParser() | 35 self.mock_parser = MockErrorParser() |
| 36 | 36 |
| 37 def testCTBenchmarks(self): | 37 def testCTBenchmarks(self): |
| 38 for benchmark in self.ct_benchmarks: | 38 for benchmark in self.ct_benchmarks: |
| (...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 107 self.fail('Expected AttributeError') | 107 self.fail('Expected AttributeError') |
| 108 except AttributeError, e: | 108 except AttributeError, e: |
| 109 self.assertEquals( | 109 self.assertEquals( |
| 110 'OptionParser instance has no attribute \'urls_list\'', | 110 'OptionParser instance has no attribute \'urls_list\'', |
| 111 e.message) | 111 e.message) |
| 112 | 112 |
| 113 # Now add an empty urls_list. | 113 # Now add an empty urls_list. |
| 114 parser.urls_list = '' | 114 parser.urls_list = '' |
| 115 benchmark.ProcessCommandLineArgs(self.mock_parser, parser) | 115 benchmark.ProcessCommandLineArgs(self.mock_parser, parser) |
| 116 self.assertEquals('Please specify --urls_list.', self.mock_parser.err_msg) | 116 self.assertEquals('Please specify --urls_list.', self.mock_parser.err_msg) |
| OLD | NEW |