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