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