1 files changed, 214 insertions, 0 deletions
diff --git a/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/PublishServletTest.java b/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/PublishServletTest.java
new file mode 100755
index 00000000..92403ac8
--- /dev/null
+++ b/datarouter-prov/src/test/java/org/onap/dmaap/datarouter/provisioning/PublishServletTest.java
@@ -0,0 +1,214 @@
+/*******************************************************************************
+ * ============LICENSE_START==================================================
+ * * org.onap.dmaap
+ * * ===========================================================================
+ * * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
+ * * ===========================================================================
+ * * Licensed under the Apache License, Version 2.0 (the "License");
+ * * you may not use this file except in compliance with the License.
+ * * You may obtain a copy of the License at
+ * *
+ *  *      http://www.apache.org/licenses/LICENSE-2.0
+ * *
+ *  * Unless required by applicable law or agreed to in writing, software
+ * * distributed under the License is distributed on an "AS IS" BASIS,
+ * * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * * See the License for the specific language governing permissions and
+ * * limitations under the License.
+ * * ============LICENSE_END====================================================
+ * *
+ * * ECOMP is a trademark and service mark of AT&T Intellectual Property.
+ * *
+ ******************************************************************************/
+
+package org.onap.dmaap.datarouter.provisioning;
+
+import org.apache.commons.lang3.reflect.FieldUtils;
+import org.json.JSONObject;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Matchers;
+import org.mockito.Mock;
+import org.onap.dmaap.datarouter.authz.AuthorizationResponse;
+import org.onap.dmaap.datarouter.authz.Authorizer;
+import org.onap.dmaap.datarouter.provisioning.beans.*;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.SuppressStaticInitializationFor;
+import org.powermock.modules.junit4.PowerMockRunner;
+
+import javax.servlet.ServletConfig;
+import javax.servlet.ServletContext;
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.util.*;
+
+import static org.hamcrest.Matchers.notNullValue;
+import static org.mockito.Matchers.*;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+import static org.onap.dmaap.datarouter.provisioning.BaseServlet.BEHALF_HEADER;
+
+/**
+ * Created by ezcoxem on 21/08/2018.
+ */
+
+@RunWith(PowerMockRunner.class)
+@SuppressStaticInitializationFor("org.onap.dmaap.datarouter.provisioning.beans.Feed")
+public class PublishServletTest extends DrServletTestBase {
+    private PublishServlet publishServlet;
+
+    private static String START_JSON_STRING = "{";
+    private static String END_JSON_STRING = "}";
+    private static String START_JSON_ARRAY = "[";
+    private static String END_JSON_ARRAY = "]";
+    private static String COMMA = ",";
+
+    @Mock
+    private HttpServletRequest request;
+
+    @Mock
+    private HttpServletResponse response;
+
+    @Before
+    public void setUp() throws Exception {
+        super.setUp();
+        publishServlet = new PublishServlet();
+    }
+
+    @Test
+    public void Given_Request_Is_HTTP_DELETE_And_There_Are_No_Nodes_Then_Service_Unavailable_Error_Is_Returned()
+            throws Exception {
+        FieldUtils.writeDeclaredStaticField(BaseServlet.class, "nodes", new String[0], true);
+        publishServlet.doDelete(request, response);
+        verify(response).sendError(eq(HttpServletResponse.SC_SERVICE_UNAVAILABLE), argThat(notNullValue(String.class)));
+    }
+
+    @Test
+    public void Given_Request_Is_HTTP_DELETE_And_Path_Is_Null_Then_Not_Found_Error_Is_Returned()
+            throws Exception {
+        FieldUtils.writeDeclaredStaticField(BaseServlet.class, "nodes", new String[1], true);
+        publishServlet.doDelete(request, response);
+        verify(response).sendError(eq(HttpServletResponse.SC_NOT_FOUND), argThat(notNullValue(String.class)));
+    }
+
+    @Test
+    public void Given_Request_Is_HTTP_DELETE_And_Ix_Is_Null_Then_Not_Found_Error_Is_Returned()
+            throws Exception {
+        FieldUtils.writeDeclaredStaticField(BaseServlet.class, "nodes", new String[1], true);
+        when(request.getPathInfo()).thenReturn("/123/");
+        publishServlet.doDelete(request, response);
+        verify(response).sendError(eq(HttpServletResponse.SC_NOT_FOUND), argThat(notNullValue(String.class)));
+    }
+
+    @Test
+    public void Given_Request_Is_HTTP_DELETE_And_Feed_Is_Not_Valid_Then_Not_Found_Error_Is_Returned()
+            throws Exception {
+        FieldUtils.writeDeclaredStaticField(BaseServlet.class, "nodes", new String[1], true);
+        when(request.getPathInfo()).thenReturn("/123/fileName.txt");
+        PowerMockito.mockStatic(Feed.class);
+        PowerMockito.when(Feed.isFeedValid(anyInt())).thenReturn(false);
+        publishServlet.doDelete(request, response);
+        verify(response).sendError(eq(HttpServletResponse.SC_NOT_FOUND), argThat(notNullValue(String.class)));
+    }
+
+    @Test
+    public void Given_Request_Is_HTTP_DELETE_And_Feed_Is_Not_A_Number_Then_Not_Found_Error_Is_Returned()
+            throws Exception {
+        FieldUtils.writeDeclaredStaticField(BaseServlet.class, "nodes", new String[1], true);
+        when(request.getPathInfo()).thenReturn("/abc/fileName.txt");
+        PowerMockito.mockStatic(Feed.class);
+        PowerMockito.when(Feed.isFeedValid(anyInt())).thenReturn(false);
+        publishServlet.doDelete(request, response);
+        verify(response).sendError(eq(HttpServletResponse.SC_NOT_FOUND), argThat(notNullValue(String.class)));
+    }
+
+
+    @Test
+    public void Given_Request_Is_HTTP_DELETE_And_All_Ok_Then_Request_succeeds()
+            throws Exception {
+        setConditionsForPositiveSuccessFlow();
+        when(request.getHeader(anyString())).thenReturn("Basic dXNlcg==");
+        publishServlet.doDelete(request, response);
+        verify(response).setStatus(eq(HttpServletResponse.SC_MOVED_PERMANENTLY));
+    }
+
+    @Test
+    public void Given_Request_Is_HTTP_PUT_And_Request_succeeds()
+            throws Exception {
+        setConditionsForPositiveSuccessFlow();
+
+        publishServlet.doPut(request, response);
+        verify(response).setStatus(eq(HttpServletResponse.SC_MOVED_PERMANENTLY));
+    }
+
+    @Test
+    public void Given_Request_Is_HTTP_POST_And_Request_succeeds()
+            throws Exception {
+        setConditionsForPositiveSuccessFlow();
+
+        publishServlet.doPost(request, response);
+        verify(response).setStatus(eq(HttpServletResponse.SC_MOVED_PERMANENTLY));
+    }
+
+    @Test
+    public void Given_Request_Is_HTTP_GET_And_Request_succeeds()
+            throws Exception {
+        setConditionsForPositiveSuccessFlow();
+
+        publishServlet.doGet(request, response);
+        verify(response).setStatus(eq(HttpServletResponse.SC_MOVED_PERMANENTLY));
+    }
+
+    private void setConditionsForPositiveSuccessFlow() throws Exception {
+        FieldUtils.writeDeclaredStaticField(BaseServlet.class, "nodes", new String[1], true);
+        FieldUtils.writeDeclaredField(publishServlet, "next_node", 0, true);
+        FieldUtils.writeDeclaredField(publishServlet, "provstring", "", true);
+        FieldUtils.writeDeclaredField(publishServlet, "irt", new ArrayList<IngressRoute>(), true);
+        FieldUtils.writeDeclaredStaticField(NodeClass.class, "map", new HashMap<String,String>(), true);
+        when(request.getPathInfo()).thenReturn("/123/fileName.txt");
+        PowerMockito.mockStatic(Feed.class);
+        PowerMockito.when(Feed.isFeedValid(anyInt())).thenReturn(true);
+        setPokerToNotCreateTimersWhenDeleteFeedIsCalled();
+    }
+
+    private void setPokerToNotCreateTimersWhenDeleteFeedIsCalled() throws Exception {
+        Poker poker = mock(Poker.class);
+        FieldUtils.writeDeclaredStaticField(Poker.class, "poker", poker, true);
+        when(poker.getProvisioningString()).thenReturn(buildProvisioningString());
+    }
+
+
+    private String buildProvisioningString(){
+        StringBuffer provisionString = new StringBuffer();
+        provisionString.append(START_JSON_STRING);
+        provisionString.append("'ingress':");
+        provisionString.append(START_JSON_ARRAY);
+        provisionString.append(buildIngressRoute());
+        provisionString.append(END_JSON_ARRAY);
+        provisionString.append(END_JSON_STRING);
+        return provisionString.toString();
+    }
+
+    private StringBuffer buildIngressRoute(){
+        StringBuffer provisionString = new StringBuffer();
+        provisionString.append(START_JSON_STRING);
+        provisionString.append("'seq':1");
+        provisionString.append(COMMA);
+        provisionString.append("'feedid':123");
+        provisionString.append(COMMA);
+        provisionString.append("'user':'user'");
+        provisionString.append(COMMA);
+        provisionString.append("'subnet':'127.0.0.1'");
+        provisionString.append(COMMA);
+        provisionString.append("'nodelist':-1");
+        provisionString.append(COMMA);
+        provisionString.append("'node':['1','2']");
+        provisionString.append(END_JSON_STRING);
+        return provisionString;
+    }
+
+
+}