| OLD | NEW |
| 1 # Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 # Use of this source code is governed by a BSD-style license that can be |
| 3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
| 4 | 4 |
| 5 """Tests for the AdbWrapper class.""" | 5 """Tests for the AdbWrapper class.""" |
| 6 | 6 |
| 7 import os | 7 import os |
| 8 import tempfile | 8 import tempfile |
| 9 import time | 9 import time |
| 10 import unittest | 10 import unittest |
| (...skipping 23 matching lines...) Expand all Loading... |
| 34 fi, path = tempfile.mkstemp() | 34 fi, path = tempfile.mkstemp() |
| 35 with os.fdopen(fi, 'wb') as f: | 35 with os.fdopen(fi, 'wb') as f: |
| 36 f.write(contents) | 36 f.write(contents) |
| 37 return path | 37 return path |
| 38 | 38 |
| 39 def testShell(self): | 39 def testShell(self): |
| 40 output = self._adb.Shell('echo test', expect_rc=0) | 40 output = self._adb.Shell('echo test', expect_rc=0) |
| 41 self.assertEqual(output.strip(), 'test') | 41 self.assertEqual(output.strip(), 'test') |
| 42 output = self._adb.Shell('echo test') | 42 output = self._adb.Shell('echo test') |
| 43 self.assertEqual(output.strip(), 'test') | 43 self.assertEqual(output.strip(), 'test') |
| 44 self.assertRaises(device_errors.AdbCommandFailedError, self._adb.Shell, | 44 self.assertRaises(device_errors.AdbShellCommandFailedError, |
| 45 'echo test', expect_rc=1) | 45 self._adb.Shell, 'echo test', expect_rc=1) |
| 46 | 46 |
| 47 def testPushPull(self): | 47 def testPushPull(self): |
| 48 path = self._MakeTempFile('foo') | 48 path = self._MakeTempFile('foo') |
| 49 device_path = '/data/local/tmp/testfile.txt' | 49 device_path = '/data/local/tmp/testfile.txt' |
| 50 local_tmpdir = os.path.dirname(path) | 50 local_tmpdir = os.path.dirname(path) |
| 51 self._adb.Push(path, device_path) | 51 self._adb.Push(path, device_path) |
| 52 self.assertEqual(self._adb.Shell('cat %s' % device_path), 'foo') | 52 self.assertEqual(self._adb.Shell('cat %s' % device_path), 'foo') |
| 53 self._adb.Pull(device_path, local_tmpdir) | 53 self._adb.Pull(device_path, local_tmpdir) |
| 54 with open(os.path.join(local_tmpdir, 'testfile.txt'), 'r') as f: | 54 with open(os.path.join(local_tmpdir, 'testfile.txt'), 'r') as f: |
| 55 self.assertEqual(f.read(), 'foo') | 55 self.assertEqual(f.read(), 'foo') |
| (...skipping 28 matching lines...) Expand all Loading... |
| 84 try: | 84 try: |
| 85 self._adb.Shell('start') | 85 self._adb.Shell('start') |
| 86 break | 86 break |
| 87 except device_errors.AdbCommandFailedError: | 87 except device_errors.AdbCommandFailedError: |
| 88 time.sleep(1) | 88 time.sleep(1) |
| 89 self._adb.Remount() | 89 self._adb.Remount() |
| 90 | 90 |
| 91 | 91 |
| 92 if __name__ == '__main__': | 92 if __name__ == '__main__': |
| 93 unittest.main() | 93 unittest.main() |
| OLD | NEW |