1 files changed, 40 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-tosca-lib/src/test/java/org/openecomp/sdc/tosca/datatypes/ToscaModelTest.java b/openecomp-be/lib/openecomp-tosca-lib/src/test/java/org/openecomp/sdc/tosca/datatypes/ToscaModelTest.java
index 5412b55129..2833cc456d 100644
--- a/openecomp-be/lib/openecomp-tosca-lib/src/test/java/org/openecomp/sdc/tosca/datatypes/ToscaModelTest.java
+++ b/openecomp-be/lib/openecomp-tosca-lib/src/test/java/org/openecomp/sdc/tosca/datatypes/ToscaModelTest.java
@@ -62,6 +62,8 @@ public class ToscaModelTest {
     private static final String DEPENDENCY_NAME = "script1.sh";
     private static final String STRING_TYPE = "string";
     private static final String ST_WITH_SERVICE_FILTER = "/serviceTemplateWithServiceFilter.yaml";
+    private static final String SUBSTITUTION_MAPPING = "/serviceTemplateWithSubstitutionMapping.yaml";
+    private static final String SUBSTITUTION_MAPPING_EXT = "/substitutionMappingExt.yaml";
 
     @Test
     public void testServiceTemplateJavaToYaml() {
@@ -448,6 +450,44 @@ public class ToscaModelTest {
 
     }
 
+    @Test
+    public void testSubstitutionMapping() throws IOException {
+        ServiceTemplate serviceTemplate = getServiceTemplateExt(BASE_DIR + SUBSTITUTION_MAPPING);
+
+        SubstitutionMapping substitutionMappings = DataModelUtil.getSubstitutionMappings(serviceTemplate);
+        Assert.assertEquals("myNodeType.node", substitutionMappings.getNode_type());
+        Assert.assertNotNull(substitutionMappings.getCapabilities());
+        Assert.assertEquals(1,substitutionMappings.getCapabilities().size());
+        Assert.assertNotNull(substitutionMappings.getRequirements());
+        Assert.assertEquals(1,substitutionMappings.getRequirements().size());
+        Assert.assertEquals(true, substitutionMappings instanceof SubstitutionMappingExt);
+        Assert.assertNull(((SubstitutionMappingExt)substitutionMappings).getSubstitution_filter());
+
+    }
+
+    @Test
+    public void testSubstitutionMappingExt() throws IOException {
+        ServiceTemplate serviceTemplate = getServiceTemplateExt(BASE_DIR + SUBSTITUTION_MAPPING_EXT);
+
+        SubstitutionMapping substitutionMappings = DataModelUtil.getSubstitutionMappings(serviceTemplate);
+        Assert.assertEquals("myNodeType.node", substitutionMappings.getNode_type());
+        Assert.assertNotNull(substitutionMappings.getCapabilities());
+        Assert.assertEquals(1,substitutionMappings.getCapabilities().size());
+        Assert.assertNotNull(substitutionMappings.getRequirements());
+        Assert.assertEquals(1,substitutionMappings.getRequirements().size());
+        Assert.assertEquals(true, substitutionMappings instanceof SubstitutionMappingExt);
+        SubstitutionFilter substitutionFilter = ((SubstitutionMappingExt) substitutionMappings).getSubstitution_filter();
+        Assert.assertNotNull(substitutionFilter);
+        Assert.assertNotNull(substitutionFilter.getProperties());
+        Assert.assertEquals(2,substitutionFilter.getProperties().size());
+        List<Constraint> vendorFilter = substitutionFilter.getProperties().get(0).get("vendor");
+        Assert.assertNotNull(vendorFilter);
+        Assert.assertNotNull(vendorFilter.get(0).getEqual());
+
+
+
+    }
+
     private ServiceTemplate getServiceTemplate(String inputPath) throws IOException {
         try (InputStream yamlFile = yamlUtil.loadYamlFileIs(inputPath)) {
             return yamlUtil.yamlToObject(yamlFile, ServiceTemplate.class);