| OLD | NEW |
| (Empty) |
| 1 # Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 # Use of this source code is governed by a BSD-style license that can be | |
| 3 # found in the LICENSE file. | |
| 4 | |
| 5 """Tests for the AdbWrapper class.""" | |
| 6 | |
| 7 import os | |
| 8 import tempfile | |
| 9 import time | |
| 10 import unittest | |
| 11 | |
| 12 from pylib.device import adb_wrapper | |
| 13 from pylib.device import device_errors | |
| 14 | |
| 15 | |
| 16 class TestAdbWrapper(unittest.TestCase): | |
| 17 | |
| 18 def setUp(self): | |
| 19 devices = adb_wrapper.AdbWrapper.Devices() | |
| 20 assert devices, 'A device must be attached' | |
| 21 self._adb = devices[0] | |
| 22 self._adb.WaitForDevice() | |
| 23 | |
| 24 @staticmethod | |
| 25 def _MakeTempFile(contents): | |
| 26 """Make a temporary file with the given contents. | |
| 27 | |
| 28 Args: | |
| 29 contents: string to write to the temporary file. | |
| 30 | |
| 31 Returns: | |
| 32 The absolute path to the file. | |
| 33 """ | |
| 34 fi, path = tempfile.mkstemp() | |
| 35 with os.fdopen(fi, 'wb') as f: | |
| 36 f.write(contents) | |
| 37 return path | |
| 38 | |
| 39 def testShell(self): | |
| 40 output = self._adb.Shell('echo test', expect_status=0) | |
| 41 self.assertEqual(output.strip(), 'test') | |
| 42 output = self._adb.Shell('echo test') | |
| 43 self.assertEqual(output.strip(), 'test') | |
| 44 with self.assertRaises(device_errors.AdbCommandFailedError): | |
| 45 self._adb.Shell('echo test', expect_status=1) | |
| 46 | |
| 47 def testPushLsPull(self): | |
| 48 path = self._MakeTempFile('foo') | |
| 49 device_path = '/data/local/tmp/testfile.txt' | |
| 50 local_tmpdir = os.path.dirname(path) | |
| 51 self._adb.Push(path, device_path) | |
| 52 files = dict(self._adb.Ls('/data/local/tmp')) | |
| 53 self.assertTrue('testfile.txt' in files) | |
| 54 self.assertEquals(3, files['testfile.txt'].st_size) | |
| 55 self.assertEqual(self._adb.Shell('cat %s' % device_path), 'foo') | |
| 56 self._adb.Pull(device_path, local_tmpdir) | |
| 57 with open(os.path.join(local_tmpdir, 'testfile.txt'), 'r') as f: | |
| 58 self.assertEqual(f.read(), 'foo') | |
| 59 | |
| 60 def testInstall(self): | |
| 61 path = self._MakeTempFile('foo') | |
| 62 with self.assertRaises(device_errors.AdbCommandFailedError): | |
| 63 self._adb.Install(path) | |
| 64 | |
| 65 def testForward(self): | |
| 66 with self.assertRaises(device_errors.AdbCommandFailedError): | |
| 67 self._adb.Forward(0, 0) | |
| 68 | |
| 69 def testUninstall(self): | |
| 70 with self.assertRaises(device_errors.AdbCommandFailedError): | |
| 71 self._adb.Uninstall('some.nonexistant.package') | |
| 72 | |
| 73 def testRebootWaitForDevice(self): | |
| 74 self._adb.Reboot() | |
| 75 print 'waiting for device to reboot...' | |
| 76 while self._adb.GetState() == 'device': | |
| 77 time.sleep(1) | |
| 78 self._adb.WaitForDevice() | |
| 79 self.assertEqual(self._adb.GetState(), 'device') | |
| 80 print 'waiting for package manager...' | |
| 81 while 'package:' not in self._adb.Shell('pm path android'): | |
| 82 time.sleep(1) | |
| 83 | |
| 84 def testRootRemount(self): | |
| 85 self._adb.Root() | |
| 86 while True: | |
| 87 try: | |
| 88 self._adb.Shell('start') | |
| 89 break | |
| 90 except device_errors.AdbCommandFailedError: | |
| 91 time.sleep(1) | |
| 92 self._adb.Remount() | |
| 93 | |
| 94 | |
| 95 if __name__ == '__main__': | |
| 96 unittest.main() | |
| OLD | NEW |