1 /*******************************************************************************
2 * ============LICENSE_START==================================================
4 * * ===========================================================================
5 * * Copyright © 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====================================================
20 * * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22 ******************************************************************************/
24 package org.onap.dmaap.datarouter.provisioning;
26 import ch.qos.logback.classic.Logger;
27 import ch.qos.logback.classic.spi.ILoggingEvent;
28 import ch.qos.logback.core.read.ListAppender;
29 import org.apache.commons.lang3.reflect.FieldUtils;
31 import org.junit.runner.RunWith;
32 import org.mockito.Mock;
33 import org.onap.dmaap.datarouter.provisioning.utils.DB;
34 import org.powermock.core.classloader.annotations.PrepareForTest;
35 import org.powermock.modules.junit4.PowerMockRunner;
36 import org.slf4j.LoggerFactory;
38 import javax.persistence.EntityManager;
39 import javax.persistence.EntityManagerFactory;
40 import javax.persistence.Persistence;
41 import javax.servlet.http.HttpServletRequest;
42 import javax.servlet.http.HttpServletResponse;
46 import java.io.FileNotFoundException;
47 import java.io.PrintWriter;
48 import java.net.InetAddress;
50 import static org.hamcrest.Matchers.notNullValue;
51 import static org.junit.Assert.assertEquals;
52 import static org.mockito.Matchers.*;
53 import static org.mockito.Mockito.verify;
54 import static org.mockito.Mockito.when;
55 import static org.powermock.api.mockito.PowerMockito.mockStatic;
58 * Created by ezcoxem on 21/08/2018.
61 @RunWith(PowerMockRunner.class)
62 @PrepareForTest({InetAddress.class })
63 public class PublishServletTest extends DrServletTestBase {
64 private PublishServlet publishServlet;
67 private HttpServletRequest request;
70 private HttpServletResponse response;
72 private static EntityManagerFactory emf;
73 private static EntityManager em;
76 ListAppender<ILoggingEvent> listAppender;
79 public static void init() {
80 emf = Persistence.createEntityManagerFactory("dr-unit-tests");
81 em = emf.createEntityManager();
83 "org.onap.dmaap.datarouter.provserver.properties",
84 "src/test/resources/h2Database.properties");
88 public static void tearDownClass() throws FileNotFoundException {
96 public void setUp() throws Exception {
97 listAppender = setTestLogger(PublishServlet.class);
98 publishServlet = new PublishServlet();
103 public void Given_Request_Is_HTTP_DELETE_And_There_Are_No_Nodes_Then_Service_Unavailable_Error_Is_Returned()
105 FieldUtils.writeDeclaredStaticField(BaseServlet.class, "nodes", new String[0], true);
106 publishServlet.doDelete(request, response);
107 verify(response).sendError(eq(HttpServletResponse.SC_SERVICE_UNAVAILABLE), argThat(notNullValue(String.class)));
108 FieldUtils.writeDeclaredStaticField(BaseServlet.class, "nodes", new String[1], true);
109 verifyEnteringExitCalled(listAppender);
113 public void Given_Request_Is_HTTP_DELETE_And_Path_Is_Null_Then_Not_Found_Error_Is_Returned()
115 publishServlet.doDelete(request, response);
116 verify(response).sendError(eq(HttpServletResponse.SC_NOT_FOUND), argThat(notNullValue(String.class)));
120 public void Given_Request_Is_HTTP_DELETE_And_Ix_Is_Null_Then_Not_Found_Error_Is_Returned()
123 when(request.getPathInfo()).thenReturn("/1/");
124 publishServlet.doDelete(request, response);
125 verify(response).sendError(eq(HttpServletResponse.SC_NOT_FOUND), argThat(notNullValue(String.class)));
129 public void Given_Request_Is_HTTP_DELETE_And_Feed_Is_Not_Valid_Then_Not_Found_Error_Is_Returned()
131 when(request.getPathInfo()).thenReturn("/122/fileName.txt");
132 publishServlet.doDelete(request, response);
133 verify(response).sendError(eq(HttpServletResponse.SC_NOT_FOUND), argThat(notNullValue(String.class)));
137 public void Given_Request_Is_HTTP_DELETE_And_Feed_Is_Not_A_Number_Then_Not_Found_Error_Is_Returned()
139 when(request.getPathInfo()).thenReturn("/abc/fileName.txt");
140 publishServlet.doDelete(request, response);
141 verify(response).sendError(eq(HttpServletResponse.SC_NOT_FOUND), argThat(notNullValue(String.class)));
146 public void Given_Request_Is_HTTP_DELETE_And_All_Ok_Then_Request_succeeds()
148 when(request.getHeader(anyString())).thenReturn("Basic dXNlcg==");
149 setConditionsForPositiveSuccessFlow();
150 publishServlet.doDelete(request, response);
151 verify(response).setStatus(eq(HttpServletResponse.SC_MOVED_PERMANENTLY));
152 verifyEnteringExitCalled(listAppender);
156 public void Given_Request_Is_HTTP_PUT_And_Request_succeeds()
158 setConditionsForPositiveSuccessFlow();
160 publishServlet.doPut(request, response);
161 verify(response).setStatus(eq(HttpServletResponse.SC_MOVED_PERMANENTLY));
162 verifyEnteringExitCalled(listAppender);
166 public void Given_Request_Is_HTTP_POST_And_Request_succeeds()
168 setConditionsForPositiveSuccessFlow();
170 publishServlet.doPost(request, response);
171 verify(response).setStatus(eq(HttpServletResponse.SC_MOVED_PERMANENTLY));
172 verifyEnteringExitCalled(listAppender);
176 public void Given_Request_Is_HTTP_GET_And_Request_succeeds_And_RequestId_Header_is_empty()
178 setConditionsForPositiveSuccessFlow();
179 mockStatic(InetAddress.class);
180 publishServlet.doGet(request, response);
181 verify(response).setStatus(eq(HttpServletResponse.SC_MOVED_PERMANENTLY));
182 verifyEnteringExitCalled(listAppender);
183 assertEquals(null, listAppender.list.get(0).getMDCPropertyMap().get("RequestId"));
184 assertEquals(null, listAppender.list.get(0).getMDCPropertyMap().get("InvocationId"));
188 public void Given_Request_Is_HTTP_GET_And_Request_succeeds_And_RequestId_Header_Is_Not_Empty()
190 setConditionsForPositiveSuccessFlow();
191 when(request.getHeader("X-ONAP-RequestID")).thenReturn("123");
192 when(request.getHeader("X-InvocationID")).thenReturn("456");
193 publishServlet.doGet(request, response);
194 verify(response).setStatus(eq(HttpServletResponse.SC_MOVED_PERMANENTLY));
195 verifyEnteringExitCalled(listAppender);
196 assertEquals("123", listAppender.list.get(0).getMDCPropertyMap().get("RequestId"));
197 assertEquals("456", listAppender.list.get(0).getMDCPropertyMap().get("InvocationId"));
200 private void setConditionsForPositiveSuccessFlow() throws Exception {
201 FieldUtils.writeDeclaredField(publishServlet, "provstring", "", true);
202 when(request.getPathInfo()).thenReturn("/1/fileName.txt");