| OLD | NEW |
| 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 | 5 |
| 6 import logging | 6 import logging |
| 7 import multiprocessing | 7 import multiprocessing |
| 8 | 8 |
| 9 from test_result import * | 9 from test_result import TestResults |
| 10 | 10 |
| 11 | 11 |
| 12 def _ShardedTestRunnable(test): | 12 def _ShardedTestRunnable(test): |
| 13 """Standalone function needed by multiprocessing.Pool.""" | 13 """Standalone function needed by multiprocessing.Pool.""" |
| 14 log_format = '[' + test.device + '] # %(asctime)-15s: %(message)s' | 14 log_format = '[' + test.device + '] # %(asctime)-15s: %(message)s' |
| 15 if logging.getLogger().handlers: | 15 if logging.getLogger().handlers: |
| 16 logging.getLogger().handlers[0].setFormatter(logging.Formatter(log_format)) | 16 logging.getLogger().handlers[0].setFormatter(logging.Formatter(log_format)) |
| 17 else: | 17 else: |
| 18 logging.basicConfig(format=log_format) | 18 logging.basicConfig(format=log_format) |
| 19 return test.Run() | 19 return test.Run() |
| (...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 69 A TestResults object. | 69 A TestResults object. |
| 70 """ | 70 """ |
| 71 logging.warning('*' * 80) | 71 logging.warning('*' * 80) |
| 72 logging.warning('Sharding in ' + str(len(self.attached_devices)) + | 72 logging.warning('Sharding in ' + str(len(self.attached_devices)) + |
| 73 ' devices.') | 73 ' devices.') |
| 74 logging.warning('Note that the output is not synchronized.') | 74 logging.warning('Note that the output is not synchronized.') |
| 75 logging.warning('Look for the "Final result" banner in the end.') | 75 logging.warning('Look for the "Final result" banner in the end.') |
| 76 logging.warning('*' * 80) | 76 logging.warning('*' * 80) |
| 77 final_results = TestResults() | 77 final_results = TestResults() |
| 78 for retry in xrange(self.retries): | 78 for retry in xrange(self.retries): |
| 79 logging.warning('Try %d of %d' % (retry + 1, self.retries)) | 79 logging.warning('Try %d of %d', retry + 1, self.retries) |
| 80 self.SetupSharding(self.tests) | 80 self.SetupSharding(self.tests) |
| 81 test_runners = [] | 81 test_runners = [] |
| 82 for index, device in enumerate(self.attached_devices): | 82 for index, device in enumerate(self.attached_devices): |
| 83 logging.warning('*' * 80) | 83 logging.warning('*' * 80) |
| 84 logging.warning('Creating shard %d for %s' % (index, device)) | 84 logging.warning('Creating shard %d for %s', index, device) |
| 85 logging.warning('*' * 80) | 85 logging.warning('*' * 80) |
| 86 test_runner = self.CreateShardedTestRunner(device, index) | 86 test_runner = self.CreateShardedTestRunner(device, index) |
| 87 test_runners += [test_runner] | 87 test_runners += [test_runner] |
| 88 logging.warning('Starting...') | 88 logging.warning('Starting...') |
| 89 pool = multiprocessing.Pool(len(self.attached_devices), | 89 pool = multiprocessing.Pool(len(self.attached_devices), |
| 90 SetTestsContainer, | 90 SetTestsContainer, |
| 91 [BaseTestSharder.tests_container]) | 91 [BaseTestSharder.tests_container]) |
| 92 results_lists = pool.map(_ShardedTestRunnable, test_runners) | 92 results_lists = pool.map(_ShardedTestRunnable, test_runners) |
| 93 test_results = TestResults.FromTestResults(results_lists) | 93 test_results = TestResults.FromTestResults(results_lists) |
| 94 if retry == self.retries - 1: | 94 if retry == self.retries - 1: |
| 95 all_passed = final_results.ok + test_results.ok | 95 all_passed = final_results.ok + test_results.ok |
| 96 final_results = test_results | 96 final_results = test_results |
| 97 final_results.ok = all_passed | 97 final_results.ok = all_passed |
| 98 break | 98 break |
| 99 else: | 99 else: |
| 100 final_results.ok += test_results.ok | 100 final_results.ok += test_results.ok |
| 101 self.tests = [] | 101 self.tests = [] |
| 102 for t in test_results.GetAllBroken(): | 102 for t in test_results.GetAllBroken(): |
| 103 self.tests += [t.name] | 103 self.tests += [t.name] |
| 104 if not self.tests: | 104 if not self.tests: |
| 105 break | 105 break |
| 106 self.OnTestsCompleted(test_runners, final_results) | 106 self.OnTestsCompleted(test_runners, final_results) |
| 107 return final_results | 107 return final_results |
| OLD | NEW |