import com.google.gson.JsonObject; import json.templateInfo.Composition; import json.templateInfo.NodeToDelete; import json.templateInfo.Relation; import json.templateInfo.TemplateInfo; import org.junit.Assert; import org.junit.Before; import org.junit.Test; import org.onap.sdc.dcae.composition.restmodels.sdc.Resource; 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 templateInfos; private Map> elementsByFolderNames; @Before @Override public void setup() { super.setup(); templateInfos = new ArrayList<>(); TemplateInfo templateInfo = new TemplateInfo(); templateInfo.setName(TEMPLATE_INFO_NAME); templateInfo.setFlowType(TEMPLATE_INFO_FLOWTYPE); Composition composition = new Composition(); composition.setType(ELEMENT_NAME3); composition.setAlias(ALIAS_NAME3); templateInfo.setComposition(Collections.singletonList(composition)); templateInfos.add(templateInfo); elementsByFolderNames = new HashMap<>(); Resource item = new Resource(); item.setName(ELEMENT_NAME3); List 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 templateInfoJsonObjectMap = templateContainer.getCdumps(); Assert.assertTrue(templateInfoJsonObjectMap.size() == 0); } @Test public void getCdumps_returnNotFoundEmptyList() { elementsByFolderNames = new HashMap<>(); Resource item = new Resource(); item.setName(ELEMENT_NAME2); List items = new ArrayList<>(Collections.singletonList(item)); elementsByFolderNames.put(TEMPLATE_INFO_NAME, items); templateContainer = new TemplateContainer(report, dcaeRestClient, templateInfos, elementsByFolderNames); Map 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 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 getCdumpsWithDeleteNode_returnOneCdumpWithDeletedNode() { NodeToDelete nodeToDelete = new NodeToDelete(); nodeToDelete.setNodeName("SomeNameFromRequirement"); nodeToDelete.setType("my element3"); templateInfos.get(0).setNodesToDelete(Collections.singletonList(nodeToDelete)); templateContainer = new TemplateContainer(report, dcaeRestClient, templateInfos, elementsByFolderNames); Map templateInfoJsonObjectMap = templateContainer.getCdumps(); JsonObject jsonObject = templateInfoJsonObjectMap.get(templateInfos.get(0)); String result = jsonObject.toString(); verifyDeletedNodeCdump(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); templateInfo.setFlowType(TEMPLATE_INFO_FLOWTYPE); List 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 itemList = new ArrayList<>(); Resource item = new Resource(); item.setName(ELEMENT_NAME3); itemList.add(item); item = new Resource(); item.setName(ELEMENT_NAME2); itemList.add(item); elementsByFolderNames.put(TEMPLATE_INFO_NAME, itemList); templateContainer = new TemplateContainer(report, dcaeRestClient, templateInfos, elementsByFolderNames); Map 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(",\"name2\":\"my alias2.SomeNameToCapability\",")); Assert.assertTrue(result.contains(",\"name1\":\"my alias3.SomeNameFromRequirement\",")); Assert.assertTrue(result.contains("\"n2\":\"n.")); Assert.assertTrue(result.contains("\"p1\":\"SomeNameFromRequirement\"")); } private void verifyCdump(String result) { String expectedResultStart = "{\"version\":0,\"flowType\":\"templateInfoFlowType\",\"nodes\":[{\"name\":\"my alias3.SomeNameFromRequirement\",\"description\":\"\",\"id\":\"e45ec9d7-01df-4cb1-896f-aff2a6ca5a8b\",\"nid\":\"n."; String expectedResultMid = "\",\"capabilities\":[{\"name\":\"SomeNameToCapability\"}],\"requirements\":[{\"name\":\"SomeNameFromRequirement\"}],\"properties\":[{\"value\":{}}],\"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)); } private void verifyDeletedNodeCdump(String result) { String expectedResult = "{\"version\":0,\"flowType\":\"templateInfoFlowType\",\"nodes\":[],\"inputs\":[],\"outputs\":[],\"relations\":[]}"; Assert.assertEquals(expectedResult, result); } }