Chromium Code Reviews| Index: presubmit_support.py |
| diff --git a/presubmit_support.py b/presubmit_support.py |
| index 4baee778831d229386738d1a42d1a3b2b5ebef88..523bb79a1245f4c30534bea2e90ac447dda49aba 100755 |
| --- a/presubmit_support.py |
| +++ b/presubmit_support.py |
| @@ -313,6 +313,12 @@ class InputApi(object): |
| self.cpu_count = multiprocessing.cpu_count() |
| + # this is done here because in RunTests, the current working directory has |
| + # changed, which causes Pool() to explode fantastically when run on windows |
| + # (because it tries to load the __main__ module, which imports lots of |
| + # things relative to the current working directory). |
| + self._run_tests_pool = multiprocessing.Pool(self.cpu_count) |
|
iannucci
2015/06/28 19:47:43
this is so we pick up the mocked cpu_count
|
| + |
| # The local path of the currently-being-processed presubmit script. |
| self._current_presubmit_path = os.path.dirname(presubmit_path) |
| @@ -492,11 +498,8 @@ class InputApi(object): |
| if self.verbose: |
| t.info = _PresubmitNotifyResult |
| if len(tests) > 1 and parallel: |
| - pool = multiprocessing.Pool() |
| # async recipe works around multiprocessing bug handling Ctrl-C |
| - msgs.extend(pool.map_async(CallCommand, tests).get(99999)) |
| - pool.close() |
| - pool.join() |
| + msgs.extend(self._run_tests_pool.map_async(CallCommand, tests).get(99999)) |
| else: |
| msgs.extend(map(CallCommand, tests)) |
| return [m for m in msgs if m] |