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.services.heattotosca.unifiedmodel.commands;
19 import org.junit.Assert;
20 import org.junit.BeforeClass;
21 import org.junit.Test;
22 import org.onap.sdc.tosca.datatypes.model.NodeTemplate;
23 import org.onap.sdc.tosca.datatypes.model.ServiceTemplate;
24 import org.onap.sdc.tosca.datatypes.model.TopologyTemplate;
25 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.composition.commands.UnifiedSubstitutionNodeTemplateIdGenerator;
26 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.composition.commands.impl.ComputeNewNodeTemplateIdGenerator;
27 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.composition.to.UnifiedCompositionTo;
29 import java.util.HashMap;
31 import java.util.Optional;
33 import static org.openecomp.sdc.translator.TestUtils.createInitServiceTemplate;
35 public class ComputeNewNodeTemplateIdGeneratorTest {
37 private static final String SERVER_PCM = "server_pcm";
38 private static UnifiedSubstitutionNodeTemplateIdGenerator unifiedSubstitutionNodeTemplateIdGenerator;
41 public static void setItUp(){
42 unifiedSubstitutionNodeTemplateIdGenerator = new ComputeNewNodeTemplateIdGenerator();
46 public void testGenerateNewComputeNodeTemplateId() {
47 UnifiedCompositionTo unifiedCompositionTo = new UnifiedCompositionTo(getTestComputeServiceTemplate(), null, null,
50 nodeTemplateId = unifiedSubstitutionNodeTemplateIdGenerator.generate(unifiedCompositionTo, SERVER_PCM);
51 if (nodeTemplateId.isPresent()) {
52 Assert.assertEquals(nodeTemplateId.get(), "pcm_server");
59 public void testGenerateInvalidOriginalNodeTemplateId() {
60 //node template with id is not present in the service template
61 UnifiedCompositionTo unifiedCompositionTo = new UnifiedCompositionTo(getTestComputeServiceTemplate(), null, null,
64 nodeTemplateId = unifiedSubstitutionNodeTemplateIdGenerator.generate(unifiedCompositionTo, SERVER_PCM +
66 Assert.assertEquals(nodeTemplateId.isPresent(), false);
70 public void testGenerateNullOriginalNodeTemplateId() {
71 UnifiedCompositionTo unifiedCompositionTo = new UnifiedCompositionTo(getTestComputeServiceTemplate(), null, null,
74 nodeTemplateId = unifiedSubstitutionNodeTemplateIdGenerator.generate(unifiedCompositionTo, null);
75 Assert.assertEquals(nodeTemplateId.isPresent(), false);
78 private static ServiceTemplate getTestComputeServiceTemplate() {
79 ServiceTemplate serviceTemplate = createInitServiceTemplate();
80 TopologyTemplate topologyTemplate = serviceTemplate.getTopology_template();
81 NodeTemplate nodeTemplate = new NodeTemplate();
82 nodeTemplate.setType("org.openecomp.resource.vfc.nodes.heat.pcm_server");
83 Map<String, NodeTemplate> nodeTemplateMap = new HashMap<>();
84 nodeTemplateMap.put(SERVER_PCM, nodeTemplate);
85 topologyTemplate.setNode_templates(nodeTemplateMap);
86 return serviceTemplate;