openecomp-be/lib/openecomp-sdc-translator-lib/openecomp-sdc-translator-core/src/test/java/org/openecomp/sdc/translator/datatypes/heattotosca/unifiedmodel/consolidation/ConsolidationDataTest.java

   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 org.junit.Assert;
  20 import org.junit.Test;
  21 import org.openecomp.sdc.translator.services.heattotosca.ConsolidationEntityType;
  22
  23 import java.util.Optional;
  24
  25 public class ConsolidationDataTest {
  26
  27     private final ConsolidationData consolidationData = new ConsolidationData();
  28
  29     @Test
  30     public void testGetConsolidationDataHandler() {
  31
  32         checkHandlerIsValid(ConsolidationEntityType.COMPUTE, ComputeConsolidationDataHandler.class);
  33         checkHandlerIsValid(ConsolidationEntityType.PORT, PortConsolidationDataHandler.class);
  34         checkHandlerIsValid(ConsolidationEntityType.SUB_INTERFACE, SubInterfaceConsolidationDataHandler.class);
  35         checkHandlerIsValid(ConsolidationEntityType.NESTED, NestedConsolidationDataHandler.class);
  36         checkHandlerIsValid(ConsolidationEntityType.VFC_NESTED, NestedConsolidationDataHandler.class);
  37     }
  38
  39     @Test
  40     public void testGetConsolidationDataHandler_Negative() {
  41         checkHandlerNotExist(ConsolidationEntityType.OTHER);
  42         checkHandlerNotExist(ConsolidationEntityType.VOLUME);
  43     }
  44
  45     private void checkHandlerIsValid(ConsolidationEntityType consolidationEntityType, Class cls) {
  46         Optional<ConsolidationDataHandler> consolidationDataHandler =
  47                 consolidationData.getConsolidationDataHandler(consolidationEntityType);
  48         Assert.assertTrue(consolidationDataHandler.isPresent());
  49         Assert.assertTrue(cls.isInstance(consolidationDataHandler.get()));
  50     }
  51
  52     private void checkHandlerNotExist(ConsolidationEntityType consolidationEntityType) {
  53         Optional<ConsolidationDataHandler> consolidationDataHandler =
  54                 consolidationData.getConsolidationDataHandler(consolidationEntityType);
  55         Assert.assertFalse(consolidationDataHandler.isPresent());
  56     }
  57
  58 }