build/android/pylib/device/adb_wrapper_test.py

   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()