| # 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 os |
| import unittest |
| |
| from core import benchmark_finders |
| from telemetry import decorators |
| |
| |
| class TestGetBenchmarkNamesForFile(unittest.TestCase): |
| |
| def setUp(self): |
| self.top_level_dir = os.path.abspath( |
| os.path.join(os.path.dirname(__file__), 'test_data')) |
| |
| def testListSimpleBenchmarksDefinedInOneFile(self): |
| self.assertEquals( |
| benchmark_finders.GetBenchmarkNamesForFile( |
| self.top_level_dir, |
| os.path.join( |
| self.top_level_dir, |
| 'simple_benchmarks_case.py')), |
| ['test_benchmark_1', 'test_benchmark_2', 'test_benchmark_subclass_1', |
| 'test_benchmark_subclass_2']) |
| |
| @decorators.Disabled('all') # http://crbug.com/637938 |
| def testListSimpleBenchmarksDefinedInOneFileComplex(self): |
| self.assertEquals( |
| benchmark_finders.GetBenchmarkNamesForFile( |
| self.top_level_dir, |
| os.path.join( |
| self.top_level_dir, |
| 'complex_benchmarks_case.py')), |
| ['test_benchmark_complex_1', 'test_benchmark_complex_subclass', |
| 'test_benchmark_complex_subclass_from_other_module']) |