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()