openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/test/java/org/openecomp/sdc/translator/services/heattotosca/impl/nameextractor/NameExtractorContrailComputeImplTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.translator.services.heattotosca.impl.nameextractor;
  22
  23 import org.junit.Assert;
  24 import org.junit.Test;
  25 import org.openecomp.sdc.heat.datatypes.model.Resource;
  26 import org.openecomp.sdc.translator.services.heattotosca.impl.nameextractor.NameExtractorContrailComputeImpl;
  27
  28 import java.util.HashMap;
  29 import java.util.Map;
  30
  31 /**
  32  * @author SHIRIA
  33  * @since December 21, 2016.
  34  */
  35 public class NameExtractorContrailComputeImplTest{
  36
  37
  38     @Test
  39     public void testStaticGetComputeNodeTypeId() throws Exception {
  40       Resource serviceTemplateResource = new Resource();
  41       serviceTemplateResource.setProperties(new HashMap<>());
  42       serviceTemplateResource.getProperties().put("image_name", "aaaa");
  43       String computeNodeTypeId =
  44           new NameExtractorContrailComputeImpl().extractNodeTypeName(serviceTemplateResource, "123", "123");
  45       Assert.assertEquals("org.openecomp.resource.vfc.nodes.heat.compute_123", computeNodeTypeId);
  46     }
  47
  48     @Test
  49     public void testComputeNodeTypeIdIsTakingFlavorNameIfImageNameIsNotAsNamingConvention()
  50         throws Exception {
  51       Resource serviceTemplateResource = new Resource();
  52       serviceTemplateResource.setProperties(new HashMap<>());
  53       serviceTemplateResource.getProperties().put("image_name", "aaaa");
  54       Map flavor = new HashMap<>();
  55       flavor.put("get_param", "bbb_flavor_name");
  56       serviceTemplateResource.getProperties().put("flavor", flavor);
  57       String computeNodeTypeId =
  58           new NameExtractorContrailComputeImpl().extractNodeTypeName(serviceTemplateResource, "123", "123");
  59       Assert.assertEquals("org.openecomp.resource.vfc.nodes.heat.bbb", computeNodeTypeId);
  60     }
  61
  62     @Test
  63     public void testStaticGetComputeNodeTypeIdByFlavor() throws Exception {
  64       Resource serviceTemplateResource = new Resource();
  65       serviceTemplateResource.setProperties(new HashMap<>());
  66       serviceTemplateResource.getProperties().put("image_name", "aaaa");
  67       serviceTemplateResource.getProperties().put("flavor", "aaaa_flavor_name");
  68       String computeNodeTypeId =
  69           new NameExtractorContrailComputeImpl().extractNodeTypeName(serviceTemplateResource, "123", "123");
  70       Assert.assertEquals("org.openecomp.resource.vfc.nodes.heat.compute_123", computeNodeTypeId);
  71     }
  72
  73
  74     @Test
  75     public void testNamingConventionGetComputeNodeTypeId() throws Exception {
  76       Resource serviceTemplateResource = new Resource();
  77       serviceTemplateResource.setProperties(new HashMap<>());
  78       Map image = new HashMap<>();
  79       image.put("get_param", "bbb_image_name");
  80       serviceTemplateResource.getProperties().put("image_name", image);
  81       String computeNodeTypeId =
  82           new NameExtractorContrailComputeImpl().extractNodeTypeName(serviceTemplateResource, "123", "123");
  83       Assert.assertEquals(computeNodeTypeId, "org.openecomp.resource.vfc.nodes.heat.bbb");
  84     }
  85
  86     @Test
  87     public void testNoNamingConventionGetComputeNodeTypeId() throws Exception {
  88       Resource serviceTemplateResource = new Resource();
  89       serviceTemplateResource.setProperties(new HashMap<>());
  90       Map image = new HashMap<>();
  91       image.put("get_file", "bbb_image");
  92       serviceTemplateResource.getProperties().put("image_name", image);
  93       String computeNodeTypeId =
  94           new NameExtractorContrailComputeImpl().extractNodeTypeName(serviceTemplateResource, "123", "123");
  95       Assert.assertEquals(computeNodeTypeId, "org.openecomp.resource.vfc.nodes.heat.compute_123");
  96     }
  97
  98 }