catalog-model/src/test/java/org/openecomp/sdc/be/model/jsonjanusgraph/operations/PolicyOperationIntegrationTest.java

   1 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
   2
   3 import fj.data.Either;
   4 import org.junit.After;
   5 import org.junit.Before;
   6 import org.junit.BeforeClass;
   7 import org.junit.Test;
   8 import org.junit.runner.RunWith;
   9 import org.openecomp.sdc.be.dao.config.JanusGraphSpringConfig;
  10 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
  11 import org.openecomp.sdc.be.dao.jsongraph.HealingJanusGraphDao;
  12 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
  13 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
  14 import org.openecomp.sdc.be.datatypes.elements.PolicyDataDefinition;
  15 import org.openecomp.sdc.be.datatypes.elements.PropertyDataDefinition;
  16 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
  17 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
  18 import org.openecomp.sdc.be.model.Component;
  19 import org.openecomp.sdc.be.model.ComponentParametersView;
  20 import org.openecomp.sdc.be.model.ModelTestBase;
  21 import org.openecomp.sdc.be.model.PolicyDefinition;
  22 import org.openecomp.sdc.be.model.config.ModelOperationsSpringConfig;
  23 import org.openecomp.sdc.be.model.jsonjanusgraph.datamodel.TopologyTemplate;
  24 import org.openecomp.sdc.be.model.jsonjanusgraph.datamodel.ToscaElement;
  25 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
  26 import org.springframework.test.context.ContextConfiguration;
  27 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
  28
  29 import javax.annotation.Resource;
  30 import java.util.Arrays;
  31 import java.util.Collections;
  32 import java.util.List;
  33 import java.util.Map;
  34
  35 import static org.apache.commons.collections.ListUtils.union;
  36 import static org.assertj.core.api.Assertions.assertThat;
  37
  38 @RunWith(SpringJUnit4ClassRunner.class)
  39 @ContextConfiguration(classes = {JanusGraphSpringConfig.class, ModelOperationsSpringConfig.class})
  40 public class PolicyOperationIntegrationTest extends ModelTestBase {
  41
  42     private static final String CONTAINER_ID = "container";
  43     public static final String POLICY_ID = "policy";
  44     @Resource
  45     private TopologyTemplateOperation topologyTemplateOperation;
  46     @Resource
  47     private HealingJanusGraphDao janusGraphDao;
  48     @Resource
  49     private PolicyOperation policyOperation;
  50     private PropertyDataDefinition prop1, prop2;
  51     private PolicyDefinition policy;
  52
  53     @BeforeClass
  54     public static void setupBeforeClass() {
  55
  56         ModelTestBase.init();
  57     }
  58
  59     @Before
  60     public void setUp() throws Exception {
  61         prop1 = new PropertyDataDefinition();
  62         prop1.setUniqueId("prop1");
  63         prop1.setName("prop1");
  64         prop1.setValue("prop1");
  65
  66         prop2 = new PropertyDataDefinition();
  67         prop2.setUniqueId("prop2");
  68         prop2.setName("prop2");
  69         prop2.setValue("prop2");
  70
  71         policy = new PolicyDefinition();
  72         policy.setUniqueId(POLICY_ID);
  73
  74         policy.setProperties(Arrays.asList(prop1, prop2));
  75
  76         GraphVertex resource = new GraphVertex(VertexTypeEnum.TOPOLOGY_TEMPLATE);
  77         resource.addMetadataProperty(GraphPropertyEnum.UNIQUE_ID, CONTAINER_ID);
  78         janusGraphDao.createVertex(resource);
  79         GraphVertex loadedResource = janusGraphDao.getVertexById(CONTAINER_ID).left().value();
  80         topologyTemplateOperation.addToscaDataToToscaElement(loadedResource, EdgeLabelEnum.POLICIES, VertexTypeEnum.POLICIES, policy, JsonPresentationFields.UNIQUE_ID);
  81 //        janusGraphDao.commit();
  82     }
  83
  84     @After
  85     public void tearDown() {
  86         janusGraphDao.rollback();
  87     }
  88
  89     @Test
  90     public void testUpdatePolicyProperties_singleProperty() {
  91         PropertyDataDefinition prop1Copy = new PropertyDataDefinition(prop1);
  92         prop1Copy.setValue("prop1NewValue");
  93         testUpdatePolicyProperties(Collections.singletonList(prop1Copy), Collections.singletonList(prop2));
  94     }
  95
  96     @Test
  97     public void testUpdatePolicyProperties_multipleProperties() {
  98         PropertyDataDefinition prop1Copy = new PropertyDataDefinition(prop1);
  99         prop1Copy.setValue("prop1NewValue");
 100
 101         PropertyDataDefinition prop2Copy = new PropertyDataDefinition(prop2);
 102         prop2Copy.setValue("prop2NewValue");
 103
 104         testUpdatePolicyProperties(Arrays.asList(prop1Copy, prop2Copy), Collections.emptyList());
 105     }
 106
 107     private void testUpdatePolicyProperties(List<PropertyDataDefinition> updatedProperties, List<PropertyDataDefinition> nonUpdatedPropeties) {
 108         Component cmpt = new org.openecomp.sdc.be.model.Resource();
 109         cmpt.setUniqueId(CONTAINER_ID);
 110         cmpt.setPolicies(Collections.singletonMap(POLICY_ID, policy));
 111         StorageOperationStatus storageOperationStatus = policyOperation.updatePolicyProperties(cmpt, POLICY_ID, updatedProperties);
 112         assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.OK);
 113
 114         ComponentParametersView componentParametersView = new ComponentParametersView();
 115         componentParametersView.disableAll();
 116         componentParametersView.setIgnorePolicies(false);
 117         Either<ToscaElement, StorageOperationStatus> loadedCmptEither = topologyTemplateOperation.getToscaElement(CONTAINER_ID, componentParametersView);
 118
 119         assertThat(loadedCmptEither.isLeft()).isTrue();
 120         ToscaElement loadedCmpt = loadedCmptEither.left().value();
 121         assertThat(loadedCmpt).isInstanceOf(TopologyTemplate.class);
 122         @SuppressWarnings("unchecked") List<PropertyDataDefinition> allProperties = union(updatedProperties, nonUpdatedPropeties);
 123         verifyPolicyPropertiesValuesUpdated((TopologyTemplate) loadedCmpt, allProperties);
 124     }
 125
 126     private void verifyPolicyPropertiesValuesUpdated(TopologyTemplate toscaElement, List<PropertyDataDefinition> expectedUpdatedProperties) {
 127         Map<String, PolicyDataDefinition> policies = toscaElement.getPolicies();
 128         PolicyDataDefinition policy = policies.get(POLICY_ID);
 129         List<PropertyDataDefinition> policyProperties = policy.getProperties();
 130         assertThat(policyProperties).usingElementComparatorOnFields("value")
 131                                     .containsAll(expectedUpdatedProperties);
 132     }
 133 }