1 files changed, 157 insertions, 0 deletions
diff --git a/dcaedt_tools/src/test/java/TemplateContainerTest.java b/dcaedt_tools/src/test/java/TemplateContainerTest.java
new file mode 100644
index 0000000..7c3d287
--- /dev/null
+++ b/dcaedt_tools/src/test/java/TemplateContainerTest.java
@@ -0,0 +1,157 @@
+import com.google.gson.JsonObject;
+import json.response.ItemsResponse.Item;
+import json.response.ItemsResponse.Model;
+import json.templateInfo.Composition;
+import json.templateInfo.Relation;
+import json.templateInfo.TemplateInfo;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import tools.TemplateContainer;
+
+import java.util.*;
+
+import static org.mockito.Matchers.anyString;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+
+public class TemplateContainerTest extends BaseTest {
+
+    private TemplateContainer templateContainer;
+
+    private List<TemplateInfo> templateInfos;
+    private Map<String, List<Item>> elementsByFolderNames;
+    @Before
+    @Override
+    public void setup() {
+        super.setup();
+        templateInfos = new ArrayList<>();
+        TemplateInfo templateInfo = new TemplateInfo();
+        templateInfo.setName(TEMPLATE_INFO_NAME);
+        Composition composition = new Composition();
+        composition.setType(ELEMENT_NAME3);
+        composition.setAlias(ALIAS_NAME3);
+        templateInfo.setComposition(Collections.singletonList(composition));
+        templateInfos.add(templateInfo);
+        elementsByFolderNames = new HashMap<>();
+        Item item = new Item();
+        item.setName(ELEMENT_NAME3);
+        item.setItemId("");
+        Model model =  new Model();
+        model.setItemId("");
+        List<Model> models = Collections.singletonList(model);
+        item.setModels(models);
+        List<Item> items = new ArrayList<>(Collections.singletonList(item));
+        elementsByFolderNames.put(TEMPLATE_INFO_NAME, items);
+
+    }
+
+    @Test
+    public void getCdumps_emptyTemplateInfo_returnEmptyMap() {
+        templateContainer = new TemplateContainer(report, dcaeRestClient, new ArrayList<>(), new HashMap<>());
+
+        Map<TemplateInfo, JsonObject> templateInfoJsonObjectMap = templateContainer.getCdumps();
+
+        Assert.assertTrue(templateInfoJsonObjectMap.size() == 0);
+    }
+
+    @Test
+    public void getCdumps_returnNotFoundEmptyList() {
+        elementsByFolderNames = new HashMap<>();
+        Item item = new Item();
+        item.setName(ELEMENT_NAME2);
+        List<Item> items = new ArrayList<>(Collections.singletonList(item));
+        elementsByFolderNames.put(TEMPLATE_INFO_NAME, items);
+        templateContainer = new TemplateContainer(report, dcaeRestClient, templateInfos, elementsByFolderNames);
+
+        Map<TemplateInfo, JsonObject> templateInfoJsonObjectMap = templateContainer.getCdumps();
+
+        verify(report).addErrorMessage(anyString());
+        Assert.assertTrue(templateInfoJsonObjectMap.size() == 0);
+    }
+
+    @Test
+    public void getCdumps_returnOneCdump() {
+        templateContainer = new TemplateContainer(report, dcaeRestClient, templateInfos, elementsByFolderNames);
+
+        Map<TemplateInfo, JsonObject> templateInfoJsonObjectMap = templateContainer.getCdumps();
+        JsonObject jsonObject = templateInfoJsonObjectMap.get(templateInfos.get(0));
+        String result = jsonObject.toString();
+
+        verifyCdump(result);
+        verify(report, times(0)).addErrorMessage(anyString());
+        Assert.assertTrue(templateInfoJsonObjectMap.size() == 1);
+    }
+
+    @Test
+    public void getChumps_returnOneChumpWithRelations() {
+        templateInfos = new ArrayList<>();
+        TemplateInfo templateInfo = new TemplateInfo();
+        templateInfo.setName(TEMPLATE_INFO_NAME);
+        List<Composition> compositionList = new ArrayList<>();
+        Composition composition = new Composition();
+        composition.setType(ELEMENT_NAME3);
+        composition.setAlias(ALIAS_NAME3);
+        compositionList.add(composition);
+        composition = new Composition();
+        composition.setType(ELEMENT_NAME2);
+        composition.setAlias(ALIAS_NAME2);
+        compositionList.add(composition);
+        templateInfo.setComposition(compositionList);
+        Relation relation = new Relation();
+        relation.setFromComponent(ALIAS_NAME3 + ".SomeNameFromRequirement");
+        relation.setToComponent(ALIAS_NAME2 + ".SomeNameToCapability");
+        relation.setFromRequirement("SomeNameFromRequirement");
+        relation.setToCapability("SomeNameToCapability");
+        templateInfo.setRelations(Collections.singletonList(relation));
+        templateInfos.add(templateInfo);
+        elementsByFolderNames = new HashMap<>();
+        List<Item> itemList = new ArrayList<>();
+        Item item = new Item();
+        item.setName(ELEMENT_NAME3);
+        item.setItemId("");
+        Model model =  new Model();
+        model.setItemId("");
+        List<Model> models = Collections.singletonList(model);
+        item.setModels(models);
+        itemList.add(item);
+        item = new Item();
+        item.setName(ELEMENT_NAME2);
+        item.setItemId("");
+        model =  new Model();
+        model.setItemId("");
+        models = Collections.singletonList(model);
+        item.setModels(models);
+        itemList.add(item);
+        elementsByFolderNames.put(TEMPLATE_INFO_NAME, itemList);
+        templateContainer = new TemplateContainer(report, dcaeRestClient, templateInfos, elementsByFolderNames);
+
+        Map<TemplateInfo, JsonObject> templateInfoJsonObjectMap = templateContainer.getCdumps();
+        JsonObject jsonObject = templateInfoJsonObjectMap.get(templateInfos.get(0));
+        String result = jsonObject.toString();
+
+        verifyCdumpRelations(result);
+        verify(report, times(0)).addErrorMessage(anyString());
+        Assert.assertTrue(templateInfoJsonObjectMap.size() == 1);
+    }
+
+    private void verifyCdumpRelations(String result) {
+        Assert.assertTrue(result.contains("p2\":\"SomeNameToCapability\""));
+        Assert.assertTrue(result.contains("\"SomeNameFromRequirement\",null,\"SomeNameToCapability\""));
+        Assert.assertTrue(result.contains("\"relationship\":["));
+        Assert.assertTrue(result.contains("\"n1\":\"n."));
+        Assert.assertTrue(result.contains("\"relations\":[{"));
+        Assert.assertTrue(result.contains("\"name1\":\"SomeNameFromRequirement\""));
+        Assert.assertTrue(result.contains("\"n2\":\"n."));
+        Assert.assertTrue(result.contains("\"p1\":\"SomeNameFromRequirement\""));
+    }
+
+    private void verifyCdump(String result) {
+        String expectedResultStart = "{\"version\":0,\"flowType\":\"templateInfoName\",\"nodes\":[{\"name\":\"SomeNameFromRequirement\",\"description\":\"\",\"id\":\"e45ec9d7-01df-4cb1-896f-aff2a6ca5a8b\",\"nid\":\"n.";
+        String expectedResultMid =  "\",\"capabilities\":[{\"name\":\"SomeNameToCapability\"}],\"requirements\":[{\"name\":\"SomeNameFromRequirement\"}],\"properties\":[{}],\"typeinfo\":{\"itemId\":\"e45ec9d7-01df-4cb1-896f-aff2a6ca5a8b/tosca.dcae.nodes.cdapApp.Map\",\"typeinfo\":\"typeInfo\"},\"type\":{\"name\":\"type\"},\"ndata\":{\"name\":\"n.";
+        String expectedResultEnd = "\",\"label\":\"SomeNameFromRequirement\",\"x\":438,\"y\":435,\"px\":437,\"py\":434,\"ports\":[],\"radius\":30}}],\"inputs\":[],\"outputs\":[],\"relations\":[]}";
+        Assert.assertTrue(result.startsWith(expectedResultStart));
+        Assert.assertTrue(result.contains(expectedResultMid));
+        Assert.assertTrue(result.endsWith(expectedResultEnd));
+    }
+}