| Index: tools/telemetry/telemetry/benchmark_runner_unittest.py
|
| diff --git a/tools/telemetry/telemetry/benchmark_runner_unittest.py b/tools/telemetry/telemetry/benchmark_runner_unittest.py
|
| deleted file mode 100644
|
| index 266f87aa17acefab2e02a6f68eeb5fd107390785..0000000000000000000000000000000000000000
|
| --- a/tools/telemetry/telemetry/benchmark_runner_unittest.py
|
| +++ /dev/null
|
| @@ -1,115 +0,0 @@
|
| -# Copyright 2015 The Chromium Authors. All rights reserved.
|
| -# Use of this source code is governed by a BSD-style license that can be
|
| -# found in the LICENSE file.
|
| -
|
| -import unittest
|
| -
|
| -from telemetry import benchmark
|
| -from telemetry import benchmark_runner
|
| -from telemetry.testing import stream
|
| -import mock
|
| -
|
| -
|
| -class BenchmarkFoo(benchmark.Benchmark):
|
| - """ Benchmark Foo for testing."""
|
| -
|
| - @classmethod
|
| - def Name(cls):
|
| - return 'FooBenchmark'
|
| -
|
| -
|
| -class BenchmarkBar(benchmark.Benchmark):
|
| - """ Benchmark Bar for testing long description line."""
|
| -
|
| - @classmethod
|
| - def Name(cls):
|
| - return 'BarBenchmarkkkkk'
|
| -
|
| -class UnusualBenchmark(benchmark.Benchmark):
|
| - @classmethod
|
| - def Name(cls):
|
| - return 'I have a very unusual name'
|
| -
|
| -
|
| -class BenchmarkRunnerUnittest(unittest.TestCase):
|
| - def setUp(self):
|
| - self._stream = stream.TestOutputStream()
|
| - self._mock_possible_browser = mock.MagicMock()
|
| - self._mock_possible_browser.browser_type = 'TestBrowser'
|
| -
|
| - def testPrintBenchmarkListWithNoDisabledBenchmark(self):
|
| - expected_printed_stream = (
|
| - 'Available benchmarks for TestBrowser are:\n'
|
| - ' BarBenchmarkkkkk Benchmark Bar for testing long description line.\n'
|
| - ' FooBenchmark Benchmark Foo for testing.\n'
|
| - 'Pass --browser to list benchmarks for another browser.\n\n')
|
| - with mock.patch('telemetry.benchmark_runner.decorators') as mock_module:
|
| - mock_module.IsEnabled.return_value = (True, None)
|
| - benchmark_runner.PrintBenchmarkList(
|
| - [BenchmarkFoo, BenchmarkBar], self._mock_possible_browser, self._stream)
|
| - self.assertEquals(expected_printed_stream, self._stream.output_data)
|
| -
|
| - def testPrintBenchmarkListWithOneDisabledBenchmark(self):
|
| - expected_printed_stream = (
|
| - 'Available benchmarks for TestBrowser are:\n'
|
| - ' FooBenchmark Benchmark Foo for testing.\n'
|
| - '\n'
|
| - 'Disabled benchmarks for TestBrowser are (force run with -d):\n'
|
| - ' BarBenchmarkkkkk Benchmark Bar for testing long description line.\n'
|
| - 'Pass --browser to list benchmarks for another browser.\n\n')
|
| - with mock.patch('telemetry.benchmark_runner.decorators') as mock_module:
|
| - def FakeIsEnabled(benchmark_class, _):
|
| - if benchmark_class is BenchmarkFoo:
|
| - return True, None
|
| - else:
|
| - return False, 'Only supported BenchmarkFoo'
|
| - mock_module.IsEnabled = FakeIsEnabled
|
| - benchmark_runner.PrintBenchmarkList(
|
| - [BenchmarkFoo, BenchmarkBar], self._mock_possible_browser, self._stream)
|
| - self.assertEquals(expected_printed_stream, self._stream.output_data)
|
| -
|
| - def testShouldDisable(self):
|
| - """Ensure that overridden ShouldDisable class methods are respected."""
|
| - expected_printed_stream = (
|
| - 'Available benchmarks for TestBrowser are:\n'
|
| - ' BarBenchmarkkkkk Benchmark Bar for testing long description line.\n'
|
| - '\n'
|
| - 'Disabled benchmarks for TestBrowser are (force run with -d):\n'
|
| - ' FooBenchmark Benchmark Foo for testing.\n'
|
| - 'Pass --browser to list benchmarks for another browser.\n\n')
|
| - @classmethod
|
| - def FakeShouldDisable(cls, possible_browser):
|
| - del possible_browser # unused
|
| - return cls is BenchmarkFoo
|
| - BenchmarkFoo.ShouldDisable = FakeShouldDisable
|
| - BenchmarkBar.ShouldDisable = FakeShouldDisable
|
| - benchmark_runner.PrintBenchmarkList(
|
| - [BenchmarkFoo, BenchmarkBar], self._mock_possible_browser, self._stream)
|
| - self.assertEquals(expected_printed_stream, self._stream.output_data)
|
| -
|
| - def testShouldDisableComplex(self):
|
| - """Ensure that browser-dependent ShouldDisable overrides are respected."""
|
| - expected_printed_stream = (
|
| - # Expected output for 'TestBrowser':
|
| - 'Available benchmarks for TestBrowser are:\n'
|
| - ' FooBenchmark Benchmark Foo for testing.\n'
|
| - '\n'
|
| - 'Disabled benchmarks for TestBrowser are (force run with -d):\n'
|
| - ' BarBenchmarkkkkk Benchmark Bar for testing long description line.\n'
|
| - 'Pass --browser to list benchmarks for another browser.\n\n'
|
| - # Expected output for 'MockBrowser':
|
| - 'Available benchmarks for MockBrowser are:\n'
|
| - ' BarBenchmarkkkkk Benchmark Bar for testing long description line.\n'
|
| - ' FooBenchmark Benchmark Foo for testing.\n'
|
| - 'Pass --browser to list benchmarks for another browser.\n\n')
|
| - @classmethod
|
| - def FakeShouldDisable(cls, possible_browser):
|
| - return cls is BenchmarkBar and not 'Mock' in possible_browser.browser_type
|
| - BenchmarkFoo.ShouldDisable = FakeShouldDisable
|
| - BenchmarkBar.ShouldDisable = FakeShouldDisable
|
| - benchmark_runner.PrintBenchmarkList(
|
| - [BenchmarkFoo, BenchmarkBar], self._mock_possible_browser, self._stream)
|
| - self._mock_possible_browser.browser_type = 'MockBrowser'
|
| - benchmark_runner.PrintBenchmarkList(
|
| - [BenchmarkFoo, BenchmarkBar], self._mock_possible_browser, self._stream)
|
| - self.assertEquals(expected_printed_stream, self._stream.output_data)
|
|
|