catalog-model/src/test/java/org/openecomp/sdc/be/model/jsonjanusgraph/operations/CapabilitiesOperationTest.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.be.model.jsonjanusgraph.operations;
  18
  19 import fj.data.Either;
  20 import org.junit.Assert;
  21 import org.junit.Before;
  22 import org.junit.Test;
  23 import org.mockito.InjectMocks;
  24 import org.mockito.Mock;
  25 import org.mockito.MockitoAnnotations;
  26 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  27 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphDao;
  28 import org.openecomp.sdc.be.dao.janusgraph.JanusGraphOperationStatus;
  29 import org.openecomp.sdc.be.datatypes.elements.MapPropertiesDataDefinition;
  30 import org.openecomp.sdc.be.datatypes.elements.PropertyDataDefinition;
  31 import org.openecomp.sdc.be.model.jsonjanusgraph.datamodel.NodeType;
  32 import org.openecomp.sdc.be.model.jsonjanusgraph.datamodel.TopologyTemplate;
  33 import org.openecomp.sdc.be.model.jsonjanusgraph.utils.CapabilityTestUtils;
  34 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  35
  36 import java.util.HashMap;
  37 import java.util.Map;
  38
  39 import static org.mockito.ArgumentMatchers.any;
  40 import static org.mockito.ArgumentMatchers.anyMap;
  41 import static org.mockito.ArgumentMatchers.anyString;
  42 import static org.mockito.Mockito.when;
  43
  44 public class CapabilitiesOperationTest {
  45
  46     @InjectMocks
  47     CapabilitiesOperation operation = new CapabilitiesOperation();
  48     @Mock
  49     private  JanusGraphDao mockJanusGraphDao;
  50     @Mock
  51     private TopologyTemplateOperation topologyTemplateOperation;
  52     @Mock
  53     private NodeTypeOperation nodeTypeOperation;
  54
  55
  56     @Before
  57     public void setUp() {
  58         MockitoAnnotations.openMocks(this);
  59
  60         when(mockJanusGraphDao.commit()).thenReturn(JanusGraphOperationStatus.OK);
  61         when(mockJanusGraphDao.getVertexById(anyString(), any())).thenReturn(Either.left(new GraphVertex()));
  62
  63         when(topologyTemplateOperation.updateFullToscaData(any(), any(), any(), anyMap())).thenReturn(StorageOperationStatus.OK);
  64         TopologyTemplate topologyTemplate = new TopologyTemplate();
  65
  66         Map<String, MapPropertiesDataDefinition> capPropsForTopologyTemplate = CapabilityTestUtils
  67                 .createCapPropsForTopologyTemplate(topologyTemplate);
  68         topologyTemplate.setCapabilitiesProperties(capPropsForTopologyTemplate);
  69
  70         when(topologyTemplateOperation.getToscaElement(anyString(), any())).thenReturn(Either.left(topologyTemplate));
  71
  72         NodeType nodeType = new NodeType();
  73         Map<String, MapPropertiesDataDefinition> capPropsForNodeType = CapabilityTestUtils
  74                 .createCapPropsForTopologyTemplate(nodeType);
  75         nodeType.setCapabilitiesProperties(capPropsForNodeType);
  76
  77         when(nodeTypeOperation.getToscaElement(anyString(), any())).thenReturn(Either.left(nodeType));
  78     }
  79
  80     @Test
  81     public void testCreateOrUpdateCapabilitiesPropertiesTopologyTemplate() {
  82
  83         Map<String, PropertyDataDefinition> mapToscaDataDefinition = new HashMap<>();
  84         PropertyDataDefinition propertyDataDefinition = new PropertyDataDefinition();
  85         propertyDataDefinition.setUniqueId("ComponentInput1_uniqueId");
  86         propertyDataDefinition.setName("propName");
  87         mapToscaDataDefinition.put(propertyDataDefinition.getUniqueId(), propertyDataDefinition);
  88         MapPropertiesDataDefinition  mapPropertiesDataDefinition = new MapPropertiesDataDefinition(mapToscaDataDefinition);
  89
  90         Map<String, MapPropertiesDataDefinition> propertiesMap = new HashMap<>();
  91         propertiesMap.put(propertyDataDefinition.getUniqueId(), mapPropertiesDataDefinition);
  92
  93         StorageOperationStatus operationStatus = operation.createOrUpdateCapabilityProperties("componentId", true,
  94                 propertiesMap);
  95
  96         Assert.assertEquals(StorageOperationStatus.OK, operationStatus);
  97     }
  98
  99     @Test
 100     public void testCreateOrUpdateCapabilitiesPropertiesToscaTemplate() {
 101
 102         Map<String, PropertyDataDefinition> mapToscaDataDefinition = new HashMap<>();
 103         PropertyDataDefinition propertyDataDefinition = new PropertyDataDefinition();
 104         propertyDataDefinition.setUniqueId("ComponentInput1_uniqueId");
 105         propertyDataDefinition.setName("propName");
 106         mapToscaDataDefinition.put(propertyDataDefinition.getUniqueId(), propertyDataDefinition);
 107         MapPropertiesDataDefinition  mapPropertiesDataDefinition = new MapPropertiesDataDefinition(mapToscaDataDefinition);
 108
 109         Map<String, MapPropertiesDataDefinition> propertiesMap = new HashMap<>();
 110         propertiesMap.put(propertyDataDefinition.getUniqueId(), mapPropertiesDataDefinition);
 111
 112         StorageOperationStatus operationStatus = operation.createOrUpdateCapabilityProperties("componentId", false,
 113                 propertiesMap);
 114
 115         Assert.assertEquals(StorageOperationStatus.OK, operationStatus);
 116     }
 117 }