| OLD | NEW |
| (Empty) |
| 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 | |
| OLD | NEW |