catalog-be/src/test/java/org/openecomp/sdc/be/distribution/servlet/DistributionServletTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.be.distribution.servlet;
  22
  23 import static org.junit.Assert.assertEquals;
  24 import static org.mockito.Mockito.when;
  25
  26 import com.google.gson.Gson;
  27 import com.google.gson.GsonBuilder;
  28 import javax.servlet.ServletContext;
  29 import javax.servlet.http.HttpServletRequest;
  30 import javax.servlet.http.HttpSession;
  31 import javax.ws.rs.client.Entity;
  32 import javax.ws.rs.core.Application;
  33 import javax.ws.rs.core.MediaType;
  34 import javax.ws.rs.core.Response;
  35 import org.apache.http.HttpStatus;
  36 import org.glassfish.hk2.utilities.binding.AbstractBinder;
  37 import org.glassfish.jersey.server.ResourceConfig;
  38 import org.glassfish.jersey.test.JerseyTest;
  39 import org.glassfish.jersey.test.TestProperties;
  40 import org.junit.BeforeClass;
  41 import org.junit.Test;
  42 import org.mockito.Mockito;
  43 import org.mockito.invocation.InvocationOnMock;
  44 import org.mockito.stubbing.Answer;
  45 import org.openecomp.sdc.be.components.distribution.engine.DistributionEngine;
  46 import org.openecomp.sdc.be.components.impl.ComponentInstanceBusinessLogic;
  47 import org.openecomp.sdc.be.components.impl.GroupBusinessLogic;
  48 import org.openecomp.sdc.be.config.SpringConfig;
  49 import org.openecomp.sdc.be.distribution.AuditHandler;
  50 import org.openecomp.sdc.be.distribution.DistributionBusinessLogic;
  51 import org.openecomp.sdc.be.distribution.api.client.RegistrationRequest;
  52 import org.openecomp.sdc.be.distribution.api.client.TopicRegistrationResponse;
  53 import org.openecomp.sdc.be.impl.ComponentsUtils;
  54 import org.openecomp.sdc.be.impl.WebAppContextWrapper;
  55 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  56 import org.openecomp.sdc.be.user.UserBusinessLogic;
  57 import org.openecomp.sdc.common.api.Constants;
  58 import org.openecomp.sdc.common.datastructure.Wrapper;
  59 import org.openecomp.sdc.common.impl.ExternalConfiguration;
  60 import org.openecomp.sdc.exception.ResponseFormat;
  61 import org.springframework.context.ApplicationContext;
  62 import org.springframework.context.annotation.AnnotationConfigApplicationContext;
  63 import org.springframework.web.context.WebApplicationContext;
  64
  65 public class DistributionServletTest extends JerseyTest {
  66
  67     public static final HttpServletRequest request = Mockito.mock(HttpServletRequest.class);
  68     public static final HttpSession session = Mockito.mock(HttpSession.class);
  69     public static final ServletContext servletContext = Mockito.mock(ServletContext.class);
  70     public static final WebAppContextWrapper webAppContextWrapper = Mockito.mock(WebAppContextWrapper.class);
  71     public static final WebApplicationContext webApplicationContext = Mockito.mock(WebApplicationContext.class);
  72     public static final ResponseFormat responseFormat = Mockito.mock(ResponseFormat.class);
  73     public static final DistributionBusinessLogic distributionBusinessLogic = Mockito.mock(DistributionBusinessLogic.class);
  74     public static final DistributionEngine distributionEngine = Mockito.mock(DistributionEngine.class);
  75
  76     public static final String ENV_NAME = "myEnv";
  77     public static final String NOTIFICATION_TOPIC = ENV_NAME + "_Notification";
  78     public static final String STATUS_TOPIC = ENV_NAME + "_Status";
  79
  80     @BeforeClass
  81     public static void setup() {
  82         ExternalConfiguration.setAppName("catalog-be");
  83         when(request.getSession()).thenReturn(session);
  84         when(request.getHeader(Constants.X_ECOMP_INSTANCE_ID_HEADER)).thenReturn("myApplicationInstanceID");
  85
  86         when(session.getServletContext()).thenReturn(servletContext);
  87         when(servletContext.getAttribute(Constants.WEB_APPLICATION_CONTEXT_WRAPPER_ATTR)).thenReturn(webAppContextWrapper);
  88         when(webAppContextWrapper.getWebAppContext(servletContext)).thenReturn(webApplicationContext);
  89         when(distributionBusinessLogic.getDistributionEngine()).thenReturn(distributionEngine);
  90         when(distributionEngine.isEnvironmentAvailable(ENV_NAME)).thenReturn(StorageOperationStatus.OK);
  91         when(distributionEngine.isEnvironmentAvailable()).thenReturn(StorageOperationStatus.OK);
  92
  93         mockBusinessLogicResponse();
  94
  95     }
  96
  97     private static void mockBusinessLogicResponse() {
  98         // Mock Register
  99         Mockito.doAnswer(new Answer<Object>() {
 100             @Override
 101             public Object answer(InvocationOnMock invocation) {
 102                 Object[] args = invocation.getArguments();
 103                 Wrapper<Response> responseWrapper = (Wrapper<Response>) args[0];
 104                 TopicRegistrationResponse okTopicResponse = new TopicRegistrationResponse();
 105                 okTopicResponse.setDistrNotificationTopicName(NOTIFICATION_TOPIC);
 106                 okTopicResponse.setDistrStatusTopicName(STATUS_TOPIC);
 107                 responseWrapper.setInnerElement(Response.status(HttpStatus.SC_OK).entity(okTopicResponse).build());
 108
 109                 return true;
 110             }
 111         }).when(distributionBusinessLogic).handleRegistration(Mockito.any(Wrapper.class), Mockito.any(RegistrationRequest.class), Mockito.any(AuditHandler.class));
 112
 113         // Mock Unregister
 114         Mockito.doAnswer(new Answer<Object>() {
 115             @Override
 116             public Object answer(InvocationOnMock invocation) {
 117                 Object[] args = invocation.getArguments();
 118                 Wrapper<Response> responseWrapper = (Wrapper<Response>) args[0];
 119                 TopicRegistrationResponse okTopicResponse = new TopicRegistrationResponse();
 120                 okTopicResponse.setDistrNotificationTopicName(NOTIFICATION_TOPIC);
 121                 okTopicResponse.setDistrStatusTopicName(STATUS_TOPIC);
 122                 responseWrapper.setInnerElement(Response.status(HttpStatus.SC_OK).entity(okTopicResponse).build());
 123
 124                 return true;
 125             }
 126         }).when(distributionBusinessLogic).handleUnRegistration(Mockito.any(Wrapper.class), Mockito.any(RegistrationRequest.class), Mockito.any(AuditHandler.class));
 127     }
 128
 129     @Test
 130     public void registerSuccessTest() {
 131         Gson gson = new GsonBuilder().setPrettyPrinting().create();
 132         RegistrationRequest registrationRequest = new RegistrationRequest("myPublicKey", ENV_NAME, false);
 133         Response response = target().path("/v1/registerForDistribution").request(MediaType.APPLICATION_JSON).post(Entity.json(gson.toJson(registrationRequest)), Response.class);
 134         assertEquals(response.getStatus(), HttpStatus.SC_OK);
 135
 136     }
 137
 138     @Test
 139     public void unRegisterSuccessTest() {
 140         Gson gson = new GsonBuilder().setPrettyPrinting().create();
 141         RegistrationRequest registrationRequest = new RegistrationRequest("myPublicKey", ENV_NAME, false);
 142         Response response = target().path("/v1/unRegisterForDistribution").request(MediaType.APPLICATION_JSON).post(Entity.json(gson.toJson(registrationRequest)), Response.class);
 143         assertEquals(response.getStatus(), HttpStatus.SC_OK);
 144
 145     }
 146
 147     @Override
 148     protected Application configure() {
 149         UserBusinessLogic userBusinessLogic = Mockito.mock(UserBusinessLogic.class);
 150         GroupBusinessLogic groupBusinessLogic = Mockito.mock(GroupBusinessLogic.class);
 151         ComponentInstanceBusinessLogic componentInstanceBusinessLogic = Mockito.mock(ComponentInstanceBusinessLogic.class);
 152         ComponentsUtils componentsUtils = Mockito.mock(ComponentsUtils.class);
 153
 154         ApplicationContext context = new AnnotationConfigApplicationContext(SpringConfig.class);
 155         forceSet(TestProperties.CONTAINER_PORT, "0");
 156         return new ResourceConfig(DistributionServlet.class)
 157                 .register(new AbstractBinder() {
 158
 159                     @Override
 160                     protected void configure() {
 161                         bind(request).to(HttpServletRequest.class);
 162                         bind(userBusinessLogic).to(UserBusinessLogic.class);
 163                         bind(groupBusinessLogic).to(GroupBusinessLogic.class);
 164                         bind(componentInstanceBusinessLogic).to(ComponentInstanceBusinessLogic.class);
 165                         bind(componentsUtils).to(ComponentsUtils.class);
 166                         bind(distributionBusinessLogic).to(DistributionBusinessLogic.class);
 167                     }
 168                 })
 169                 .property("contextConfig", context);
 170     }
 171 }