aboutsummaryrefslogtreecommitdiffstats
path: root/participant/participant-impl/participant-impl-kubernetes/src/test/java/org/onap/policy/clamp/acm/participant/kubernetes/handler/AutomationCompositionElementHandlerTest.java
blob: d9702abc5eb0ab5da686243e6fa0633a4411d94f (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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
/*-
 * ============LICENSE_START=======================================================
 *  Copyright (C) 2021-2023 Nordix Foundation.
 *  Modifications Copyright (C) 2021 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * SPDX-License-Identifier: Apache-2.0
 * ============LICENSE_END=========================================================
 */

package org.onap.policy.clamp.acm.participant.kubernetes.handler;

import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.assertDoesNotThrow;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.doThrow;

import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.Spy;
import org.onap.policy.clamp.acm.participant.intermediary.api.ParticipantIntermediaryApi;
import org.onap.policy.clamp.acm.participant.kubernetes.exception.ServiceException;
import org.onap.policy.clamp.acm.participant.kubernetes.models.ChartInfo;
import org.onap.policy.clamp.acm.participant.kubernetes.models.ChartList;
import org.onap.policy.clamp.acm.participant.kubernetes.parameters.CommonTestData;
import org.onap.policy.clamp.acm.participant.kubernetes.service.ChartService;
import org.onap.policy.clamp.acm.participant.kubernetes.utils.TestUtils;
import org.onap.policy.clamp.models.acm.concepts.AcElementDeploy;
import org.onap.policy.clamp.models.acm.concepts.AutomationCompositionElement;
import org.onap.policy.clamp.models.acm.messages.rest.instantiation.DeployOrder;
import org.onap.policy.common.utils.coder.Coder;
import org.onap.policy.common.utils.coder.CoderException;
import org.onap.policy.common.utils.coder.StandardCoder;
import org.onap.policy.models.base.PfModelException;
import org.onap.policy.models.tosca.authorative.concepts.ToscaConceptIdentifier;
import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
import org.springframework.test.context.junit.jupiter.SpringExtension;

@ExtendWith(SpringExtension.class)
class AutomationCompositionElementHandlerTest {

    private static final Coder CODER = new StandardCoder();
    private static final String CHART_INFO_YAML = "src/test/resources/ChartList.json";
    private static final String KEY_NAME =
        "org.onap.domain.database.HelloWorld_K8SMicroserviceAutomationCompositionElement";
    private static List<ChartInfo> charts;
    private static ToscaServiceTemplate toscaServiceTemplate;
    private static final String K8S_AUTOMATION_COMPOSITION_ELEMENT =
        "org.onap.domain.database.PMSH_K8SMicroserviceAutomationCompositionElement";
    private final CommonTestData commonTestData = new CommonTestData();

    @InjectMocks
    @Spy
    private AutomationCompositionElementHandler automationCompositionElementHandler =
        new AutomationCompositionElementHandler();

    @Mock
    private ChartService chartService;

    @Mock
    private ParticipantIntermediaryApi participantIntermediaryApi;

    @Mock
    private ExecutorService executor;
    @Mock
    private Future<String> result;

    @BeforeAll
    static void init() throws CoderException {
        charts = CODER.decode(new File(CHART_INFO_YAML), ChartList.class).getCharts();
        toscaServiceTemplate = TestUtils.testAutomationCompositionRead();
    }

    @Test
    void test_AutomationCompositionElementStateChange() throws ServiceException {
        var automationCompositionElementId1 = UUID.randomUUID();
        var automationCompositionElementId2 = UUID.randomUUID();

        automationCompositionElementHandler.getChartMap().put(automationCompositionElementId1, charts.get(0));
        automationCompositionElementHandler.getChartMap().put(automationCompositionElementId2, charts.get(1));

        doNothing().when(chartService).uninstallChart(charts.get(0));

        automationCompositionElementHandler.undeploy(
            commonTestData.getAutomationCompositionId(), automationCompositionElementId1);

        doThrow(new ServiceException("Error uninstalling the chart")).when(chartService).uninstallChart(charts.get(0));

        assertDoesNotThrow(() -> automationCompositionElementHandler.undeploy(
            commonTestData.getAutomationCompositionId(), automationCompositionElementId1));
    }

    @Test
    void test_AutomationCompositionElementUpdate() throws PfModelException, IOException, ServiceException,
        ExecutionException, InterruptedException {
        doReturn(true).when(chartService).installChart(any());
        doNothing().when(automationCompositionElementHandler).checkPodStatus(any(), any(), any(), anyInt(), anyInt());
        var elementId1 = UUID.randomUUID();
        var element = new AcElementDeploy();
        element.setId(elementId1);
        element.setDefinition(new ToscaConceptIdentifier(KEY_NAME, "1.0.1"));
        element.setOrderedState(DeployOrder.DEPLOY);

        var nodeTemplatesMap =
            toscaServiceTemplate.getToscaTopologyTemplate().getNodeTemplates();
        automationCompositionElementHandler.deploy(
            commonTestData.getAutomationCompositionId(), element,
            nodeTemplatesMap.get(K8S_AUTOMATION_COMPOSITION_ELEMENT).getProperties());

        assertThat(automationCompositionElementHandler.getChartMap()).hasSize(1).containsKey(elementId1);

        doThrow(new ServiceException("Error installing the chart")).when(chartService).installChart(Mockito.any());

        var elementId2 = UUID.randomUUID();
        element.setId(elementId2);
        automationCompositionElementHandler.deploy(
            commonTestData.getAutomationCompositionId(), element,
            nodeTemplatesMap.get(K8S_AUTOMATION_COMPOSITION_ELEMENT).getProperties());

        assertThat(automationCompositionElementHandler.getChartMap().containsKey(elementId2)).isFalse();
    }

    @Test
    void test_checkPodStatus() throws ExecutionException, InterruptedException {
        doReturn(result).when(executor).submit(any(Runnable.class), any());
        doReturn("Done").when(result).get();
        doReturn(true).when(result).isDone();
        var chartInfo = charts.get(0);
        var automationCompositionId = UUID.randomUUID();
        var element = new AutomationCompositionElement();
        assertDoesNotThrow(
            () -> automationCompositionElementHandler.checkPodStatus(automationCompositionId,
                    element.getId(), chartInfo, 1, 1));
    }

    @Test
    void testUpdate() throws PfModelException {
        var elementId1 = UUID.randomUUID();
        var element = new AcElementDeploy();
        element.setId(elementId1);
        element.setDefinition(new ToscaConceptIdentifier(KEY_NAME, "1.0.1"));
        element.setOrderedState(DeployOrder.DEPLOY);
        var automationCompositionId = commonTestData.getAutomationCompositionId();
        assertDoesNotThrow(
                () -> automationCompositionElementHandler.update(automationCompositionId, element, Map.of()));
    }

    @Test
    void testLock() throws PfModelException {
        assertDoesNotThrow(() -> automationCompositionElementHandler.lock(UUID.randomUUID(), UUID.randomUUID()));
    }

    @Test
    void testUnlock() throws PfModelException {
        assertDoesNotThrow(() -> automationCompositionElementHandler.unlock(UUID.randomUUID(), UUID.randomUUID()));
    }

    @Test
    void testDelete() throws PfModelException {
        assertDoesNotThrow(() -> automationCompositionElementHandler.delete(UUID.randomUUID(), UUID.randomUUID()));
    }

    @Test
    void testPrime() throws PfModelException {
        assertDoesNotThrow(() -> automationCompositionElementHandler.prime(UUID.randomUUID(), List.of()));
    }

    @Test
    void testDeprime() throws PfModelException {
        assertDoesNotThrow(() -> automationCompositionElementHandler.deprime(UUID.randomUUID()));
    }
}