def test_read_container_extract(self, mock_copy, mock_ip, mock_storage_obj, mock_extract): disk_storage_backend = DiskStorageBackend() full_path_to_src = "the/full/path/to/src" container_path = "container/path" mock_storage_obj.container = True mock_storage_obj.get_full_path.return_value = full_path_to_src mock_storage_obj.storage_medium.storage_target.target = container_path mock_storage_obj.ip = mock_ip mock_ip.aic.pk = 1234 disk_storage_backend.read(storage_object=mock_storage_obj, dst="some_dest", extract=True) expected_copy_calls = [ mock.call(f"{full_path_to_src}.xml", "some_dest", block_size=65536), mock.call(os.path.join(container_path, "1234.xml"), "some_dest", block_size=65536), ] self.assertEqual(mock_copy.call_count, 2) mock_copy.assert_has_calls(expected_copy_calls) mock_extract.assert_called_once_with(mock_storage_obj, "some_dest")
def test_read_container_default(self, mock_copy, mock_ip, mock_storage_obj): disk_storage_backend = DiskStorageBackend() full_path_to_src = "the/full/path/to/src" container_path = "container/path" mock_storage_obj.container = True mock_storage_obj.get_full_path.return_value = full_path_to_src mock_storage_obj.storage_medium.storage_target.target = container_path mock_storage_obj.ip = mock_ip mock_ip.aic.pk = 1234 disk_storage_backend.read(storage_object=mock_storage_obj, dst="some_dest") expected_copy_calls = [ mock.call(f"{full_path_to_src}.xml", "some_dest", block_size=DEFAULT_BLOCK_SIZE), mock.call(os.path.join(container_path, "1234.xml"), "some_dest", block_size=DEFAULT_BLOCK_SIZE), mock.call(f"{full_path_to_src}", "some_dest", block_size=DEFAULT_BLOCK_SIZE) ] self.assertEqual(mock_copy.call_count, 3) mock_copy.assert_has_calls(expected_copy_calls)
def test_read_not_container_should_call_copy(self, mock_copy, mock_storage_obj): disk_storage_backend = DiskStorageBackend() mock_storage_obj.container = False mock_storage_obj.get_full_path.return_value = "the_full_path" disk_storage_backend.read(storage_object=mock_storage_obj, dst="some_dest") mock_copy.assert_called_once_with("the_full_path", "some_dest", block_size=65536)
def test_read_container_no_xml(self, mock_copy, mock_ip, mock_storage_obj): disk_storage_backend = DiskStorageBackend() full_path_to_src = "the/full/path/to/src" container_path = "container/path" mock_storage_obj.container = True mock_storage_obj.get_full_path.return_value = full_path_to_src mock_storage_obj.storage_medium.storage_target.target = container_path mock_storage_obj.ip = mock_ip mock_ip.aic.pk = 1234 disk_storage_backend.read(storage_object=mock_storage_obj, dst="some_dest", include_xml=False) mock_copy.assert_called_once_with(f"{full_path_to_src}", "some_dest", block_size=65536)