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.components.impl.ComponentInstanceBusinessLogic;
37 import org.openecomp.sdc.be.components.impl.GroupBusinessLogic;
38 import org.openecomp.sdc.be.config.ConfigurationManager;
39 import org.openecomp.sdc.be.config.SpringConfig;
40 import org.openecomp.sdc.be.distribution.AuditHandler;
41 import org.openecomp.sdc.be.distribution.DistributionBusinessLogic;
42 import org.openecomp.sdc.be.distribution.api.client.RegistrationRequest;
43 import org.openecomp.sdc.be.distribution.api.client.TopicRegistrationResponse;
44 import org.openecomp.sdc.be.impl.ComponentsUtils;
45 import org.openecomp.sdc.be.impl.WebAppContextWrapper;
46 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
47 import org.openecomp.sdc.be.user.UserBusinessLogic;
48 import org.openecomp.sdc.common.api.ConfigurationSource;
49 import org.openecomp.sdc.common.api.Constants;
50 import org.openecomp.sdc.common.api.FilterDecisionEnum;
51 import org.openecomp.sdc.common.datastructure.Wrapper;
52 import org.openecomp.sdc.common.impl.ExternalConfiguration;
53 import org.openecomp.sdc.common.impl.FSConfigurationSource;
54 import org.openecomp.sdc.common.util.ThreadLocalsHolder;
55 import org.openecomp.sdc.exception.ResponseFormat;
56 import org.springframework.context.ApplicationContext;
57 import org.springframework.context.annotation.AnnotationConfigApplicationContext;
58 import org.springframework.web.context.WebApplicationContext;
60 import javax.servlet.ServletContext;
61 import javax.servlet.http.HttpServletRequest;
62 import javax.servlet.http.HttpSession;
63 import javax.ws.rs.client.Entity;
64 import javax.ws.rs.core.Application;
65 import javax.ws.rs.core.MediaType;
66 import javax.ws.rs.core.Response;
67 import java.util.Arrays;
69 import static org.junit.Assert.assertEquals;
70 import static org.mockito.ArgumentMatchers.anyString;
71 import static org.mockito.Mockito.when;
73 public class DistributionServletTest extends JerseyTest {
75 public static final HttpServletRequest request = Mockito.mock(HttpServletRequest.class);
76 public static final HttpSession session = Mockito.mock(HttpSession.class);
77 public static final ServletContext servletContext = Mockito.mock(ServletContext.class);
78 public static final WebAppContextWrapper webAppContextWrapper = Mockito.mock(WebAppContextWrapper.class);
79 public static final WebApplicationContext webApplicationContext = Mockito.mock(WebApplicationContext.class);
80 public static final ResponseFormat responseFormat = Mockito.mock(ResponseFormat.class);
81 public static final DistributionBusinessLogic distributionBusinessLogic = Mockito.mock(DistributionBusinessLogic.class);
82 public static final DistributionEngine distributionEngine = Mockito.mock(DistributionEngine.class);
83 private static ConfigurationSource configurationSource = new FSConfigurationSource(
84 ExternalConfiguration.getChangeListener(), "src/test/resources/config/catalog-be");
85 static ConfigurationManager configurationManager = new ConfigurationManager(configurationSource);
88 public static final String ENV_NAME = "myEnv";
89 public static final String NOTIFICATION_TOPIC = ENV_NAME + "_Notification";
90 public static final String STATUS_TOPIC = ENV_NAME + "_Status";
93 public static void setup() {
94 ThreadLocalsHolder.setApiType(FilterDecisionEnum.EXTERNAL);
95 ExternalConfiguration.setAppName("catalog-be");
96 when(request.getSession()).thenReturn(session);
97 when(request.getHeader(Constants.X_ECOMP_INSTANCE_ID_HEADER)).thenReturn("myApplicationInstanceID");
99 when(session.getServletContext()).thenReturn(servletContext);
100 when(servletContext.getAttribute(Constants.WEB_APPLICATION_CONTEXT_WRAPPER_ATTR)).thenReturn(webAppContextWrapper);
101 when(webAppContextWrapper.getWebAppContext(servletContext)).thenReturn(webApplicationContext);
102 when(webApplicationContext.getBean(DistributionBusinessLogic.class)).thenReturn(distributionBusinessLogic);
103 when(distributionBusinessLogic.getDistributionEngine()).thenReturn(distributionEngine);
104 when(distributionEngine.isEnvironmentAvailable(ENV_NAME)).thenReturn(StorageOperationStatus.OK);
105 when(distributionEngine.isEnvironmentAvailable()).thenReturn(StorageOperationStatus.OK);
107 when(request.isUserInRole(anyString())).thenReturn(true);
109 mockBusinessLogicResponse();
113 private static void mockBusinessLogicResponse() {
115 Mockito.doAnswer(new Answer<Object>() {
117 public Object answer(InvocationOnMock invocation) {
118 Object[] args = invocation.getArguments();
119 Wrapper<Response> responseWrapper = (Wrapper<Response>) args[0];
120 TopicRegistrationResponse okTopicResponse = new TopicRegistrationResponse();
121 okTopicResponse.setDistrNotificationTopicName(NOTIFICATION_TOPIC);
122 okTopicResponse.setDistrStatusTopicName(STATUS_TOPIC);
123 responseWrapper.setInnerElement(Response.status(HttpStatus.SC_OK).entity(okTopicResponse).build());
127 }).when(distributionBusinessLogic).handleRegistration(Mockito.any(Wrapper.class), Mockito.any(RegistrationRequest.class), Mockito.any(AuditHandler.class));
130 Mockito.doAnswer(new Answer<Object>() {
132 public Object answer(InvocationOnMock invocation) {
133 Object[] args = invocation.getArguments();
134 Wrapper<Response> responseWrapper = (Wrapper<Response>) args[0];
135 TopicRegistrationResponse okTopicResponse = new TopicRegistrationResponse();
136 okTopicResponse.setDistrNotificationTopicName(NOTIFICATION_TOPIC);
137 okTopicResponse.setDistrStatusTopicName(STATUS_TOPIC);
138 responseWrapper.setInnerElement(Response.status(HttpStatus.SC_OK).entity(okTopicResponse).build());
142 }).when(distributionBusinessLogic).handleUnRegistration(Mockito.any(Wrapper.class), Mockito.any(RegistrationRequest.class), Mockito.any(AuditHandler.class));
146 public void registerSuccessTest() {
147 Gson gson = new GsonBuilder().setPrettyPrinting().create();
148 RegistrationRequest registrationRequest = new RegistrationRequest("myPublicKey", ENV_NAME, false);
149 Response response = target().path("/v1/registerForDistribution").request(MediaType.APPLICATION_JSON).post(Entity.json(gson.toJson(registrationRequest)), Response.class);
150 assertEquals(response.getStatus(), HttpStatus.SC_OK);
155 public void registerSuccessOnTenantTest() {
156 Gson gson = new GsonBuilder().setPrettyPrinting().create();
157 RegistrationRequest registrationRequest = new RegistrationRequest("myPublicKey", ENV_NAME, Arrays.asList("11","22"),false);
158 Response response = target().path("/v1/registerForDistribution").request(MediaType.APPLICATION_JSON).post(Entity.json(gson.toJson(registrationRequest)), Response.class);
159 assertEquals(response.getStatus(), HttpStatus.SC_OK);
164 public void unRegisterSuccessTest() {
165 Gson gson = new GsonBuilder().setPrettyPrinting().create();
166 RegistrationRequest registrationRequest = new RegistrationRequest("myPublicKey", ENV_NAME, false);
167 Response response = target().path("/v1/unRegisterForDistribution").request(MediaType.APPLICATION_JSON).post(Entity.json(gson.toJson(registrationRequest)), Response.class);
168 assertEquals(response.getStatus(), HttpStatus.SC_OK);
173 public void unRegisterSuccessOnTenantTest() {
174 Gson gson = new GsonBuilder().setPrettyPrinting().create();
175 RegistrationRequest registrationRequest = new RegistrationRequest("myPublicKey", ENV_NAME, false);
176 Response response = target().path("/v1/unRegisterForDistribution").request(MediaType.APPLICATION_JSON).post(Entity.json(gson.toJson(registrationRequest)), Response.class);
177 assertEquals(response.getStatus(), HttpStatus.SC_OK);
182 protected Application configure() {
183 UserBusinessLogic userBusinessLogic = Mockito.mock(UserBusinessLogic.class);
184 GroupBusinessLogic groupBusinessLogic = Mockito.mock(GroupBusinessLogic.class);
185 ComponentInstanceBusinessLogic componentInstanceBusinessLogic = Mockito.mock(ComponentInstanceBusinessLogic.class);
186 ComponentsUtils componentsUtils = Mockito.mock(ComponentsUtils.class);
188 ApplicationContext context = new AnnotationConfigApplicationContext(SpringConfig.class);
189 forceSet(TestProperties.CONTAINER_PORT, "0");
190 enable(TestProperties.LOG_TRAFFIC);
191 enable(TestProperties.DUMP_ENTITY);
192 return new ResourceConfig(DistributionServlet.class)
193 .register(new AbstractBinder() {
196 protected void configure() {
197 bind(request).to(HttpServletRequest.class);
198 bind(userBusinessLogic).to(UserBusinessLogic.class);
199 bind(groupBusinessLogic).to(GroupBusinessLogic.class);
200 bind(componentInstanceBusinessLogic).to(ComponentInstanceBusinessLogic.class);
201 bind(componentsUtils).to(ComponentsUtils.class);
202 bind(distributionBusinessLogic).to(DistributionBusinessLogic.class);
205 .property("contextConfig", context);