Index: build/android/pylib/utils/test_environment.py |
diff --git a/build/android/pylib/utils/test_environment.py b/build/android/pylib/utils/test_environment.py |
index e78eb5ce801ffa9185fd5c0b69e5f56a7eaef570..c9b56769c2f871c34ca95ba68cc22af0a099fcd2 100644 |
--- a/build/android/pylib/utils/test_environment.py |
+++ b/build/android/pylib/utils/test_environment.py |
@@ -30,8 +30,12 @@ def _KillWebServers(): |
logging.warning('Failed waiting for %s to die. %s', p.pid, e) |
-def CleanupLeftoverProcesses(): |
- """Clean up the test environment, restarting fresh adb and HTTP daemons.""" |
+def CleanupLeftoverProcesses(devices): |
+ """Clean up the test environment, restarting fresh adb and HTTP daemons. |
+ |
+ Args: |
+ devices: The devices to clean. |
+ """ |
_KillWebServers() |
device_utils.RestartServer() |
@@ -43,5 +47,5 @@ def CleanupLeftoverProcesses(): |
logging.error(str(e)) |
d.WaitUntilFullyBooted() |
- device_utils.DeviceUtils.parallel().pMap(cleanup_device) |
+ device_utils.DeviceUtils.parallel(devices).pMap(cleanup_device) |