Added AAI Query Tool

Change-Id: I8110ad860ac4fb4e83a864ab7ab9c31d7f84da60
Issue-ID: HOLMES-160
Signed-off-by: GuangrongFu <fu.guangrong@zte.com.cn>

2 files changed, 323 insertions, 4 deletions

diff --git a/holmes-actions/src/test/java/org/onap/holmes/common/aai/AaiQuery4CcvpnTest.java b/holmes-actions/src/test/java/org/onap/holmes/common/aai/AaiQuery4CcvpnTest.java
new file mode 100644
index 0000000..51cba1c
--- /dev/null
+++ b/holmes-actions/src/test/java/org/onap/holmes/common/aai/AaiQuery4CcvpnTest.java
@@ -0,0 +1,319 @@
+/**
+ * Copyright 2018 ZTE Corporation.
+ * <p>
+ * 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
+ * <p>
+ * http://www.apache.org/licenses/LICENSE-2.0
+ * <p>
+ * 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.
+ */
+
+package org.onap.holmes.common.aai;
+
+import com.alibaba.fastjson.JSONArray;
+import com.alibaba.fastjson.JSONObject;
+import org.easymock.EasyMock;
+import org.junit.*;
+import org.junit.rules.ExpectedException;
+import org.junit.runner.RunWith;
+import org.onap.holmes.common.aai.config.AaiConfig;
+import org.onap.holmes.common.exception.CorrelationException;
+import org.powermock.api.easymock.PowerMock;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+import org.powermock.reflect.Whitebox;
+
+import javax.ws.rs.client.*;
+import javax.ws.rs.client.Invocation.Builder;
+import javax.ws.rs.core.MultivaluedHashMap;
+import javax.ws.rs.core.MultivaluedMap;
+import javax.ws.rs.core.Response;
+import java.io.*;
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+import java.util.HashMap;
+
+import static org.hamcrest.CoreMatchers.equalTo;
+import static org.junit.Assert.assertThat;
+import static org.onap.holmes.common.config.MicroServiceConfig.MSB_ADDR;
+
+
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({ClientBuilder.class, Client.class, Builder.class, WebTarget.class, Response.class})
+public class AaiQuery4CcvpnTest {
+
+    @Rule
+    public ExpectedException thrown = ExpectedException.none();
+
+    private static JSONObject data;
+
+    private static AaiQuery4Ccvpn aai = new AaiQuery4Ccvpn();
+
+    private static MultivaluedMap<String, Object> headers = new MultivaluedHashMap<>();
+    private static Client client;
+    private static WebTarget webTarget;
+    private static Builder builder;
+    private static Response response;
+
+    @BeforeClass
+    static public void beforeClass() {
+        System.setProperty(MSB_ADDR, "127.0.0.1:80");
+
+        File file = new File(AaiQuery4CcvpnTest.class.getClassLoader().getResource("./ccvpn.data.json").getFile());
+        BufferedReader reader = null;
+        try {
+            reader = new BufferedReader(new FileReader(file));
+            StringBuilder sb = new StringBuilder();
+            reader.lines().forEach(l -> sb.append(l));
+            data = JSONObject.parseObject(sb.toString());
+        } catch (FileNotFoundException e) {
+            // Do nothing
+        } catch (IOException e) {
+            // Do nothing
+        } finally {
+            if (reader != null) {
+                try {
+                    reader.close();
+                } catch (IOException e) {
+                    // Do nothing
+                }
+            }
+        }
+
+        headers.add("X-TransactionId", AaiConfig.X_TRANSACTION_ID);
+        headers.add("X-FromAppId", AaiConfig.X_FROMAPP_ID);
+        headers.add("Authorization", AaiConfig.getAuthenticationCredentials());
+        headers.add("Accept", "application/json");
+        Whitebox.setInternalState(aai, "headers", headers);
+    }
+
+    @Before
+    public void before() {
+        PowerMock.mockStatic(ClientBuilder.class);
+        client = PowerMock.createMock(Client.class);
+        webTarget = PowerMock.createMock(WebTarget.class);
+        builder = PowerMock.createMock(Builder.class);
+        response = PowerMock.createMock(Response.class);
+    }
+
+    @After
+    public void after() {
+        PowerMock.resetAll();
+    }
+
+    @Test
+    public void test_getPath() throws NoSuchMethodException, InvocationTargetException, IllegalAccessException {
+        String path = "/aai/v14/business/customers/customer/{global-customer-id}/service-subscriptions/service-subscription/{service-type}/service-instances?service-instance-id={servId}";
+
+        Method method = AaiQuery4Ccvpn.class.getDeclaredMethod("getPath", String.class);
+        method.setAccessible(true);
+
+        String ret = (String) method.invoke(aai, path);
+
+        assertThat(ret, equalTo("/api/aai-business/v14/customers/customer/{global-customer-id}/service-subscriptions/service-subscription/{service-type}/service-instances?service-instance-id={servId}"));
+
+    }
+
+    @Test
+    public void test_getLogicLink_exception() {
+        mockGetMethod();
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.NOT_FOUND).times(2);
+
+        thrown.expect(RuntimeException.class);
+
+        PowerMock.replayAll();
+
+        String linkId = aai.getLogicLink("network-1", "pnf-1", "interface-1", "DOWN");
+
+        PowerMock.verifyAll();
+
+        assertThat(linkId, equalTo("logic-link-1"));
+
+    }
+
+    @Test
+    public void test_getLogicLink() {
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("logic-link"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        PowerMock.replayAll();
+
+        String linkId = aai.getLogicLink("network-1", "pnf-1", "interface-1", "DOWN");
+
+        PowerMock.verifyAll();
+
+        assertThat(linkId, equalTo("logic-link-1"));
+
+    }
+
+    @Test
+    public void test_getServiceInstances_exception() {
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("vpn-binding"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("connectivity"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("service-instance-by-connectivity"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("service-instances-by-service-type"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.NOT_FOUND).times(2);
+
+        thrown.expect(RuntimeException.class);
+
+        PowerMock.replayAll();
+
+        JSONArray instances = aai.getServiceInstances("network-1", "pnf-1", "interface-1", "DOWN");
+
+        PowerMock.verifyAll();
+
+        assertThat(instances, equalTo("logic-link-1"));
+
+    }
+
+    @Test
+    public void test_getServiceInstances() {
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("vpn-binding"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("connectivity"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("service-instance-by-connectivity"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("service-instances-by-service-type"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        PowerMock.replayAll();
+
+        JSONArray instances = aai.getServiceInstances("network-1", "pnf-1", "interface-1", "DOWN");
+
+        PowerMock.verifyAll();
+
+        assertThat(instances.getJSONObject(0).getString("service-instance-id"), equalTo("some id 1"));
+        assertThat(instances.getJSONObject(1).getString("service-instance-id"), equalTo("some id 2"));
+        assertThat(instances.getJSONObject(2).getString("service-instance-id"), equalTo("some id 3"));
+    }
+
+    @Test
+    public void test_getServiceInstances_1() throws CorrelationException {
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("service-instances-by-service-type"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        PowerMock.replayAll();
+
+        JSONArray instances = aai.getServiceInstances("custom-1", "service-type-1");
+
+        PowerMock.verifyAll();
+
+        assertThat(instances.getJSONObject(0).getString("service-instance-id"), equalTo("some id 1"));
+        assertThat(instances.getJSONObject(1).getString("service-instance-id"), equalTo("some id 2"));
+        assertThat(instances.getJSONObject(2).getString("service-instance-id"), equalTo("some id 3"));
+    }
+
+    @Test
+    public void test_getServiceInstances_1_exception() throws CorrelationException {
+        mockGetMethod();
+        EasyMock.expect(response.getEntity()).andReturn(data.getJSONObject("service-instances-by-service-type"));
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.NOT_FOUND).times(2);
+
+        thrown.expect(CorrelationException.class);
+
+        PowerMock.replayAll();
+
+        JSONArray instances = aai.getServiceInstances("custom-1", "service-type-1");
+
+        PowerMock.verifyAll();
+
+        assertThat(instances.getJSONObject(0).getString("service-instance-id"), equalTo("some id 1"));
+        assertThat(instances.getJSONObject(1).getString("service-instance-id"), equalTo("some id 2"));
+        assertThat(instances.getJSONObject(2).getString("service-instance-id"), equalTo("some id 3"));
+    }
+
+    @Test
+    public void test_updateTerminalPointStatus() throws CorrelationException {
+        mockPatchMethod();
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        PowerMock.replayAll();
+
+        aai.updateTerminalPointStatus("network-1", "pnf-1", "if-1", new HashMap<>());
+
+        PowerMock.verifyAll();
+    }
+
+    @Test
+    public void test_updateTerminalPointStatus_exception() throws CorrelationException {
+        mockPatchMethod();
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.NOT_FOUND).times(2);
+
+        thrown.expect(CorrelationException.class);
+
+        PowerMock.replayAll();
+
+        aai.updateTerminalPointStatus("network-1", "pnf-1", "if-1", new HashMap<>());
+
+        PowerMock.verifyAll();
+    }
+
+    @Test
+    public void test_updateLogicLinkStatus() throws CorrelationException {
+        mockPatchMethod();
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.OK);
+
+        PowerMock.replayAll();
+
+        aai.updateLogicLinkStatus("link-1", new HashMap<>());
+
+        PowerMock.verifyAll();
+    }
+
+    @Test
+    public void test_updateLogicLinkStatus_exception() throws CorrelationException {
+        mockPatchMethod();
+        EasyMock.expect(response.getStatusInfo()).andReturn(Response.Status.NOT_FOUND).times(2);
+
+        thrown.expect(CorrelationException.class);
+
+        PowerMock.replayAll();
+
+        aai.updateLogicLinkStatus("link-1", new HashMap<>());
+
+        PowerMock.verifyAll();
+
+    }
+
+    private void mockGetMethod() {
+        initCommonMock();
+        EasyMock.expect(builder.get()).andReturn(response);
+    }
+
+    private void mockPatchMethod() {
+        initCommonMock();
+        EasyMock.expect(builder.method(EasyMock.anyObject(String.class), EasyMock.anyObject(Entity.class))).andReturn(response);
+    }
+
+    private void initCommonMock() {
+        EasyMock.expect(ClientBuilder.newClient()).andReturn(client);
+        EasyMock.expect(client.target(EasyMock.anyObject(String.class))).andReturn(webTarget);
+        EasyMock.expect(webTarget.path(EasyMock.anyObject(String.class))).andReturn(webTarget);
+        EasyMock.expect(webTarget.request()).andReturn(builder);
+        EasyMock.expect(builder.headers(headers)).andReturn(builder);
+    }
+}
\ No newline at end of file
diff --git a/holmes-actions/src/test/java/org/onap/holmes/common/aai/config/AaiConfigTest.java b/holmes-actions/src/test/java/org/onap/holmes/common/aai/config/AaiConfigTest.java
index 6cbc56e..1d3d2e9 100644
--- a/holmes-actions/src/test/java/org/onap/holmes/common/aai/config/AaiConfigTest.java
+++ b/holmes-actions/src/test/java/org/onap/holmes/common/aai/config/AaiConfigTest.java
@@ -24,12 +24,12 @@ public class AaiConfigTest {
 
     @Test
     public void testAaiConfig_get_static_fields() {
-        String aaiVnfAddr = "/aai/v11/network/generic-vnfs/generic-vnf";
-        String aaiVmAddr = "/aai/v11/search/nodes-query?search-node-type=vserver&filter=";
+        String aaiVnfAddr = "/aai/v14/network/generic-vnfs/generic-vnf";
+        String aaiVmAddr = "/aai/v14/search/nodes-query?search-node-type=vserver&filter=";
         String xTransactionId = "9999";
         String xFromAppId = "jimmy-postman";
-        assertThat(aaiVnfAddr, equalTo(AaiConfig.AAI_VNF_ADDR));
-        assertThat(aaiVmAddr, equalTo(AaiConfig.AAI_VM_ADDR));
+        assertThat(aaiVnfAddr, equalTo(AaiConfig.AaiConsts.AAI_VNF_ADDR));
+        assertThat(aaiVmAddr, equalTo(AaiConfig.AaiConsts.AAI_VM_ADDR));
         assertThat(xTransactionId, equalTo(AaiConfig.X_TRANSACTION_ID));
         assertThat(xFromAppId, equalTo(AaiConfig.X_FROMAPP_ID));
     }