| # Copyright 2014 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import os |
| import shutil |
| import sys |
| import unittest |
| |
| sys.path.append(os.path.join(os.path.dirname(__file__), os.pardir)) |
| |
| from pylib import android_commands |
| |
| # pylint: disable=W0212,W0702 |
| |
| class TestDeviceTempFile(unittest.TestCase): |
| def setUp(self): |
| if not os.getenv('BUILDTYPE'): |
| os.environ['BUILDTYPE'] = 'Debug' |
| |
| devices = android_commands.GetAttachedDevices() |
| self.assertGreater(len(devices), 0, 'No device attached!') |
| self.ac = android_commands.AndroidCommands(device=devices[0]) |
| |
| def testTempFileDeleted(self): |
| """Tests that DeviceTempFile deletes files when closed.""" |
| temp_file = android_commands.DeviceTempFile(self.ac) |
| self.assertFalse(self.ac.FileExistsOnDevice(temp_file.name)) |
| self.ac.SetFileContents(temp_file.name, "contents") |
| self.assertTrue(self.ac.FileExistsOnDevice(temp_file.name)) |
| temp_file.close() |
| self.assertFalse(self.ac.FileExistsOnDevice(temp_file.name)) |
| |
| with android_commands.DeviceTempFile(self.ac) as with_temp_file: |
| self.assertFalse(self.ac.FileExistsOnDevice(with_temp_file.name)) |
| self.ac.SetFileContents(with_temp_file.name, "contents") |
| self.assertTrue(self.ac.FileExistsOnDevice(with_temp_file.name)) |
| |
| self.assertFalse(self.ac.FileExistsOnDevice(with_temp_file.name)) |
| |
| def testTempFileNotWritten(self): |
| """Tests that device temp files work successfully even if not written to.""" |
| temp_file = android_commands.DeviceTempFile(self.ac) |
| temp_file.close() |
| self.assertFalse(self.ac.FileExistsOnDevice(temp_file.name)) |
| |
| with android_commands.DeviceTempFile(self.ac) as with_temp_file: |
| pass |
| self.assertFalse(self.ac.FileExistsOnDevice(with_temp_file.name)) |
| |
| def testNaming(self): |
| """Tests that returned filenames are as requested.""" |
| temp_file = android_commands.DeviceTempFile(self.ac, prefix="cat") |
| self.assertTrue(os.path.basename(temp_file.name).startswith("cat")) |
| |
| temp_file = android_commands.DeviceTempFile(self.ac, suffix="dog") |
| self.assertTrue(temp_file.name.endswith("dog")) |
| |
| temp_file = android_commands.DeviceTempFile( |
| self.ac, prefix="cat", suffix="dog") |
| self.assertTrue(os.path.basename(temp_file.name).startswith("cat")) |
| self.assertTrue(temp_file.name.endswith("dog")) |
| |
| |
| class TestGetFilesChanged(unittest.TestCase): |
| |
| def setUp(self): |
| if not os.getenv('BUILDTYPE'): |
| os.environ['BUILDTYPE'] = 'Debug' |
| |
| devices = android_commands.GetAttachedDevices() |
| self.assertGreater(len(devices), 0, 'No device attached!') |
| self.ac = android_commands.AndroidCommands(device=devices[0]) |
| self.host_data_dir = os.path.realpath('test_push_data') |
| self.device_data_dir = '%s/test_push_data' % ( |
| self.ac.RunShellCommand('realpath %s' % |
| self.ac.GetExternalStorage())[0]) |
| |
| os.mkdir(self.host_data_dir) |
| for i in xrange(1, 10): |
| with open('%s/%d.txt' % (self.host_data_dir, i), 'w') as f: |
| f.write('file #%d' % i) |
| |
| self.ac.RunShellCommand('mkdir %s' % self.device_data_dir) |
| |
| def testGetFilesChangedAllNeeded(self): |
| """ Tests GetFilesChanged when none of the files are on the device. |
| """ |
| expected = [('%s/%d.txt' % (self.host_data_dir, i), |
| '%s/%d.txt' % (self.device_data_dir, i)) |
| for i in xrange(1, 10)] |
| actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir) |
| self.assertSequenceEqual(expected, actual) |
| |
| def testGetFilesChangedSomeIdentical(self): |
| """ Tests GetFilesChanged when some of the files are on the device. |
| """ |
| for i in xrange(1, 5): |
| self.ac._adb.Push('%s/%d.txt' % (self.host_data_dir, i), |
| self.device_data_dir) |
| expected = [('%s/%d.txt' % (self.host_data_dir, i), |
| '%s/%d.txt' % (self.device_data_dir, i)) |
| for i in xrange(5, 10)] |
| actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir) |
| self.assertSequenceEqual(expected, actual) |
| |
| def testGetFilesChangedAllIdentical(self): |
| """ Tests GetFilesChanged when all of the files are on the device. |
| """ |
| for i in xrange(1, 10): |
| self.ac._adb.Push('%s/%d.txt' % (self.host_data_dir, i), |
| self.device_data_dir) |
| expected = [] |
| actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir) |
| self.assertSequenceEqual(expected, actual) |
| |
| def testGetFilesChangedRename(self): |
| """ Tests GetFilesChanged when one of the files has been renamed. |
| |
| This tests both with and without the ignore_filenames flag set. |
| """ |
| for i in xrange(5, 10): |
| self.ac._adb.Push('%s/%d.txt' % (self.host_data_dir, i), |
| self.device_data_dir) |
| os.rename('%s/5.txt' % (self.host_data_dir), |
| '%s/99.txt' % (self.host_data_dir)) |
| |
| expected = [('%s/%d.txt' % (self.host_data_dir, i), |
| '%s/%d.txt' % (self.device_data_dir, i)) |
| for i in xrange(1, 5)] |
| actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir, |
| ignore_filenames=True) |
| self.assertSequenceEqual(expected, actual) |
| |
| expected.append(('%s/99.txt' % self.host_data_dir, |
| '%s/99.txt' % self.device_data_dir)) |
| actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir) |
| self.assertSequenceEqual(expected, actual) |
| |
| def testGetFilesChangedCopy(self): |
| """ Tests GetFilesChanged when one of the files has been copied. |
| |
| This tests both with and without the ignore_filenames flag set. |
| """ |
| for i in xrange(5, 10): |
| self.ac._adb.Push('%s/%d.txt' % (self.host_data_dir, i), |
| self.device_data_dir) |
| shutil.copy('%s/5.txt' % self.host_data_dir, |
| '%s/99.txt' % self.host_data_dir) |
| |
| expected = [('%s/%d.txt' % (self.host_data_dir, i), |
| '%s/%d.txt' % (self.device_data_dir, i)) |
| for i in xrange(1, 5)] |
| actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir, |
| ignore_filenames=True) |
| self.assertSequenceEqual(expected, actual) |
| |
| expected.append(('%s/99.txt' % self.host_data_dir, |
| '%s/99.txt' % self.device_data_dir)) |
| actual = self.ac.GetFilesChanged(self.host_data_dir, self.device_data_dir) |
| self.assertSequenceEqual(expected, actual) |
| |
| def testGetFilesChangedIndividual(self): |
| """ Tests GetFilesChanged when provided one file. |
| """ |
| expected = [('%s/1.txt' % self.host_data_dir, |
| '%s/1.txt' % self.device_data_dir)] |
| actual = self.ac.GetFilesChanged('%s/1.txt' % self.host_data_dir, |
| '%s/1.txt' % self.device_data_dir) |
| self.assertSequenceEqual(expected, actual) |
| |
| def testGetFilesChangedFileToDirectory(self): |
| """ Tests GetFilesChanged when provided a file from the host and a |
| directory on the device. |
| """ |
| expected = [('%s/1.txt' % self.host_data_dir, |
| '%s' % self.device_data_dir)] |
| actual = self.ac.GetFilesChanged('%s/1.txt' % self.host_data_dir, |
| '%s' % self.device_data_dir) |
| self.assertSequenceEqual(expected, actual) |
| |
| def tearDown(self): |
| try: |
| shutil.rmtree(self.host_data_dir) |
| self.ac.RunShellCommand('rm -rf %s' % self.device_data_dir) |
| except: |
| pass |
| |
| if __name__ == '__main__': |
| unittest.main() |
| |
