| OLD | NEW |
| (Empty) |
| 1 # Copyright (C) 2012 Google, Inc. | |
| 2 # | |
| 3 # Redistribution and use in source and binary forms, with or without | |
| 4 # modification, are permitted provided that the following conditions | |
| 5 # are met: | |
| 6 # 1. Redistributions of source code must retain the above copyright | |
| 7 # notice, this list of conditions and the following disclaimer. | |
| 8 # 2. Redistributions in binary form must reproduce the above copyright | |
| 9 # notice, this list of conditions and the following disclaimer in the | |
| 10 # documentation and/or other materials provided with the distribution. | |
| 11 # | |
| 12 # THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' AND | |
| 13 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED | |
| 14 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE | |
| 15 # DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR | |
| 16 # ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
| 17 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR | |
| 18 # SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER | |
| 19 # CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, | |
| 20 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE | |
| 21 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
| 22 | |
| 23 """code to actually run a list of python tests.""" | |
| 24 | |
| 25 import re | |
| 26 import sys | |
| 27 import time | |
| 28 import unittest | |
| 29 | |
| 30 from webkitpy.common import message_pool | |
| 31 | |
| 32 _test_description = re.compile("(\w+) \(([\w.]+)\)") | |
| 33 | |
| 34 | |
| 35 def unit_test_name(test): | |
| 36 m = _test_description.match(str(test)) | |
| 37 return "%s.%s" % (m.group(2), m.group(1)) | |
| 38 | |
| 39 | |
| 40 class Runner(object): | |
| 41 def __init__(self, printer, loader, webkit_finder): | |
| 42 self.printer = printer | |
| 43 self.loader = loader | |
| 44 self.webkit_finder = webkit_finder | |
| 45 self.tests_run = 0 | |
| 46 self.errors = [] | |
| 47 self.failures = [] | |
| 48 self.worker_factory = lambda caller: _Worker(caller, self.loader, self.w
ebkit_finder) | |
| 49 | |
| 50 def run(self, test_names, num_workers): | |
| 51 if not test_names: | |
| 52 return | |
| 53 num_workers = min(num_workers, len(test_names)) | |
| 54 with message_pool.get(self, self.worker_factory, num_workers) as pool: | |
| 55 pool.run(('test', test_name) for test_name in test_names) | |
| 56 | |
| 57 def handle(self, message_name, source, test_name, delay=None, failures=None,
errors=None): | |
| 58 if message_name == 'started_test': | |
| 59 self.printer.print_started_test(source, test_name) | |
| 60 return | |
| 61 | |
| 62 self.tests_run += 1 | |
| 63 if failures: | |
| 64 self.failures.append((test_name, failures)) | |
| 65 if errors: | |
| 66 self.errors.append((test_name, errors)) | |
| 67 self.printer.print_finished_test(source, test_name, delay, failures, err
ors) | |
| 68 | |
| 69 | |
| 70 class _Worker(object): | |
| 71 def __init__(self, caller, loader, webkit_finder): | |
| 72 self._caller = caller | |
| 73 self._loader = loader | |
| 74 | |
| 75 # FIXME: coverage needs to be in sys.path for its internal imports to wo
rk. | |
| 76 thirdparty_path = webkit_finder.path_from_webkit_base('Tools', 'Scripts'
, 'webkitpy', 'thirdparty') | |
| 77 if not thirdparty_path in sys.path: | |
| 78 sys.path.append(thirdparty_path) | |
| 79 | |
| 80 | |
| 81 def handle(self, message_name, source, test_name): | |
| 82 assert message_name == 'test' | |
| 83 result = unittest.TestResult() | |
| 84 start = time.time() | |
| 85 self._caller.post('started_test', test_name) | |
| 86 | |
| 87 # We will need to rework this if a test_name results in multiple tests. | |
| 88 self._loader.loadTestsFromName(test_name, None).run(result) | |
| 89 self._caller.post('finished_test', test_name, time.time() - start, | |
| 90 [failure[1] for failure in result.failures], [error[1] for error in
result.errors]) | |
| OLD | NEW |