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 com.google.gson.Gson;
  24 import com.google.gson.GsonBuilder;
  25 import org.apache.http.HttpStatus;
  26 import org.glassfish.hk2.utilities.binding.AbstractBinder;
  27 import org.glassfish.jersey.server.ResourceConfig;
  28 import org.glassfish.jersey.test.JerseyTest;
  29 import org.glassfish.jersey.test.TestProperties;
  30 import org.junit.BeforeClass;
  31 import org.junit.Test;
  32 import org.mockito.Mockito;
  33 import org.mockito.invocation.InvocationOnMock;
  34 import org.mockito.stubbing.Answer;
  35 import org.openecomp.sdc.be.components.distribution.engine.DistributionEngine;
  36 import org.openecomp.sdc.be.config.SpringConfig;
  37 import org.openecomp.sdc.be.distribution.AuditHandler;
  38 import org.openecomp.sdc.be.distribution.DistributionBusinessLogic;
  39 import org.openecomp.sdc.be.distribution.api.client.RegistrationRequest;
  40 import org.openecomp.sdc.be.distribution.api.client.TopicRegistrationResponse;
  41 import org.openecomp.sdc.be.impl.WebAppContextWrapper;
  42 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  43 import org.openecomp.sdc.common.api.Constants;
  44 import org.openecomp.sdc.common.datastructure.Wrapper;
  45 import org.openecomp.sdc.common.impl.ExternalConfiguration;
  46 import org.openecomp.sdc.exception.ResponseFormat;
  47 import org.springframework.context.ApplicationContext;
  48 import org.springframework.context.annotation.AnnotationConfigApplicationContext;
  49 import org.springframework.web.context.WebApplicationContext;
  50
  51 import javax.servlet.ServletContext;
  52 import javax.servlet.http.HttpServletRequest;
  53 import javax.servlet.http.HttpSession;
  54 import javax.ws.rs.client.Entity;
  55 import javax.ws.rs.core.Application;
  56 import javax.ws.rs.core.MediaType;
  57 import javax.ws.rs.core.Response;
  58
  59 import static org.junit.Assert.assertEquals;
  60 import static org.junit.Assert.assertTrue;
  61 import static org.mockito.Mockito.when;
  62
  63 public class DistributionServletTest extends JerseyTest {
  64
  65     public static final HttpServletRequest request = Mockito.mock(HttpServletRequest.class);
  66     public static final HttpSession session = Mockito.mock(HttpSession.class);
  67     public static final ServletContext servletContext = Mockito.mock(ServletContext.class);
  68     public static final WebAppContextWrapper webAppContextWrapper = Mockito.mock(WebAppContextWrapper.class);
  69     public static final WebApplicationContext webApplicationContext = Mockito.mock(WebApplicationContext.class);
  70     public static final ResponseFormat responseFormat = Mockito.mock(ResponseFormat.class);
  71     public static final DistributionBusinessLogic distributionBusinessLogic = Mockito.mock(DistributionBusinessLogic.class);
  72     public static final DistributionEngine distributionEngine = Mockito.mock(DistributionEngine.class);
  73
  74     public static final String ENV_NAME = "myEnv";
  75     public static final String NOTIFICATION_TOPIC = ENV_NAME + "_Notification";
  76     public static final String STATUS_TOPIC = ENV_NAME + "_Status";
  77
  78     @BeforeClass
  79     public static void setup() {
  80         ExternalConfiguration.setAppName("catalog-be");
  81         when(request.getSession()).thenReturn(session);
  82         when(request.getHeader(Constants.X_ECOMP_INSTANCE_ID_HEADER)).thenReturn("myApplicationInstanceID");
  83
  84         when(session.getServletContext()).thenReturn(servletContext);
  85         when(servletContext.getAttribute(Constants.WEB_APPLICATION_CONTEXT_WRAPPER_ATTR)).thenReturn(webAppContextWrapper);
  86         when(webAppContextWrapper.getWebAppContext(servletContext)).thenReturn(webApplicationContext);
  87         when(webApplicationContext.getBean(DistributionBusinessLogic.class)).thenReturn(distributionBusinessLogic);
  88         when(distributionBusinessLogic.getDistributionEngine()).thenReturn(distributionEngine);
  89         when(distributionEngine.isEnvironmentAvailable(ENV_NAME)).thenReturn(StorageOperationStatus.OK);
  90         when(distributionEngine.isEnvironmentAvailable()).thenReturn(StorageOperationStatus.OK);
  91
  92         mockBusinessLogicResponse();
  93
  94     }
  95
  96     private static void mockBusinessLogicResponse() {
  97         // Mock Register
  98         Mockito.doAnswer(new Answer<Object>() {
  99             @Override
 100             public Object answer(InvocationOnMock invocation) {
 101                 Object[] args = invocation.getArguments();
 102                 Wrapper<Response> responseWrapper = (Wrapper<Response>) args[0];
 103                 TopicRegistrationResponse okTopicResponse = new TopicRegistrationResponse();
 104                 okTopicResponse.setDistrNotificationTopicName(NOTIFICATION_TOPIC);
 105                 okTopicResponse.setDistrStatusTopicName(STATUS_TOPIC);
 106                 responseWrapper.setInnerElement(Response.status(HttpStatus.SC_OK).entity(okTopicResponse).build());
 107
 108                 return true;
 109             }
 110         }).when(distributionBusinessLogic).handleRegistration(Mockito.any(Wrapper.class), Mockito.any(RegistrationRequest.class), Mockito.any(AuditHandler.class));
 111
 112         // Mock Unregister
 113         Mockito.doAnswer(new Answer<Object>() {
 114             @Override
 115             public Object answer(InvocationOnMock invocation) {
 116                 Object[] args = invocation.getArguments();
 117                 Wrapper<Response> responseWrapper = (Wrapper<Response>) args[0];
 118                 TopicRegistrationResponse okTopicResponse = new TopicRegistrationResponse();
 119                 okTopicResponse.setDistrNotificationTopicName(NOTIFICATION_TOPIC);
 120                 okTopicResponse.setDistrStatusTopicName(STATUS_TOPIC);
 121                 responseWrapper.setInnerElement(Response.status(HttpStatus.SC_OK).entity(okTopicResponse).build());
 122
 123                 return true;
 124             }
 125         }).when(distributionBusinessLogic).handleUnRegistration(Mockito.any(Wrapper.class), Mockito.any(RegistrationRequest.class), Mockito.any(AuditHandler.class));
 126     }
 127
 128     @Test
 129     public void registerSuccessTest() {
 130         Gson gson = new GsonBuilder().setPrettyPrinting().create();
 131         RegistrationRequest registrationRequest = new RegistrationRequest("myPublicKey", ENV_NAME, false);
 132         Response response = target().path("/v1/registerForDistribution").request(MediaType.APPLICATION_JSON).post(Entity.json(gson.toJson(registrationRequest)), Response.class);
 133         assertEquals(response.getStatus(), HttpStatus.SC_OK);
 134
 135     }
 136
 137     @Test
 138     public void unRegisterSuccessTest() {
 139         Gson gson = new GsonBuilder().setPrettyPrinting().create();
 140         RegistrationRequest registrationRequest = new RegistrationRequest("myPublicKey", ENV_NAME, false);
 141         Response response = target().path("/v1/unRegisterForDistribution").request(MediaType.APPLICATION_JSON).post(Entity.json(gson.toJson(registrationRequest)), Response.class);
 142         assertEquals(response.getStatus(), HttpStatus.SC_OK);
 143
 144     }
 145
 146     @Override
 147     protected Application configure() {
 148         ApplicationContext context = new AnnotationConfigApplicationContext(SpringConfig.class);
 149         forceSet(TestProperties.CONTAINER_PORT, "0");
 150         return new ResourceConfig(DistributionServlet.class)
 151                 .register(new AbstractBinder() {
 152
 153                     @Override
 154                     protected void configure() {
 155                         bind(request).to(HttpServletRequest.class);
 156                     }
 157                 })
 158                 .property("contextConfig", context);
 159     }
 160 }