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 |