# Copyright 2016 the V8 project 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

from testrunner.local import testsuite
from testrunner.objects import testcase


class VariantsGenerator(testsuite.VariantsGenerator):
  def _get_variants(self, test):
    return self._standard_variant


class TestSuite(testsuite.TestSuite):
  SUB_TESTS = ( 'json', 'parser', 'regexp', 'multi_return', 'wasm',
          'wasm_async', 'wasm_call', 'wasm_code', 'wasm_compile',
          'wasm_data_section', 'wasm_function_sigs_section',
          'wasm_globals_section', 'wasm_imports_section', 'wasm_memory_section',
          'wasm_names_section', 'wasm_types_section' )

  def ListTests(self, context):
    tests = []
    for subtest in TestSuite.SUB_TESTS:
      for fname in os.listdir(os.path.join(self.root, subtest)):
        if not os.path.isfile(os.path.join(self.root, subtest, fname)):
          continue
        test = self._create_test('%s/%s' % (subtest, fname))
        tests.append(test)
    tests.sort()
    return tests

  def _test_class(self):
    return TestCase

  def _variants_gen_class(self):
    return VariantsGenerator

  def _LegacyVariantsGeneratorFactory(self):
    return testsuite.StandardLegacyVariantsGenerator


class TestCase(testcase.TestCase):
  def _get_files_params(self, ctx):
    suite, name = self.path.split('/')
    return [os.path.join(self.suite.root, suite, name)]

  def _get_variant_flags(self):
    return []

  def _get_statusfile_flags(self):
    return []

  def _get_mode_flags(self, ctx):
    return []

  def get_shell(self):
    group, _ = self.path.split('/', 1)
    return 'v8_simple_%s_fuzzer' % group


def GetSuite(name, root):
  return TestSuite(name, root)