build/android/pylib/android_commands_unittest.py

   1 # Copyright 2014 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 import os
   6 import shutil
   7 import sys
   8 import unittest
   9
  10 sys.path.append(os.path.join(os.path.dirname(__file__), os.pardir))
  11
  12 from pylib import android_commands
  13
  14 # pylint: disable=W0212,W0702
  15
  16 class TestDeviceTempFile(unittest.TestCase):
  17   def setUp(self):
  18     if not os.getenv('BUILDTYPE'):
  19       os.environ['BUILDTYPE'] = 'Debug'
  20
  21     devices = android_commands.GetAttachedDevices()
  22     self.assertGreater(len(devices), 0, 'No device attached!')
  23     self.ac = android_commands.AndroidCommands(device=devices[0])
  24
  25   def testTempFileDeleted(self):
  26     """Tests that DeviceTempFile deletes files when closed."""
  27     temp_file = android_commands.DeviceTempFile(self.ac)
  28     self.assertFalse(self.ac.FileExistsOnDevice(temp_file.name))
  29     self.ac.SetFileContents(temp_file.name, "contents")
  30     self.assertTrue(self.ac.FileExistsOnDevice(temp_file.name))
  31     temp_file.close()
  32     self.assertFalse(self.ac.FileExistsOnDevice(temp_file.name))
  33
  34     with android_commands.DeviceTempFile(self.ac) as with_temp_file:
  35       self.assertFalse(self.ac.FileExistsOnDevice(with_temp_file.name))
  36       self.ac.SetFileContents(with_temp_file.name, "contents")
  37       self.assertTrue(self.ac.FileExistsOnDevice(with_temp_file.name))
  38
  39     self.assertFalse(self.ac.FileExistsOnDevice(with_temp_file.name))
  40
  41   def testTempFileNotWritten(self):
  42     """Tests that device temp files work successfully even if not written to."""
  43     temp_file = android_commands.DeviceTempFile(self.ac)
  44     temp_file.close()
  45     self.assertFalse(self.ac.FileExistsOnDevice(temp_file.name))
  46
  47     with android_commands.DeviceTempFile(self.ac) as with_temp_file:
  48       pass
  49     self.assertFalse(self.ac.FileExistsOnDevice(with_temp_file.name))
  50
  51   def testNaming(self):
  52     """Tests that returned filenames are as requested."""
  53     temp_file = android_commands.DeviceTempFile(self.ac, prefix="cat")
  54     self.assertTrue(os.path.basename(temp_file.name).startswith("cat"))
  55
  56     temp_file = android_commands.DeviceTempFile(self.ac, suffix="dog")
  57     self.assertTrue(temp_file.name.endswith("dog"))
  58
  59     temp_file = android_commands.DeviceTempFile(
  60         self.ac, prefix="cat", suffix="dog")
  61     self.assertTrue(os.path.basename(temp_file.name).startswith("cat"))
  62     self.assertTrue(temp_file.name.endswith("dog"))
  63
  64
  65 class TestGetFilesChanged(unittest.TestCase):
  66
  67   def setUp(self):
  68     if not os.getenv('BUILDTYPE'):
  69       os.environ['BUILDTYPE'] = 'Debug'
  70
  71     devices = android_commands.GetAttachedDevices()
  72     self.assertGreater(len(devices), 0, 'No device attached!')
  73     self.ac = android_commands.AndroidCommands(device=devices[0])
  74     self.host_data_dir = os.path.realpath('test_push_data')
  75     self.device_data_dir = '%s/test_push_data' % (
  76         self.ac.RunShellCommand('realpath %s' %
  77             self.ac.GetExternalStorage())[0])
  78
  79     os.mkdir(self.host_data_dir)
  80     for i in xrange(1, 10):
  81       with open('%s/%d.txt' % (self.host_data_dir, i), 'w') as f:
  82         f.write('file #%d' % i)
  83
  84     self.ac.RunShellCommand('mkdir %s' % self.device_data_dir)
  85
  86   def testGetFilesChangedAllNeeded(self):
  87     """ Tests GetFilesChanged when none of the files are on the device.
  88     """
  89     expected = [('%s/%d.txt' % (self.host_data_dir, i),
  90                  '%s/%d.txt' % (self.device_data_dir, i))
  91                 for i in xrange(1, 10)]
  92     actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir)
  93     self.assertSequenceEqual(expected, actual)
  94
  95   def testGetFilesChangedSomeIdentical(self):
  96     """ Tests GetFilesChanged when some of the files are on the device.
  97     """
  98     for i in xrange(1, 5):
  99       self.ac._adb.Push('%s/%d.txt' % (self.host_data_dir, i),
 100                         self.device_data_dir)
 101     expected = [('%s/%d.txt' % (self.host_data_dir, i),
 102                  '%s/%d.txt' % (self.device_data_dir, i))
 103                 for i in xrange(5, 10)]
 104     actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir)
 105     self.assertSequenceEqual(expected, actual)
 106
 107   def testGetFilesChangedAllIdentical(self):
 108     """ Tests GetFilesChanged when all of the files are on the device.
 109     """
 110     for i in xrange(1, 10):
 111       self.ac._adb.Push('%s/%d.txt' % (self.host_data_dir, i),
 112                         self.device_data_dir)
 113     expected = []
 114     actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir)
 115     self.assertSequenceEqual(expected, actual)
 116
 117   def testGetFilesChangedRename(self):
 118     """ Tests GetFilesChanged when one of the files has been renamed.
 119
 120         This tests both with and without the ignore_filenames flag set.
 121     """
 122     for i in xrange(5, 10):
 123       self.ac._adb.Push('%s/%d.txt' % (self.host_data_dir, i),
 124                         self.device_data_dir)
 125     os.rename('%s/5.txt' % (self.host_data_dir),
 126               '%s/99.txt' % (self.host_data_dir))
 127
 128     expected = [('%s/%d.txt' % (self.host_data_dir, i),
 129                  '%s/%d.txt' % (self.device_data_dir, i))
 130                 for i in xrange(1, 5)]
 131     actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir,
 132                                      ignore_filenames=True)
 133     self.assertSequenceEqual(expected, actual)
 134
 135     expected.append(('%s/99.txt' % self.host_data_dir,
 136                      '%s/99.txt' % self.device_data_dir))
 137     actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir)
 138     self.assertSequenceEqual(expected, actual)
 139
 140   def testGetFilesChangedCopy(self):
 141     """ Tests GetFilesChanged when one of the files has been copied.
 142
 143         This tests both with and without the ignore_filenames flag set.
 144     """
 145     for i in xrange(5, 10):
 146       self.ac._adb.Push('%s/%d.txt' % (self.host_data_dir, i),
 147                         self.device_data_dir)
 148     shutil.copy('%s/5.txt' % self.host_data_dir,
 149                 '%s/99.txt' % self.host_data_dir)
 150
 151     expected = [('%s/%d.txt' % (self.host_data_dir, i),
 152                  '%s/%d.txt' % (self.device_data_dir, i))
 153                 for i in xrange(1, 5)]
 154     actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir,
 155                                      ignore_filenames=True)
 156     self.assertSequenceEqual(expected, actual)
 157
 158     expected.append(('%s/99.txt' % self.host_data_dir,
 159                      '%s/99.txt' % self.device_data_dir))
 160     actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir)
 161     self.assertSequenceEqual(expected, actual)
 162
 163   def testGetFilesChangedIndividual(self):
 164     """ Tests GetFilesChanged when provided one file.
 165     """
 166     expected = [('%s/1.txt' % self.host_data_dir,
 167                  '%s/1.txt' % self.device_data_dir)]
 168     actual = self.ac.GetFilesChanged('%s/1.txt' % self.host_data_dir,
 169                                      '%s/1.txt' % self.device_data_dir)
 170     self.assertSequenceEqual(expected, actual)
 171
 172   def testGetFilesChangedFileToDirectory(self):
 173     """ Tests GetFilesChanged when provided a file from the host and a
 174         directory on the device.
 175     """
 176     expected = [('%s/1.txt' % self.host_data_dir,
 177                  '%s' % self.device_data_dir)]
 178     actual = self.ac.GetFilesChanged('%s/1.txt' % self.host_data_dir,
 179                                      '%s' % self.device_data_dir)
 180     self.assertSequenceEqual(expected, actual)
 181
 182   def tearDown(self):
 183     try:
 184       shutil.rmtree(self.host_data_dir)
 185       self.ac.RunShellCommand('rm -rf %s' % self.device_data_dir)
 186     except:
 187       pass
 188
 189 if __name__ == '__main__':
 190   unittest.main()
 191