1 files changed, 95 insertions, 11 deletions

diff --git a/common/onap-tosca-datatype/src/test/java/org/onap/sdc/tosca/datatypes/model/NodeTemplateTest.java b/common/onap-tosca-datatype/src/test/java/org/onap/sdc/tosca/datatypes/model/NodeTemplateTest.java
index e265d4bd8b..3f8e100147 100644
--- a/common/onap-tosca-datatype/src/test/java/org/onap/sdc/tosca/datatypes/model/NodeTemplateTest.java
+++ b/common/onap-tosca-datatype/src/test/java/org/onap/sdc/tosca/datatypes/model/NodeTemplateTest.java
@@ -14,6 +14,7 @@
  * limitations under the License.
  *
  * Modifications copyright (c) 2019 Nokia
+ * Modifications copyright (c) 2021 AT&T Intellectual Property
  */
 
 package org.onap.sdc.tosca.datatypes.model;
@@ -21,10 +22,14 @@ package org.onap.sdc.tosca.datatypes.model;
 import java.io.IOException;
 import java.io.InputStream;
 import java.util.HashMap;
+import java.util.LinkedList;
+import java.util.List;
 import java.util.Map;
 
-import org.junit.Assert;
-import org.junit.Test;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import org.junit.jupiter.api.Test;
 import org.onap.sdc.tosca.services.ToscaExtensionYamlUtil;
 
 import static com.google.code.beanmatchers.BeanMatchers.hasValidBeanEqualsExcluding;
@@ -70,7 +75,7 @@ public class NodeTemplateTest {
 
         String expectedServiceTemplate = toscaExtensionYamlUtil.objectToYaml(expectedServiceTemplateFromYaml);
         String actualServiceTemplate = toscaExtensionYamlUtil.objectToYaml(serviceTemplateForUpdate);
-        Assert.assertEquals(expectedServiceTemplate, actualServiceTemplate);
+        assertEquals(expectedServiceTemplate, actualServiceTemplate);
     }
 
     @Test
@@ -88,6 +93,85 @@ public class NodeTemplateTest {
         assertThat(NodeTemplate.class, hasValidBeanHashCodeExcluding("requirements", "normalizeInterfaces"));
     }
 
