# Copyright 2015 The Chromium Authors
# 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.assertEqual(
        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.assertEqual(
        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'
            ])