diff options
Diffstat (limited to 'openecomp-be/lib/openecomp-sdc-enrichment-lib/openecomp-sdc-enrichment-impl/src/test/java/org/openecomp/sdc/enrichment/impl/tosca/ComponentQuestionnaireDataTest.java')
-rw-r--r-- | openecomp-be/lib/openecomp-sdc-enrichment-lib/openecomp-sdc-enrichment-impl/src/test/java/org/openecomp/sdc/enrichment/impl/tosca/ComponentQuestionnaireDataTest.java | 115 |
1 files changed, 115 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-sdc-enrichment-lib/openecomp-sdc-enrichment-impl/src/test/java/org/openecomp/sdc/enrichment/impl/tosca/ComponentQuestionnaireDataTest.java b/openecomp-be/lib/openecomp-sdc-enrichment-lib/openecomp-sdc-enrichment-impl/src/test/java/org/openecomp/sdc/enrichment/impl/tosca/ComponentQuestionnaireDataTest.java new file mode 100644 index 0000000000..d281604227 --- /dev/null +++ b/openecomp-be/lib/openecomp-sdc-enrichment-lib/openecomp-sdc-enrichment-impl/src/test/java/org/openecomp/sdc/enrichment/impl/tosca/ComponentQuestionnaireDataTest.java @@ -0,0 +1,115 @@ +package org.openecomp.sdc.enrichment.impl.tosca; + +import static org.mockito.Mockito.doReturn; +import static org.openecomp.sdc.enrichment.impl.util.EnrichmentConstants.HIGH_AVAIL_MODE; +import static org.openecomp.sdc.enrichment.impl.util.EnrichmentConstants.MANDATORY; +import static org.openecomp.sdc.enrichment.impl.util.EnrichmentConstants.MAX_INSTANCES; +import static org.openecomp.sdc.enrichment.impl.util.EnrichmentConstants.MIN_INSTANCES; +import static org.openecomp.sdc.enrichment.impl.util.EnrichmentConstants.VFC_NAMING_CODE; + +import org.mockito.InjectMocks; +import org.mockito.Mock; +import org.mockito.MockitoAnnotations; +import org.openecomp.sdc.vendorsoftwareproduct.dao.ComponentDao; +import org.openecomp.sdc.vendorsoftwareproduct.dao.ComponentDependencyModelDao; +import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentDependencyModelEntity; +import org.openecomp.sdc.vendorsoftwareproduct.dao.type.ComponentEntity; +import org.openecomp.sdc.versioning.VersioningManager; +import org.openecomp.sdc.versioning.dao.types.Version; +import org.testng.Assert; +import org.testng.annotations.BeforeMethod; +import org.testng.annotations.Test; + +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +public class ComponentQuestionnaireDataTest { + private static String VSP_ID = "vspId"; + public static final Version VERSION01 = new Version(0, 1); + private static final Version VERSION10 = new Version(1, 0); + + @Mock + private ComponentDao componentDaoMock; + + @Mock + private ComponentDependencyModelDao componentDependencyDaoMock; + + @InjectMocks + private static ComponentQuestionnaireData componentQuestionnaireData; + + @BeforeMethod + public void setUp() throws Exception { + MockitoAnnotations.initMocks(this); + } + + @Test + public void testGetData() { + ComponentEntity componentEntity = new ComponentEntity(VSP_ID, VERSION01,"ID1" ); + componentEntity.setCompositionData("{\"name\": \"org.openecomp.resource.vfc.nodes.heat.be\"," + + "\"displayName\": \"be\", \"vfcCode\": \"be_1\"}"); + componentEntity.setQuestionnaireData + ("{\"highAvailabilityAndLoadBalancing\":{\"isComponentMandatory\" : \"NO\"," + + "\"highAvailabilityMode\":\"geo-activeactive\"},\"compute\":{\"numOfVMs\" " + + ":{\"maximum\" : 5, \"minimum\" : 0}}}"); + + List<ComponentEntity> entitites = new ArrayList<ComponentEntity>(); + entitites.add(componentEntity); + + doReturn(entitites).when(componentDaoMock).listCompositionAndQuestionnaire(VSP_ID, VERSION01); + + final Map<String, Map<String, Object>> propertiesfromCompQuestionnaire = + componentQuestionnaireData.getPropertiesfromCompQuestionnaire(VSP_ID, VERSION01); + + final Map<String, Object> be = propertiesfromCompQuestionnaire.get("be"); + Assert.assertEquals(be.get(VFC_NAMING_CODE) , "be_1"); + Assert.assertEquals(be.get(MANDATORY) ,"NO"); + Assert.assertEquals(be.get(HIGH_AVAIL_MODE) ,"geo-activeactive"); + Assert.assertEquals(be.get(MIN_INSTANCES) ,null); + Assert.assertEquals(be.get(MAX_INSTANCES) ,5); + + final Map<String, String> sourceToTargetComponent = + componentQuestionnaireData.getSourceToTargetComponent(); + + Assert.assertEquals("be", sourceToTargetComponent.get("ID1")); + } + + + @Test + public void testPopulateDepnendency() { + ComponentDependencyModelEntity sourceComponent = new ComponentDependencyModelEntity(VSP_ID, VERSION01,"ID1" ); + sourceComponent.setSourceComponentId("Comp1"); + sourceComponent.setTargetComponentId("Comp2"); + sourceComponent.setRelation("dependsOn"); + + ComponentDependencyModelEntity targetComponent = new ComponentDependencyModelEntity(VSP_ID, + VERSION01,"ID2" ); + targetComponent.setSourceComponentId("Comp1"); + targetComponent.setTargetComponentId("Comp3"); + targetComponent.setRelation("dependsOn"); + + List<ComponentDependencyModelEntity> entitites = new ArrayList<ComponentDependencyModelEntity>(); + entitites.add(sourceComponent); + entitites.add(targetComponent); + + doReturn(entitites).when(componentDependencyDaoMock).list(new ComponentDependencyModelEntity + (VSP_ID, VERSION01, null)); + + final Map<String, String> sourceToTargetComponent = new HashMap<String, String>(); + sourceToTargetComponent.put("Comp1", "fe"); + sourceToTargetComponent.put("Comp2", "be"); + sourceToTargetComponent.put("Comp3", "smp"); + final Map<String, List<String>> dependencies = + componentQuestionnaireData.populateDependencies(VSP_ID, VERSION01, sourceToTargetComponent); + + List<String> expectedTargets = new ArrayList<String>(); + expectedTargets.add("be"); expectedTargets.add("smp"); + + Assert.assertEquals(dependencies.get("fe"), expectedTargets); + + + + } + +} |