2 * ============LICENSE_START=======================================================
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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.openecomp.sdc.be.distribution.servlet;
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;
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;
59 import static org.junit.Assert.assertEquals;
60 import static org.junit.Assert.assertTrue;
61 import static org.mockito.Mockito.when;
63 public class DistributionServletTest extends JerseyTest {
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);
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";
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");
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);
92 mockBusinessLogicResponse();
96 private static void mockBusinessLogicResponse() {
98 Mockito.doAnswer(new Answer<Object>() {
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());
110 }).when(distributionBusinessLogic).handleRegistration(Mockito.any(Wrapper.class), Mockito.any(RegistrationRequest.class), Mockito.any(AuditHandler.class));
113 Mockito.doAnswer(new Answer<Object>() {
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());
125 }).when(distributionBusinessLogic).handleUnRegistration(Mockito.any(Wrapper.class), Mockito.any(RegistrationRequest.class), Mockito.any(AuditHandler.class));
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);
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);
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() {
154 protected void configure() {
155 bind(request).to(HttpServletRequest.class);
158 .property("contextConfig", context);