Chromium Code Reviews| 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 android_commands | |
| 6 import logging | 7 import logging |
| 7 import multiprocessing | 8 import multiprocessing |
| 8 | 9 |
| 9 from test_result import TestResults | 10 from test_result import TestResults |
| 10 | 11 |
| 11 | 12 |
| 12 def _ShardedTestRunnable(test): | 13 def _ShardedTestRunnable(test): |
| 13 """Standalone function needed by multiprocessing.Pool.""" | 14 """Standalone function needed by multiprocessing.Pool.""" |
| 14 log_format = '[' + test.device + '] # %(asctime)-15s: %(message)s' | 15 log_format = '[' + test.device + '] # %(asctime)-15s: %(message)s' |
| 15 if logging.getLogger().handlers: | 16 if logging.getLogger().handlers: |
| (...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 79 logging.warning('*' * 80) | 80 logging.warning('*' * 80) |
| 80 final_results = TestResults() | 81 final_results = TestResults() |
| 81 for retry in xrange(self.retries): | 82 for retry in xrange(self.retries): |
| 82 logging.warning('Try %d of %d', retry + 1, self.retries) | 83 logging.warning('Try %d of %d', retry + 1, self.retries) |
| 83 self.SetupSharding(self.tests) | 84 self.SetupSharding(self.tests) |
| 84 test_runners = [] | 85 test_runners = [] |
| 85 for index, device in enumerate(self.attached_devices): | 86 for index, device in enumerate(self.attached_devices): |
| 86 logging.warning('*' * 80) | 87 logging.warning('*' * 80) |
| 87 logging.warning('Creating shard %d for %s', index, device) | 88 logging.warning('Creating shard %d for %s', index, device) |
| 88 logging.warning('*' * 80) | 89 logging.warning('*' * 80) |
| 89 test_runner = self.CreateShardedTestRunner(device, index) | 90 test_runner = self.CreateShardedTestRunner(device, index) |
|
Yaron
2012/10/26 18:48:44
Based on: http://build.chromium.org/p/chromium.lin
yongsheng
2012/10/28 01:41:03
yes, i don't try to catch exceptions here for i'm
bulach
2012/10/29 14:45:42
yaron: just to clarify, I kindly asked Yongsheng (
bulach
2012/10/29 14:51:23
btw, the issue mentioned above I guess is in run_t
| |
| 90 test_runners += [test_runner] | 91 test_runners += [test_runner] |
| 91 logging.warning('Starting...') | 92 logging.warning('Starting...') |
| 92 pool = multiprocessing.Pool(len(self.attached_devices), | 93 pool = multiprocessing.Pool(len(self.attached_devices), |
| 93 SetTestsContainer, | 94 SetTestsContainer, |
| 94 [BaseTestSharder.tests_container]) | 95 [BaseTestSharder.tests_container]) |
| 95 # map can't handle KeyboardInterrupt exception. It's a python bug. | 96 # map can't handle KeyboardInterrupt exception. It's a python bug. |
| 96 # So use map_async instead. | 97 # So use map_async instead. |
| 97 async_results = pool.map_async(_ShardedTestRunnable, test_runners) | 98 async_results = pool.map_async(_ShardedTestRunnable, test_runners) |
| 98 results_lists = async_results.get(999999) | 99 results_lists = async_results.get(999999) |
| 100 | |
| 99 test_results = TestResults.FromTestResults(results_lists) | 101 test_results = TestResults.FromTestResults(results_lists) |
| 100 if retry == self.retries - 1: | 102 # Re-check the attached devices for some devices may |
| 103 # become offline | |
| 104 retry_devices = set(android_commands.GetAttachedDevices()) | |
| 105 # Remove devices that had exceptions. | |
| 106 retry_devices -= TestResults.DeviceExceptions(results_lists) | |
| 107 # Retry on devices that didn't have any exception. | |
| 108 self.attached_devices = list(retry_devices) | |
| 109 if (retry == self.retries - 1 or | |
| 110 len(self.attached_devices) == 0): | |
| 101 all_passed = final_results.ok + test_results.ok | 111 all_passed = final_results.ok + test_results.ok |
| 102 final_results = test_results | 112 final_results = test_results |
| 103 final_results.ok = all_passed | 113 final_results.ok = all_passed |
| 104 break | 114 break |
| 105 else: | 115 else: |
| 106 final_results.ok += test_results.ok | 116 final_results.ok += test_results.ok |
| 107 self.tests = [] | 117 self.tests = [] |
| 108 for t in test_results.GetAllBroken(): | 118 for t in test_results.GetAllBroken(): |
| 109 self.tests += [t.name] | 119 self.tests += [t.name] |
| 110 if not self.tests: | 120 if not self.tests: |
| 111 break | 121 break |
| 112 self.OnTestsCompleted(test_runners, final_results) | 122 self.OnTestsCompleted(test_runners, final_results) |
| 113 return final_results | 123 return final_results |
| OLD | NEW |