gerrit.onap Code Review - sdc.git/blob

   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.Collection;
  20 import java.util.HashSet;
  21 import java.util.Set;
  22 import org.junit.Assert;
  23 import org.junit.Test;
  24
  25 import org.openecomp.sdc.heat.datatypes.model.HeatResourcesTypes;
  26
  27 public class FilePortConsolidationDataTest {
  28
  29     private static final String PORT_NODE_TEMPLATE_ID_1 = "portNodeTemplateId1";
  30     private static final String PORT_NODE_TEMPLATE_ID_2 = "portNodeTemplateId2";
  31     private static final String PORT_RESOURCE_ID_1 = "pcm_port_01";
  32     private static final String PORT_RESOURCE_ID_2 = "pcm_port_02";
  33     private static final String PORT_RESOURCE_TYPE = HeatResourcesTypes.NEUTRON_PORT_RESOURCE_TYPE.getHeatResource();
  34
  35     private final FilePortConsolidationData consolidationData = new FilePortConsolidationData();
  36
  37     @Test
  38     public void testAddPortConsolidationData() {
  39         consolidationData.addPortTemplateConsolidationData(PORT_NODE_TEMPLATE_ID_1, PORT_RESOURCE_ID_1,
  40                 PORT_RESOURCE_TYPE);
  41         Set<String> expectedComputeNodeTypes = new HashSet<>();
  42         expectedComputeNodeTypes.add(PORT_NODE_TEMPLATE_ID_1);
  43         checkComputeConsolidationData(consolidationData, PORT_NODE_TEMPLATE_ID_1, expectedComputeNodeTypes);
  44     }
  45
  46     @Test
  47     public void testAddSameConsolidationDataTwice_noNewCreated() {
  48         consolidationData.addPortTemplateConsolidationData(PORT_NODE_TEMPLATE_ID_1, PORT_RESOURCE_ID_1,
  49                 PORT_RESOURCE_TYPE);
  50         PortTemplateConsolidationData firstPortTemplateConsolidationData =
  51                 consolidationData.getPortTemplateConsolidationData(PORT_NODE_TEMPLATE_ID_1);
  52
  53         consolidationData.addPortTemplateConsolidationData(PORT_NODE_TEMPLATE_ID_1, PORT_RESOURCE_ID_1,
  54                 PORT_RESOURCE_TYPE);
  55         PortTemplateConsolidationData secondPortTemplateConsolidationData =
  56                 consolidationData.getPortTemplateConsolidationData(PORT_NODE_TEMPLATE_ID_1);
  57         Assert.assertEquals(firstPortTemplateConsolidationData, secondPortTemplateConsolidationData);
  58     }
  59
  60     @Test
  61     public void testAddDiffConsolidationData_diffNodeTemplateIds() {
  62         consolidationData.addPortTemplateConsolidationData(PORT_NODE_TEMPLATE_ID_1, PORT_RESOURCE_ID_1,
  63                 PORT_RESOURCE_TYPE);
  64         Set<String> expectedComputeNodeTypes = new HashSet<>();
  65         expectedComputeNodeTypes.add(PORT_NODE_TEMPLATE_ID_1);
  66         checkComputeConsolidationData(consolidationData, PORT_NODE_TEMPLATE_ID_1, expectedComputeNodeTypes);
  67
  68         consolidationData.addPortTemplateConsolidationData(PORT_NODE_TEMPLATE_ID_2, PORT_RESOURCE_ID_2,
  69                 PORT_RESOURCE_TYPE);
  70         expectedComputeNodeTypes.add(PORT_NODE_TEMPLATE_ID_2);
  71         checkComputeConsolidationData(consolidationData, PORT_NODE_TEMPLATE_ID_2, expectedComputeNodeTypes);
  72     }
  73
  74     @Test
  75     public void testAddDiffConsolidationData_diffResourceIds() {
  76         consolidationData.addPortTemplateConsolidationData(PORT_NODE_TEMPLATE_ID_1, PORT_RESOURCE_ID_1,
  77                 PORT_RESOURCE_TYPE);
  78         Set<String> expectedComputeNodeTypes = new HashSet<>();
  79         expectedComputeNodeTypes.add(PORT_NODE_TEMPLATE_ID_1);
  80         checkComputeConsolidationData(consolidationData, PORT_NODE_TEMPLATE_ID_1, expectedComputeNodeTypes);
  81
  82         consolidationData.addPortTemplateConsolidationData(PORT_NODE_TEMPLATE_ID_1, PORT_RESOURCE_ID_2,
  83                 PORT_RESOURCE_TYPE);
  84         checkComputeConsolidationData(consolidationData, PORT_NODE_TEMPLATE_ID_1, expectedComputeNodeTypes);
  85     }
  86
  87     private void checkComputeConsolidationData(FilePortConsolidationData filePortConsolidationData,
  88             String computeNodeTemplateId, Set<String> expectedPortNodeTemplateIds) {
  89         PortTemplateConsolidationData consolidationData =
  90                 filePortConsolidationData.getPortTemplateConsolidationData(computeNodeTemplateId);
  91         Assert.assertNotNull(consolidationData);
  92         checkGetTemplateIds(filePortConsolidationData, expectedPortNodeTemplateIds);
  93     }
  94
  95     private void checkGetTemplateIds(FilePortConsolidationData filePortConsolidationData,
  96             Set<String> expectedPortNodeTemplateIds) {
  97         Collection<String> allPortNodeTemplateIds = filePortConsolidationData.getAllPortNodeTemplateIds();
  98         Assert.assertNotNull(allPortNodeTemplateIds);
  99         Assert.assertEquals(allPortNodeTemplateIds.size(), expectedPortNodeTemplateIds.size());
 100         Assert.assertTrue(allPortNodeTemplateIds.containsAll(expectedPortNodeTemplateIds));
 101     }
 102 }