Add DataRouter Subscriber

Change-Id: I7f81c3d7249dcfcf00e2c7f3272f478d2346397d
Issue-ID: DCAEGEN2-1079
Signed-off-by: JoeOLeary <joseph.o.leary@est.tech>

1 files changed, 225 insertions, 0 deletions

diff --git a/src/test/java/org/onap/dcaegen2/services/pmmapper/datarouter/DataRouterSubscriberTest.java b/src/test/java/org/onap/dcaegen2/services/pmmapper/datarouter/DataRouterSubscriberTest.java
new file mode 100644
index 0000000..8f73c91
--- /dev/null
+++ b/src/test/java/org/onap/dcaegen2/services/pmmapper/datarouter/DataRouterSubscriberTest.java
@@ -0,0 +1,225 @@
+/*-
+ * ============LICENSE_START=======================================================
+ *  Copyright (C) 2019 Nordix Foundation.
+ * ================================================================================
+ * 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.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ * ============LICENSE_END=========================================================
+ */
+package org.onap.dcaegen2.services.pmmapper.datarouter;
+
+import com.google.gson.GsonBuilder;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParser;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.jupiter.api.Assertions;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.onap.dcaegen2.services.pmmapper.config.BusControllerConfig;
+import org.onap.dcaegen2.services.pmmapper.exceptions.TooManyTriesException;
+import org.onap.dcaegen2.services.pmmapper.model.Event;
+import org.onap.dcaegen2.services.pmmapper.model.EventMetadata;
+import io.undertow.io.Receiver;
+import io.undertow.io.Sender;
+import io.undertow.server.HttpServerExchange;
+import io.undertow.util.StatusCodes;
+
+import org.mockito.invocation.InvocationOnMock;
+import org.mockito.stubbing.Answer;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+
+import java.io.IOException;
+import java.net.HttpURLConnection;
+import java.net.URL;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+
+import static org.mockito.Mockito.RETURNS_DEEP_STUBS;
+import static org.mockito.Mockito.any;
+import static org.mockito.Mockito.anyInt;
+import static org.mockito.Mockito.doAnswer;
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+
+
+@RunWith(PowerMockRunner.class)
+@PrepareForTest(DataRouterSubscriber.class)
+public class DataRouterSubscriberTest {
+
+
+    @Mock
+    private EventReceiver eventReceiver;
+
+    private DataRouterSubscriber objUnderTest;
+
+    @Before
+    public void setUp() {
+        objUnderTest = new DataRouterSubscriber(eventReceiver);
+    }
+
+    @Test
+    public void testStartTooManyTriesWithResponse() throws IOException {
+        PowerMockito.mockStatic(Thread.class);
+
+        URL subURL = mock(URL.class);
+        BusControllerConfig config = new BusControllerConfig();
+        config.setDataRouterSubscribeEndpoint(subURL);
+        HttpURLConnection huc = mock(HttpURLConnection.class, RETURNS_DEEP_STUBS);
+        when(subURL.openConnection()).thenReturn(huc);
+        when(huc.getResponseCode()).thenReturn(300);
+        Assertions.assertThrows(TooManyTriesException.class, () -> objUnderTest.start(config));
+    }
+
+    @Test
+    public void testStartImmediateSuccess() throws IOException, TooManyTriesException, InterruptedException {
+        URL subURL = mock(URL.class);
+        BusControllerConfig config = new BusControllerConfig();
+        config.setDataRouterSubscribeEndpoint(subURL);
+        HttpURLConnection huc = mock(HttpURLConnection.class, RETURNS_DEEP_STUBS);
+        when(subURL.openConnection()).thenReturn(huc);
+        when(huc.getResponseCode()).thenReturn(200);
+        objUnderTest.start(config);
+        verify(huc, times(1)).getResponseCode();
+    }
+
+    @Test
+    public void testStartDelayedSuccess() throws IOException, TooManyTriesException, InterruptedException {
+        PowerMockito.mockStatic(Thread.class);
+
+        URL subURL = mock(URL.class);
+        BusControllerConfig config = new BusControllerConfig();
+        config.setDataRouterSubscribeEndpoint(subURL);
+        HttpURLConnection huc = mock(HttpURLConnection.class, RETURNS_DEEP_STUBS);
+        when(subURL.openConnection()).thenReturn(huc);
+        doAnswer(new Answer() {
+            boolean forceRetry = true;
+
+            @Override
+            public Object answer(InvocationOnMock invocationOnMock) throws Throwable {
+                if (forceRetry) {
+                    forceRetry = false;
+                    throw new IOException();
+                }
+                return 200;
+            }
+        }).when(huc).getResponseCode();
+        objUnderTest.start(config);
+        verify(huc, times(2)).getResponseCode();
+    }
+
+    @Test
+    public void testStartReadTimeout() throws IOException {
+        PowerMockito.mockStatic(Thread.class);
+
+        URL subURL = mock(URL.class);
+        BusControllerConfig config = new BusControllerConfig();
+        config.setDataRouterSubscribeEndpoint(subURL);
+        HttpURLConnection huc = mock(HttpURLConnection.class, RETURNS_DEEP_STUBS);
+        when(subURL.openConnection()).thenReturn(huc);
+        doThrow(new IOException()).when(huc).getResponseCode();
+        Assertions.assertThrows(TooManyTriesException.class, () -> objUnderTest.start(config));
+    }
+
+    @Test
+    public void testRequestInboundLimitedStateServiceUnavailable() throws Exception {
+        HttpServerExchange httpServerExchange = mock(HttpServerExchange.class);
+        Sender responseSender = mock(Sender.class);
+        when(httpServerExchange.setStatusCode(anyInt())).thenReturn(httpServerExchange);
+        when(httpServerExchange.getResponseSender()).thenReturn(responseSender);
+        objUnderTest.setLimited(true);
+        objUnderTest.handleRequest(httpServerExchange);
+        verify(httpServerExchange).setStatusCode(StatusCodes.SERVICE_UNAVAILABLE);
+    }
+
+    @Test
+    public void testRequestInboundLimitedStateServiceNoEmission() throws Exception {
+        HttpServerExchange httpServerExchange = mock(HttpServerExchange.class);
+        Sender responseSender = mock(Sender.class);
+        when(httpServerExchange.setStatusCode(anyInt())).thenReturn(httpServerExchange);
+        when(httpServerExchange.getResponseSender()).thenReturn(responseSender);
+        objUnderTest.setLimited(true);
+        objUnderTest.handleRequest(httpServerExchange);
+        verify(eventReceiver, times(0)).receive(any());
+    }
+
+
+
+    @Test
+    public void testRequestInboundInvalidMetadata() throws Exception {
+        HttpServerExchange httpServerExchange = mock(HttpServerExchange.class, RETURNS_DEEP_STUBS);
+        JsonObject metadata = new JsonParser().parse(new String(Files.readAllBytes(Paths.get("src/test/resources/invalid_metadata.json")))).getAsJsonObject();
+        when(httpServerExchange.getRequestHeaders().get(any(String.class)).get(anyInt())).thenReturn(metadata.toString());
+        when(httpServerExchange.setStatusCode(anyInt())).thenReturn(httpServerExchange);
+        objUnderTest.handleRequest(httpServerExchange);
+        verify(httpServerExchange, times(1)).setStatusCode(StatusCodes.BAD_REQUEST);
+        verify(httpServerExchange.getResponseSender(), times(1)).send("Malformed Metadata.");
+
+    }
+
+    @Test
+    public void testRequestInboundNoMetadata() throws Exception{
+        HttpServerExchange httpServerExchange = mock(HttpServerExchange.class, RETURNS_DEEP_STUBS);
+        Receiver receiver = mock(Receiver.class);
+        when(httpServerExchange.getRequestReceiver()).thenReturn(receiver);
+        when(httpServerExchange.setStatusCode(anyInt())).thenReturn(httpServerExchange);
+
+        doAnswer((Answer<Void>) invocationOnMock -> {
+            Receiver.FullStringCallback callback = invocationOnMock.getArgument(0);
+            callback.handle(httpServerExchange, "");
+            return null;
+        }).when(receiver).receiveFullString(any());
+        doAnswer((Answer<Void>) invocationOnMock -> {
+            Runnable runnable = invocationOnMock.getArgument(0);
+            runnable.run();
+            return null;
+        }).when(httpServerExchange).dispatch(any(Runnable.class));
+        objUnderTest.handleRequest(httpServerExchange);
+        verify(httpServerExchange, times(1)).setStatusCode(StatusCodes.BAD_REQUEST);
+        verify(httpServerExchange.getResponseSender(), times(1)).send("Missing Metadata.");
+
+    }
+
+    @Test
+    public void testRequestInboundSuccess() throws Exception {
+        HttpServerExchange httpServerExchange = mock(HttpServerExchange.class, RETURNS_DEEP_STUBS);
+        Receiver receiver = mock(Receiver.class);
+        when(httpServerExchange.getRequestReceiver()).thenReturn(receiver);
+        String testString = "MESSAGE BODY";
+        JsonObject metadata = new JsonParser().parse(new String(Files.readAllBytes(Paths.get("src/test/resources/valid_metadata.json")))).getAsJsonObject();
+        EventMetadata metadataObj = new GsonBuilder().create().fromJson(metadata, EventMetadata.class);
+
+        when(httpServerExchange.getRequestHeaders().get(any(String.class)).get(anyInt())).thenReturn(metadata.toString());
+        doAnswer((Answer<Void>) invocationOnMock -> {
+            Receiver.FullStringCallback callback = invocationOnMock.getArgument(0);
+            callback.handle(httpServerExchange, testString);
+            return null;
+        }).when(receiver).receiveFullString(any());
+
+        doAnswer((Answer<Void>) invocationOnMock -> {
+            Runnable runnable = invocationOnMock.getArgument(0);
+            runnable.run();
+            return null;
+        }).when(httpServerExchange).dispatch(any(Runnable.class));
+
+        objUnderTest.handleRequest(httpServerExchange);
+        verify(eventReceiver, times(1)).receive(new Event(httpServerExchange, testString, metadataObj));
+    }
+}