6388620e7c7b8c8943b353f9904947f8cdce8649
[sdc.git] /
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.translator.datatypes.heattotosca.unifiedmodel.consolidation;
18
19 import java.util.Arrays;
20 import java.util.Collection;
21 import java.util.HashMap;
22 import java.util.HashSet;
23 import java.util.List;
24 import java.util.Map;
25 import java.util.Objects;
26 import java.util.Set;
27
28 import org.apache.commons.collections4.MapUtils;
29 import org.junit.Assert;
30 import org.junit.Test;
31
32 public class TypeComputeConsolidationDataTest {
33
34     private static final String COMPUTE_NODE_TEMPLATE_ID_1 = "computeNodeTemplateId1";
35     private static final String COMPUTE_NODE_TEMPLATE_ID_2 = "computeNodeTemplateId2";
36     private static final String SERVER_NETWORKROLE_1_PORT = "server_networkrole_1_port";
37     private static final String SERVER_NETWORKROLE_2_PORT = "server_networkrole_2_port";
38
39     @Test
40     public void testCollectAllPortsOfEachTypeFromComputesNoPorts() {
41         TypeComputeConsolidationData typeComputeConsolidationData = new TypeComputeConsolidationData();
42
43         ComputeTemplateConsolidationData computeTemplateConsolidationData = new ComputeTemplateConsolidationData();
44
45         typeComputeConsolidationData.setComputeTemplateConsolidationData("server_type",
46                 computeTemplateConsolidationData);
47
48         Map<String, List<String>> stringListMap = typeComputeConsolidationData.collectAllPortsOfEachTypeFromComputes();
49         Assert.assertTrue(Objects.nonNull(stringListMap) && MapUtils.isEmpty(stringListMap));
50
51     }
52
53     @Test
54     public void testCollectAllPortsOfEachTypeFromComputes() {
55         TypeComputeConsolidationData typeComputeConsolidationData = new TypeComputeConsolidationData();
56         Map<String, List<String>> ports = new HashMap<>();
57         ports.put(SERVER_NETWORKROLE_1_PORT,
58                 Arrays.asList("server_0_networkrole_1_port", "server_1_networkrole_1_port"));
59
60         ComputeTemplateConsolidationData computeTemplateConsolidationData = new ComputeTemplateConsolidationData();
61         computeTemplateConsolidationData.setPorts(ports);
62
63         typeComputeConsolidationData.setComputeTemplateConsolidationData("server_type",
64                 computeTemplateConsolidationData);
65
66         Map<String, List<String>> stringListMap = typeComputeConsolidationData.collectAllPortsOfEachTypeFromComputes();
67         Assert.assertTrue(stringListMap.containsKey(SERVER_NETWORKROLE_1_PORT)
68                                   && stringListMap.get(SERVER_NETWORKROLE_1_PORT).size() == 2);
69
70     }
71
72     @Test
73     public void testCollectAllPortsOfEachTypeFromComputesWithMultipleCompute() {
74         Map<String, List<String>> ports = new HashMap<>();
75         ports.put(SERVER_NETWORKROLE_1_PORT,
76                 Arrays.asList("server_0_networkrole_1_port_1", "server_1_networkrole_1_port_2"));
77
78         Map<String, List<String>> ports1 = new HashMap<>();
79         ports1.put(SERVER_NETWORKROLE_2_PORT,
80                 Arrays.asList("server_0_networkrole_2_port_1", "server_1_networkrole_2_port_2"));
81
82         ComputeTemplateConsolidationData computeTemplateConsolidationData = new ComputeTemplateConsolidationData();
83         computeTemplateConsolidationData.setPorts(ports);
84
85         ComputeTemplateConsolidationData computeTemplateConsolidationData1 = new ComputeTemplateConsolidationData();
86         computeTemplateConsolidationData1.setPorts(ports1);
87
88         TypeComputeConsolidationData typeComputeConsolidationData = new TypeComputeConsolidationData();
89         typeComputeConsolidationData.setComputeTemplateConsolidationData("server_type",
90                 computeTemplateConsolidationData);
91
92         typeComputeConsolidationData.setComputeTemplateConsolidationData("server_type1",
93                 computeTemplateConsolidationData1);
94
95         Map<String, List<String>> stringListMap = typeComputeConsolidationData.collectAllPortsOfEachTypeFromComputes();
96         Assert.assertTrue(stringListMap.size() == 2);
97         Assert.assertTrue(stringListMap.get(SERVER_NETWORKROLE_1_PORT).size() == 2);
98         Assert.assertTrue(stringListMap.get(SERVER_NETWORKROLE_1_PORT).contains("server_0_networkrole_1_port_1")
99                                   && stringListMap.get(SERVER_NETWORKROLE_1_PORT).contains("server_1_networkrole_1_port_2"));
100
101         Assert.assertTrue(stringListMap.get(SERVER_NETWORKROLE_2_PORT).size() == 2);
102         Assert.assertTrue(stringListMap.get(SERVER_NETWORKROLE_2_PORT).contains("server_0_networkrole_2_port_1")
103                                   && stringListMap.get(SERVER_NETWORKROLE_2_PORT).contains("server_1_networkrole_2_port_2"));
104     }
105
106     @Test
107     public void testAddComputeTemplateConsolidationData() {
108         TypeComputeConsolidationData consolidationData = new TypeComputeConsolidationData();
109         consolidationData.addComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_1);
110
111         Set<String> expectedNodeTemplateIds =  new HashSet<>();
112         expectedNodeTemplateIds.add(COMPUTE_NODE_TEMPLATE_ID_1);
113         verifyComputeTemplateConsolidationData(consolidationData, COMPUTE_NODE_TEMPLATE_ID_1, expectedNodeTemplateIds);
114     }
115
116     @Test
117     public void testAddSameConsolidationDataTwice_noNewCreated() {
118         TypeComputeConsolidationData consolidationData = new TypeComputeConsolidationData();
119         consolidationData.addComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_1);
120         ComputeTemplateConsolidationData firstComputeTemplateConsolidationData =
121                 consolidationData.getComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_1);
122
123         consolidationData.addComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_1);
124         ComputeTemplateConsolidationData secondComputeTemplateConsolidationData =
125                 consolidationData.getComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_1);
126         Assert.assertEquals(firstComputeTemplateConsolidationData, secondComputeTemplateConsolidationData);
127     }
128
129     @Test
130     public void testAddDiffConsolidationData_DiffNodeTemplateId() {
131         TypeComputeConsolidationData consolidationData = new TypeComputeConsolidationData();
132
133         final ComputeTemplateConsolidationData firstComputeTemplateConsolidationData = consolidationData
134                 .addComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_1);
135         Set<String> expectedNodeTemplateIds1 =  new HashSet<>();
136         expectedNodeTemplateIds1.add(COMPUTE_NODE_TEMPLATE_ID_1);
137         verifyComputeTemplateConsolidationData(consolidationData, COMPUTE_NODE_TEMPLATE_ID_1, expectedNodeTemplateIds1);
138
139         final ComputeTemplateConsolidationData secondComputeTemplateConsolidationData = consolidationData
140                 .addComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_2);
141         Set<String> expectedNodeTemplateIds2 =  new HashSet<>();
142         expectedNodeTemplateIds2.add(COMPUTE_NODE_TEMPLATE_ID_2);
143         verifyComputeTemplateConsolidationData(consolidationData, COMPUTE_NODE_TEMPLATE_ID_2, expectedNodeTemplateIds2);
144
145         Assert.assertNotEquals(firstComputeTemplateConsolidationData, secondComputeTemplateConsolidationData);
146     }
147
148     private void verifyComputeTemplateConsolidationData(TypeComputeConsolidationData typeComputeConsolidationData,
149             String computeNodeTemplateId, Set<String> expectedComputeNodeTemplateIds) {
150         ComputeTemplateConsolidationData computeTemplateConsolidationData =
151                 typeComputeConsolidationData.getComputeTemplateConsolidationData(computeNodeTemplateId);
152         Assert.assertNotNull(computeTemplateConsolidationData);
153         Assert.assertEquals(computeTemplateConsolidationData.getNodeTemplateId(), computeNodeTemplateId);
154
155         verifyGetAllComputeNodeTemplateId(typeComputeConsolidationData, expectedComputeNodeTemplateIds);
156     }
157
158     private void verifyGetAllComputeNodeTemplateId(TypeComputeConsolidationData typeComputeConsolidationData,
159             Set<String> expectedComputeNodeTemplateIds) {
160         Collection<String> allComputeNodeTemplateIds  = typeComputeConsolidationData.getAllComputeNodeTemplateIds();
161         Assert.assertNotNull(allComputeNodeTemplateIds);
162         Assert.assertTrue(allComputeNodeTemplateIds.containsAll(expectedComputeNodeTemplateIds));
163     }
164 }