aboutsummaryrefslogtreecommitdiff
path: root/catapult/devil/devil/android/md5sum_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'catapult/devil/devil/android/md5sum_test.py')
-rwxr-xr-xcatapult/devil/devil/android/md5sum_test.py126
1 files changed, 48 insertions, 78 deletions
diff --git a/catapult/devil/devil/android/md5sum_test.py b/catapult/devil/devil/android/md5sum_test.py
index c9b49545..548b2d02 100755
--- a/catapult/devil/devil/android/md5sum_test.py
+++ b/catapult/devil/devil/android/md5sum_test.py
@@ -19,17 +19,16 @@ MD5_DIST = os.path.join(TEST_OUT_DIR, 'md5sum_dist')
class Md5SumTest(unittest.TestCase):
-
def setUp(self):
mocked_attrs = {
- 'md5sum_host': HOST_MD5_EXECUTABLE,
- 'md5sum_device': MD5_DIST,
+ 'md5sum_host': HOST_MD5_EXECUTABLE,
+ 'md5sum_device': MD5_DIST,
}
self._patchers = [
- mock.patch('devil.devil_env._Environment.FetchPath',
- mock.Mock(side_effect=lambda a, device=None: mocked_attrs[a])),
- mock.patch('os.path.exists',
- new=mock.Mock(return_value=True)),
+ mock.patch(
+ 'devil.devil_env._Environment.FetchPath',
+ mock.Mock(side_effect=lambda a, device=None: mocked_attrs[a])),
+ mock.patch('os.path.exists', new=mock.Mock(return_value=True)),
]
for p in self._patchers:
p.start()
@@ -41,53 +40,30 @@ class Md5SumTest(unittest.TestCase):
def testCalculateHostMd5Sums_singlePath(self):
test_path = '/test/host/file.dat'
mock_get_cmd_output = mock.Mock(
- return_value='0123456789abcdeffedcba9876543210 /test/host/file.dat')
- with mock.patch('devil.utils.cmd_helper.GetCmdOutput',
- new=mock_get_cmd_output):
+ return_value='0123456789abcdef')
+ with mock.patch(
+ 'devil.utils.cmd_helper.GetCmdOutput', new=mock_get_cmd_output):
out = md5sum.CalculateHostMd5Sums(test_path)
self.assertEquals(1, len(out))
self.assertTrue('/test/host/file.dat' in out)
- self.assertEquals('0123456789abcdeffedcba9876543210',
- out['/test/host/file.dat'])
+ self.assertEquals('0123456789abcdef', out['/test/host/file.dat'])
mock_get_cmd_output.assert_called_once_with(
- [HOST_MD5_EXECUTABLE, '/test/host/file.dat'])
+ [HOST_MD5_EXECUTABLE, "-gz", mock.ANY])
def testCalculateHostMd5Sums_list(self):
test_paths = ['/test/host/file0.dat', '/test/host/file1.dat']
mock_get_cmd_output = mock.Mock(
- return_value='0123456789abcdeffedcba9876543210 /test/host/file0.dat\n'
- '123456789abcdef00fedcba987654321 /test/host/file1.dat\n')
- with mock.patch('devil.utils.cmd_helper.GetCmdOutput',
- new=mock_get_cmd_output):
+ return_value='0123456789abcdef\n123456789abcdef0\n')
+ with mock.patch(
+ 'devil.utils.cmd_helper.GetCmdOutput', new=mock_get_cmd_output):
out = md5sum.CalculateHostMd5Sums(test_paths)
self.assertEquals(2, len(out))
self.assertTrue('/test/host/file0.dat' in out)
- self.assertEquals('0123456789abcdeffedcba9876543210',
- out['/test/host/file0.dat'])
+ self.assertEquals('0123456789abcdef', out['/test/host/file0.dat'])
self.assertTrue('/test/host/file1.dat' in out)
- self.assertEquals('123456789abcdef00fedcba987654321',
- out['/test/host/file1.dat'])
+ self.assertEquals('123456789abcdef0', out['/test/host/file1.dat'])
mock_get_cmd_output.assert_called_once_with(
- [HOST_MD5_EXECUTABLE, '/test/host/file0.dat',
- '/test/host/file1.dat'])
-
- def testCalculateHostMd5Sums_generator(self):
- test_paths = ('/test/host/' + p for p in ['file0.dat', 'file1.dat'])
- mock_get_cmd_output = mock.Mock(
- return_value='0123456789abcdeffedcba9876543210 /test/host/file0.dat\n'
- '123456789abcdef00fedcba987654321 /test/host/file1.dat\n')
- with mock.patch('devil.utils.cmd_helper.GetCmdOutput',
- new=mock_get_cmd_output):
- out = md5sum.CalculateHostMd5Sums(test_paths)
- self.assertEquals(2, len(out))
- self.assertTrue('/test/host/file0.dat' in out)
- self.assertEquals('0123456789abcdeffedcba9876543210',
- out['/test/host/file0.dat'])
- self.assertTrue('/test/host/file1.dat' in out)
- self.assertEquals('123456789abcdef00fedcba987654321',
- out['/test/host/file1.dat'])
- mock_get_cmd_output.assert_called_once_with(
- [HOST_MD5_EXECUTABLE, '/test/host/file0.dat', '/test/host/file1.dat'])
+ [HOST_MD5_EXECUTABLE, "-gz", mock.ANY])
def testCalculateDeviceMd5Sums_noPaths(self):
device = mock.NonCallableMock()
@@ -100,29 +76,26 @@ class Md5SumTest(unittest.TestCase):
test_path = '/storage/emulated/legacy/test/file.dat'
device = mock.NonCallableMock()
- device_md5sum_output = [
- '0123456789abcdeffedcba9876543210 '
- '/storage/emulated/legacy/test/file.dat',
- ]
+ device_md5sum_output = ['0123456789abcdef',]
device.RunShellCommand = mock.Mock(return_value=device_md5sum_output)
with mock.patch('os.path.getsize', return_value=1337):
out = md5sum.CalculateDeviceMd5Sums(test_path, device)
self.assertEquals(1, len(out))
self.assertTrue('/storage/emulated/legacy/test/file.dat' in out)
- self.assertEquals('0123456789abcdeffedcba9876543210',
+ self.assertEquals('0123456789abcdef',
out['/storage/emulated/legacy/test/file.dat'])
self.assertEquals(1, len(device.RunShellCommand.call_args_list))
def testCalculateDeviceMd5Sums_list(self):
- test_path = ['/storage/emulated/legacy/test/file0.dat',
- '/storage/emulated/legacy/test/file1.dat']
+ test_path = [
+ '/storage/emulated/legacy/test/file0.dat',
+ '/storage/emulated/legacy/test/file1.dat'
+ ]
device = mock.NonCallableMock()
device_md5sum_output = [
- '0123456789abcdeffedcba9876543210 '
- '/storage/emulated/legacy/test/file0.dat',
- '123456789abcdef00fedcba987654321 '
- '/storage/emulated/legacy/test/file1.dat',
+ '0123456789abcdef',
+ '123456789abcdef0',
]
device.RunShellCommand = mock.Mock(return_value=device_md5sum_output)
@@ -130,10 +103,10 @@ class Md5SumTest(unittest.TestCase):
out = md5sum.CalculateDeviceMd5Sums(test_path, device)
self.assertEquals(2, len(out))
self.assertTrue('/storage/emulated/legacy/test/file0.dat' in out)
- self.assertEquals('0123456789abcdeffedcba9876543210',
+ self.assertEquals('0123456789abcdef',
out['/storage/emulated/legacy/test/file0.dat'])
self.assertTrue('/storage/emulated/legacy/test/file1.dat' in out)
- self.assertEquals('123456789abcdef00fedcba987654321',
+ self.assertEquals('123456789abcdef0',
out['/storage/emulated/legacy/test/file1.dat'])
self.assertEquals(1, len(device.RunShellCommand.call_args_list))
@@ -143,10 +116,8 @@ class Md5SumTest(unittest.TestCase):
device = mock.NonCallableMock()
device_md5sum_output = [
- '0123456789abcdeffedcba9876543210 '
- '/storage/emulated/legacy/test/file0.dat',
- '123456789abcdef00fedcba987654321 '
- '/storage/emulated/legacy/test/file1.dat',
+ '0123456789abcdef',
+ '123456789abcdef0',
]
device.RunShellCommand = mock.Mock(return_value=device_md5sum_output)
@@ -154,10 +125,10 @@ class Md5SumTest(unittest.TestCase):
out = md5sum.CalculateDeviceMd5Sums(test_path, device)
self.assertEquals(2, len(out))
self.assertTrue('/storage/emulated/legacy/test/file0.dat' in out)
- self.assertEquals('0123456789abcdeffedcba9876543210',
+ self.assertEquals('0123456789abcdef',
out['/storage/emulated/legacy/test/file0.dat'])
self.assertTrue('/storage/emulated/legacy/test/file1.dat' in out)
- self.assertEquals('123456789abcdef00fedcba987654321',
+ self.assertEquals('123456789abcdef0',
out['/storage/emulated/legacy/test/file1.dat'])
self.assertEquals(1, len(device.RunShellCommand.call_args_list))
@@ -168,10 +139,9 @@ class Md5SumTest(unittest.TestCase):
device = mock.NonCallableMock()
device_md5sum_output = [
'WARNING: linker: /data/local/tmp/md5sum/md5sum_bin: '
- 'unused DT entry: type 0x1d arg 0x15db',
+ 'unused DT entry: type 0x1d arg 0x15db',
'THIS_IS_NOT_A_VALID_CHECKSUM_ZZZ some random text',
- '0123456789abcdeffedcba9876543210 '
- '/storage/emulated/legacy/test/file.dat',
+ '0123456789abcdef',
]
device.RunShellCommand = mock.Mock(return_value=device_md5sum_output)
@@ -179,17 +149,18 @@ class Md5SumTest(unittest.TestCase):
out = md5sum.CalculateDeviceMd5Sums(test_path, device)
self.assertEquals(1, len(out))
self.assertTrue('/storage/emulated/legacy/test/file.dat' in out)
- self.assertEquals('0123456789abcdeffedcba9876543210',
+ self.assertEquals('0123456789abcdef',
out['/storage/emulated/legacy/test/file.dat'])
self.assertEquals(1, len(device.RunShellCommand.call_args_list))
def testCalculateDeviceMd5Sums_list_fileMissing(self):
- test_path = ['/storage/emulated/legacy/test/file0.dat',
- '/storage/emulated/legacy/test/file1.dat']
+ test_path = [
+ '/storage/emulated/legacy/test/file0.dat',
+ '/storage/emulated/legacy/test/file1.dat'
+ ]
device = mock.NonCallableMock()
device_md5sum_output = [
- '0123456789abcdeffedcba9876543210 '
- '/storage/emulated/legacy/test/file0.dat',
+ '0123456789abcdef',
'[0819/203513:ERROR:md5sum.cc(25)] Could not open file asdf',
'',
]
@@ -199,7 +170,7 @@ class Md5SumTest(unittest.TestCase):
out = md5sum.CalculateDeviceMd5Sums(test_path, device)
self.assertEquals(1, len(out))
self.assertTrue('/storage/emulated/legacy/test/file0.dat' in out)
- self.assertEquals('0123456789abcdeffedcba9876543210',
+ self.assertEquals('0123456789abcdef',
out['/storage/emulated/legacy/test/file0.dat'])
self.assertEquals(1, len(device.RunShellCommand.call_args_list))
@@ -211,27 +182,26 @@ class Md5SumTest(unittest.TestCase):
device.adb.Push = mock.Mock()
device_md5sum_output = [
'WARNING: linker: /data/local/tmp/md5sum/md5sum_bin: '
- 'unused DT entry: type 0x1d arg 0x15db',
+ 'unused DT entry: type 0x1d arg 0x15db',
'THIS_IS_NOT_A_VALID_CHECKSUM_ZZZ some random text',
- '0123456789abcdeffedcba9876543210 '
- '/storage/emulated/legacy/test/file.dat',
+ '0123456789abcdef',
]
error = device_errors.AdbShellCommandFailedError('cmd', 'out', 2)
device.RunShellCommand = mock.Mock(
side_effect=(error, '', device_md5sum_output))
- with mock.patch('os.path.isdir', return_value=True), (
- mock.patch('os.path.getsize', return_value=1337)):
+ with mock.patch(
+ 'os.path.isdir', return_value=True), (mock.patch(
+ 'os.path.getsize', return_value=1337)):
out = md5sum.CalculateDeviceMd5Sums(test_path, device)
self.assertEquals(1, len(out))
self.assertTrue('/storage/emulated/legacy/test/file.dat' in out)
- self.assertEquals('0123456789abcdeffedcba9876543210',
+ self.assertEquals('0123456789abcdef',
out['/storage/emulated/legacy/test/file.dat'])
self.assertEquals(3, len(device.RunShellCommand.call_args_list))
- device.adb.Push.assert_called_once_with(
- 'test/out/directory/md5sum_dist', '/data/local/tmp/md5sum')
+ device.adb.Push.assert_called_once_with('test/out/directory/md5sum_dist',
+ '/data/local/tmp/md5sum')
if __name__ == '__main__':
unittest.main(verbosity=2)
-
generated by cgit v1.2.3 (git 2.25.1) at 2024-05-23 06:43:41 +0000