Index: build/android/devil/android/device_utils_test.py |
diff --git a/build/android/pylib/device/device_utils_test.py b/build/android/devil/android/device_utils_test.py |
similarity index 97% |
rename from build/android/pylib/device/device_utils_test.py |
rename to build/android/devil/android/device_utils_test.py |
index 5fd4b2f9353d2ac8c2464899285f2f7861973a7c..4e78e69f43a96df085cb080ba36812189d3dcd6b 100755 |
--- a/build/android/pylib/device/device_utils_test.py |
+++ b/build/android/devil/android/device_utils_test.py |
@@ -19,16 +19,16 @@ import re |
import sys |
import unittest |
-from pylib import cmd_helper |
+from devil.android import device_signal |
+from devil.android import device_blacklist |
+from devil.android import device_errors |
+from devil.android import device_utils |
+from devil.android.sdk import adb_wrapper |
+from devil.android.sdk import intent |
+from devil.android.sdk import split_select |
+from devil.utils import cmd_helper |
+from devil.utils import mock_calls |
from pylib import constants |
-from pylib import device_signal |
-from pylib.device import adb_wrapper |
-from pylib.device import device_blacklist |
-from pylib.device import device_errors |
-from pylib.device import device_utils |
-from pylib.device import intent |
-from pylib.sdk import split_select |
-from pylib.utils import mock_calls |
sys.path.append(os.path.join( |
constants.DIR_SOURCE_ROOT, 'third_party', 'pymock')) |
@@ -64,7 +64,8 @@ class DeviceUtilsGetAVDsTest(mock_calls.TestCase): |
def testGetAVDs(self): |
with self.assertCall( |
- mock.call.pylib.cmd_helper.GetCmdOutput([mock.ANY, 'list', 'avd']), |
+ mock.call.devil.utils.cmd_helper.GetCmdOutput( |
+ [mock.ANY, 'list', 'avd']), |
'Available Android Virtual Devices:\n' |
' Name: my_android5.0\n' |
' Path: /some/path/to/.android/avd/my_android5.0.avd\n' |
@@ -80,13 +81,16 @@ class DeviceUtilsRestartServerTest(mock_calls.TestCase): |
@mock.patch('time.sleep', mock.Mock()) |
def testRestartServer_succeeds(self): |
with self.assertCalls( |
- mock.call.pylib.device.adb_wrapper.AdbWrapper.KillServer(), |
- (mock.call.pylib.cmd_helper.GetCmdStatusAndOutput(['pgrep', 'adb']), |
+ mock.call.devil.android.sdk.adb_wrapper.AdbWrapper.KillServer(), |
+ (mock.call.devil.utils.cmd_helper.GetCmdStatusAndOutput( |
+ ['pgrep', 'adb']), |
(1, '')), |
- mock.call.pylib.device.adb_wrapper.AdbWrapper.StartServer(), |
- (mock.call.pylib.cmd_helper.GetCmdStatusAndOutput(['pgrep', 'adb']), |
+ mock.call.devil.android.sdk.adb_wrapper.AdbWrapper.StartServer(), |
+ (mock.call.devil.utils.cmd_helper.GetCmdStatusAndOutput( |
+ ['pgrep', 'adb']), |
(1, '')), |
- (mock.call.pylib.cmd_helper.GetCmdStatusAndOutput(['pgrep', 'adb']), |
+ (mock.call.devil.utils.cmd_helper.GetCmdStatusAndOutput( |
+ ['pgrep', 'adb']), |
(0, '123\n'))): |
device_utils.RestartServer() |
@@ -548,7 +552,8 @@ class DeviceUtilsInstallTest(DeviceUtilsTest): |
def testInstall_noPriorInstall(self): |
with self.assertCalls( |
- (mock.call.pylib.utils.apk_helper.GetPackageName('/fake/test/app.apk'), |
+ (mock.call.devil.android.apk_helper.GetPackageName( |
+ '/fake/test/app.apk'), |
'test.package'), |
(self.call.device._GetApplicationPathsInternal('test.package'), []), |
self.call.adb.Install('/fake/test/app.apk', reinstall=False)): |
@@ -556,7 +561,8 @@ class DeviceUtilsInstallTest(DeviceUtilsTest): |
def testInstall_differentPriorInstall(self): |
with self.assertCalls( |
- (mock.call.pylib.utils.apk_helper.GetPackageName('/fake/test/app.apk'), |
+ (mock.call.devil.android.apk_helper.GetPackageName( |
+ '/fake/test/app.apk'), |
'test.package'), |
(self.call.device._GetApplicationPathsInternal('test.package'), |
['/fake/data/app/test.package.apk']), |
@@ -569,7 +575,8 @@ class DeviceUtilsInstallTest(DeviceUtilsTest): |
def testInstall_differentPriorInstall_reinstall(self): |
with self.assertCalls( |
- (mock.call.pylib.utils.apk_helper.GetPackageName('/fake/test/app.apk'), |
+ (mock.call.devil.android.apk_helper.GetPackageName( |
+ '/fake/test/app.apk'), |
'test.package'), |
(self.call.device._GetApplicationPathsInternal('test.package'), |
['/fake/data/app/test.package.apk']), |
@@ -581,7 +588,8 @@ class DeviceUtilsInstallTest(DeviceUtilsTest): |
def testInstall_identicalPriorInstall(self): |
with self.assertCalls( |
- (mock.call.pylib.utils.apk_helper.GetPackageName('/fake/test/app.apk'), |
+ (mock.call.devil.android.apk_helper.GetPackageName( |
+ '/fake/test/app.apk'), |
'test.package'), |
(self.call.device._GetApplicationPathsInternal('test.package'), |
['/fake/data/app/test.package.apk']), |
@@ -592,7 +600,8 @@ class DeviceUtilsInstallTest(DeviceUtilsTest): |
def testInstall_fails(self): |
with self.assertCalls( |
- (mock.call.pylib.utils.apk_helper.GetPackageName('/fake/test/app.apk'), |
+ (mock.call.devil.android.apk_helper.GetPackageName( |
+ '/fake/test/app.apk'), |
'test.package'), |
(self.call.device._GetApplicationPathsInternal('test.package'), []), |
(self.call.adb.Install('/fake/test/app.apk', reinstall=False), |
@@ -605,11 +614,11 @@ class DeviceUtilsInstallSplitApkTest(DeviceUtilsTest): |
def testInstallSplitApk_noPriorInstall(self): |
with self.assertCalls( |
(self.call.device._CheckSdkLevel(21)), |
- (mock.call.pylib.sdk.split_select.SelectSplits( |
+ (mock.call.devil.android.sdk.split_select.SelectSplits( |
self.device, 'base.apk', |
['split1.apk', 'split2.apk', 'split3.apk']), |
['split2.apk']), |
- (mock.call.pylib.utils.apk_helper.GetPackageName('base.apk'), |
+ (mock.call.devil.android.apk_helper.GetPackageName('base.apk'), |
'test.package'), |
(self.call.device._GetApplicationPathsInternal('test.package'), []), |
(self.call.adb.InstallMultiple( |
@@ -620,11 +629,11 @@ class DeviceUtilsInstallSplitApkTest(DeviceUtilsTest): |
def testInstallSplitApk_partialInstall(self): |
with self.assertCalls( |
(self.call.device._CheckSdkLevel(21)), |
- (mock.call.pylib.sdk.split_select.SelectSplits( |
+ (mock.call.devil.android.sdk.split_select.SelectSplits( |
self.device, 'base.apk', |
['split1.apk', 'split2.apk', 'split3.apk']), |
['split2.apk']), |
- (mock.call.pylib.utils.apk_helper.GetPackageName('base.apk'), |
+ (mock.call.devil.android.apk_helper.GetPackageName('base.apk'), |
'test.package'), |
(self.call.device._GetApplicationPathsInternal('test.package'), |
['base-on-device.apk', 'split2-on-device.apk']), |
@@ -704,7 +713,7 @@ class DeviceUtilsRunShellCommandTest(DeviceUtilsTest): |
payload = 'hi! ' * 1024 |
expected_cmd = "echo '%s'" % payload |
with self.assertCalls( |
- (mock.call.pylib.utils.device_temp_file.DeviceTempFile( |
+ (mock.call.devil.android.device_temp_file.DeviceTempFile( |
self.adb, suffix='.sh'), MockTempFile('/sdcard/temp-123.sh')), |
self.call.device._WriteFileWithPush('/sdcard/temp-123.sh', expected_cmd), |
(self.call.adb.Shell('sh /sdcard/temp-123.sh'), payload + '\n')): |
@@ -718,7 +727,7 @@ class DeviceUtilsRunShellCommandTest(DeviceUtilsTest): |
with self.assertCalls( |
(self.call.device.NeedsSU(), True), |
(self.call.device._Su(expected_cmd_without_su), expected_cmd), |
- (mock.call.pylib.utils.device_temp_file.DeviceTempFile( |
+ (mock.call.devil.android.device_temp_file.DeviceTempFile( |
self.adb, suffix='.sh'), MockTempFile('/sdcard/temp-123.sh')), |
self.call.device._WriteFileWithPush('/sdcard/temp-123.sh', expected_cmd), |
(self.call.adb.Shell('sh /sdcard/temp-123.sh'), payload + '\n')): |
@@ -798,7 +807,7 @@ class DeviceUtilsRunShellCommandTest(DeviceUtilsTest): |
temp_file = MockTempFile('/sdcard/temp-123') |
cmd_redirect = '%s > %s' % (cmd, temp_file.name) |
with self.assertCalls( |
- (mock.call.pylib.utils.device_temp_file.DeviceTempFile(self.adb), |
+ (mock.call.devil.android.device_temp_file.DeviceTempFile(self.adb), |
temp_file), |
(self.call.adb.Shell(cmd_redirect)), |
(self.call.device.ReadFile(temp_file.name, force_pull=True), |
@@ -820,7 +829,7 @@ class DeviceUtilsRunShellCommandTest(DeviceUtilsTest): |
cmd_redirect = '%s > %s' % (cmd, temp_file.name) |
with self.assertCalls( |
(self.call.adb.Shell(cmd), self.ShellError('', None)), |
- (mock.call.pylib.utils.device_temp_file.DeviceTempFile(self.adb), |
+ (mock.call.devil.android.device_temp_file.DeviceTempFile(self.adb), |
temp_file), |
(self.call.adb.Shell(cmd_redirect)), |
(self.call.device.ReadFile(mock.ANY, force_pull=True), |
@@ -1518,7 +1527,7 @@ class DeviceUtilsReadFileTest(DeviceUtilsTest): |
as_root=True, check_return=True), |
['-rw------- root root 123456 1970-01-01 00:00 can.be.read.with.su']), |
(self.call.device.NeedsSU(), True), |
- (mock.call.pylib.utils.device_temp_file.DeviceTempFile(self.adb), |
+ (mock.call.devil.android.device_temp_file.DeviceTempFile(self.adb), |
MockTempFile('/sdcard/tmp/on.device')), |
self.call.device.RunShellCommand( |
['cp', '/this/big/file/can.be.read.with.su', |
@@ -1578,7 +1587,7 @@ class DeviceUtilsWriteFileTest(DeviceUtilsTest): |
contents = 'some large contents ' * 26 # 20 * 26 = 520 chars |
with self.assertCalls( |
(self.call.device.NeedsSU(), True), |
- (mock.call.pylib.utils.device_temp_file.DeviceTempFile(self.adb), |
+ (mock.call.devil.android.device_temp_file.DeviceTempFile(self.adb), |
MockTempFile('/sdcard/tmp/on.device')), |
self.call.device._WriteFileWithPush('/sdcard/tmp/on.device', contents), |
self.call.device.RunShellCommand( |
@@ -1808,7 +1817,7 @@ class DeviceUtilsTakeScreenshotTest(DeviceUtilsTest): |
def testTakeScreenshot_fileNameProvided(self): |
with self.assertCalls( |
- (mock.call.pylib.utils.device_temp_file.DeviceTempFile( |
+ (mock.call.devil.android.device_temp_file.DeviceTempFile( |
self.adb, suffix='.png'), |
MockTempFile('/tmp/path/temp-123.png')), |
(self.call.adb.Shell('/system/bin/screencap -p /tmp/path/temp-123.png'), |
@@ -1958,7 +1967,7 @@ class DeviceUtilsHealthyDevicesTest(mock_calls.TestCase): |
def testHealthyDevices_emptyBlacklist(self): |
test_serials = ['0123456789abcdef', 'fedcba9876543210'] |
with self.assertCalls( |
- (mock.call.pylib.device.adb_wrapper.AdbWrapper.Devices(), |
+ (mock.call.devil.android.sdk.adb_wrapper.AdbWrapper.Devices(), |
[self._createAdbWrapperMock(s) for s in test_serials])): |
blacklist = mock.NonCallableMock(**{'Read.return_value': []}) |
devices = device_utils.DeviceUtils.HealthyDevices(blacklist) |
@@ -1969,7 +1978,7 @@ class DeviceUtilsHealthyDevicesTest(mock_calls.TestCase): |
def testHealthyDevices_blacklist(self): |
test_serials = ['0123456789abcdef', 'fedcba9876543210'] |
with self.assertCalls( |
- (mock.call.pylib.device.adb_wrapper.AdbWrapper.Devices(), |
+ (mock.call.devil.android.sdk.adb_wrapper.AdbWrapper.Devices(), |
[self._createAdbWrapperMock(s) for s in test_serials])): |
blacklist = mock.NonCallableMock( |
**{'Read.return_value': ['fedcba9876543210']}) |
@@ -1984,7 +1993,7 @@ class DeviceUtilsRestartAdbdTest(DeviceUtilsTest): |
def testAdbdRestart(self): |
mock_temp_file = '/sdcard/temp-123.sh' |
with self.assertCalls( |
- (mock.call.pylib.utils.device_temp_file.DeviceTempFile( |
+ (mock.call.devil.android.device_temp_file.DeviceTempFile( |
self.adb, suffix='.sh'), MockTempFile(mock_temp_file)), |
self.call.device.WriteFile(mock.ANY, mock.ANY), |
(self.call.device.RunShellCommand( |