def test_calculate_checksum_updates_hasher_until_empty(self):
file_handle = self.mock_open.return_value.__enter__.return_value
file_handle.read.side_effect = ['1', '2', '3', '']
file_hash = self.hlib.md5.return_value
calculate_checksum('/some/path/file.txt')
self.assertEqual(map(call, ['1', '2', '3']), file_hash.update.mock_calls)
def test_calculate_checksum_uses__calculate_checksum_from_remote_namespace_if_conn_specified(self):
conn = MagicMock()
mock_calc = Mock()
conn.namespace = {'_calculate_checksum': mock_calc}
calculate_checksum('/some/path/file.txt', conn)
mock_calc.assert_called_once_with('/some/path/file.txt')
def test_calculate_checksum_reads_bytes_until_empty(self):
file_handle = self.mock_open.return_value.__enter__.return_value
file_handle.read.side_effect = ['1', '2', '3', '']
file_hash = self.hlib.md5.return_value
calculate_checksum('/some/path/file.txt')
self.assertEqual(4, len(file_handle.read.mock_calls))
def test_calculate_checksum_returns_hexdigest(self):
file_hash = self.hlib.md5.return_value
ret = calculate_checksum('/some/path/file.txt')
file_hash.hexdigest.assert_called_once_with()
self.assertEqual(ret, file_hash.hexdigest.return_value)
def file_in_buffer(self, filename, checksum):
dst = os.path.join(MESSOR_BUFFER, checksum)
return os.path.isfile(dst) and calculate_checksum(dst) == checksum
def calculate_checksum(self, filename): return calculate_checksum(filename, self.rpyc_conn)
def test_calculate_checksum_defines_calculate_checksum_function_in_remote_namespace_if_conn_specified(self):
conn = MagicMock()
calculate_checksum('/some/path/file.txt', conn)
self.assertEqual(2, len(conn.execute.mock_calls))
def test_calculate_checksum_opens_path_rb(self):
calculate_checksum('/some/path/file.txt')
self.mock_open.assert_called_once_with('/some/path/file.txt', 'rb')
def test_calculate_checksum_instantiates_hasher(self):
calculate_checksum('/some/path/file.txt')
self.hlib.md5.assert_called_once_with()
