1 package org.openecomp.sdc.be.model.jsontitan.operations;
4 import org.junit.After;
5 import org.junit.Before;
6 import org.junit.BeforeClass;
8 import org.junit.runner.RunWith;
9 import org.openecomp.sdc.be.dao.config.TitanSpringConfig;
10 import org.openecomp.sdc.be.dao.jsongraph.GraphVertex;
11 import org.openecomp.sdc.be.dao.jsongraph.HealingTitanDao;
12 import org.openecomp.sdc.be.dao.jsongraph.TitanDao;
13 import org.openecomp.sdc.be.dao.titan.TitanOperationStatus;
14 import org.openecomp.sdc.be.datatypes.elements.PolicyTargetType;
15 import org.openecomp.sdc.be.model.Component;
16 import org.openecomp.sdc.be.model.ComponentParametersView;
17 import org.openecomp.sdc.be.model.ModelTestBase;
18 import org.openecomp.sdc.be.model.PolicyDefinition;
19 import org.openecomp.sdc.be.model.config.ModelOperationsSpringConfig;
20 import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;
21 import org.springframework.beans.factory.annotation.Autowired;
22 import org.springframework.test.context.ContextConfiguration;
23 import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
25 import java.util.ArrayList;
26 import java.util.HashMap;
27 import java.util.List;
29 import java.util.stream.Collectors;
31 import static java.util.Arrays.asList;
32 import static java.util.Collections.singletonList;
33 import static org.assertj.core.api.Assertions.assertThat;
34 import static org.junit.Assert.assertTrue;
36 @RunWith(SpringJUnit4ClassRunner.class)
37 @ContextConfiguration(classes = {TitanSpringConfig.class, ModelOperationsSpringConfig.class})
38 public class ToscaOperationFacadePoliciesTest extends ModelTestBase {
41 private ToscaOperationFacade toscaOperationFacade;
43 private HealingTitanDao titanDao;
45 private PolicyDefinition policy1, policy2;
48 public static void setupBeforeClass() {
53 public void setUp() throws Exception {
54 policy1 = createPolicyDefinition("type1");
55 policy2 = createPolicyDefinition("type2");
56 createContainerVertexInDB();
57 createPoliciesOnGraph(policy1, policy2);
60 private void createPoliciesOnGraph(PolicyDefinition ... policies) {
61 for (int i = 0; i < policies.length; i++) {
62 PolicyDefinition policy = policies[i];
63 Either<PolicyDefinition, StorageOperationStatus> createdPolicy = toscaOperationFacade.associatePolicyToComponent(CONTAINER_ID, policy, i);
64 assertTrue(createdPolicy.isLeft());
69 public void tearDown() {
74 public void updatePoliciesTargetsOfComponent_updateSinglePolicy() {
75 List<String> updatedTargetIds = asList("instance1new", "instance2");
76 PolicyDefinition originalPolicy2 = clonePolicyWithTargets(policy2);
77 updatePolicyTypeTargetsIds(policy1, PolicyTargetType.COMPONENT_INSTANCES, updatedTargetIds);
78 updatePolicyTypeTargetsIds(policy2, PolicyTargetType.COMPONENT_INSTANCES, updatedTargetIds);
80 StorageOperationStatus storageOperationStatus = toscaOperationFacade.updatePoliciesOfComponent(CONTAINER_ID, singletonList(policy1));
81 assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.OK);
82 Component updatedComponent = fetchComponentFromDB();
83 verifyPolicyTargets(updatedComponent.getPolicyById(policy1.getUniqueId()), policy1);
84 verifyPolicyTargets(updatedComponent.getPolicyById(policy2.getUniqueId()), originalPolicy2);
88 public void updatePoliciesTargetsOfComponent_updateMultiplePolicies() {
89 List<String> updatedTargetIds = asList("instance1new", "instance2");
90 updatePolicyTypeTargetsIds(policy1, PolicyTargetType.COMPONENT_INSTANCES, updatedTargetIds);
91 updatePolicyTypeTargetsIds(policy2, PolicyTargetType.COMPONENT_INSTANCES, updatedTargetIds);
92 StorageOperationStatus storageOperationStatus = toscaOperationFacade.updatePoliciesOfComponent(CONTAINER_ID, asList(policy1, policy2));
93 assertThat(storageOperationStatus).isEqualTo(StorageOperationStatus.OK);
94 Component updatedComponent = fetchComponentFromDB();
95 verifyPolicyTargets(updatedComponent.getPolicyById(policy1.getUniqueId()), policy1);
96 verifyPolicyTargets(updatedComponent.getPolicyById(policy2.getUniqueId()), policy2);
99 private PolicyDefinition clonePolicyWithTargets(PolicyDefinition policy) {
100 PolicyDefinition originalPolicy = new PolicyDefinition(policy);
101 Map<PolicyTargetType, List<String>> clonedTargetMap = policy.getTargets().entrySet().stream().collect(Collectors.toMap(Map.Entry::getKey, entry -> new ArrayList<>(entry.getValue())));
102 originalPolicy.setTargets(clonedTargetMap);
103 return originalPolicy;
106 private void verifyPolicyTargets(PolicyDefinition updatedPolicy, PolicyDefinition expectedPolicy) {
107 assertThat(updatedPolicy.getTargets())
108 .isEqualTo(expectedPolicy.getTargets());
111 private void updatePolicyTypeTargetsIds(PolicyDefinition policy, PolicyTargetType targetType, List<String> updatedTargetIds) {
112 policy.getTargets().put(targetType, updatedTargetIds);
115 private Component fetchComponentFromDB() {
116 ComponentParametersView componentParametersView = new ComponentParametersView();
117 componentParametersView.disableAll();
118 componentParametersView.setIgnorePolicies(false);
119 return toscaOperationFacade.getToscaElement(CONTAINER_ID, componentParametersView).left().value();
122 private void createContainerVertexInDB() {
123 GraphVertex resource = createBasicContainerGraphVertex();
124 Either<GraphVertex, TitanOperationStatus> container = titanDao.createVertex(resource);
125 assertTrue(container.isLeft());
128 private PolicyDefinition createPolicyDefinition(String type) {
129 PolicyDefinition policy = new PolicyDefinition();
130 policy.setPolicyTypeName(type);
131 policy.setTargets(new HashMap<>());
132 policy.getTargets().put(PolicyTargetType.COMPONENT_INSTANCES, asList("instance1", "instance2"));