Interface operation feature enhancements
[sdc.git] / catalog-be / src / test / java / org / openecomp / sdc / test / utils / InterfaceOperationTestUtils.java
1 /*
2  * Copyright © 2016-2018 European Support Limited
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
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
17 package org.openecomp.sdc.test.utils;
18
19 import java.util.HashMap;
20 import java.util.Map;
21 import org.openecomp.sdc.be.datatypes.elements.ListDataDefinition;
22 import org.openecomp.sdc.be.datatypes.elements.OperationInputDefinition;
23 import org.openecomp.sdc.be.datatypes.elements.OperationOutputDefinition;
24 import org.openecomp.sdc.be.model.ArtifactDefinition;
25 import org.openecomp.sdc.be.model.InterfaceDefinition;
26 import org.openecomp.sdc.be.model.Operation;
27
28 public class InterfaceOperationTestUtils {
29
30     public static Map<String, InterfaceDefinition> createMockInterfaceDefinitionMap(String interfaceId,
31             String operationId) {
32         Map<String, InterfaceDefinition> interfaceDefinitionMap = new HashMap<>();
33         interfaceDefinitionMap.put(interfaceId, createMockInterface(interfaceId, operationId));
34         return interfaceDefinitionMap;
35     }
36
37     public static InterfaceDefinition createMockInterface(String interfaceId, String operationId) {
38         return createInterface(interfaceId, interfaceId, interfaceId, interfaceId, createMockOperationMap(operationId));
39     }
40
41     private static InterfaceDefinition createInterface(String uniqueId, String description, String type,
42             String toscaResourceName, Map<String, Operation> op) {
43         InterfaceDefinition id = new InterfaceDefinition();
44         id.setType(type);
45         id.setDescription(description);
46         id.setUniqueId(uniqueId);
47         id.setToscaResourceName(toscaResourceName);
48         id.setOperationsMap(op);
49         return id;
50     }
51
52     public static Map<String, Operation> createMockOperationMap(String operationId) {
53         Map<String, Operation> operationMap = new HashMap<>();
54         operationMap.put(operationId, createMockOperation(operationId));
55         return operationMap;
56     }
57
58     public static Operation createMockOperation(String operationId) {
59         Operation operation = new Operation();
60         ListDataDefinition<OperationInputDefinition> operationInputDefinitionList = new ListDataDefinition<>();
61         operationInputDefinitionList.add(createMockOperationInputDefinition("inputId"));
62         operation.setInputs(operationInputDefinitionList);
63
64         ListDataDefinition<OperationOutputDefinition> operationOutputDefList = new ListDataDefinition<>();
65         operationOutputDefList.add(createMockOperationOutputDefinition("outputId"));
66         operation.setOutputs(operationOutputDefList);
67
68         operation.setDefinition(false);
69         operation.setName(operationId);
70         operation.setUniqueId(operationId);
71         ArtifactDefinition implementation = new ArtifactDefinition();
72         implementation.setUniqueId("uniqId");
73         implementation.setArtifactUUID("artifactId");
74         operation.setImplementation(implementation);
75         operation.setWorkflowId("workflowId");
76         operation.setWorkflowVersionId("workflowVersionId");
77         return operation;
78     }
79
80     public static OperationInputDefinition createMockOperationInputDefinition(String inputId) {
81         OperationInputDefinition operationInputDefinition = new OperationInputDefinition();
82         operationInputDefinition.setName(inputId);
83         operationInputDefinition.setUniqueId(inputId);
84         operationInputDefinition.setInputId(inputId);
85         operationInputDefinition.setValue(inputId);
86         operationInputDefinition.setDefaultValue(inputId);
87         return operationInputDefinition;
88     }
89
90     public static OperationOutputDefinition createMockOperationOutputDefinition(String outputId) {
91         OperationOutputDefinition operationOutputDefinition = new OperationOutputDefinition();
92         operationOutputDefinition.setName(outputId);
93         operationOutputDefinition.setUniqueId(outputId);
94         operationOutputDefinition.setInputId(outputId);
95         operationOutputDefinition.setValue(outputId);
96         operationOutputDefinition.setDefaultValue(outputId);
97         return operationOutputDefinition;
98     }
99
100     public static Map<String, InterfaceDefinition> createMockInterfaceTypeMap(String interfaceType,
101             String operationType) {
102         Map<String, Operation> operationMap = createMockOperationTypeMap(operationType);
103         Map<String, InterfaceDefinition> interfaceDefinitionMap = new HashMap<>();
104         interfaceDefinitionMap.put(interfaceType,
105                 createInterface(interfaceType, interfaceType, interfaceType, interfaceType, operationMap));
106         return interfaceDefinitionMap;
107     }
108
109     private static Map<String, Operation> createMockOperationTypeMap(String operationType) {
110         Operation operation = new Operation();
111         operation.setUniqueId(operationType);
112         Map<String, Operation> operationMap = new HashMap<>();
113         operationMap.put(operationType, operation);
114         return operationMap;
115     }
116
117 }