Add UI support for adding tosca artifact types
[sdc.git] / catalog-be / src / test / java / org / openecomp / sdc / be / tosca / builder / ToscaRelationshipBuilderTest.java
1 /*
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2021 Nordix Foundation
4  *  ================================================================================
5  *  Licensed under the Apache License, Version 2.0 (the "License");
6  *  you may not use this file except in compliance with the License.
7  *  You may obtain a copy of the License at
8  *
9  *        http://www.apache.org/licenses/LICENSE-2.0
10  *  Unless required by applicable law or agreed to in writing, software
11  *  distributed under the License is distributed on an "AS IS" BASIS,
12  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  *  See the License for the specific language governing permissions and
14  *  limitations under the License.
15  *
16  *  SPDX-License-Identifier: Apache-2.0
17  *  ============LICENSE_END=========================================================
18  */
19
20 package org.openecomp.sdc.be.tosca.builder;
21
22 import static org.junit.jupiter.api.Assertions.assertEquals;
23 import static org.junit.jupiter.api.Assertions.assertFalse;
24 import static org.junit.jupiter.api.Assertions.assertNotNull;
25 import static org.junit.jupiter.api.Assertions.assertNull;
26 import static org.junit.jupiter.api.Assertions.assertTrue;
27
28 import java.util.ArrayList;
29 import java.util.List;
30 import java.util.Map;
31 import org.junit.jupiter.api.Test;
32 import org.openecomp.sdc.be.datatypes.elements.ArtifactDataDefinition;
33 import org.openecomp.sdc.be.model.CapabilityRequirementRelationship;
34 import org.openecomp.sdc.be.model.RelationshipImpl;
35 import org.openecomp.sdc.be.model.RelationshipInfo;
36 import org.openecomp.sdc.be.tosca.model.ToscaInterfaceDefinition;
37 import org.openecomp.sdc.be.tosca.model.ToscaOperationAssignment;
38 import org.openecomp.sdc.be.tosca.model.ToscaPropertyAssignment;
39 import org.openecomp.sdc.be.tosca.model.ToscaRelationship;
40 import org.openecomp.sdc.be.ui.model.OperationUi;
41 import org.openecomp.sdc.be.ui.model.PropertyAssignmentUi;
42
43 class ToscaRelationshipBuilderTest {
44
45     @Test
46     void testBuildFromCapabilityRequirementRelationshipSuccess() {
47         final CapabilityRequirementRelationship relationship = new CapabilityRequirementRelationship();
48         final RelationshipImpl relationshipImpl = new RelationshipImpl();
49         final String relationshipType = "relationshipType";
50         relationshipImpl.setType(relationshipType);
51         final RelationshipInfo relationshipInfo = new RelationshipInfo();
52         relationshipInfo.setRelationships(relationshipImpl);
53         relationship.setRelation(relationshipInfo);
54
55         final List<OperationUi> operationList = new ArrayList<>();
56         final OperationUi operationUi1 = new OperationUi();
57         operationUi1.setInterfaceType("interfaceType1");
58         operationUi1.setOperationType("operation1");
59         operationUi1.setImplementation(new ArtifactDataDefinition());
60         operationList.add(operationUi1);
61
62         final OperationUi operationUi2 = new OperationUi();
63         operationUi2.setInterfaceType("interfaceType1");
64         operationUi2.setOperationType("operation2");
65         operationUi2.setImplementation(new ArtifactDataDefinition());
66         operationList.add(operationUi2);
67         final List<PropertyAssignmentUi> operation2InputList = new ArrayList<>();
68         final PropertyAssignmentUi propertyAssignmentUi1 = new PropertyAssignmentUi();
69         propertyAssignmentUi1.setValue("propertyAssignmentUi1Value");
70         propertyAssignmentUi1.setType("string");
71         propertyAssignmentUi1.setName("propertyAssignmentUi1");
72         operation2InputList.add(propertyAssignmentUi1);
73         final PropertyAssignmentUi propertyAssignmentUi2 = new PropertyAssignmentUi();
74         propertyAssignmentUi2.setValue("propertyAssignmentUi2Value");
75         propertyAssignmentUi2.setType("string");
76         propertyAssignmentUi2.setName("propertyAssignmentUi2");
77         operation2InputList.add(propertyAssignmentUi2);
78         operationUi2.setInputs(operation2InputList);
79
80         final OperationUi operationUi3 = new OperationUi();
81         operationUi3.setInterfaceType("interfaceType2");
82         operationUi3.setOperationType("operation1");
83         operationUi3.setImplementation(new ArtifactDataDefinition());
84         operationList.add(operationUi3);
85
86         relationship.setOperations(operationList);
87
88
89         final ToscaRelationship toscaRelationship = new ToscaRelationshipBuilder().from(relationship);
90         assertEquals(toscaRelationship.getType(), relationshipType);
91         final Map<String, ToscaInterfaceDefinition> interfaceMap = toscaRelationship.getInterfaces();
92         assertNotNull(interfaceMap);
93         assertFalse(interfaceMap.isEmpty());
94         assertEquals(2, interfaceMap.size());
95         final ToscaInterfaceDefinition toscaInterfaceDefinition = interfaceMap.get(operationUi1.getInterfaceType());
96         assertNull(toscaInterfaceDefinition.getType());
97         assertNotNull(toscaInterfaceDefinition.getOperations());
98         assertEquals(2, toscaInterfaceDefinition.getOperations().size());
99         final Object actualOperation1Obj = toscaInterfaceDefinition.getOperations().get(operationUi1.getOperationType());
100         assertTrue(actualOperation1Obj instanceof ToscaOperationAssignment);
101         final ToscaOperationAssignment actualOperation1 = (ToscaOperationAssignment) actualOperation1Obj;
102         assertOperationUi(actualOperation1, operationUi1);
103     }
104
105     private void assertOperationUi(final ToscaOperationAssignment toscaOperationAssignment, final OperationUi operationUi1) {
106         if (operationUi1 == null) {
107             assertNull(toscaOperationAssignment);
108             return;
109         }
110         assertEquals(toscaOperationAssignment.getImplementation(), operationUi1.getImplementation());
111         if (operationUi1.getInputs() == null) {
112             assertNull(toscaOperationAssignment.getInputs());
113             return;
114         }
115         assertEquals(toscaOperationAssignment.getInputs().size(), operationUi1.getInputs().size());
116         operationUi1.getInputs().forEach(propertyAssignmentUi -> {
117             final ToscaPropertyAssignment toscaPropertyAssignment = toscaOperationAssignment.getInputs()
118                 .get(propertyAssignmentUi.getName());
119             assertNotNull(toscaPropertyAssignment);
120             assertEquals(propertyAssignmentUi.getValue(), toscaPropertyAssignment.getValue());
121         });
122     }
123 }