2 * Copyright © 2016-2018 European Support Limited
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
8 * 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.
17 package org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.consolidation;
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;
25 import java.util.Objects;
28 import org.apache.commons.collections4.MapUtils;
29 import org.junit.Assert;
30 import org.junit.Test;
32 public class TypeComputeConsolidationDataTest {
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";
40 public void testCollectAllPortsOfEachTypeFromComputesNoPorts() {
41 TypeComputeConsolidationData typeComputeConsolidationData = new TypeComputeConsolidationData();
43 ComputeTemplateConsolidationData computeTemplateConsolidationData = new ComputeTemplateConsolidationData();
45 typeComputeConsolidationData.setComputeTemplateConsolidationData("server_type",
46 computeTemplateConsolidationData);
48 Map<String, List<String>> stringListMap = typeComputeConsolidationData.collectAllPortsOfEachTypeFromComputes();
49 Assert.assertTrue(Objects.nonNull(stringListMap) && MapUtils.isEmpty(stringListMap));
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"));
60 ComputeTemplateConsolidationData computeTemplateConsolidationData = new ComputeTemplateConsolidationData();
61 computeTemplateConsolidationData.setPorts(ports);
63 typeComputeConsolidationData.setComputeTemplateConsolidationData("server_type",
64 computeTemplateConsolidationData);
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);
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"));
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"));
82 ComputeTemplateConsolidationData computeTemplateConsolidationData = new ComputeTemplateConsolidationData();
83 computeTemplateConsolidationData.setPorts(ports);
85 ComputeTemplateConsolidationData computeTemplateConsolidationData1 = new ComputeTemplateConsolidationData();
86 computeTemplateConsolidationData1.setPorts(ports1);
88 TypeComputeConsolidationData typeComputeConsolidationData = new TypeComputeConsolidationData();
89 typeComputeConsolidationData.setComputeTemplateConsolidationData("server_type",
90 computeTemplateConsolidationData);
92 typeComputeConsolidationData.setComputeTemplateConsolidationData("server_type1",
93 computeTemplateConsolidationData1);
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"));
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"));
107 public void testAddComputeTemplateConsolidationData() {
108 TypeComputeConsolidationData consolidationData = new TypeComputeConsolidationData();
109 consolidationData.addComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_1);
111 Set<String> expectedNodeTemplateIds = new HashSet<>();
112 expectedNodeTemplateIds.add(COMPUTE_NODE_TEMPLATE_ID_1);
113 verifyComputeTemplateConsolidationData(consolidationData, COMPUTE_NODE_TEMPLATE_ID_1, expectedNodeTemplateIds);
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);
123 consolidationData.addComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_1);
124 ComputeTemplateConsolidationData secondComputeTemplateConsolidationData =
125 consolidationData.getComputeTemplateConsolidationData(COMPUTE_NODE_TEMPLATE_ID_1);
126 Assert.assertEquals(firstComputeTemplateConsolidationData, secondComputeTemplateConsolidationData);
130 public void testAddDiffConsolidationData_DiffNodeTemplateId() {
131 TypeComputeConsolidationData consolidationData = new TypeComputeConsolidationData();
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);
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);
145 Assert.assertNotEquals(firstComputeTemplateConsolidationData, secondComputeTemplateConsolidationData);
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);
155 verifyGetAllComputeNodeTemplateId(typeComputeConsolidationData, expectedComputeNodeTemplateIds);
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));