1 files changed, 456 insertions, 0 deletions
diff --git a/src/test/java/org/onap/pomba/contextbuilder/aai/test/RestUtilTest.java b/src/test/java/org/onap/pomba/contextbuilder/aai/test/RestUtilTest.java
new file mode 100644
index 0000000..bbf4c8f
--- /dev/null
+++ b/src/test/java/org/onap/pomba/contextbuilder/aai/test/RestUtilTest.java
@@ -0,0 +1,456 @@
+/*
+ * ============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.aai.test;
+
+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.tomakehurst.wiremock.junit.WireMockRule;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.text.MessageFormat;
+import java.util.List;
+import java.util.UUID;
+import javax.servlet.http.HttpServletRequest;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.onap.aai.restclient.client.RestClient;
+import org.onap.pomba.common.datatypes.ModelContext;
+import org.onap.pomba.common.datatypes.Network;
+import org.onap.pomba.common.datatypes.VFModule;
+import org.onap.pomba.common.datatypes.VM;
+import org.onap.pomba.common.datatypes.VNF;
+import org.onap.pomba.contextbuilder.aai.Application;
+import org.onap.pomba.contextbuilder.aai.exception.AuditException;
+import org.onap.pomba.contextbuilder.aai.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(classes = Application.class)
+@TestPropertySource(properties = { "aai.serviceName=localhost","aai.servicePort=9808", "aai.httpProtocol=http" })
+public class RestUtilTest {
+    @Autowired
+    private RestClient aaiClient;
+    @Autowired
+    private String aaiBaseUrl;
+    @Autowired
+    private String aaiPathToSearchNodeQuery;
+    @Autowired
+    private String aaiBasicAuthorization;
+
+    HttpServletRequest mockHttpServletRequest = mock(HttpServletRequest.class);
+
+    private String partnerName = "RestUtilTest";
+
+    private static final String DEPTH = "?depth=2";
+
+    @Rule
+    public WireMockRule aaiEnricherRule = new WireMockRule(wireMockConfig().port(9808));
+
+    @Test
+    public void testValidateServiceInstanceId() {
+        // Missing ServiceInstanceId or it is null
+        try {
+            RestUtil.validateServiceInstanceId("");
+        } catch (AuditException e) {
+            assertTrue(e.getMessage().contains("Invalid request URL, missing parameter: serviceInstanceId"));
+        }
+
+        try {
+            RestUtil.validateServiceInstanceId(null);
+        } catch (AuditException e) {
+            assertTrue(e.getMessage().contains("Invalid request URL, missing parameter: serviceInstanceId"));
+        }
+
+    }
+
+    @Test
+    public void testIsEmptyJson() {
+        assertTrue(RestUtil.isEmptyJson("{}"));
+        assertTrue(!RestUtil.isEmptyJson("{Not Empty}"));
+    }
+
+    private void addResponse(String path, String classpathResource, WireMockRule thisMock) throws IOException {
+        String payload = readFully(ClassLoader.getSystemResourceAsStream(classpathResource));
+        thisMock.stubFor(get(path).willReturn(okJson(payload)));
+    }
+
+    private String readFully(InputStream in) throws IOException {
+        char[] cbuf = new char[1024];
+        StringBuilder content = new StringBuilder();
+        try (InputStreamReader reader = new InputStreamReader(in, "UTF-8")) {
+            int count;
+            while ((count = reader.read(cbuf)) >= 0) {
+                content.append(cbuf, 0, count);
+            }
+        }
+        return content.toString();
+    }
+
+    private static String generateGetCustomerInfoUrl (String baseURL, String aaiPathToSearchNodeQuery,String serviceInstanceId) {
+        return baseURL + MessageFormat.format(aaiPathToSearchNodeQuery, serviceInstanceId);
+    }
+
+    ////
+    @Test
+    public void testretrieveAaiModelDataFromAaiPnf() throws Exception {
+
+        String requestId = UUID.randomUUID().toString();
+        String serviceInstanceId = "adc3cc2a-c73e-414f-8ddb-367de81300cb"; //match to the test data in junit/queryNodeData-1.json
+
+        String queryNodeUrl = generateGetCustomerInfoUrl("", aaiPathToSearchNodeQuery, serviceInstanceId);
+
+        // 2. simulate the response of AAI (1 vnf and 1 pnf)
+        addResponse( queryNodeUrl,
+        "junit/aai-service-instance.json", aaiEnricherRule);
+
+        // 3. simulate the rsp of VNF
+        addResponse( "/aai/v13/network/generic-vnfs/generic-vnf/8a9ddb25-2e79-449c-a40d-5011bac0da39" + DEPTH,
+        "junit/genericVnfInput.json", aaiEnricherRule);
+
+        // 4. simulate the response of PNF based on the resourceLink in (2)
+        //note: match pnf_id in junit/aai-service-instance.json
+        addResponse( "/aai/v13/network/pnfs/pnf/amdocsPnfName" + DEPTH,
+        "junit/pnfSampleInput.json", aaiEnricherRule);
+
+        when(mockHttpServletRequest.getRemoteAddr()).thenReturn("testretrieveAaiModelDataFromAaiPnf");
+
+        ModelContext modelCtx = RestUtil.retrieveAAIModelData(aaiClient,
+                                                              aaiBaseUrl,
+                                                              aaiPathToSearchNodeQuery,
+                                                              mockHttpServletRequest,
+                                                              requestId,
+                                                              partnerName,
+                                                              serviceInstanceId,
+                                                              aaiBasicAuthorization);
+        assertEquals(modelCtx.getVnfs().size(), 1);
+        assertEquals(modelCtx.getPnfs().size(), 1);
+
+    }
+    ///Verify the relationship serviceInstanceId -> vnf -> vserver
+    @Test
+    public void testretrieveAaiModelDataFromAaiVserverPserver() throws Exception {
+
+        String requestId = UUID.randomUUID().toString();
+        String serviceInstanceId = "adc3cc2a-c73e-414f-8ddb-367de81300cb"; //match to the test data in junit/queryNodeData-1.json
+
+        String queryNodeUrl = generateGetCustomerInfoUrl("", aaiPathToSearchNodeQuery, serviceInstanceId);
+
+        // 2. simulate the response of AAI (1 vnf)
+        addResponse( queryNodeUrl,
+        "junit/aai-service-instance_set2.json", aaiEnricherRule);
+
+        // 3. simulate the rsp of VNF (with 1 vserver)
+        // note: match vnf_id in (2)
+        addResponse( "/aai/v13/network/generic-vnfs/generic-vnf/8a9ddb25-2e79-449c-a40d-5011bac0da39" + DEPTH,
+        "junit/genericVnfInput_set2.json", aaiEnricherRule);
+
+        // 4. simulate the rsp of vserer
+        // note: match to vserver-id to the path of "vserver" in (3)
+        addResponse(
+                "/aai/v13/cloud-infrastructure/cloud-regions/cloud-region/CloudOwner/RegionOne/tenants/tenant"
+                        + "/b49b830686654191bb1e952a74b014ad/vservers/vserver/b494cd6e-b9f3-45e0-afe7-e1d1a5f5d74a" + DEPTH,
+                "junit/aai-vserver.json", aaiEnricherRule);
+
+        // 5. simulate the rsp of pserver
+        // note: match pserver hostname to the path of "pserver" in (4)
+        addResponse(
+                "/aai/v13/cloud-infrastructure/pservers/pserver/mtn96compute.cci.att.com" + DEPTH,
+                "junit/pserverInput_set2.json", aaiEnricherRule);
+
+        when(mockHttpServletRequest.getRemoteAddr()).thenReturn("testretrieveAaiModelDataFromAaiVserverPserver");
+
+        ModelContext modelCtx = RestUtil.retrieveAAIModelData(aaiClient,
+                                                              aaiBaseUrl,
+                                                              aaiPathToSearchNodeQuery,
+                                                              mockHttpServletRequest,
+                                                              requestId,
+                                                              partnerName,
+                                                              serviceInstanceId,
+                                                              aaiBasicAuthorization);
+        // verify results
+        List<VNF> vnfList = modelCtx.getVnfs();
+        assertEquals(vnfList.size(), 1);
+        List<VFModule>  vfModuleList = vnfList.get(0).getVfModules();
+        assertEquals(vfModuleList.size(), 1);
+        List<VM> vmList = vfModuleList.get(0).getVms();
+        assertEquals(vmList.size(), 1);
+        assertEquals(vmList.get(0).getUuid(), "b494cd6e-b9f3-45e0-afe7-e1d1a5f5d74a"); //vserver-id
+        assertEquals(vmList.get(0).getPServer().getName(), "mtn96compute.cci.att.com"); //pserver-name
+    }
+
+    ////
+    @Test
+    public void testretrieveAaiModelDataFromAaiPinterfaceLinterfaceWithPnf() throws Exception {
+
+        String requestId = UUID.randomUUID().toString();
+        String serviceInstanceId = "adc3cc2a-c73e-414f-8ddb-367de81300cb"; //match to the test data in junit/queryNodeData-1.json
+        String queryNodeUrl = generateGetCustomerInfoUrl("", aaiPathToSearchNodeQuery, serviceInstanceId);
+        // 2. simulate the response of AAI (1 vnf and 1 pnf)
+        addResponse( queryNodeUrl,
+        "junit/aai-service-instance.json", aaiEnricherRule);
+
+        // 3. simulate the rsp of VNF
+        addResponse( "/aai/v13/network/generic-vnfs/generic-vnf/8a9ddb25-2e79-449c-a40d-5011bac0da39" + DEPTH,
+        "junit/genericVnfInput.json", aaiEnricherRule);
+
+        // 4. simulate the response of PNF based on the resourceLink in (2)
+        //note: match pnf_id in junit/aai-service-instance.json
+        addResponse( "/aai/v13/network/pnfs/pnf/amdocsPnfName" + DEPTH,
+        "junit/pnfInput_w_pInterface_LInterface.json", aaiEnricherRule);
+
+        when(mockHttpServletRequest
+                .getRemoteAddr())
+                .thenReturn("testretrieveAaiModelDataFromAaiPinterfaceLinterfaceWithPnf");
+
+        ModelContext modelCtx = RestUtil.retrieveAAIModelData(aaiClient,
+                aaiBaseUrl,
+                aaiPathToSearchNodeQuery,
+                mockHttpServletRequest,
+                requestId,
+                partnerName,
+                serviceInstanceId,
+                aaiBasicAuthorization);
+
+        assertEquals(modelCtx.getVnfs().size(), 1);
+        assertEquals(modelCtx.getPnfs().size(), 1);
+        assertEquals(modelCtx.getPnfs().get(0).getPInterfaceList().size(), 1);
+        assertEquals(modelCtx.getPnfs().get(0).getPInterfaceList().get(0).getLInterfaceList().size(), 2);
+        assertEquals(modelCtx.getPnfs().get(0).getPInterfaceList().get(0).getLInterfaceList().get(0).getName(), "junit-l-interface-name5"); //l-interface-name
+        assertEquals(modelCtx.getPnfs().get(0).getPInterfaceList().get(0).getLInterfaceList().get(1).getName(), "junit-l-interface-name6"); //l-interface-name
+    }
+
+    @Test
+    public void testretrieveAaiModelDataFromAaiPinterfaceLinterfaceWithPserver() throws Exception {
+
+        String requestId = UUID.randomUUID().toString();
+        String serviceInstanceId = "adc3cc2a-c73e-414f-8ddb-367de81300cb"; //match to the test data in junit/queryNodeData-1.json
+        String queryNodeUrl = generateGetCustomerInfoUrl("", aaiPathToSearchNodeQuery, serviceInstanceId);
+        // 2. simulate the response of AAI (1 vnf)
+        addResponse( queryNodeUrl,
+        "junit/aai-service-instance_set2.json", aaiEnricherRule);
+
+        // 3. simulate the rsp of VNF (with 1 vserver)
+        // note: match vnf_id in (2)
+        addResponse( "/aai/v13/network/generic-vnfs/generic-vnf/8a9ddb25-2e79-449c-a40d-5011bac0da39" + DEPTH,
+        "junit/genericVnfInput_set2.json", aaiEnricherRule);
+
+        // 4. simulate the rsp of vserer
+        // note: match to vserver-id to the path of "vserver" in (3)
+        addResponse(
+                "/aai/v13/cloud-infrastructure/cloud-regions/cloud-region/CloudOwner/RegionOne/tenants/tenant"
+                        + "/b49b830686654191bb1e952a74b014ad/vservers/vserver/b494cd6e-b9f3-45e0-afe7-e1d1a5f5d74a" + DEPTH,
+                "junit/aai-vserver.json", aaiEnricherRule);
+
+        // 5. simulate the rsp of pserver with P-interface which also contains L-interface.
+        // note: match pserver hostname to the path of "pserver" in (4)
+        addResponse(
+                "/aai/v13/cloud-infrastructure/pservers/pserver/mtn96compute.cci.att.com" + DEPTH,
+                "junit/pserverInput_with_pInterface_LInterface.json", aaiEnricherRule);
+
+        when(mockHttpServletRequest
+                .getRemoteAddr())
+                .thenReturn("testretrieveAaiModelDataFromAaiPinterfaceLinterfaceWithPserver");
+
+        ModelContext modelCtx = RestUtil.retrieveAAIModelData(aaiClient,
+                aaiBaseUrl,
+                aaiPathToSearchNodeQuery,
+                mockHttpServletRequest,
+                requestId,
+                partnerName,
+                serviceInstanceId,
+                aaiBasicAuthorization);
+
+        // verify results
+        List<VNF> vnfList = modelCtx.getVnfs();
+        assertEquals(vnfList.size(), 1);
+        List<VFModule>  vfModuleList = vnfList.get(0).getVfModules();
+        assertEquals(vfModuleList.size(), 1);
+        List<VM> vmList = vfModuleList.get(0).getVms();
+        assertEquals(vmList.size(), 1);
+        assertEquals(vmList.get(0).getPServer().getPInterfaceList().size(), 1);
+        assertEquals(vmList.get(0).getPServer().getPInterfaceList().get(0).getName(), "bdc3cc2a-c73e-414f-7ddb-367de92801cb"); //interface-name
+        assertEquals(vmList.get(0).getPServer().getPInterfaceList().get(0).getLInterfaceList().size(), 2);
+        assertEquals(vmList.get(0).getPServer().getPInterfaceList().get(0).getLInterfaceList().get(0).getName(), "junit-l-interface-name7"); //l-interface-name
+        assertEquals(vmList.get(0).getPServer().getPInterfaceList().get(0).getLInterfaceList().get(1).getName(), "junit-l-interface-name8"); //l-interface-name
+    }
+
+    ///Verify the relationship serviceInstanceId -> l3network
+    @Test
+    public void testretrieveAaiModelDataFromAaiL3NetworkInServiceLevel () throws Exception {
+
+        String requestId = UUID.randomUUID().toString();
+        String serviceInstanceId = "adc3cc2a-c73e-414f-8ddb-367de81300cb"; //match to the test data in junit/queryNodeData-1.json
+        String queryNodeUrl = generateGetCustomerInfoUrl("", aaiPathToSearchNodeQuery, serviceInstanceId);
+        // 2. simulate the response of AAI (1 vnf)
+        addResponse( queryNodeUrl,
+        "junit/aai-service-instance_set3.json", aaiEnricherRule);
+
+        // 3. simulate the rsp of l3-network
+        // note: match to network-id to the path of "l3network" in (2:  aai-service-instance_set3)
+        addResponse(
+                "/aai/v13/network/l3-networks/l3-network/01e8d84a-l3-network-1" + DEPTH,
+                "junit/l3-network-1.json", aaiEnricherRule);
+        addResponse(
+                "/aai/v13/network/l3-networks/l3-network/01e8d84a-l3-network-2" + DEPTH,
+                "junit/l3-network-2.json", aaiEnricherRule);
+
+
+        when(mockHttpServletRequest
+                .getRemoteAddr())
+                .thenReturn("testretrieveAaiModelDataFromAaiL3NetworkInServiceLevel");
+
+        ModelContext modelCtx = RestUtil.retrieveAAIModelData(aaiClient,
+                aaiBaseUrl,
+                aaiPathToSearchNodeQuery,
+                mockHttpServletRequest,
+                requestId,
+                partnerName,
+                serviceInstanceId,
+                aaiBasicAuthorization);
+
+        // verify results
+        List<Network> networkList = modelCtx.getNetworkList();
+        assertEquals(networkList.size(), 2);
+        assertEquals(networkList.get(0).getUuid(), "01e8d84a-l3-network-1");
+        assertEquals(networkList.get(1).getUuid(), "01e8d84a-l3-network-2");
+    }
+
+    ///Verify the relationship serviceInstanceId -> vnf -> l3network
+    @Test
+    public void testretrieveAaiModelDataFromAaiL3NetworkInVnfLevel() throws Exception {
+
+        String requestId = UUID.randomUUID().toString();
+        String serviceInstanceId = "adc3cc2a-c73e-414f-8ddb-367de81300cb"; //match to the test data in junit/queryNodeData-1.json
+        String queryNodeUrl = generateGetCustomerInfoUrl("", aaiPathToSearchNodeQuery, serviceInstanceId);
+        // 2. simulate the response of AAI (1 vnf)
+        addResponse( queryNodeUrl,
+        "junit/aai-service-instance_set2.json", aaiEnricherRule);
+
+        // 3. simulate the rsp of VNF (with 1 vserver)
+        // note: match vnf_id in (2)
+        addResponse( "/aai/v13/network/generic-vnfs/generic-vnf/8a9ddb25-2e79-449c-a40d-5011bac0da39" + DEPTH,
+        "junit/genericVnfInput_set3.json", aaiEnricherRule);
+
+        // 4. simulate the rsp of vserer
+        // note: match to vserver-id to the path of "vserver" in (3)
+        addResponse(
+                "/aai/v13/cloud-infrastructure/cloud-regions/cloud-region/CloudOwner/RegionOne/tenants/tenant"
+                        + "/b49b830686654191bb1e952a74b014ad/vservers/vserver/b494cd6e-b9f3-45e0-afe7-e1d1a5f5d74a" + DEPTH,
+                "junit/aai-vserver-set2.json", aaiEnricherRule);
+
+        // 5. simulate the rsp of l3-network
+        // note: match to network-id to the path of "l3network" in (3:  genericVnfInput_set3)
+        addResponse(
+                "/aai/v13/network/l3-networks/l3-network/01e8d84a-l3-network-1" + DEPTH,
+                "junit/l3-network-1.json", aaiEnricherRule);
+
+        when(mockHttpServletRequest
+                .getRemoteAddr())
+                .thenReturn("testretrieveAaiModelDataFromAaiL3NetworkInVnfLevel");
+
+        ModelContext modelCtx = RestUtil.retrieveAAIModelData(aaiClient,
+                aaiBaseUrl,
+                aaiPathToSearchNodeQuery,
+                mockHttpServletRequest,
+                requestId,
+                partnerName,
+                serviceInstanceId,
+                aaiBasicAuthorization);
+
+        // verify results
+        List<VNF> vnfList = modelCtx.getVnfs();
+        assertEquals(vnfList.size(), 1);
+        List<Network> networkList = vnfList.get(0).getNetworks();
+        assertEquals(networkList.size(), 1);
+        assertEquals(networkList.get(0).getUuid(), "01e8d84a-l3-network-1");
+    }
+
+    @Test
+    public void testretrieveAaiModelDataFromAaiL3NetworkInvModuleLevel() throws Exception {
+
+        String requestId = UUID.randomUUID().toString();
+        String serviceInstanceId = "adc3cc2a-c73e-414f-8ddb-367de81300cb"; //match to the test data in junit/queryNodeData-1.json
+        String queryNodeUrl = generateGetCustomerInfoUrl("", aaiPathToSearchNodeQuery, serviceInstanceId);
+        // 2. simulate the response of AAI (1 vnf)
+        addResponse( queryNodeUrl,
+        "junit/aai-service-instance_set2.json", aaiEnricherRule);
+
+        // 3. simulate the rsp of VNF (with 1 vserver)
+        // note: match vnf_id in (2)
+        addResponse( "/aai/v13/network/generic-vnfs/generic-vnf/8a9ddb25-2e79-449c-a40d-5011bac0da39" + DEPTH,
+        "junit/genericVnfInput_set4.json", aaiEnricherRule);
+
+        // 4. simulate the rsp of vserer
+        // note: match to vserver-id to the path of "vserver" in (3)
+        addResponse(
+                "/aai/v13/cloud-infrastructure/cloud-regions/cloud-region/CloudOwner/RegionOne/tenants/tenant"
+                        + "/b49b830686654191bb1e952a74b014ad/vservers/vserver/b494cd6e-b9f3-45e0-afe7-e1d1a5f5d74a" + DEPTH,
+                "junit/aai-vserver-set2.json", aaiEnricherRule);
+
+        // 5. simulate the rsp of l3-network
+        // note: match to network-id to the path of "l3network" in (3:  genericVnfInput_set4)
+        addResponse(
+                "/aai/v13/network/l3-networks/l3-network/01e8d84a-l3-network-1" + DEPTH,
+                "junit/l3-network-1.json", aaiEnricherRule);
+        addResponse(
+                "/aai/v13/network/l3-networks/l3-network/01e8d84a-l3-network-2" + DEPTH,
+                "junit/l3-network-2.json", aaiEnricherRule);
+
+
+        when(mockHttpServletRequest
+                .getRemoteAddr())
+                .thenReturn("testretrieveAaiModelDataFromAaiL3NetworkInvModuleLevel");
+
+        ModelContext modelCtx = RestUtil.retrieveAAIModelData(aaiClient,
+                aaiBaseUrl,
+                aaiPathToSearchNodeQuery,
+                mockHttpServletRequest,
+                requestId,
+                partnerName,
+                serviceInstanceId,
+                aaiBasicAuthorization);
+        // verify results
+        List<VNF> vnfList = modelCtx.getVnfs();
+        assertEquals(vnfList.size(), 1);
+        List<VFModule>  vfModuleList = vnfList.get(0).getVfModules();
+        assertEquals(vfModuleList.size(), 1);
+
+        List<Network> networkList = vfModuleList.get(0).getNetworks();
+        assertEquals(networkList.size(), 2);
+        assertEquals(networkList.get(0).getUuid(), "01e8d84a-l3-network-1");
+        assertEquals(networkList.get(1).getUuid(), "01e8d84a-l3-network-2");
+    }
+
+}