aboutsummaryrefslogtreecommitdiffstats
path: root/catalog-be/src/test/java/org/openecomp/sdc/be/components/impl/version/PolicyTargetsUpdateOperationTest.java
blob: 5da782e484fb0fd13d4df7de88fc948eaccfc4ef (plain)
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
package org.openecomp.sdc.be.components.impl.version;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.anyList;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.verifyZeroInteractions;
import static org.mockito.Mockito.when;
import static org.openecomp.sdc.be.components.impl.utils.TestGenerationUtils.getComponentsUtils;

import java.util.List;
import java.util.stream.Stream;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
import org.openecomp.sdc.be.components.utils.ComponentInstanceBuilder;
import org.openecomp.sdc.be.components.utils.PolicyDefinitionBuilder;
import org.openecomp.sdc.be.components.utils.ResourceBuilder;
import org.openecomp.sdc.be.dao.api.ActionStatus;
import org.openecomp.sdc.be.model.Component;
import org.openecomp.sdc.be.model.ComponentInstance;
import org.openecomp.sdc.be.model.PolicyDefinition;
import org.openecomp.sdc.be.model.Resource;
import org.openecomp.sdc.be.model.jsontitan.operations.ToscaOperationFacade;
import org.openecomp.sdc.be.model.operations.api.StorageOperationStatus;

@RunWith(MockitoJUnitRunner.class)
public class PolicyTargetsUpdateOperationTest {

    private static final String CONTAINER_ID = "containerId";

    private PolicyTargetsUpdateOperation policyTargetsUpdateOperation;
    @Mock
    private ToscaOperationFacade toscaOperationFacade;

    @Captor
    private ArgumentCaptor<List<PolicyDefinition>> updatedPoliciesCaptor;
    private PolicyDefinition policy1, policy2, policy3;
    private Resource container;

    @Before
    public void setUp() throws Exception {
        policy1 = createPolicyWithCmptInstAsTargets("policy1", "inst1", "inst2");
        policy2 = createPolicyWithCmptInstAsTargets("policy2", "inst1", "inst2", "inst3");
        policy3 = createPolicyWithCmptInstAsTargets("policy3", "inst1", "inst3");
        container = new ResourceBuilder()
                .addPolicy(policy1)
                .addPolicy(policy2)
                .addPolicy(policy3)
                .setUniqueId(CONTAINER_ID)
                .build();
        policyTargetsUpdateOperation = new PolicyTargetsUpdateOperation(toscaOperationFacade, getComponentsUtils());
    }

    @Test
    public void whenNoPolicies_returnActionOk() {
        Component container = new Resource();
        ComponentInstance prevVersion = new ComponentInstanceBuilder().setId("prevVersion").build();
        ComponentInstance newVersion = new ComponentInstanceBuilder().setId("newVersion").build();
        ActionStatus operationStatus = policyTargetsUpdateOperation.onChangeVersion(container, prevVersion, newVersion);
        assertThat(operationStatus).isEqualTo(ActionStatus.OK);
        verifyZeroInteractions(toscaOperationFacade);
    }

    @Test
    public void whenNoCmptInstancePolicies_returnOk() {
        PolicyDefinition policy = PolicyDefinitionBuilder.create().addGroupTarget("someGroup").build();
        Component container = new ResourceBuilder().addPolicy(policy).build();
        ComponentInstance prevVersion = new ComponentInstanceBuilder().setId("prevVersion").build();
        ComponentInstance newVersion = new ComponentInstanceBuilder().setId("newVersion").build();
        ActionStatus operationStatus = policyTargetsUpdateOperation.onChangeVersion(container, prevVersion, newVersion);
        assertThat(operationStatus).isEqualTo(ActionStatus.OK);
        verifyZeroInteractions(toscaOperationFacade);
    }

    @Test
    public void whenNoPoliciesWithPrevInstanceAsTarget_returnActionOk() {
        ComponentInstance prevVersion = new ComponentInstanceBuilder().setId("prevVersion").build();
        ComponentInstance newVersion = new ComponentInstanceBuilder().setId("newVersion").build();
        ActionStatus operationStatus = policyTargetsUpdateOperation.onChangeVersion(container, prevVersion, newVersion);
        assertThat(operationStatus).isEqualTo(ActionStatus.OK);
        verifyZeroInteractions(toscaOperationFacade);
    }

    @Test
    public void replacePrevCmptInstanceIdWithNewInstanceIdAndUpdatePolicies() {
        ComponentInstance prevVersion = new ComponentInstanceBuilder().setId("inst2").build();
        ComponentInstance newVersion = new ComponentInstanceBuilder().setId("inst2New").build();
        when(toscaOperationFacade.updatePoliciesOfComponent(eq(CONTAINER_ID), updatedPoliciesCaptor.capture())).thenReturn(StorageOperationStatus.OK);
        ActionStatus updatePoliciesRes = policyTargetsUpdateOperation.onChangeVersion(container, prevVersion, newVersion);
        assertThat(updatePoliciesRes).isEqualTo(ActionStatus.OK);
        List<PolicyDefinition> updatedPolicies = updatedPoliciesCaptor.getValue();
        verifyUpdatedPolicies(updatedPolicies, policy1, policy2);//policy3 does not have "inst2" as target, no update needed
        verifyUpdatedPolicyTargets(policy1, "inst1", "inst2New");
        verifyUpdatedPolicyTargets(policy2, "inst1", "inst2New", "inst3");
    }

    @Test
    public void whenUpdateOfPoliciesFails_propagateTheFailure() {
        ComponentInstance prevVersion = new ComponentInstanceBuilder().setId("inst2").build();
        ComponentInstance newVersion = new ComponentInstanceBuilder().setId("inst2New").build();
        when(toscaOperationFacade.updatePoliciesOfComponent(eq(CONTAINER_ID), anyList())).thenReturn(StorageOperationStatus.GENERAL_ERROR);
        ActionStatus updatePoliciesRes = policyTargetsUpdateOperation.onChangeVersion(container, prevVersion, newVersion);
        assertThat(updatePoliciesRes).isEqualTo(ActionStatus.GENERAL_ERROR);
    }

    private void verifyUpdatedPolicyTargets(PolicyDefinition updatedPolicy, String ... expectedCmptInstanceTargetIds) {
        assertThat(updatedPolicy.resolveComponentInstanceTargets())
                .containsExactlyInAnyOrder(expectedCmptInstanceTargetIds);
    }

    private void verifyUpdatedPolicies(List<PolicyDefinition> updatedPolicies, PolicyDefinition ... expectedUpdatedPolicies) {
        assertThat(updatedPolicies)
                .usingElementComparatorOnFields("targets")
                .containsExactlyInAnyOrder(expectedUpdatedPolicies);
    }

    private PolicyDefinition createPolicyWithCmptInstAsTargets(String uniqueId, String ... instId) {
        PolicyDefinitionBuilder policyDefinitionBuilder = PolicyDefinitionBuilder.create();
        Stream.of(instId).forEach(policyDefinitionBuilder::addComponentInstanceTarget);
        return policyDefinitionBuilder.setUniqueId(uniqueId).build();
    }
}