+    @Test
+    public void setRequirementsTest() throws IOException {
+        ServiceTemplate serviceTemplateForUpdate = getServiceTemplate(INTERFACE_DEFINITION_FOR_UPD);
+        NodeTemplate nodeTemplate =
+                serviceTemplateForUpdate.getTopology_template().getNode_templates().get(NODE_WITH_INTERFACE);
+        nodeTemplate.addInterface(INTERFACE_KEY, createInterfaceDefinitionTemplate());
+
+        List<RequirementAssignment> requirementAssignmentList = new LinkedList<>();
+        RequirementAssignment requirement1 = new RequirementAssignment();
+        requirement1.setNode("node1");
+        requirement1.setCapability("cap1");
+        requirementAssignmentList.add(requirement1);
+        nodeTemplate.setRequirements(requirementAssignmentList);
+
+        List<Map<String, RequirementAssignment>> res = nodeTemplate.getRequirements();
+        assertNotNull(res);
+        assertEquals(res.size(), 0);
+
+        RequirementAssignment requirement2 = new RequirementAssignment();
+        requirement2.setNode("node2");
+        requirement2.setCapability("cap2");
+        HashMap<String, RequirementAssignment> map = new HashMap<>();
+        map.put("value2", requirement2);
+        nodeTemplate.addRequirements(map);
+        List<Map<String, RequirementAssignment>> res2 = nodeTemplate.getRequirements();
+        assertNotNull(res2);
+        assertEquals(res2.size(), 1);
+        assertEquals(res2.get(0).get("value2"), requirement2);
+    }
+
+    @Test
+    public void addInterfaceTest() throws IOException {
+        ServiceTemplate serviceTemplateForUpdate = getServiceTemplate(INTERFACE_DEFINITION_FOR_UPD);
+        NodeTemplate nodeTemplate =
+                serviceTemplateForUpdate.getTopology_template().getNode_templates().get(NODE_WITH_INTERFACE);
+        nodeTemplate.addInterface(INTERFACE_KEY, createInterfaceDefinitionTemplate());
+
+        Map<String, Object> res = nodeTemplate.getInterfaces();
+        assertEquals(res.size(), 2);
+        assertNotNull(res.get("Standard"));
+        assertNotNull(res.get(INTERFACE_KEY));
+    }
+
+    @Test
+    public void cloneTest() throws IOException {
+        ServiceTemplate serviceTemplateForUpdate = getServiceTemplate(INTERFACE_DEFINITION_FOR_UPD);
+        NodeTemplate nodeTemplate =
+                serviceTemplateForUpdate.getTopology_template().getNode_templates().get(NODE_WITH_INTERFACE);
+
+        NodeTemplate res = nodeTemplate.clone();
+        assertEquals(res, nodeTemplate);
+    }
+
+    @Test
+    public void convertToscaRequirementAssignmentTest() throws IOException {
+        List<?> requirementAssignmentObj = new LinkedList<>();
+        List<Map<String, RequirementAssignment>> res = NodeTemplate.convertToscaRequirementAssignment(requirementAssignmentObj);
+        assertNull(res);
+
+        Map<String, Object> map = new HashMap<>();
+        map.put("value1", new RequirementAssignment());
+        Map<String, Object> requirementMap = new HashMap<>();
+        requirementMap.put("capability", "capabilityValue");
+        requirementMap.put("node", "nodeValue");
+        requirementMap.put("relationship", "relationshipValue");
+        requirementMap.put("node_filter", new NodeFilter());
+        Object[] objectArr = {};
+        requirementMap.put("occurrences", objectArr);
+        map.put("value2", requirementMap);
+        ((List<Map<String, Object>>)requirementAssignmentObj).add(map);
+        List<Map<String, RequirementAssignment>> res2 = NodeTemplate.convertToscaRequirementAssignment(requirementAssignmentObj);
+        assertNotNull(res2);
+        assertEquals(res2.size(), 2);
+        assertEquals(res2.get(0).get("value2").getNode(), "nodeValue");
+        assertEquals(res2.get(0).get("value2").getCapability(), "capabilityValue");
+        assertEquals(res2.get(0).get("value2").getRelationship(), "relationshipValue");
+        assertEquals(res2.get(0).get("value2").getOccurrences().length, 0 );
+    }
+
     private InterfaceDefinitionTemplate createInterfaceDefinitionTemplate() {
         InterfaceDefinitionTemplate interfaceDefinitionTemplate = new InterfaceDefinitionTemplate();
         interfaceDefinitionTemplate.setInputs(new HashMap<>());
@@ -110,16 +194,16 @@ public class NodeTemplateTest {
     }
 
     protected InterfaceDefinitionTemplate chkData(Map<String, InterfaceDefinitionTemplate> normalizeInterfaces) {
-        Assert.assertNotNull(normalizeInterfaces);
+        assertNotNull(normalizeInterfaces);
         InterfaceDefinitionTemplate interfaceDefinitionTemplate = normalizeInterfaces.get(STANDARD_INTERFACE_KEY);
-        Assert.assertNotNull(interfaceDefinitionTemplate);
-        Assert.assertNotNull(interfaceDefinitionTemplate.getInputs());
-        Assert.assertEquals(1, interfaceDefinitionTemplate.getInputs().size());
-        Assert.assertNotNull(interfaceDefinitionTemplate.getOperations());
-        Assert.assertEquals(1, interfaceDefinitionTemplate.getOperations().size());
+        assertNotNull(interfaceDefinitionTemplate);
+        assertNotNull(interfaceDefinitionTemplate.getInputs());
+        assertEquals(1, interfaceDefinitionTemplate.getInputs().size());
+        assertNotNull(interfaceDefinitionTemplate.getOperations());
+        assertEquals(1, interfaceDefinitionTemplate.getOperations().size());
         OperationDefinitionTemplate createOperation = interfaceDefinitionTemplate.getOperations().get(CREATE_OPER);
-        Assert.assertNotNull(createOperation);
-        Assert.assertNotNull(createOperation.getInputs());
+        assertNotNull(createOperation);
+        assertNotNull(createOperation.getInputs());
         return interfaceDefinitionTemplate;
     }