Index: build/android/pylib/device/device_utils_device_test.py |
diff --git a/build/android/pylib/device/device_utils_device_test.py b/build/android/pylib/device/device_utils_device_test.py |
new file mode 100755 |
index 0000000000000000000000000000000000000000..daae2b65fbf33d2aeda673d86e40e2caf330d1ec |
--- /dev/null |
+++ b/build/android/pylib/device/device_utils_device_test.py |
@@ -0,0 +1,211 @@ |
+#!/usr/bin/env python |
+# Copyright 2015 The Chromium Authors. All rights reserved. |
+# Use of this source code is governed by a BSD-style license that can be |
+# found in the LICENSE file. |
+ |
+""" |
+Unit tests for the contents of device_utils.py (mostly DeviceUtils). |
+The test will invoke real devices |
+""" |
+ |
+import os |
+import tempfile |
+import unittest |
+ |
+from pylib import cmd_helper |
+from pylib import constants |
+from pylib.device import adb_wrapper |
+from pylib.device import device_utils |
+from pylib.utils import md5sum |
+ |
+_OLD_CONTENTS = "foo" |
+_NEW_CONTENTS = "bar" |
+_DEVICE_DIR = "/data/local/tmp/device_utils_test" |
+_SUB_DIR = "sub" |
+_SUB_DIR1 = "sub1" |
+_SUB_DIR2 = "sub2" |
+ |
+class DeviceUtilsPushDeleteFilesTest(unittest.TestCase): |
+ |
+ def setUp(self): |
+ devices = adb_wrapper.AdbWrapper.Devices() |
+ assert devices, 'A device must be attached' |
+ self.adb = devices[0] |
+ self.adb.WaitForDevice() |
+ self.device = device_utils.DeviceUtils( |
+ self.adb, default_timeout=10, default_retries=0) |
+ default_build_type = os.environ.get('BUILDTYPE', 'Debug') |
+ constants.SetBuildType(default_build_type) |
+ |
+ @staticmethod |
+ def _MakeTempFile(contents): |
+ """Make a temporary file with the given contents. |
+ |
+ Args: |
+ contents: string to write to the temporary file. |
+ |
+ Returns: |
+ the tuple contains the absolute path to the file and the file name |
+ """ |
+ fi, path = tempfile.mkstemp(text=True) |
+ with os.fdopen(fi, 'w') as f: |
+ f.write(contents) |
+ file_name = os.path.basename(path) |
+ return (path, file_name) |
+ |
+ @staticmethod |
+ def _MakeTempFileGivenDir(directory, contents): |
+ """Make a temporary file under the given directory |
+ with the given contents |
+ |
+ Args: |
+ directory: the temp directory to create the file |
+ contents: string to write to the temp file |
+ |
+ Returns: |
+ the list contains the absolute path to the file and the file name |
+ """ |
+ fi, path = tempfile.mkstemp(dir=directory, text=True) |
+ with os.fdopen(fi, 'w') as f: |
+ f.write(contents) |
+ file_name = os.path.basename(path) |
+ return (path, file_name) |
+ |
+ @staticmethod |
+ def _ChangeTempFile(path, contents): |
+ with os.open(path, 'w') as f: |
+ f.write(contents) |
+ |
+ @staticmethod |
+ def _DeleteTempFile(path): |
+ os.remove(path) |
+ |
+ def testPushChangedFiles_noFileChange(self): |
+ (host_file_path, file_name) = self._MakeTempFile(_OLD_CONTENTS) |
+ device_file_path = "%s/%s" % (_DEVICE_DIR, file_name) |
+ self.adb.Push(host_file_path, device_file_path) |
+ self.device.PushChangedFiles([(host_file_path, device_file_path)]) |
+ result = self.device.RunShellCommand(['cat', device_file_path], |
+ single_line=True) |
+ self.assertEqual(_OLD_CONTENTS, result) |
+ |
+ cmd_helper.RunCmd(['rm', host_file_path]) |
+ self.device.RunShellCommand(['rm', '-rf', _DEVICE_DIR]) |
+ |
+ def testPushChangedFiles_singleFileChange(self): |
+ (host_file_path, file_name) = self._MakeTempFile(_OLD_CONTENTS) |
+ device_file_path = "%s/%s" % (_DEVICE_DIR, file_name) |
+ self.adb.Push(host_file_path, device_file_path) |
+ |
+ with open(host_file_path, 'w') as f: |
+ f.write(_NEW_CONTENTS) |
+ self.device.PushChangedFiles([(host_file_path, device_file_path)]) |
+ result = self.device.RunShellCommand(['cat', device_file_path], |
+ single_line=True) |
+ self.assertEqual(_NEW_CONTENTS, result) |
+ |
+ cmd_helper.RunCmd(['rm', host_file_path]) |
+ self.device.RunShellCommand(['rm', '-rf', _DEVICE_DIR]) |
+ |
+ def testDeleteFiles(self): |
+ host_tmp_dir = tempfile.mkdtemp() |
+ (host_file_path, file_name) = self._MakeTempFileGivenDir( |
+ host_tmp_dir, _OLD_CONTENTS) |
+ |
+ device_file_path = "%s/%s" % (_DEVICE_DIR, file_name) |
+ self.adb.Push(host_file_path, device_file_path) |
+ |
+ cmd_helper.RunCmd(['rm', host_file_path]) |
+ self.device.PushChangedFiles([(host_tmp_dir, _DEVICE_DIR)], |
+ delete_device_stale=True) |
+ result = self.device.RunShellCommand(['ls', _DEVICE_DIR], single_line=True) |
+ self.assertEqual('', result) |
+ |
+ cmd_helper.RunCmd(['rm', '-rf', host_tmp_dir]) |
+ self.device.RunShellCommand(['rm', '-rf', _DEVICE_DIR]) |
+ |
+ def testPushAndDeleteFiles_noSubDir(self): |
+ host_tmp_dir = tempfile.mkdtemp() |
+ (host_file_path1, file_name1) = self._MakeTempFileGivenDir( |
+ host_tmp_dir, _OLD_CONTENTS) |
+ (host_file_path2, file_name2) = self._MakeTempFileGivenDir( |
+ host_tmp_dir, _OLD_CONTENTS) |
+ |
+ device_file_path1 = "%s/%s" % (_DEVICE_DIR, file_name1) |
+ device_file_path2 = "%s/%s" % (_DEVICE_DIR, file_name2) |
+ self.adb.Push(host_file_path1, device_file_path1) |
+ self.adb.Push(host_file_path2, device_file_path2) |
+ |
+ with open(host_file_path1, 'w') as f: |
+ f.write(_NEW_CONTENTS) |
+ cmd_helper.RunCmd(['rm', host_file_path2]) |
+ |
+ self.device.PushChangedFiles([(host_tmp_dir, _DEVICE_DIR)], |
+ delete_device_stale=True) |
+ result = self.device.RunShellCommand(['cat', device_file_path1], |
+ single_line=True) |
+ self.assertEqual(_NEW_CONTENTS, result) |
+ result = self.device.RunShellCommand(['ls', _DEVICE_DIR], single_line=True) |
+ self.assertEqual(file_name1, result) |
+ |
+ self.device.RunShellCommand(['rm', '-rf', _DEVICE_DIR]) |
+ cmd_helper.RunCmd(['rm', '-rf', host_tmp_dir]) |
+ |
+ def testPushAndDeleteFiles_SubDir(self): |
+ host_tmp_dir = tempfile.mkdtemp() |
+ host_sub_dir1 = "%s/%s" % (host_tmp_dir, _SUB_DIR1) |
+ host_sub_dir2 = "%s/%s/%s" % (host_tmp_dir, _SUB_DIR, _SUB_DIR2) |
+ cmd_helper.RunCmd(['mkdir', '-p', host_sub_dir1]) |
+ cmd_helper.RunCmd(['mkdir', '-p', host_sub_dir2]) |
+ |
+ (host_file_path1, file_name1) = self._MakeTempFileGivenDir( |
+ host_tmp_dir, _OLD_CONTENTS) |
+ (host_file_path2, file_name2) = self._MakeTempFileGivenDir( |
+ host_tmp_dir, _OLD_CONTENTS) |
+ (host_file_path3, file_name3) = self._MakeTempFileGivenDir( |
+ host_sub_dir1, _OLD_CONTENTS) |
+ (host_file_path4, file_name4) = self._MakeTempFileGivenDir( |
+ host_sub_dir2, _OLD_CONTENTS) |
+ |
+ device_file_path1 = "%s/%s" % (_DEVICE_DIR, file_name1) |
+ device_file_path2 = "%s/%s" % (_DEVICE_DIR, file_name2) |
+ device_file_path3 = "%s/%s/%s" % (_DEVICE_DIR, _SUB_DIR1, file_name3) |
+ device_file_path4 = "%s/%s/%s/%s" % (_DEVICE_DIR, _SUB_DIR, |
+ _SUB_DIR2, file_name4) |
+ |
+ self.adb.Push(host_file_path1, device_file_path1) |
+ self.adb.Push(host_file_path2, device_file_path2) |
+ self.adb.Push(host_file_path3, device_file_path3) |
+ self.adb.Push(host_file_path4, device_file_path4) |
+ |
+ with open(host_file_path1, 'w') as f: |
+ f.write(_NEW_CONTENTS) |
+ cmd_helper.RunCmd(['rm', host_file_path2]) |
+ cmd_helper.RunCmd(['rm', host_file_path4]) |
+ |
+ self.device.PushChangedFiles([(host_tmp_dir, _DEVICE_DIR)], |
+ delete_device_stale=True) |
+ result = self.device.RunShellCommand(['cat', device_file_path1], |
+ single_line=True) |
+ self.assertEqual(_NEW_CONTENTS, result) |
+ |
+ result = self.device.RunShellCommand(['ls', _DEVICE_DIR]) |
+ self.assertIn(file_name1, result) |
+ self.assertIn(_SUB_DIR1, result) |
+ self.assertIn(_SUB_DIR, result) |
+ self.assertEqual(3, len(result)) |
+ |
+ result = self.device.RunShellCommand(['cat', device_file_path3], |
+ single_line=True) |
+ self.assertEqual(_OLD_CONTENTS, result) |
+ |
+ result = self.device.RunShellCommand(["ls", "%s/%s/%s" |
+ % (_DEVICE_DIR, _SUB_DIR, _SUB_DIR2)], |
+ single_line=True) |
+ self.assertEqual('', result) |
+ |
+ self.device.RunShellCommand(['rm', '-rf', _DEVICE_DIR]) |
+ cmd_helper.RunCmd(['rm', '-rf', host_tmp_dir]) |
+ |
+if __name__ == '__main__': |
+ unittest.main() |