1 files changed, 131 insertions, 0 deletions
diff --git a/vid-app-common/src/test/java/org/onap/vid/services/AAITreeNodeBuilderTest.java b/vid-app-common/src/test/java/org/onap/vid/services/AAITreeNodeBuilderTest.java
new file mode 100644
index 000000000..5eac5f924
--- /dev/null
+++ b/vid-app-common/src/test/java/org/onap/vid/services/AAITreeNodeBuilderTest.java
@@ -0,0 +1,131 @@
+package org.onap.vid.services;
+
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.onap.vid.aai.AaiClientInterface;
+import org.onap.vid.model.aaiTree.AAITreeNode;
+import org.onap.vid.utils.Unchecked;
+import org.testng.Assert;
+import org.testng.annotations.BeforeTest;
+import org.testng.annotations.Test;
+
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.concurrent.ConcurrentLinkedQueue;
+import java.util.concurrent.ConcurrentSkipListSet;
+import java.util.concurrent.ThreadPoolExecutor;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import static java.util.Comparator.comparing;
+import static org.mockito.Mockito.when;
+import static org.onap.vid.services.AAIServiceTree.AAI_TREE_PATHS;
+
+public class AAITreeNodeBuilderTest {
+
+    AAITreeNodeBuilder aaiTreeNodeBuilder;
+
+    @Mock
+    AaiClientInterface aaiClientMock;
+
+    @Mock
+    ThreadPoolExecutor threadPoolMock;
+
+
+    @BeforeTest
+    public void initMocks() {
+        MockitoAnnotations.initMocks(this);
+        aaiTreeNodeBuilder = new AAITreeNodeBuilder(aaiClientMock);
+    }
+
+    @Test
+    public void buildNode_buildGroupNode_NodeIsAsExpected(){
+        ConcurrentSkipListSet<AAITreeNode> nodesAccumulator = new ConcurrentSkipListSet<>(comparing(AAITreeNode::getUniqueNodeKey));
+        ConcurrentLinkedQueue<String> visitedNodes = new ConcurrentLinkedQueue<>();
+        when(aaiClientMock.typedAaiGet(Unchecked.toURI("anyUrl"), JsonNode.class)).thenReturn(createGroupJson());
+
+        AAITreeNode groupNode = aaiTreeNodeBuilder.buildNode("instance-group",
+                "anyUrl",
+                nodesAccumulator,
+                threadPoolMock,
+                visitedNodes,
+                new AtomicInteger(0),
+                AAI_TREE_PATHS).get(0);
+
+        AAITreeNode expectedGroupNode = createExpectedGroupNode();
+        assertNodeIsAsExpected(expectedGroupNode,groupNode);
+    }
+
+    private void assertNodeIsAsExpected(AAITreeNode expectedGroupNode, AAITreeNode groupNode) {
+        Assert.assertEquals(groupNode.getId(), expectedGroupNode.getId());
+        Assert.assertEquals(groupNode.getType(), expectedGroupNode.getType());
+        Assert.assertEquals(groupNode.getName(), expectedGroupNode.getName());
+        Assert.assertEquals(groupNode.getModelVersionId(), expectedGroupNode.getModelVersionId());
+        Assert.assertEquals(groupNode.getModelInvariantId(), expectedGroupNode.getModelInvariantId());
+        Assert.assertEquals(groupNode.getInMaint(), expectedGroupNode.getInMaint());
+        Assert.assertEquals(groupNode.getAdditionalProperties(), expectedGroupNode.getAdditionalProperties());
+    }
+
+    private AAITreeNode createExpectedGroupNode() {
+        AAITreeNode expectedNode = new AAITreeNode();
+        expectedNode.setId("c4fcf022-31a0-470a-b5b8-c18335b7af32");
+        expectedNode.setType("instance-group");
+        expectedNode.setName("Test vE-Flex");
+        expectedNode.setModelVersionId("Test vE-Flex");
+        expectedNode.setModelInvariantId("dd182d7d-6949-4b90-b3cc-5befe400742e");
+        expectedNode.setInMaint(false);
+        HashMap<String,Object> additionalProperties = new HashMap<>();
+        additionalProperties.put("inMaint","false");
+        additionalProperties.put("description","Test vE-Flex instance-group");
+        additionalProperties.put("instance-group-type","ha");
+        additionalProperties.put("instance-group-role","test-IG-role");
+        additionalProperties.put("resource-version","1533315433086");
+        additionalProperties.put("instance-group-function","vTSBC Customer Landing Network Collection");
+        expectedNode.setAdditionalProperties(additionalProperties);
+
+
+        return expectedNode;
+    }
+
+    private JsonNode createGroupJson() {
+        ObjectMapper objectMapper = new ObjectMapper();
+        JsonNode groupNode = null;
+        try {
+            groupNode = objectMapper.readTree("" +
+                    "{" +
+                    "      \"id\": \"c4fcf022-31a0-470a-b5b8-c18335b7af32\"," +
+                    "      \"instance-group-role\": \"test-IG-role\"," +
+                    "      \"description\": \"Test vE-Flex instance-group\"," +
+                    "      \"instance-group-type\": \"ha\"," +
+                    "      \"resource-version\": \"1533315433086\"," +
+                    "      \"instance-group-name\": \"Test vE-Flex\"," +
+                    "      \"model-invariant-id\": \"dd182d7d-6949-4b90-b3cc-5befe400742e\"," +
+                    "      \"model-version-id\": \"Test vE-Flex\"," +
+                    "      \"inMaint\": \"false\"," +
+                    "      \"instance-group-function\": \"vTSBC Customer Landing Network Collection\"," +
+                    "      \"relationship-list\": {" +
+                    "      \"relationship\": [{" +
+                    "       \"related-to\": \"generic-vnf\"," +
+                    "       \"relationship-label\": \"org.onap.relationships.inventory.MemberOf\" ," +
+                    "       \"related-link\": \"/aai/v14/network/generic-vnfs/generic-vnf/8c54c369-2876-4423-9b33-80f783f29082\" ," +
+                    "       \"relationship-data\": [{" +
+                    "        \"relationship-key\": \"generic-vnf.vnf-id\"," +
+                    "        \"relationship-value\": \"8c54c369-2876-4423-9b33-80f783f29082\"" +
+                    "      }" +
+                    "    ]," +
+                    "      \"related-to-property\": [{" +
+                    "      \"property-key\": \"generic-vnf.vnf-name\"," +
+                    "      \"property-value\": \"zrdm5bffad01\"" +
+                    "    }" +
+                    "    ]" +
+                    "    }" +
+                    "    ]" +
+                    "    }" +
+                    "    }");
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+        return groupNode;
+    }
+}