2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2019 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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.openecomp.sdc.be.model.jsonjanusgraph.operations;
23 import static org.apache.commons.collections.ListUtils.union;
24 import static org.assertj.core.api.Assertions.assertThat;
26 import fj.data.Either;
27 import java.util.Arrays;
28 import java.util.Collections;
29 import java.util.List;
31 import javax.annotation.Resource;
32 import org.junit.jupiter.api.AfterEach;
33 import org.junit.jupiter.api.BeforeAll;
34 import org.junit.jupiter.api.BeforeEach;
35 import org.junit.jupiter.api.Test;
36 import org.openecomp.sdc.be.dao.config.JanusGraphSpringConfig;
37 import org.openecomp.sdc.be.dao.janusgraph.HealingJanusGraphDao;
38 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
39 import org.openecomp.sdc.be.dao.jsongraph.types.EdgeLabelEnum;
40 import org.openecomp.sdc.be.dao.jsongraph.types.VertexTypeEnum;
41 import org.openecomp.sdc.be.datatypes.elements.PolicyDataDefinition;
42 import org.openecomp.sdc.be.datatypes.elements.PropertyDataDefinition;
43 import org.openecomp.sdc.be.datatypes.enums.GraphPropertyEnum;
44 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
45 import org.openecomp.sdc.be.model.Component;
46 import org.openecomp.sdc.be.model.ComponentParametersView;
47 import org.openecomp.sdc.be.model.ModelTestBase;
48 import org.openecomp.sdc.be.model.PolicyDefinition;
49 import org.openecomp.sdc.be.model.config.ModelOperationsSpringConfig;
50 import org.openecomp.sdc.be.model.jsonjanusgraph.datamodel.TopologyTemplate;
51 import org.openecomp.sdc.be.model.jsonjanusgraph.datamodel.ToscaElement;
52 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
53 import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;
55 @SpringJUnitConfig(classes = {JanusGraphSpringConfig.class, ModelOperationsSpringConfig.class})
56 public class PolicyOperationIntegrationTest extends ModelTestBase {
58 public static final String POLICY_ID = "policy";
59 private static final String CONTAINER_ID = "container";
61 private TopologyTemplateOperation topologyTemplateOperation;
63 private HealingJanusGraphDao janusGraphDao;
65 private PolicyOperation policyOperation;
66 private PropertyDataDefinition prop1, prop2;
67 private PolicyDefinition policy;
70 public static void setupBeforeClass() {
76 public void setUp() throws Exception {
77 prop1 = new PropertyDataDefinition();
78 prop1.setUniqueId("prop1");
79 prop1.setName("prop1");
80 prop1.setValue("prop1");
82 prop2 = new PropertyDataDefinition();
83 prop2.setUniqueId("prop2");
84 prop2.setName("prop2");
85 prop2.setValue("prop2");
87 policy = new PolicyDefinition();
88 policy.setUniqueId(POLICY_ID);
90 policy.setProperties(Arrays.asList(prop1, prop2));
92 GraphVertex resource = new GraphVertex(VertexTypeEnum.TOPOLOGY_TEMPLATE);
93 resource.addMetadataProperty(GraphPropertyEnum.UNIQUE_ID, CONTAINER_ID);
94 janusGraphDao.createVertex(resource);
95 GraphVertex loadedResource = janusGraphDao.getVertexById(CONTAINER_ID).left().value();
96 topologyTemplateOperation.addToscaDataToToscaElement(loadedResource, EdgeLabelEnum.POLICIES, VertexTypeEnum.POLICIES, policy,
97 JsonPresentationFields.UNIQUE_ID);
98 // janusGraphDao.commit();
102 public void tearDown() {
103 janusGraphDao.rollback();
107 public void testUpdatePolicyProperties_singleProperty() {
108 PropertyDataDefinition prop1Copy = new PropertyDataDefinition(prop1);
109 prop1Copy.setValue("prop1NewValue");
110 testUpdatePolicyProperties(Collections.singletonList(prop1Copy), Collections.singletonList(prop2));
114 public void testUpdatePolicyProperties_multipleProperties() {
115 PropertyDataDefinition prop1Copy = new PropertyDataDefinition(prop1);
116 prop1Copy.setValue("prop1NewValue");
118 PropertyDataDefinition prop2Copy = new PropertyDataDefinition(prop2);
119 prop2Copy.setValue("prop2NewValue");
121 testUpdatePolicyProperties(Arrays.asList(prop1Copy, prop2Copy), Collections.emptyList());
124 private void testUpdatePolicyProperties(List<PropertyDataDefinition> updatedProperties, List<PropertyDataDefinition> nonUpdatedPropeties) {
125 Component cmpt = new org.openecomp.sdc.be.model.Resource();
126 cmpt.setUniqueId(CONTAINER_ID);
127 cmpt.setPolicies(Collections.singletonMap(POLICY_ID, policy));
128 StorageOperationStatus storageOperationStatus = policyOperation.updatePolicyProperties(cmpt, POLICY_ID, updatedProperties);
129 assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.OK);
131 ComponentParametersView componentParametersView = new ComponentParametersView();
132 componentParametersView.disableAll();
133 componentParametersView.setIgnorePolicies(false);
134 Either<ToscaElement, StorageOperationStatus> loadedCmptEither = topologyTemplateOperation.getToscaElement(CONTAINER_ID,
135 componentParametersView);
137 assertThat(loadedCmptEither.isLeft()).isTrue();
138 ToscaElement loadedCmpt = loadedCmptEither.left().value();
139 assertThat(loadedCmpt).isInstanceOf(TopologyTemplate.class);
140 @SuppressWarnings("unchecked") List<PropertyDataDefinition> allProperties = union(updatedProperties, nonUpdatedPropeties);
141 verifyPolicyPropertiesValuesUpdated((TopologyTemplate) loadedCmpt, allProperties);
144 private void verifyPolicyPropertiesValuesUpdated(TopologyTemplate toscaElement, List<PropertyDataDefinition> expectedUpdatedProperties) {
145 Map<String, PolicyDataDefinition> policies = toscaElement.getPolicies();
146 PolicyDataDefinition policy = policies.get(POLICY_ID);
147 List<PropertyDataDefinition> policyProperties = policy.getProperties();
148 assertThat(policyProperties).usingElementComparatorOnFields("value")
149 .containsAll(expectedUpdatedProperties);