Index: build/android/pylib/gtest/setup.py |
diff --git a/build/android/pylib/gtest/setup.py b/build/android/pylib/gtest/setup.py |
index e62b64811aaee30b04a749f73d98aa684f9b3bba..62889761e2907bf94e555eb4bcf87de5c7dbacc8 100644 |
--- a/build/android/pylib/gtest/setup.py |
+++ b/build/android/pylib/gtest/setup.py |
@@ -12,10 +12,9 @@ import os |
import shutil |
import sys |
-from pylib import android_commands |
from pylib import cmd_helper |
from pylib import constants |
- |
+from pylib.device import adb_wrapper |
from pylib.gtest import test_package_apk |
from pylib.gtest import test_package_exe |
from pylib.gtest import test_runner |
@@ -219,8 +218,8 @@ def _GetTestsFromDevice(runner_factory, devices): |
try: |
logging.info('Obtaining tests from %s', device) |
return runner_factory(device, 0).GetAllTests() |
- except (android_commands.errors.WaitForResponseTimedOutError, |
- android_commands.errors.DeviceUnresponsiveError), e: |
+ except (adb_wrapper.CommandTimeoutError, |
craigdh
2014/04/09 15:55:02
doesn't this need rebasing? I thought you changed
|
+ adb_wrapper.DeviceUnreachableError), e: |
logging.warning('Failed obtaining test list from %s with exception: %s', |
device, e) |
raise Exception('Failed to obtain test list from devices.') |