1 files changed, 155 insertions, 0 deletions
diff --git a/src/test/java/org/onap/pomba/contextbuilder/sdnc/unittest/service/SdncContextBuilderTest.java b/src/test/java/org/onap/pomba/contextbuilder/sdnc/unittest/service/SdncContextBuilderTest.java
new file mode 100644
index 0000000..91ab1cf
--- /dev/null
+++ b/src/test/java/org/onap/pomba/contextbuilder/sdnc/unittest/service/SdncContextBuilderTest.java
@@ -0,0 +1,155 @@
+/*
+ * ============LICENSE_START===================================================
+ * Copyright (c) 2018 Amdocs
+ * ============================================================================
+ * 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=====================================================
+ */
+
+package org.onap.pomba.contextbuilder.sdnc.unittest.service;
+
+import static com.github.tomakehurst.wiremock.client.WireMock.get;
+import static com.github.tomakehurst.wiremock.client.WireMock.okJson;
+import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.wireMockConfig;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
+
+import com.github.jknack.handlebars.internal.Files;
+import com.github.tomakehurst.wiremock.junit.WireMockRule;
+import java.io.File;
+import java.util.Collections;
+import javax.ws.rs.core.HttpHeaders;
+import javax.ws.rs.core.MultivaluedMap;
+import javax.ws.rs.core.MultivaluedHashMap;
+import javax.ws.rs.core.Response;
+import javax.ws.rs.core.Response.Status;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.onap.pomba.contextbuilder.sdnc.service.rs.RestService;
+import org.onap.pomba.contextbuilder.sdnc.util.RestUtil;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
+import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
+import org.springframework.boot.autoconfigure.orm.jpa.HibernateJpaAutoConfiguration;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.TestPropertySource;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.test.context.web.WebAppConfiguration;
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@EnableAutoConfiguration(exclude = {DataSourceAutoConfiguration.class, HibernateJpaAutoConfiguration.class})
+@WebAppConfiguration
+@SpringBootTest
+@TestPropertySource(properties = {"sdnc.host=localhost", "sdnc.port=30202"})
+public class SdncContextBuilderTest {
+
+    private String serviceInstanceId = "c6456519-6acf-4adb-997c-3c363dd4caaf";
+    private String testRestHeaders = "testRestHeaders";
+    @Autowired
+    RestService service;
+    @Autowired
+    private String sdncCtxBuilderBasicAuthorization;
+
+    @Rule
+    public WireMockRule sdncRule = new WireMockRule(wireMockConfig().port(30202));
+
+    @Before
+    public void setUp() throws Exception {
+    }
+
+    @After
+    public void tearDown() throws Exception {
+    }
+
+    @Test
+    public void testRestHeaders() throws Exception {
+        HttpHeaders mockHttpHeaders = mock( HttpHeaders.class);
+
+        // Test with No Partner Name
+        final MultivaluedMap<String, String> multivaluedMapImpl = buildHeaders(
+                null, testRestHeaders, sdncCtxBuilderBasicAuthorization);
+
+        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl);
+        Response response = service.getContext(mockHttpHeaders, serviceInstanceId);
+        assertTrue(response.getEntity().toString().contains("Missing header parameter: " + RestUtil.FROM_APP_ID));
+
+        // Test with no Authorization
+        final MultivaluedMap<String, String> multivaluedMapImpl1 = buildHeaders(
+                "test2", testRestHeaders, null);
+
+        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl1);
+        response = service.getContext(mockHttpHeaders, serviceInstanceId);
+        assertTrue(response.getEntity().toString().contains("Missing header parameter: " + RestUtil.AUTHORIZATION));
+
+        // Test with garbage Authorization
+        final MultivaluedMap<String, String> multivaluedMapImpl2 = buildHeaders(
+                "test2", testRestHeaders, "garbage");
+
+        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl2);
+        response = service.getContext(mockHttpHeaders, serviceInstanceId);
+        assertTrue(response.getEntity().toString().contains("Failed Basic " + RestUtil.AUTHORIZATION));
+    }
+
+    @Test
+    public void testRestParameterServiceInstanceId() throws Exception {
+        HttpHeaders mockHttpHeaders = mock( HttpHeaders.class);
+        final MultivaluedMap<String, String> multivaluedMapImpl = buildHeaders(
+                "testRestParameterServiceInstanceId", "test1", sdncCtxBuilderBasicAuthorization);
+
+        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl);
+        Response response = service.getContext(mockHttpHeaders, null);
+        assertTrue(response.getEntity().toString().contains("Invalid request URL, missing parameter: serviceInstanceId"));
+    }
+
+    @Test
+    public void testVerifySdncContextBuilder() throws Exception {
+
+        String urlStr = "/restconf/config/GENERIC-RESOURCE-API:services/service/" + serviceInstanceId;
+        File file = new File(ClassLoader.getSystemResource("sdncResponse.json").getFile());
+        String sdResonse = Files.read(file);
+        this.sdncRule.stubFor(get(urlStr).willReturn(okJson(sdResonse)));
+
+        HttpHeaders mockHttpHeaders = mock( HttpHeaders.class);
+        final MultivaluedMap<String, String> multivaluedMapImpl = buildHeaders(
+                "testVerifyServiceDecomposition", "test1", sdncCtxBuilderBasicAuthorization);
+
+        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl);
+        Response response = this.service.getContext(mockHttpHeaders, serviceInstanceId);
+        assertEquals(Status.OK.getStatusCode(), response.getStatus());
+
+        // Try again with no transcactionId
+        final MultivaluedMap<String, String> multivaluedMapImpl1 = buildHeaders(
+                "testVerifyServiceDecomposition", null, sdncCtxBuilderBasicAuthorization);
+
+        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl1);
+        response = this.service.getContext(mockHttpHeaders, serviceInstanceId);
+        assertEquals(Status.OK.getStatusCode(), response.getStatus());
+    }
+
+    private static MultivaluedMap<String, String> buildHeaders(
+            String partnerName, String transactionId, String authorization) {
+
+        MultivaluedMap<String, String> headers = new MultivaluedHashMap<>();
+        headers.put(RestUtil.FROM_APP_ID, Collections.singletonList(partnerName));
+        headers.put(RestUtil.TRANSACTION_ID, Collections.singletonList(transactionId));
+        if (null != authorization) {
+            headers.put(RestUtil.AUTHORIZATION, Collections.singletonList(authorization));
+        }
+        return headers;
+    }
+}