1# Copyright 2008 the V8 project authors. All rights reserved.
2# Redistribution and use in source and binary forms, with or without
3# modification, are permitted provided that the following conditions are
4# met:
5#
6#     * Redistributions of source code must retain the above copyright
7#       notice, this list of conditions and the following disclaimer.
8#     * Redistributions in binary form must reproduce the above
9#       copyright notice, this list of conditions and the following
10#       disclaimer in the documentation and/or other materials provided
11#       with the distribution.
12#     * Neither the name of Google Inc. nor the names of its
13#       contributors may be used to endorse or promote products derived
14#       from this software without specific prior written permission.
15#
16# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
17# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
18# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
19# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
20# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
26# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27
28import os
29import re
30
31from testrunner.local import testsuite
32from testrunner.objects import testcase
33
34FLAGS_PATTERN = re.compile(r"//\s+Flags:(.*)")
35FILES_PATTERN = re.compile(r"//\s+Files:(.*)")
36SELF_SCRIPT_PATTERN = re.compile(r"//\s+Env: TEST_FILE_NAME")
37
38
39class MjsunitTestSuite(testsuite.TestSuite):
40
41  def __init__(self, name, root):
42    super(MjsunitTestSuite, self).__init__(name, root)
43
44  def ListTests(self, context):
45    tests = []
46    for dirname, dirs, files in os.walk(self.root):
47      for dotted in [x for x in dirs if x.startswith('.')]:
48        dirs.remove(dotted)
49      dirs.sort()
50      files.sort()
51      for filename in files:
52        if filename.endswith(".js") and filename != "mjsunit.js":
53          testname = os.path.join(dirname[len(self.root) + 1:], filename[:-3])
54          test = testcase.TestCase(self, testname)
55          tests.append(test)
56    return tests
57
58  def GetFlagsForTestCase(self, testcase, context):
59    source = self.GetSourceForTest(testcase)
60    flags = [] + context.mode_flags
61    flags_match = re.findall(FLAGS_PATTERN, source)
62    for match in flags_match:
63      flags += match.strip().split()
64
65    files_list = []  # List of file names to append to command arguments.
66    files_match = FILES_PATTERN.search(source);
67    # Accept several lines of 'Files:'.
68    while True:
69      if files_match:
70        files_list += files_match.group(1).strip().split()
71        files_match = FILES_PATTERN.search(source, files_match.end())
72      else:
73        break
74    files = [ os.path.normpath(os.path.join(self.root, '..', '..', f))
75              for f in files_list ]
76    testfilename = os.path.join(self.root, testcase.path + self.suffix())
77    if SELF_SCRIPT_PATTERN.search(source):
78      env = ["-e", "TEST_FILE_NAME=\"%s\"" % testfilename.replace("\\", "\\\\")]
79      files = env + files
80    files.append(os.path.join(self.root, "mjsunit.js"))
81    files.append(testfilename)
82
83    flags += files
84    if context.isolates:
85      flags.append("--isolate")
86      flags += files
87
88    return testcase.flags + flags
89
90  def GetSourceForTest(self, testcase):
91    filename = os.path.join(self.root, testcase.path + self.suffix())
92    with open(filename) as f:
93      return f.read()
94
95
96def GetSuite(name, root):
97  return MjsunitTestSuite(name, root)
98