1 files changed, 41 insertions, 3 deletions

diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/utils/YamlTemplateParsingHandlerTest.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/utils/YamlTemplateParsingHandlerTest.java
index 65ad5fc1a6..5f85b5134a 100644
--- a/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/utils/YamlTemplateParsingHandlerTest.java
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/utils/YamlTemplateParsingHandlerTest.java
@@ -47,6 +47,7 @@ import org.openecomp.sdc.be.model.ParsedToscaYamlInfo;
 import org.openecomp.sdc.be.model.PolicyDefinition;
 import org.openecomp.sdc.be.model.PolicyTypeDefinition;
 import org.openecomp.sdc.be.model.Resource;
+import org.openecomp.sdc.be.model.Service;
 import org.openecomp.sdc.be.model.UploadArtifactInfo;
 import org.openecomp.sdc.be.model.UploadComponentInstanceInfo;
 import org.openecomp.sdc.be.model.User;
@@ -65,7 +66,9 @@ import java.util.stream.Collectors;
 import org.springframework.test.util.ReflectionTestUtils;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.when;
@@ -149,18 +152,37 @@ public class YamlTemplateParsingHandlerTest {
 
         Resource resource = new Resource();
         ParsedToscaYamlInfo parsedYaml = handler.parseResourceInfoFromYAML(FILE_NAME, resourceYml, new HashMap<>(),
-                csarInfo.extractTypesInfo(), NODE_NAME, resource);
+                csarInfo.extractTypesInfo(), NODE_NAME, resource, getInterfaceTemplateYaml(csarInfo).get());
 
         validateParsedYaml(parsedYaml, NESTED_GROUP_NAME,
                 Lists.newArrayList("heat_file", "description"));
     }
 
     @Test
+    public void parseServicePropertiesInfoFromYamlTest() {
+        String main_template_content = new String(csar.get(MAIN_TEMPLATE_NAME));
+        CsarInfo csarInfo = new CsarInfo(user, CSAR_UUID, csar, RESOURCE_NAME,
+            MAIN_TEMPLATE_NAME, main_template_content, true);
+
+        Service service = new Service();
+        ParsedToscaYamlInfo parsedYaml = handler.parseResourceInfoFromYAML(FILE_NAME, resourceYml, new HashMap<>(),
+            csarInfo.extractTypesInfo(), NODE_NAME, service, getInterfaceTemplateYaml(csarInfo).get());
+
+        assertThat(parsedYaml.getProperties()).isNotNull();
+        assertEquals(5, parsedYaml.getProperties().size());
+        assertTrue(parsedYaml.getProperties().containsKey("skip_post_instantiation_configuration"));
+        assertTrue(parsedYaml.getProperties().containsKey("controller_actor"));
+        assertTrue(parsedYaml.getProperties().containsKey("cds_model_version"));
+        assertTrue(parsedYaml.getProperties().containsKey("cds_model_name"));
+        assertTrue(parsedYaml.getProperties().containsKey("default_software_version"));
+    }
+
+    @Test
     public void parseResourceInfoFromYAMLTest() {
 
         Resource resource = new Resource();
         ParsedToscaYamlInfo parsedYaml = handler.parseResourceInfoFromYAML(FILE_NAME, resourceYml, new HashMap<>(),
-                new HashMap<>(), "", resource);
+                new HashMap<>(), "", resource, null);
         validateParsedYamlWithCapability(parsedYaml);
     }
 
@@ -234,7 +256,7 @@ public class YamlTemplateParsingHandlerTest {
     public void parseResourceWithPoliciesDefined() {
         Resource resource = new Resource();
         ParsedToscaYamlInfo parsedYaml = handler.parseResourceInfoFromYAML(FILE_NAME, resourceYml, new HashMap<>(),
-                new HashMap<>(), "", resource);
+                new HashMap<>(), "", resource, "");
         validateParsedYamlWithPolicies(parsedYaml);
     }
 
@@ -393,4 +415,20 @@ public class YamlTemplateParsingHandlerTest {
         }
         return policyTypeDefinition;
     }
+
+    private Optional<String> getInterfaceTemplateYaml(CsarInfo csarInfo) {
+        String[] yamlFile;
+        String interfaceTemplateYaml = "";
+        if (csarInfo.getMainTemplateName().contains(".yml")) {
+            yamlFile = csarInfo.getMainTemplateName().split(".yml");
+            interfaceTemplateYaml = yamlFile[0] + "-interface.yml";
+        } else if (csarInfo.getMainTemplateName().contains(".yaml")) {
+            yamlFile = csarInfo.getMainTemplateName().split(".yaml");
+            interfaceTemplateYaml = yamlFile[0] + "-interface.yaml";
+        }
+        if (csarInfo.getCsar().containsKey(interfaceTemplateYaml)) {
+            return Optional.of(new String(csarInfo.getCsar().get(interfaceTemplateYaml)));
+        }
+        return Optional.empty();
+    }
 }