tools/perf/core/benchmark_finders_unittest.py

# 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


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'])

  def testListSimpleBenchmarksDefinedInOneFile(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'])