/******************************************************************************* * Copyright French Prime minister Office/SGMAP/DINSIC/Vitam Program (2015-2019) * * contact.vitam@culture.gouv.fr * * This software is a computer program whose purpose is to implement a digital archiving back-office system managing * high volumetry securely and efficiently. * * This software is governed by the CeCILL 2.1 license under French law and abiding by the rules of distribution of free * software. You can use, modify and/ or redistribute the software under the terms of the CeCILL 2.1 license as * circulated by CEA, CNRS and INRIA at the following URL "http://www.cecill.info". * * As a counterpart to the access to the source code and rights to copy, modify and redistribute granted by the license, * users are provided only with a limited warranty and the software's author, the holder of the economic rights, and the * successive licensors have only limited liability. * * In this respect, the user's attention is drawn to the risks associated with loading, using, modifying and/or * developing or reproducing the software by the user in light of its specific status of free software, that may mean * that it is complicated to manipulate, and that also therefore means that it is reserved for developers and * experienced professionals having in-depth computer knowledge. Users are therefore encouraged to load and test the * software's suitability as regards their requirements in conditions enabling the security of their systems and/or data * to be ensured and, more generally, to use and operate it in the same conditions as regards security. * * The fact that you are presently reading this means that you have had knowledge of the CeCILL 2.1 license and that you * accept its terms. *******************************************************************************/ package fr.gouv.vitam.access.internal.core; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNotNull; import static org.junit.Assert.assertTrue; import static org.mockito.Matchers.anyObject; import static org.mockito.Matchers.anyString; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; import java.io.ByteArrayInputStream; import java.io.InputStream; import javax.ws.rs.ProcessingException; import javax.ws.rs.core.Response; import org.apache.commons.io.IOUtils; import org.junit.AfterClass; import org.junit.Before; import org.junit.BeforeClass; import org.junit.Rule; import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.ArgumentCaptor; import org.mockito.Mockito; import org.powermock.api.mockito.PowerMockito; import org.powermock.core.classloader.annotations.PowerMockIgnore; import org.powermock.core.classloader.annotations.PrepareForTest; import org.powermock.modules.junit4.PowerMockRunner; import com.fasterxml.jackson.databind.JsonNode; import fr.gouv.vitam.access.internal.api.AccessBinaryData; import fr.gouv.vitam.access.internal.common.exception.AccessInternalExecutionException; import fr.gouv.vitam.common.PropertiesUtils; import fr.gouv.vitam.common.database.builder.query.action.SetAction; import fr.gouv.vitam.common.database.builder.request.multiple.Update; import fr.gouv.vitam.common.exception.InvalidParseOperationException; import fr.gouv.vitam.common.json.JsonHandler; import fr.gouv.vitam.common.junit.JunitHelper; import fr.gouv.vitam.common.server.application.junit.AsyncResponseJunitTest; import fr.gouv.vitam.common.thread.RunWithCustomExecutor; import fr.gouv.vitam.common.thread.RunWithCustomExecutorRule; import fr.gouv.vitam.common.thread.VitamThreadPoolExecutor; import fr.gouv.vitam.common.thread.VitamThreadUtils; import fr.gouv.vitam.logbook.common.exception.LogbookClientNotFoundException; import fr.gouv.vitam.logbook.common.parameters.LogbookLifeCycleUnitParameters; import fr.gouv.vitam.logbook.common.parameters.LogbookParameterName; import fr.gouv.vitam.logbook.lifecycles.client.LogbookLifeCyclesClient; import fr.gouv.vitam.logbook.lifecycles.client.LogbookLifeCyclesClientFactory; import fr.gouv.vitam.logbook.operations.client.LogbookOperationsClient; import fr.gouv.vitam.logbook.operations.client.LogbookOperationsClientFactory; import fr.gouv.vitam.metadata.api.exception.MetaDataDocumentSizeException; import fr.gouv.vitam.metadata.api.exception.MetaDataExecutionException; import fr.gouv.vitam.metadata.client.MetaDataClient; import fr.gouv.vitam.metadata.client.MetaDataClientFactory; import fr.gouv.vitam.metadata.client.MetaDataClientRest; import fr.gouv.vitam.storage.engine.client.StorageClient; import fr.gouv.vitam.storage.engine.client.StorageClientFactory; import fr.gouv.vitam.storage.engine.client.exception.StorageServerClientException; import fr.gouv.vitam.workspace.client.WorkspaceClient; import fr.gouv.vitam.workspace.client.WorkspaceClientFactory; @RunWith(PowerMockRunner.class) @PowerMockIgnore("javax.net.ssl.*") @PrepareForTest({MetaDataClientFactory.class, LogbookOperationsClientFactory.class, LogbookLifeCyclesClientFactory.class, StorageClientFactory.class, WorkspaceClientFactory.class}) public class AccessInternalModuleImplTest { @Rule public RunWithCustomExecutorRule runInThread = new RunWithCustomExecutorRule(VitamThreadPoolExecutor.getDefaultExecutor()); private static String HOST = "http:\\localhost:"; private AccessInternalModuleImpl accessModuleImpl; private MetaDataClient metaDataClient; private WorkspaceClient workspaceClient; private LogbookOperationsClient logbookOperationClient; private LogbookLifeCyclesClient logbookLifeCycleClient; private StorageClient storageClient; private static AsyncResponseJunitTest asynResponse = new AsyncResponseJunitTest(); private static JunitHelper junitHelper; private static int serverPort; private static final String SAMPLE_OBJECTGROUP_FILENAME = "sample_objectGroup_document.json"; private static JsonNode sampleObjectGroup; private static final String QUERY = "{\"$queries\": [{ \"$path\": \"aaaaa\" }],\"$filter\": { },\"$projection\": {}}"; private static final String QUERY_UPDATE = "{\"$root\": { },\"$queries\": [{ \"$path\": \"aaaaa\" }],\"$filter\": { },\"$action\": {}}"; // Note: this is an incorrect response since missing Qualifiers but _id is correct but within private static final String FAKE_METADATA_RESULT = "{$results:[{'_id':123}]}"; private static final String FAKE_METADATA_MULTIPLE_RESULT = "{$results:[{'_id':123}, {'_id':124}]}"; private static final String REAL_DATA_RESULT_PATH = "sample_data_results.json"; private static final String REAL_DATA_RESULT_MULTI_PATH = "sample_data_multi_results.json"; private static final Update updateQuery = new Update(); private static final Integer TENANT_ID = 0; @BeforeClass public static void setUpBeforeClass() throws Exception { junitHelper = JunitHelper.getInstance(); serverPort = junitHelper.findAvailablePort(); HOST += serverPort; sampleObjectGroup = JsonHandler.getFromFile(PropertiesUtils.findFile(SAMPLE_OBJECTGROUP_FILENAME)); updateQuery.addActions(new SetAction("name", "test")); } @AfterClass public static void tearDownAfterClass() { junitHelper.releasePort(serverPort); } private static final String ID = "aeaqaaaaaitxll67abarqaktftcfyniaaaaq"; private static final String REQUEST_ID = "aeaqaaaaaitxll67abarqaktftcfyniaaaaq"; /** * @param query * @return * @throws InvalidParseOperationException */ public JsonNode fromStringToJson(String query) throws InvalidParseOperationException { return JsonHandler.getFromString(query); } @Before public void setUp() { final MetaDataClientFactory mockedMetadataFactory = mock(MetaDataClientFactory.class); PowerMockito.mockStatic(MetaDataClientFactory.class); metaDataClient = mock(MetaDataClientRest.class); PowerMockito.when(MetaDataClientFactory.getInstance()).thenReturn(mockedMetadataFactory); PowerMockito.when(mockedMetadataFactory.getClient()).thenReturn(metaDataClient); logbookLifeCycleClient = mock(LogbookLifeCyclesClient.class); logbookOperationClient = mock(LogbookOperationsClient.class); final LogbookLifeCyclesClientFactory factorylc = mock(LogbookLifeCyclesClientFactory.class); final LogbookOperationsClientFactory factoryop = mock(LogbookOperationsClientFactory.class); PowerMockito.mockStatic(LogbookLifeCyclesClientFactory.class); PowerMockito.when(LogbookLifeCyclesClientFactory.getInstance()).thenReturn(factorylc); PowerMockito.when(factorylc.getClient()).thenReturn(logbookLifeCycleClient); PowerMockito.mockStatic(LogbookOperationsClientFactory.class); PowerMockito.when(LogbookOperationsClientFactory.getInstance()).thenReturn(factoryop); PowerMockito.when(factoryop.getClient()).thenReturn(logbookOperationClient); storageClient = mock(StorageClient.class); final StorageClientFactory factoryst = mock(StorageClientFactory.class); PowerMockito.mockStatic(StorageClientFactory.class); PowerMockito.when(StorageClientFactory.getInstance()).thenReturn(factoryst); PowerMockito.when(factoryst.getClient()).thenReturn(storageClient); final WorkspaceClientFactory workspaceClientFactory = mock(WorkspaceClientFactory.class); PowerMockito.mockStatic(WorkspaceClientFactory.class); workspaceClient = mock(WorkspaceClient.class); PowerMockito.when(WorkspaceClientFactory.getInstance()).thenReturn(workspaceClientFactory); PowerMockito.when(workspaceClientFactory.getClient()).thenReturn(workspaceClient); accessModuleImpl = new AccessInternalModuleImpl(storageClient, logbookOperationClient, logbookLifeCycleClient, workspaceClient); } @Test public void given_correct_dsl_When_select_thenOK() throws Exception { when(metaDataClient.selectUnits(anyObject())).thenReturn(JsonHandler.createObjectNode()); accessModuleImpl.selectUnit(fromStringToJson(QUERY)); } @Test(expected = InvalidParseOperationException.class) public void given_empty_dsl_When_select_thenTrows_IllegalArgumentException() throws Exception { accessModuleImpl.selectUnit(fromStringToJson("")); } @Test(expected = IllegalArgumentException.class) public void given_test_AccessExecutionException() throws Exception { Mockito.doThrow(new IllegalArgumentException("")).when(metaDataClient).selectUnits(anyObject()); accessModuleImpl.selectUnit(fromStringToJson(QUERY)); } @Test(expected = InvalidParseOperationException.class) public void given_empty_DSLWhen_select_units_ThenThrows_InvalidParseOperationException() throws Exception { PowerMockito.doThrow(new InvalidParseOperationException("")).when(metaDataClient).selectUnits(anyObject()); accessModuleImpl.selectUnit(fromStringToJson(QUERY)); } @Test(expected = IllegalArgumentException.class) public void given__DSLWhen_select_units_ThenThrows_MetadataInvalidSelectException() throws Exception { Mockito.doThrow(new IllegalArgumentException("")).when(metaDataClient).selectUnits(anyObject()); accessModuleImpl.selectUnit(fromStringToJson(QUERY)); } @Test(expected = AccessInternalExecutionException.class) public void given_DSLWhen_select_units_ThenThrows_MetaDataDocumentSizeException() throws Exception { Mockito.doThrow(new MetaDataDocumentSizeException("")).when(metaDataClient).selectUnits(anyObject()); accessModuleImpl.selectUnit(fromStringToJson(QUERY)); } @Test(expected = AccessInternalExecutionException.class) public void given_clientProblem_When_select_units_ThenThrows_AccessExecutionException() throws Exception { Mockito.doThrow(new ProcessingException("")).when(metaDataClient).selectUnits(anyObject()); accessModuleImpl.selectUnit(fromStringToJson(QUERY)); } // select Unit Id @Test public void given_correct_dsl_When_selectunitById_thenOK() throws Exception { when(metaDataClient.selectUnitbyId(anyObject(), anyObject())).thenReturn(JsonHandler.createObjectNode()); accessModuleImpl.selectUnitbyId(fromStringToJson(QUERY), ID); } @Test(expected = InvalidParseOperationException.class) public void given_empty_dsl_When_selectunitById_thenTrows_IllegalArgumentException() throws Exception { accessModuleImpl.selectUnitbyId(fromStringToJson(""), ID); } @Test(expected = IllegalArgumentException.class) public void given_test_AccessExecutionException_unitById() throws Exception { Mockito.doThrow(new IllegalArgumentException("")).when(metaDataClient).selectUnitbyId(anyObject(), anyObject()); accessModuleImpl.selectUnitbyId(fromStringToJson(QUERY), ID); } @Test(expected = InvalidParseOperationException.class) public void given_empty_DSLWhen_select_unitById_ThenThrows_InvalidParseOperationException() throws Exception { Mockito.doThrow(new InvalidParseOperationException("")).when(metaDataClient).selectUnitbyId(anyObject(), anyObject()); accessModuleImpl.selectUnitbyId(fromStringToJson(QUERY), ID); } @Test(expected = AccessInternalExecutionException.class) public void given__DSLWhen_select_unitById_ThenThrows_MetadataInvalidSelectException() throws Exception { Mockito.doThrow(new MetaDataDocumentSizeException("")).when(metaDataClient).selectUnitbyId(anyObject(), anyObject()); accessModuleImpl.selectUnitbyId(fromStringToJson(QUERY), ID); } @Test(expected = IllegalArgumentException.class) public void given_emptyOrNullIdUnit_when_selectUnitbyId_thenthrows_IllegalArgumentException() throws Exception { Mockito.doThrow(new IllegalArgumentException("")).when(metaDataClient) .selectUnitbyId(fromStringToJson(QUERY), ""); accessModuleImpl.selectUnitbyId(fromStringToJson(QUERY), ""); } @Test(expected = InvalidParseOperationException.class) public void given_empty_DSLWhen_selectUnitById_ThenThrows_InvalidParseOperationException() throws Exception { Mockito.doThrow(new InvalidParseOperationException("")).when(metaDataClient) .selectUnitbyId(fromStringToJson(QUERY), ID); accessModuleImpl.selectUnitbyId(fromStringToJson(QUERY), ID); } @Test(expected = InvalidParseOperationException.class) public void given_empty_DSLWhen_selectUnit_ThenThrows_InvalidParseOperationException() throws Exception { final JsonNode jsonQuery = JsonHandler.getFromString(QUERY); Mockito.doThrow(new InvalidParseOperationException("")).when(metaDataClient) .selectUnits(jsonQuery); accessModuleImpl.selectUnit(fromStringToJson(QUERY)); } @Test(expected = IllegalArgumentException.class) public void given_emptyOrNullIdUnit_when_selectOGbyId_thenthrows_IllegalArgumentException() throws Exception { Mockito.doThrow(new IllegalArgumentException("")).when(metaDataClient) .selectObjectGrouptbyId(fromStringToJson(QUERY), ""); accessModuleImpl.selectObjectGroupById(fromStringToJson(QUERY), ""); } @Test(expected = AccessInternalExecutionException.class) public void given_metadataAccessProblem_throw_AccessExecutionException() throws Exception { Mockito.doThrow(new ProcessingException("Fake error")).when(metaDataClient) .selectObjectGrouptbyId(fromStringToJson(QUERY), "ds"); accessModuleImpl.selectObjectGroupById(fromStringToJson(QUERY), "ds"); } @Test public void given_selectObjectGroupById_OK() throws Exception { when(metaDataClient.selectObjectGrouptbyId(fromStringToJson(QUERY), ID)) .thenReturn(sampleObjectGroup); final JsonNode result = accessModuleImpl.selectObjectGroupById(fromStringToJson(QUERY), ID); assertEquals(sampleObjectGroup, result); } @Test(expected = InvalidParseOperationException.class) public void given_empty_DSLWhen_selectOGById_ThenThrows_InvalidParseOperationException() throws Exception { Mockito.doThrow(new InvalidParseOperationException("")).when(metaDataClient) .selectObjectGrouptbyId(fromStringToJson(QUERY), ID); accessModuleImpl.selectObjectGroupById(fromStringToJson(QUERY), ID); } // update by id - start @Test @RunWithCustomExecutor public void given_correct_dsl_When_updateUnitById_thenOK() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); final ArgumentCaptor<LogbookLifeCycleUnitParameters> logbookLFCUnitParametersArgsCaptor = ArgumentCaptor.forClass(LogbookLifeCycleUnitParameters.class); Mockito.doNothing().when(logbookOperationClient).update(anyObject()); Mockito.doNothing().when(logbookLifeCycleClient).update(logbookLFCUnitParametersArgsCaptor.capture()); final String id = "aeaqaaaaaaaaaaabaasdaakxocodoiyaaaaq"; // Mock select unit response when(metaDataClient.selectUnitbyId(anyObject(), anyObject())).thenReturn(JsonHandler.getFromString("{\"$hits" + "\":{\"total\":1,\"size\":1,\"limit\":1,\"time_out\":false},\"$context\":{}," + "\"$results\":[{\"#id\":\"aeaqaaaaaaaaaaabaasdaakxocodoiyaaaaq\",\"Title\":\"MyTitle\"," + "\"Description\":\"Ma description est bien détaillée\",\"CreatedDate\":\"2016-09-28T11:44:28.548\"," + "\"MyInt\":20,\"MyBoolean\":false,\"MyFloat\":2.0,\"ArrayVar\":[\"val1\",\"val2\"]," + "\"Array2Var\":[\"val1\",\"val2\"],\"#tenant\":0,\"#max\":1,\"#min\":1,\"#unitups\":[],\"#nbunits\":0}]}")); // Mock update unit response when(metaDataClient.updateUnitbyId(anyObject(), anyObject())).thenReturn(JsonHandler.getFromString("{\"$hits" + "\":{\"total\":1,\"size\":1,\"limit\":1,\"time_out\":false},\"$context\":{}," + "\"$results\":[{\"#id\":\"aeaqaaaaaaaaaaabaasdaakxocodoiyaaaaq\",\"#diff\":\"- \\\"Title\\\" : " + "\\\"MyTitle\\\",\\n+ \\\"Title\\\" : \\\"Modified title\\\",\\n- \\\"MyBoolean\\\" : false,\\n+ " + " \\\"MyBoolean\\\" : true,\"}]}")); accessModuleImpl.updateUnitbyId(new Update().getFinalUpdate(), id, REQUEST_ID); // check if diff for update sent to lfc is correct final LogbookLifeCycleUnitParameters capture = logbookLFCUnitParametersArgsCaptor.getValue(); assertNotNull(capture.getParameterValue(LogbookParameterName.eventDetailData)); JsonNode lfcParams = JsonHandler.getFromString(capture.getParameterValue(LogbookParameterName.eventDetailData)); assertEquals( "- \"Title\" : \"MyTitle\",\n+ \"Title\" : \"Modified title\",\n- \"MyBoolean\" : false,\n+ \"MyBoolean\" : true,", lfcParams.get("diff").textValue()); } @Test @RunWithCustomExecutor public void given_dsl_nodiff_When_updateUnitById_thenOK() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); final ArgumentCaptor<LogbookLifeCycleUnitParameters> logbookLFCUnitParametersArgsCaptor = ArgumentCaptor.forClass(LogbookLifeCycleUnitParameters.class); Mockito.doNothing().when(logbookOperationClient).update(anyObject()); Mockito.doNothing().when(logbookLifeCycleClient).update(logbookLFCUnitParametersArgsCaptor.capture()); final String id = "aeaqaaaaaaaaaaabaasdaakxocodoiyaaaaq"; // Mock select unit response when(metaDataClient.selectUnitbyId(anyObject(), anyObject())).thenReturn(JsonHandler.getFromString("{\"$hits" + "\":{\"total\":1,\"size\":1,\"limit\":1,\"time_out\":false},\"$context\":{}," + "\"$results\":[{\"#id\":\"aeaqaaaaaaaaaaabaasdaakxocodoiyaaaaq\",\"Title\":\"MyTitle\"," + "\"Description\":\"Ma description est bien détaillée\",\"CreatedDate\":\"2016-09-28T11:44:28.548\"," + "\"MyInt\":20,\"MyBoolean\":false,\"MyFloat\":2.0,\"ArrayVar\":[\"val1\",\"val2\"]," + "\"Array2Var\":[\"val1\",\"val2\"],\"#tenant\":0,\"#max\":1,\"#min\":1,\"#unitups\":[],\"#nbunits\":0}]}")); // Mock update unit response when(metaDataClient.updateUnitbyId(anyObject(), anyObject())).thenReturn(JsonHandler.getFromString("{\"$hits" + "\":{\"total\":1,\"size\":1,\"limit\":1,\"time_out\":false},\"$context\":{}," + "\"$results\":[{\"#id\":\"aeaqaaaaaaaaaaabaasdaakxocodoiyaaaaq\"}]}")); accessModuleImpl.updateUnitbyId(new Update().getFinalUpdate(), id, REQUEST_ID); // check if diff for update sent to lfc is null final LogbookLifeCycleUnitParameters capture = logbookLFCUnitParametersArgsCaptor.getValue(); assertNotNull(capture.getParameterValue(LogbookParameterName.eventDetailData)); JsonNode lfcParams = JsonHandler.getFromString(capture.getParameterValue(LogbookParameterName.eventDetailData)); assertTrue(lfcParams.get("diff").isNull()); } @Test(expected = InvalidParseOperationException.class) public void given_empty_dsl_When_updateUnitById_thenTrows_IllegalArgumentException() throws Exception { Mockito.doNothing().when(logbookOperationClient).update(anyObject()); Mockito.doNothing().when(logbookLifeCycleClient).update(anyObject()); accessModuleImpl.updateUnitbyId(fromStringToJson(""), ID, REQUEST_ID); } @Test(expected = IllegalArgumentException.class) @RunWithCustomExecutor public void given_test_AccessExecutionException_updateUnitById() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); Mockito.doNothing().when(logbookOperationClient).update(anyObject()); Mockito.doNothing().when(logbookLifeCycleClient).update(anyObject()); Mockito.doThrow(new IllegalArgumentException("")).when(metaDataClient) .updateUnitbyId(fromStringToJson(QUERY), ID); accessModuleImpl.updateUnitbyId(fromStringToJson(QUERY), ID, REQUEST_ID); } @Test(expected = InvalidParseOperationException.class) @RunWithCustomExecutor public void given_empty_DSLWhen_updateUnitById_ThenThrows_InvalidParseOperationException() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); Mockito.doNothing().when(logbookOperationClient).update(anyObject()); Mockito.doNothing().when(logbookLifeCycleClient).update(anyObject()); Mockito.doThrow(new InvalidParseOperationException("")).when(metaDataClient) .updateUnitbyId(anyObject(), anyObject()); accessModuleImpl.updateUnitbyId(fromStringToJson(QUERY_UPDATE), ID, REQUEST_ID); } @Test(expected = AccessInternalExecutionException.class) @RunWithCustomExecutor public void given_DSLWhen_updateUnitById_ThenThrows_MetaDataDocumentSizeException() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); Mockito.doNothing().when(logbookOperationClient).update(anyObject()); Mockito.doNothing().when(logbookLifeCycleClient).update(anyObject()); when(metaDataClient.updateUnitbyId(anyObject(), anyObject())).thenThrow(new MetaDataDocumentSizeException("")); accessModuleImpl.updateUnitbyId(updateQuery.getFinalUpdate(), ID, REQUEST_ID); } @Test(expected = AccessInternalExecutionException.class) @RunWithCustomExecutor public void given_DSL_When_updateUnitById_ThenThrows_MetaDataExecutionException() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); Mockito.doNothing().when(logbookOperationClient).update(anyObject()); Mockito.doNothing().when(logbookLifeCycleClient).update(anyObject()); when(metaDataClient.updateUnitbyId(anyObject(), anyObject())).thenReturn(JsonHandler.createObjectNode()); Mockito.doThrow(new MetaDataExecutionException("")).when(metaDataClient) .updateUnitbyId(anyObject(), anyObject()); accessModuleImpl.updateUnitbyId(updateQuery.getFinalUpdate(), ID, REQUEST_ID); } @Test(expected = AccessInternalExecutionException.class) @RunWithCustomExecutor public void given_LogbookProblem_When_updateUnitById_ThenThrows_AccessExecutionException() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); Mockito.doThrow(new LogbookClientNotFoundException("")).when(logbookOperationClient).update(anyObject()); Mockito.doNothing().when(logbookLifeCycleClient).update(anyObject()); when(metaDataClient.updateUnitbyId(anyObject(), anyObject())).thenReturn(JsonHandler.createObjectNode()); Mockito.doThrow(new MetaDataExecutionException("")).when(metaDataClient) .updateUnitbyId(fromStringToJson(QUERY), ID); accessModuleImpl.updateUnitbyId(updateQuery.getFinalUpdate(), ID, REQUEST_ID); } @Test(expected = IllegalArgumentException.class) public void given_null_params_conf_When_updateUnitById_ThenNotThrowAnyException() throws Exception { accessModuleImpl = new AccessInternalModuleImpl(null); } @Test(expected = IllegalArgumentException.class) public void given_emptyOrNullIdUnit_when_updateUnitbyId_thenthrows_IllegalArgumentException() throws Exception { Mockito.doNothing().when(logbookOperationClient).update(anyObject()); Mockito.doNothing().when(logbookLifeCycleClient).update(anyObject()); Mockito.doThrow(new IllegalArgumentException("")).when(metaDataClient) .updateUnitbyId(fromStringToJson(QUERY), ""); accessModuleImpl.updateUnitbyId(fromStringToJson(QUERY), "", REQUEST_ID); } @Test @RunWithCustomExecutor public void testGetOneObjectFromObjectGroup_OK() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); when(metaDataClient.selectObjectGrouptbyId(anyObject(), anyString())) .thenReturn(fromStringToJson(FAKE_METADATA_RESULT)); final Response responseMock = mock(Response.class); when(responseMock.readEntity(InputStream.class)) .thenReturn(new ByteArrayInputStream(FAKE_METADATA_RESULT.getBytes())); when(storageClient.getContainerAsync(anyString(), anyString(), anyObject())) .thenReturn(responseMock); final AccessBinaryData abd = accessModuleImpl.getOneObjectFromObjectGroup(asynResponse, ID, fromStringToJson(QUERY), "BinaryMaster", 0); assertNotNull(abd); final Response binaryMasterResponse = abd.getOriginalResponse(); assertNotNull(binaryMasterResponse); final InputStream binaryMaster = binaryMasterResponse.readEntity(InputStream.class); final InputStream stream2 = IOUtils.toInputStream(FAKE_METADATA_RESULT); assertTrue(IOUtils.contentEquals(binaryMaster, stream2)); } @Test @RunWithCustomExecutor public void testGetOneObjectFromObjectGroupRealData_OK() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); when(metaDataClient.selectObjectGrouptbyId(anyObject(), anyString())) .thenReturn(JsonHandler.getFromFile(PropertiesUtils.getResourceFile(REAL_DATA_RESULT_PATH))); final Response responseMock = mock(Response.class); when(responseMock.readEntity(InputStream.class)) .thenReturn(PropertiesUtils.getResourceAsStream(REAL_DATA_RESULT_PATH)); when(storageClient.getContainerAsync(anyString(), anyString(), anyObject())) .thenReturn(responseMock); final AccessBinaryData abd = accessModuleImpl.getOneObjectFromObjectGroup(asynResponse, ID, fromStringToJson(QUERY), "BinaryMaster", 0); assertNotNull(abd); final Response binaryMasterResponse = abd.getOriginalResponse(); assertNotNull(binaryMasterResponse); assertEquals("image/png", abd.getMimetype()); assertEquals("Vitam-S\u00E9nsibilisation de l' API-V1.0.png", abd.getFilename()); } @Test @RunWithCustomExecutor public void testGetOneObjectFromObjectGroupRealData_WARN() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); when(metaDataClient.selectObjectGrouptbyId(anyObject(), anyString())) .thenReturn(JsonHandler.getFromFile(PropertiesUtils.getResourceFile(REAL_DATA_RESULT_MULTI_PATH))); final Response responseMock = mock(Response.class); when(responseMock.readEntity(InputStream.class)) .thenReturn(PropertiesUtils.getResourceAsStream(REAL_DATA_RESULT_MULTI_PATH)); when(storageClient.getContainerAsync(anyString(), anyString(), anyObject())) .thenReturn(responseMock); final AccessBinaryData abd = accessModuleImpl.getOneObjectFromObjectGroup(asynResponse, ID, fromStringToJson(QUERY), "BinaryMaster", 0); assertNotNull(abd); final Response binaryMasterResponse = abd.getOriginalResponse(); assertNotNull(binaryMasterResponse); assertEquals("image/png", abd.getMimetype()); assertEquals("Wrong name", abd.getFilename()); } @Test(expected = AccessInternalExecutionException.class) @RunWithCustomExecutor public void testGetOneObjectFromObjectGroup_With_Multiple_Result() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); when(metaDataClient.selectObjectGrouptbyId(anyObject(), anyString())) .thenReturn(fromStringToJson(FAKE_METADATA_MULTIPLE_RESULT)); accessModuleImpl.getOneObjectFromObjectGroup(asynResponse, ID, fromStringToJson(QUERY), "BinaryMaster", 0); } @Test(expected = AccessInternalExecutionException.class) @RunWithCustomExecutor public void testGetOneObjectFromObjectGroup_With_Result_Null() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); when(metaDataClient.selectObjectGrouptbyId(anyObject(), anyString())).thenReturn(null); accessModuleImpl.getOneObjectFromObjectGroup(asynResponse, ID, fromStringToJson(QUERY), "BinaryMaster", 0); } @Test(expected = AccessInternalExecutionException.class) @RunWithCustomExecutor public void testGetOneObjectFromObjectGroup_With_StorageClient_Error() throws Exception { VitamThreadUtils.getVitamSession().setTenantId(TENANT_ID); when(metaDataClient.selectObjectGrouptbyId(anyObject(), anyString())) .thenReturn(fromStringToJson(FAKE_METADATA_RESULT)); when(storageClient.getContainerAsync(anyString(), anyString(), anyObject())) .thenThrow(new StorageServerClientException("Test wanted exception")); accessModuleImpl.getOneObjectFromObjectGroup(asynResponse, ID, fromStringToJson(QUERY), "BinaryMaster", 0); } }