catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/CapabilityTypeImportManagerTest.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.be.components.impl;
  22
  23 import fj.data.Either;
  24 import org.junit.Before;
  25 import org.junit.BeforeClass;
  26 import org.junit.Test;
  27 import org.mockito.InjectMocks;
  28 import org.mockito.Mockito;
  29 import org.mockito.MockitoAnnotations;
  30 import org.mockito.invocation.InvocationOnMock;
  31 import org.mockito.stubbing.Answer;
  32 import org.openecomp.sdc.be.impl.ComponentsUtils;
  33 import org.openecomp.sdc.be.model.CapabilityTypeDefinition;
  34 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  35 import org.openecomp.sdc.be.model.operations.impl.CapabilityTypeOperation;
  36 import org.openecomp.sdc.common.util.CapabilityTypeNameEnum;
  37 import org.openecomp.sdc.exception.ResponseFormat;
  38
  39 import java.io.IOException;
  40 import java.nio.file.Files;
  41 import java.nio.file.Path;
  42 import java.nio.file.Paths;
  43 import java.util.HashMap;
  44 import java.util.List;
  45 import java.util.Map;
  46
  47 import static org.junit.Assert.assertTrue;
  48 import static org.mockito.Mockito.when;
  49
  50 public class CapabilityTypeImportManagerTest {
  51     @InjectMocks
  52     private CapabilityTypeImportManager manager = new CapabilityTypeImportManager();
  53     public static final CommonImportManager commonImportManager = Mockito.mock(CommonImportManager.class);
  54     public static final CapabilityTypeOperation capabilityTypeOperation = Mockito.mock(CapabilityTypeOperation.class);
  55     public static final ComponentsUtils componentsUtils = Mockito.mock(ComponentsUtils.class);
  56
  57     @BeforeClass
  58     public static void beforeClass() {
  59         when(capabilityTypeOperation.addCapabilityType(Mockito.any(CapabilityTypeDefinition.class))).thenAnswer(new Answer<Either<CapabilityTypeDefinition, StorageOperationStatus>>() {
  60             public Either<CapabilityTypeDefinition, StorageOperationStatus> answer(InvocationOnMock invocation) {
  61                 Object[] args = invocation.getArguments();
  62                 Either<CapabilityTypeDefinition, StorageOperationStatus> ans = Either.left((CapabilityTypeDefinition) args[0]);
  63                 return ans;
  64             }
  65
  66         });
  67         when(commonImportManager.createElementTypesFromYml(Mockito.anyString(), Mockito.any())).thenCallRealMethod();
  68         when(commonImportManager.createElementTypesFromToscaJsonMap(Mockito.any(), Mockito.any())).thenCallRealMethod();
  69     }
  70
  71     @Before
  72     public void initMocks() {
  73         MockitoAnnotations.initMocks(this);
  74     }
  75
  76     @Test
  77     public void testCreateCapabilityTypes() throws IOException {
  78         String ymlContent = getCapabilityTypesYml();
  79         Either<List<CapabilityTypeDefinition>, ResponseFormat> createCapabilityTypes = manager.createCapabilityTypes(ymlContent);
  80         assertTrue(createCapabilityTypes.isLeft());
  81
  82         List<CapabilityTypeDefinition> capabilityTypesList = createCapabilityTypes.left().value();
  83         assertTrue(capabilityTypesList.size() == 14);
  84         Map<String, CapabilityTypeDefinition> capibilityTypeMap = new HashMap<>();
  85         for (CapabilityTypeDefinition capType : capabilityTypesList) {
  86             capibilityTypeMap.put(capType.getType(), capType);
  87         }
  88         assertTrue(capabilityTypesList.size() == 14);
  89
  90         for (CapabilityTypeNameEnum curr : CapabilityTypeNameEnum.values()) {
  91             assertTrue(capibilityTypeMap.containsKey(curr.getCapabilityName()));
  92         }
  93
  94     }
  95
  96     private String getCapabilityTypesYml() throws IOException {
  97         Path filePath = Paths.get("src/test/resources/types/capabilityTypes.yml");
  98         byte[] fileContent = Files.readAllBytes(filePath);
  99         String ymlContent = new String(fileContent);
 100         return ymlContent;
 101     }
 102
 103 }