diff options
Diffstat (limited to 'catalog-be/src/test/java/org/openecomp/sdc/be/servlets/TypesUploadEndpointTest.java')
| -rw-r--r-- | catalog-be/src/test/java/org/openecomp/sdc/be/servlets/TypesUploadEndpointTest.java | 205 |
1 files changed, 205 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/servlets/TypesUploadEndpointTest.java b/catalog-be/src/test/java/org/openecomp/sdc/be/servlets/TypesUploadEndpointTest.java new file mode 100644 index 0000000000..8638de334a --- /dev/null +++ b/catalog-be/src/test/java/org/openecomp/sdc/be/servlets/TypesUploadEndpointTest.java @@ -0,0 +1,205 @@ +package org.openecomp.sdc.be.servlets; + +import com.fasterxml.jackson.databind.DeserializationFeature; +import fj.data.Either; +import org.apache.commons.lang3.tuple.ImmutablePair; +import org.eclipse.jetty.http.HttpStatus; +import org.glassfish.jersey.client.ClientConfig; +import org.glassfish.jersey.jackson.internal.jackson.jaxrs.json.JacksonJaxbJsonProvider; +import org.glassfish.jersey.jackson.internal.jackson.jaxrs.json.JacksonJsonProvider; +import org.glassfish.jersey.logging.LoggingFeature; +import org.glassfish.jersey.media.multipart.FormDataMultiPart; +import org.glassfish.jersey.media.multipart.MultiPart; +import org.glassfish.jersey.media.multipart.MultiPartFeature; +import org.glassfish.jersey.media.multipart.file.FileDataBodyPart; +import org.glassfish.jersey.server.ResourceConfig; +import org.junit.BeforeClass; +import org.junit.Test; +import org.mockito.Mockito; +import org.openecomp.sdc.be.components.impl.CommonImportManager; +import org.openecomp.sdc.be.components.validation.AccessValidations; +import org.openecomp.sdc.be.dao.titan.TitanGenericDao; +import org.openecomp.sdc.be.dao.titan.TitanOperationStatus; +import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum; +import org.openecomp.sdc.be.impl.ComponentsUtils; +import org.openecomp.sdc.be.model.AnnotationTypeDefinition; +import org.openecomp.sdc.be.model.operations.StorageException; +import org.openecomp.sdc.be.model.operations.impl.AnnotationTypeOperations; +import org.openecomp.sdc.be.model.operations.impl.CommonTypeOperations; +import org.openecomp.sdc.be.model.operations.impl.OperationUtils; +import org.openecomp.sdc.be.model.operations.impl.PropertyOperation; +import org.openecomp.sdc.be.resources.data.AnnotationTypeData; +import org.openecomp.sdc.common.api.Constants; +import org.springframework.context.annotation.Bean; +import org.springframework.context.annotation.Import; + +import javax.ws.rs.client.Entity; +import javax.ws.rs.core.MediaType; +import javax.ws.rs.core.Response; +import java.io.File; +import java.util.ArrayList; +import java.util.List; + +import static java.util.Collections.emptyMap; +import static org.assertj.core.api.Assertions.assertThat; +import static org.mockito.ArgumentMatchers.any; +import static org.mockito.ArgumentMatchers.anyList; +import static org.mockito.ArgumentMatchers.anyMap; +import static org.mockito.ArgumentMatchers.anyString; +import static org.mockito.ArgumentMatchers.eq; +import static org.mockito.ArgumentMatchers.isA; +import static org.mockito.Mockito.*; + +public class TypesUploadEndpointTest extends JerseySpringBaseTest { + + static final String userId = "jh0003"; + + private static AccessValidations accessValidations; + private static TitanGenericDao titanGenericDao; + private static PropertyOperation propertyOperation; + private static ComponentsUtils componentUtils; + private static OperationUtils operationUtils; + + @org.springframework.context.annotation.Configuration + @Import(BaseTestConfig.class) + static class TypesUploadTestConfig { + + @Bean + TypesUploadEndpoint typesUploadEndpoint() { + return new TypesUploadEndpoint(commonImportManager(), annotationTypeOperations(), accessValidations); + } + + @Bean + CommonImportManager commonImportManager() { + return new CommonImportManager(componentUtils, propertyOperation); + } + + @Bean + AnnotationTypeOperations annotationTypeOperations() { + return new AnnotationTypeOperations(commonTypeOperations()); + } + + @Bean + CommonTypeOperations commonTypeOperations() { + return new CommonTypeOperations(titanGenericDao, propertyOperation, operationUtils); + } + } + + @BeforeClass + public static void initClass() { + titanGenericDao = mock(TitanGenericDao.class); + accessValidations = mock(AccessValidations.class); + propertyOperation = mock(PropertyOperation.class); + componentUtils = Mockito.mock(ComponentsUtils.class); + operationUtils = Mockito.mock(OperationUtils.class); + } + + @Override + protected void configureClient(ClientConfig config) { + final JacksonJsonProvider jacksonJsonProvider = new JacksonJaxbJsonProvider() + .configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false); + config.register(jacksonJsonProvider); + config.register(MultiPartFeature.class); + } + + @Override + protected ResourceConfig configure() { + return super.configure(TypesUploadEndpointTest.TypesUploadTestConfig.class) + .register(TypesUploadEndpoint.class) + .property(LoggingFeature.LOGGING_FEATURE_LOGGER_LEVEL_SERVER, "WARNING"); + } + + @Test + public void creatingAnnotationTypeSuccessTest() { + doNothing().when(accessValidations).validateUserExists(eq(userId), anyString()); + when(titanGenericDao.createNode(isA(AnnotationTypeData.class), eq(AnnotationTypeData.class))).thenReturn(Either.left(new AnnotationTypeData())); + when(titanGenericDao.getNode(anyString(), eq("org.openecomp.annotations.source.1.0.annotationtype"), eq(AnnotationTypeData.class))).thenReturn(Either.left(new AnnotationTypeData())); + when(titanGenericDao.getByCriteria(eq(NodeTypeEnum.AnnotationType), anyMap(), eq(AnnotationTypeData.class))).thenReturn(Either.right(TitanOperationStatus.NOT_FOUND)); + when(propertyOperation.addPropertiesToElementType(anyString(), eq(NodeTypeEnum.AnnotationType), anyList())).thenReturn(Either.left(emptyMap())); + when(propertyOperation.fillPropertiesList(anyString(), eq(NodeTypeEnum.AnnotationType), any())).thenReturn(TitanOperationStatus.OK); + when(propertyOperation.getTitanGenericDao()).thenReturn(titanGenericDao); + when(titanGenericDao.commit()).thenReturn(TitanOperationStatus.OK); + when(titanGenericDao.rollback()).thenReturn(TitanOperationStatus.OK); + FileDataBodyPart filePart = new FileDataBodyPart("annotationTypesZip", new File("src/test/resources/types/annotationTypes.zip")); + MultiPart multipartEntity = new FormDataMultiPart(); + multipartEntity.bodyPart(filePart); + + Response response = target().path("/v1/catalog/uploadType/annotationtypes") + .request(MediaType.APPLICATION_JSON) + .header(Constants.USER_ID_HEADER, userId) + .post(Entity.entity(multipartEntity, MediaType.MULTIPART_FORM_DATA), Response.class); + assertThat(response.getStatus()).isEqualTo(HttpStatus.CREATED_201); + assertThat(Boolean.valueOf(getTypeActionResult(response))).isTrue(); + } + + private String getTypeActionResult(Response response) { + String typeResult = ""; + String body = response.readEntity(String.class); + int indexColon = body.lastIndexOf(':'); + if (indexColon > 0) { + int indexCurly = body.indexOf('}', indexColon); + if (indexCurly > 0) { + typeResult = body.substring(indexColon+1, indexCurly); + } + } + return typeResult; + } + + @Test + public void creatingAnnotationTypeFailureTest() { + doNothing().when(accessValidations).validateUserExists(eq(userId), anyString()); + when(titanGenericDao.createNode(isA(AnnotationTypeData.class), eq(AnnotationTypeData.class))).thenReturn(Either.left(new AnnotationTypeData())); + when(titanGenericDao.getNode(anyString(), eq("org.openecomp.annotations.source.1.0.annotationtype"), eq(AnnotationTypeData.class))).thenReturn(Either.left(new AnnotationTypeData())); + when(titanGenericDao.getByCriteria(eq(NodeTypeEnum.AnnotationType), anyMap(), eq(AnnotationTypeData.class))).thenReturn(Either.right(TitanOperationStatus.NOT_FOUND)); + when(propertyOperation.addPropertiesToElementType(anyString(), eq(NodeTypeEnum.AnnotationType), anyList())).thenThrow(new StorageException(TitanOperationStatus.MATCH_NOT_FOUND)); + when(propertyOperation.fillPropertiesList(anyString(), eq(NodeTypeEnum.AnnotationType), any())).thenReturn(TitanOperationStatus.OK); + when(propertyOperation.getTitanGenericDao()).thenReturn(titanGenericDao); + when(titanGenericDao.commit()).thenReturn(TitanOperationStatus.OK); + when(titanGenericDao.rollback()).thenReturn(TitanOperationStatus.OK); + FileDataBodyPart filePart = new FileDataBodyPart("annotationTypesZip", new File("src/test/resources/types/annotationTypes.zip")); + MultiPart multipartEntity = new FormDataMultiPart(); + multipartEntity.bodyPart(filePart); + + Response response = target().path("/v1/catalog/uploadType/annotationtypes") + .request(MediaType.APPLICATION_JSON) + .header(Constants.USER_ID_HEADER, userId) + .post(Entity.entity(multipartEntity, MediaType.MULTIPART_FORM_DATA), Response.class); + assertThat(response.getStatus()).isEqualTo(HttpStatus.BAD_REQUEST_400); + assertThat(Boolean.valueOf(getTypeActionResult(response))).isFalse(); + } + + @Test + public void singleTypeSucceeded_statusIsCreated() { + List<ImmutablePair<AnnotationTypeDefinition, Boolean>> typeActionResults = new ArrayList<>(); + AnnotationTypeDefinition dummyDefition = new AnnotationTypeDefinition(); + typeActionResults.add(new ImmutablePair(dummyDefition, true)); + assertThat(TypesUploadEndpoint.getHttpStatus(typeActionResults).value()).isEqualTo(HttpStatus.CREATED_201); + } + + @Test + public void singleTypeExists_statusIsConflict() { + List<ImmutablePair<AnnotationTypeDefinition, Boolean>> typeActionResults = new ArrayList<>(); + AnnotationTypeDefinition dummyDefition = new AnnotationTypeDefinition(); + typeActionResults.add(new ImmutablePair(dummyDefition, null)); + assertThat(TypesUploadEndpoint.getHttpStatus(typeActionResults).value()).isEqualTo(HttpStatus.CONFLICT_409); + } + + @Test + public void mixedSuccessAndExists_statusIsCreated() { + List<ImmutablePair<AnnotationTypeDefinition, Boolean>> typeActionResults = new ArrayList<>(); + AnnotationTypeDefinition dummyDefition = new AnnotationTypeDefinition(); + typeActionResults.add(new ImmutablePair(dummyDefition, true)); + typeActionResults.add(new ImmutablePair(dummyDefition, null)); + assertThat(TypesUploadEndpoint.getHttpStatus(typeActionResults).value()).isEqualTo(HttpStatus.CREATED_201); + } + + @Test + public void mixedSuccessAndFailure_statusIsBadRequest() { + List<ImmutablePair<AnnotationTypeDefinition, Boolean>> typeActionResults = new ArrayList<>(); + AnnotationTypeDefinition dummyDefition = new AnnotationTypeDefinition(); + typeActionResults.add(new ImmutablePair(dummyDefition, true)); + typeActionResults.add(new ImmutablePair(dummyDefition, false)); + typeActionResults.add(new ImmutablePair(dummyDefition, null)); + assertThat(TypesUploadEndpoint.getHttpStatus(typeActionResults).value()).isEqualTo(HttpStatus.BAD_REQUEST_400); + } +}
\ No newline at end of file |