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 static org.openecomp.sdc.translator.TestUtils.createInitServiceTemplate;
21 import org.openecomp.sdc.tosca.datatypes.model.NodeTemplate;
22 import org.openecomp.sdc.tosca.datatypes.model.ServiceTemplate;
23 import org.openecomp.sdc.tosca.datatypes.model.TopologyTemplate;
24 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.composition.commands.UnifiedSubstitutionNodeTemplateIdGenerator;
25 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.composition.commands.impl.ComputeNewNodeTemplateIdGenerator;
26 import org.openecomp.sdc.translator.datatypes.heattotosca.unifiedmodel.composition.to.UnifiedCompositionTo;
27 import org.openecomp.sdc.translator.services.heattotosca.impl.resourcetranslation.BaseResourceTranslationTest;
28 import org.junit.Assert;
29 import org.junit.BeforeClass;
30 import org.junit.Test;
32 import java.util.HashMap;
34 import java.util.Optional;
36 public class ComputeNewNodeTemplateIdGeneratorTest {
38 private static final String SERVER_PCM = "server_pcm";
39 private static UnifiedSubstitutionNodeTemplateIdGenerator unifiedSubstitutionNodeTemplateIdGenerator;
42 public static void setItUp(){
43 unifiedSubstitutionNodeTemplateIdGenerator = new ComputeNewNodeTemplateIdGenerator();
47 public void testGenerateNewComputeNodeTemplateId() {
48 UnifiedCompositionTo unifiedCompositionTo = new UnifiedCompositionTo(getTestComputeServiceTemplate(), null, null,
51 nodeTemplateId = unifiedSubstitutionNodeTemplateIdGenerator.generate(unifiedCompositionTo, SERVER_PCM);
52 if (nodeTemplateId.isPresent()) {
53 Assert.assertEquals(nodeTemplateId.get(), "pcm_server");
60 public void testGenerateInvalidOriginalNodeTemplateId() {
61 //node template with id is not present in the service template
62 UnifiedCompositionTo unifiedCompositionTo = new UnifiedCompositionTo(getTestComputeServiceTemplate(), null, null,
65 nodeTemplateId = unifiedSubstitutionNodeTemplateIdGenerator.generate(unifiedCompositionTo, SERVER_PCM +
67 Assert.assertEquals(nodeTemplateId.isPresent(), false);
71 public void testGenerateNullOriginalNodeTemplateId() {
72 UnifiedCompositionTo unifiedCompositionTo = new UnifiedCompositionTo(getTestComputeServiceTemplate(), null, null,
75 nodeTemplateId = unifiedSubstitutionNodeTemplateIdGenerator.generate(unifiedCompositionTo, null);
76 Assert.assertEquals(nodeTemplateId.isPresent(), false);
79 private static ServiceTemplate getTestComputeServiceTemplate() {
80 ServiceTemplate serviceTemplate = createInitServiceTemplate();
81 TopologyTemplate topologyTemplate = serviceTemplate.getTopology_template();
82 NodeTemplate nodeTemplate = new NodeTemplate();
83 nodeTemplate.setType("org.openecomp.resource.vfc.nodes.heat.pcm_server");
84 Map<String, NodeTemplate> nodeTemplateMap = new HashMap<>();
85 nodeTemplateMap.put(SERVER_PCM, nodeTemplate);
86 topologyTemplate.setNode_templates(nodeTemplateMap);
87 return serviceTemplate;