OLD | NEW |
(Empty) | |
| 1 # Copyright 2012 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 |
| 28 |
| 29 import multiprocessing |
| 30 import os |
| 31 import threading |
| 32 import time |
| 33 |
| 34 from . import commands |
| 35 from . import utils |
| 36 |
| 37 |
| 38 class Job(object): |
| 39 def __init__(self, command, dep_command, test_id, timeout, verbose): |
| 40 self.command = command |
| 41 self.dep_command = dep_command |
| 42 self.id = test_id |
| 43 self.timeout = timeout |
| 44 self.verbose = verbose |
| 45 |
| 46 |
| 47 def RunTest(job): |
| 48 try: |
| 49 start_time = time.time() |
| 50 if job.dep_command is not None: |
| 51 dep_output = commands.Execute(job.dep_command, job.verbose, job.timeout) |
| 52 # TODO(jkummerow): We approximate the test suite specific function |
| 53 # IsFailureOutput() by just checking the exit code here. Currently |
| 54 # only cctests define dependencies, for which this simplification is |
| 55 # correct. |
| 56 if dep_output.exit_code != 0: |
| 57 return (job.id, dep_output, time.time() - start_time) |
| 58 output = commands.Execute(job.command, job.verbose, job.timeout) |
| 59 return (job.id, output, time.time() - start_time) |
| 60 except Exception, e: |
| 61 print(">>> EXCEPTION: %s" % e) |
| 62 return (-1, -1, 0) |
| 63 |
| 64 |
| 65 class Runner(object): |
| 66 |
| 67 def __init__(self, suites, progress_indicator, context): |
| 68 self.tests = [ t for s in suites for t in s.tests ] |
| 69 self._CommonInit(len(self.tests), progress_indicator, context) |
| 70 |
| 71 def _CommonInit(self, num_tests, progress_indicator, context): |
| 72 self.indicator = progress_indicator |
| 73 progress_indicator.runner = self |
| 74 self.context = context |
| 75 self.succeeded = 0 |
| 76 self.total = num_tests |
| 77 self.remaining = num_tests |
| 78 self.failed = [] |
| 79 self.crashed = 0 |
| 80 self.terminate = False |
| 81 self.lock = threading.Lock() |
| 82 |
| 83 def Run(self, jobs): |
| 84 self.indicator.Starting() |
| 85 self._RunInternal(jobs) |
| 86 self.indicator.Done() |
| 87 return not self.failed |
| 88 |
| 89 def _RunInternal(self, jobs): |
| 90 pool = multiprocessing.Pool(processes=jobs) |
| 91 test_map = {} |
| 92 queue = [] |
| 93 for test in self.tests: |
| 94 assert test.id >= 0 |
| 95 test_map[test.id] = test |
| 96 command = self.GetCommand(test) |
| 97 timeout = self.context.timeout |
| 98 if ("--stress-opt" in test.flags or |
| 99 "--stress-opt" in self.context.mode_flags or |
| 100 "--stress-opt" in self.context.extra_flags): |
| 101 timeout *= 4 |
| 102 if test.dependency is not None: |
| 103 dep_command = [ c.replace(test.path, test.dependency) for c in command ] |
| 104 else: |
| 105 dep_command = None |
| 106 job = Job(command, dep_command, test.id, timeout, self.context.verbose) |
| 107 queue.append(job) |
| 108 try: |
| 109 kChunkSize = 1 |
| 110 it = pool.imap_unordered(RunTest, queue, kChunkSize) |
| 111 for result in it: |
| 112 test_id = result[0] |
| 113 if test_id < 0: |
| 114 raise BreakNowException("User pressed Ctrl+C or IO went wrong") |
| 115 test = test_map[test_id] |
| 116 self.indicator.AboutToRun(test) |
| 117 test.output = result[1] |
| 118 test.duration = result[2] |
| 119 if test.suite.HasUnexpectedOutput(test): |
| 120 self.failed.append(test) |
| 121 if test.output.HasCrashed(): |
| 122 self.crashed += 1 |
| 123 else: |
| 124 self.succeeded += 1 |
| 125 self.remaining -= 1 |
| 126 self.indicator.HasRun(test) |
| 127 except: |
| 128 pool.terminate() |
| 129 pool.join() |
| 130 raise |
| 131 return |
| 132 |
| 133 |
| 134 def GetCommand(self, test): |
| 135 d8testflag = [] |
| 136 shell = test.suite.shell() |
| 137 if shell == "d8": |
| 138 d8testflag = ["--test"] |
| 139 if utils.IsWindows(): |
| 140 shell += ".exe" |
| 141 cmd = ([self.context.command_prefix] + |
| 142 [os.path.join(self.context.shell_dir, shell)] + |
| 143 d8testflag + |
| 144 test.suite.GetFlagsForTestCase(test, self.context) + |
| 145 [self.context.extra_flags]) |
| 146 cmd = [ c for c in cmd if c != "" ] |
| 147 return cmd |
| 148 |
| 149 |
| 150 class BreakNowException(Exception): |
| 151 def __init__(self, value): |
| 152 self.value = value |
| 153 def __str__(self): |
| 154 return repr(self.value) |
OLD | NEW |