| Index: tools/perf/core/perf_data_generator_unittest.py
|
| diff --git a/tools/perf/core/perf_data_generator_unittest.py b/tools/perf/core/perf_data_generator_unittest.py
|
| index 4aad5c30d22b13d07292ddba75268917644a75c2..91bd6518f060b67bcb07afeea3491eddf8c66091 100644
|
| --- a/tools/perf/core/perf_data_generator_unittest.py
|
| +++ b/tools/perf/core/perf_data_generator_unittest.py
|
| @@ -7,6 +7,14 @@ from core import perf_data_generator
|
| from core.perf_data_generator import BenchmarkMetadata
|
|
|
|
|
| +class _MockBenchmark(object):
|
| + def __init__(self, name):
|
| + self.name = name
|
| +
|
| + def Name(self):
|
| + return self.name
|
| +
|
| +
|
| class PerfDataGeneratorTest(unittest.TestCase):
|
| def setUp(self):
|
| # Test config can be big, so set maxDiff to None to see the full comparision
|
| @@ -111,3 +119,71 @@ class PerfDataGeneratorTest(unittest.TestCase):
|
| 'isolate_name': 'telemetry_perf_tests',
|
| }
|
| self.assertEquals(test, expected_generated_test)
|
| +
|
| + def testGenerateTelemetryTestsBlacklistedReferenceBuildTest(self):
|
| + swarming_dimensions = [
|
| + {'os': 'SkyNet', 'id': 'T-850', 'pool': 'T-RIP', 'device_ids': ['a']}
|
| + ]
|
| + test_config = {
|
| + 'platform': 'android',
|
| + 'swarming_dimensions': swarming_dimensions,
|
| + }
|
| + benchmarks = [
|
| + _MockBenchmark('blacklisted'),
|
| + _MockBenchmark('not_blacklisted')
|
| + ]
|
| + tests = perf_data_generator.generate_telemetry_tests(
|
| + test_config, benchmarks, None, False, ['blacklisted'])
|
| + expected_generated_tests = [
|
| + {'args': ['blacklisted',
|
| + '-v',
|
| + '--upload-results',
|
| + '--output-format=chartjson',
|
| + '--browser=android-chromium'],
|
| + 'isolate_name': 'telemetry_perf_tests',
|
| + 'name': 'blacklisted',
|
| + 'override_compile_targets': ['telemetry_perf_tests'],
|
| + 'swarming': {'can_use_on_swarming_builders': True,
|
| + 'dimension_sets': [{'id': 'a',
|
| + 'os': 'SkyNet',
|
| + 'pool': 'Chrome-perf'}],
|
| + 'expiration': 36000,
|
| + 'hard_timeout': 7200,
|
| + 'ignore_task_failure': False,
|
| + 'io_timeout': 3600}},
|
| + {'args': ['not_blacklisted',
|
| + '-v',
|
| + '--upload-results',
|
| + '--output-format=chartjson',
|
| + '--browser=android-chromium'],
|
| + 'isolate_name': 'telemetry_perf_tests',
|
| + 'name': 'not_blacklisted',
|
| + 'override_compile_targets': ['telemetry_perf_tests'],
|
| + 'swarming': {'can_use_on_swarming_builders': True,
|
| + 'dimension_sets': [{'id': 'a',
|
| + 'os': 'SkyNet',
|
| + 'pool': 'Chrome-perf'}],
|
| + 'expiration': 36000,
|
| + 'hard_timeout': 7200,
|
| + 'ignore_task_failure': False,
|
| + 'io_timeout': 3600}},
|
| + {'args': ['not_blacklisted',
|
| + '-v',
|
| + '--upload-results',
|
| + '--output-format=chartjson',
|
| + '--browser=reference',
|
| + '--output-trace-tag=_ref'],
|
| + 'isolate_name': 'telemetry_perf_tests',
|
| + 'name': 'not_blacklisted.reference',
|
| + 'override_compile_targets': ['telemetry_perf_tests'],
|
| + 'swarming': {'can_use_on_swarming_builders': True,
|
| + 'dimension_sets': [{'id': 'a',
|
| + 'os': 'SkyNet',
|
| + 'pool': 'Chrome-perf'}],
|
| + 'expiration': 36000,
|
| + 'hard_timeout': 7200,
|
| + 'ignore_task_failure': True,
|
| + 'io_timeout': 3600}
|
| + }
|
| + ]
|
| + self.assertEquals(tests, expected_generated_tests)
|
|
|