catalog-model/src/test/java/org/openecomp/sdc/be/model/jsontitan/operations/TopologyTemplateOperationCapabilityIntegrationTest.java

   1 package org.openecomp.sdc.be.model.jsontitan.operations;
   2
   3 import org.junit.After;
   4 import org.junit.Before;
   5 import org.junit.BeforeClass;
   6 import org.junit.Test;
   7 import org.junit.runner.RunWith;
   8 import org.openecomp.sdc.be.dao.config.TitanSpringConfig;
   9 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  10 import org.openecomp.sdc.be.dao.jsongraph.TitanDao;
  11 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
  12 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  13 import org.openecomp.sdc.be.datatypes.elements.MapCapabiltyProperty;
  14 import org.openecomp.sdc.be.datatypes.elements.MapPropertiesDataDefinition;
  15 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
  16 import org.openecomp.sdc.be.model.Component;
  17 import org.openecomp.sdc.be.model.ComponentParametersView;
  18 import org.openecomp.sdc.be.model.ModelTestBase;
  19 import org.openecomp.sdc.be.model.PropertyDefinition;
  20 import org.openecomp.sdc.be.model.config.ModelOperationsSpringConfig;
  21 import org.openecomp.sdc.be.model.jsontitan.datamodel.TopologyTemplate;
  22 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  23 import org.springframework.test.context.ContextConfiguration;
  24 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  25
  26 import javax.annotation.Resource;
  27 import java.util.Arrays;
  28 import java.util.HashMap;
  29 import java.util.List;
  30 import java.util.Map;
  31
  32 import static org.assertj.core.api.Assertions.assertThat;
  33
  34 @RunWith(SpringJUnit4ClassRunner.class)
  35 @ContextConfiguration(classes = {TitanSpringConfig.class, ModelOperationsSpringConfig.class})
  36 public class TopologyTemplateOperationCapabilityIntegrationTest extends ModelTestBase {
  37
  38     private static final String CONTAINER_ID = "id";
  39     private Map<String, MapCapabiltyProperty> capabilitiesPropsMap;
  40     private List<CapabilityPropertyDataObject> capabilitiesProperties;
  41
  42     @Resource
  43     private TopologyTemplateOperation topologyTemplateOperation;
  44
  45     @Resource
  46     private TitanDao titanDao;
  47     private CapabilityPropertyDataObject capabilityProperty1;
  48     private CapabilityPropertyDataObject capabilityProperty2;
  49     private CapabilityPropertyDataObject capabilityProperty3;
  50
  51     @BeforeClass
  52     public static void setupBeforeClass() {
  53
  54         ModelTestBase.init();
  55     }
  56
  57     @Before
  58     public void setUp() throws Exception {
  59
  60         capabilitiesPropsMap = new HashMap<>();
  61         capabilityProperty1 = new CapabilityPropertyDataObject("instance1", "capability1", "prop1", "val1");
  62         capabilityProperty2 = new CapabilityPropertyDataObject("instance1", "capability2", "prop2", "val2");
  63         capabilityProperty3 = new CapabilityPropertyDataObject("instance2", "capability3", "prop3", "val3");
  64         capabilitiesProperties = Arrays.asList(capabilityProperty1, capabilityProperty2, capabilityProperty3);
  65
  66
  67         //capablities props == Map<instance id, Map<capability id, Map<prop id, property>>>
  68         capabilitiesProperties.forEach(capabilitiesProperty -> {
  69             capabilitiesPropsMap.computeIfAbsent(capabilitiesProperty.getInstanceId(), k -> new MapCapabiltyProperty(new HashMap<>()))
  70                              .getMapToscaDataDefinition().computeIfAbsent(capabilitiesProperty.getCapabilityId(), k -> new MapPropertiesDataDefinition(new HashMap<>()))
  71                              .getMapToscaDataDefinition().computeIfAbsent(capabilitiesProperty.getPropName(), k -> new PropertyDefinition(createPropWithValue(capabilitiesProperty.getPropValue())));
  72         });
  73
  74         GraphVertex resource = new GraphVertex(VertexTypeEnum.TOPOLOGY_TEMPLATE);
  75         resource.addMetadataProperty(GraphPropertyEnum.UNIQUE_ID, CONTAINER_ID);
  76         titanDao.createVertex(resource);
  77         GraphVertex loadedResource = titanDao.getVertexById(CONTAINER_ID).left().value();
  78         topologyTemplateOperation.assosiateElementToData(loadedResource, VertexTypeEnum.CALCULATED_CAP_PROPERTIES, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES, capabilitiesPropsMap).left().value();
  79         titanDao.commit();
  80
  81     }
  82
  83     @After
  84     public void tearDown() throws Exception {
  85         titanDao.rollback();
  86     }
  87
  88     @Test
  89     public void overrideCalculatedCapabilityProperties() {
  90         Map<String, MapCapabiltyProperty> loadedCapPropsMap = fetchCapabilitiesProps(CONTAINER_ID);
  91         compareCapabilitiesProperties(capabilitiesProperties, loadedCapPropsMap);
  92         capabilityProperty1.setPropValue("newVal1");
  93         capabilityProperty3.setPropValue("newVal3");
  94         setPropertyValue(capabilitiesPropsMap, capabilityProperty1);
  95         setPropertyValue(capabilitiesPropsMap, capabilityProperty3);
  96         StorageOperationStatus storageOperationStatus = topologyTemplateOperation.overrideToscaDataOfToscaElement(CONTAINER_ID, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES, capabilitiesPropsMap);
  97         assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.OK);
  98         Map<String, MapCapabiltyProperty> updatedCapPropsMap = fetchCapabilitiesProps(CONTAINER_ID);
  99         compareCapabilitiesProperties(capabilitiesProperties, updatedCapPropsMap);
 100     }
 101
 102     @Test
 103     public void updateToscaDataDeepElementsBlockToToscaElement() {
 104         assertCapabilityPropValue(capabilityProperty1, "val1");
 105         assertCapabilityPropValue(capabilityProperty2, "val2");
 106         assertCapabilityPropValue(capabilityProperty3, "val3");
 107
 108         MapCapabiltyProperty instance1Props = capabilitiesPropsMap.get("instance1");
 109         capabilityProperty1.setPropValue("newVal1");
 110         setPropertyValue(capabilitiesPropsMap, capabilityProperty1);
 111
 112         capabilityProperty3.setPropValue("newVal3");
 113         setPropertyValue(capabilitiesPropsMap, capabilityProperty3);
 114
 115         Component component = new org.openecomp.sdc.be.model.Resource();
 116         component.setUniqueId(CONTAINER_ID);
 117         StorageOperationStatus updateStatus = topologyTemplateOperation.updateToscaDataDeepElementsBlockToToscaElement(CONTAINER_ID, EdgeLabelEnum.CALCULATED_CAP_PROPERTIES, instance1Props, "instance1");
 118
 119         assertThat(updateStatus).isEqualTo(StorageOperationStatus.OK);
 120         assertCapabilityPropValue(capabilityProperty1, "newVal1");
 121         assertCapabilityPropValue(capabilityProperty2, "val2");
 122         assertCapabilityPropValue(capabilityProperty3, "val3");//only instance1 props should be updated
 123     }
 124
 125     private Map<String, MapCapabiltyProperty> fetchCapabilitiesProps(String containerId) {
 126         ComponentParametersView capabilityPropsFilter = new ComponentParametersView(true);
 127         capabilityPropsFilter.setIgnoreCapabiltyProperties(false);
 128         return ((TopologyTemplate) topologyTemplateOperation.getToscaElement(containerId, capabilityPropsFilter).left().value()).getCalculatedCapabilitiesProperties();
 129     }
 130
 131     private void compareCapabilitiesProperties(List<CapabilityPropertyDataObject> expected, Map<String, MapCapabiltyProperty> actual) {
 132         expected.forEach(expectedCapabilityProp -> {
 133             assertThat(getPropertyValue(actual, expectedCapabilityProp.instanceId, expectedCapabilityProp.capabilityId, expectedCapabilityProp.propName))
 134                     .isEqualTo(expectedCapabilityProp.propValue);
 135         });
 136     }
 137
 138     private String getPropertyValue(Map<String, MapCapabiltyProperty> capabilityPropertyMap, String instance, String capability, String prop) {
 139         return capabilityPropertyMap.get(instance).getMapToscaDataDefinition().get(capability).getMapToscaDataDefinition().get(prop).getValue();
 140     }
 141
 142     private void setPropertyValue(Map<String, MapCapabiltyProperty> capabilityPropertyMap, CapabilityPropertyDataObject capabilityProperty) {
 143         setPropertyValue(capabilityPropertyMap.get(capabilityProperty.getInstanceId()), capabilityProperty);
 144
 145     }
 146
 147     private void setPropertyValue(MapCapabiltyProperty capabilitiesInstanceProperties, CapabilityPropertyDataObject capabilityProperty) {
 148         capabilitiesInstanceProperties.getMapToscaDataDefinition().get(capabilityProperty.getCapabilityId())
 149                                       .getMapToscaDataDefinition().get(capabilityProperty.getPropName())
 150                                       .setValue(capabilityProperty.getPropValue());
 151     }
 152
 153     private void assertCapabilityPropValue(CapabilityPropertyDataObject prop, String expectedValue) {
 154         Map<String, MapCapabiltyProperty> loadedCapPropsMap = fetchCapabilitiesProps(CONTAINER_ID);
 155         String propertyValue = getPropertyValue(loadedCapPropsMap, prop.getInstanceId(), prop.getCapabilityId(), prop.getPropName());
 156         assertThat(propertyValue).isEqualTo(expectedValue);
 157     }
 158
 159
 160     private PropertyDefinition createPropWithValue(String val) {
 161         PropertyDefinition propertyDefinition = new PropertyDefinition();
 162         propertyDefinition.setValue(val);
 163         return propertyDefinition;
 164     }
 165
 166     private static class CapabilityPropertyDataObject {
 167         private String instanceId;
 168         private String capabilityId;
 169         private String propName;
 170         private String propValue;
 171
 172         CapabilityPropertyDataObject(String instanceId, String capabilityId, String propName, String propValue) {
 173             this.instanceId = instanceId;
 174             this.capabilityId = capabilityId;
 175             this.propName = propName;
 176             this.propValue = propValue;
 177         }
 178
 179         String getInstanceId() {
 180             return instanceId;
 181         }
 182
 183         String getCapabilityId() {
 184             return capabilityId;
 185         }
 186
 187         String getPropName() {
 188             return propName;
 189         }
 190
 191         String getPropValue() {
 192             return propValue;
 193         }
 194
 195         void setPropValue(String propValue) {
 196             this.propValue = propValue;
 197         }
 198     }
 199 }