8a7f403f4666f5ff97fce33cc4f61a4e3fcc32dc
[policy/clamp.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2021 Nordix Foundation.
4  * ================================================================================
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  *      http://www.apache.org/licenses/LICENSE-2.0
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  * SPDX-License-Identifier: Apache-2.0
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.onap.policy.clamp.controlloop.participant.dcae.main.handler;
22
23 import static org.assertj.core.api.Assertions.assertThat;
24 import static org.mockito.ArgumentMatchers.any;
25 import static org.mockito.ArgumentMatchers.eq;
26 import static org.mockito.Mockito.mock;
27 import static org.mockito.Mockito.spy;
28 import static org.mockito.Mockito.times;
29 import static org.mockito.Mockito.verify;
30 import static org.mockito.Mockito.when;
31
32 import java.util.UUID;
33 import org.json.JSONException;
34 import org.junit.jupiter.api.BeforeAll;
35 import org.junit.jupiter.api.Test;
36 import org.junit.jupiter.api.extension.ExtendWith;
37 import org.onap.policy.clamp.controlloop.models.controlloop.concepts.ControlLoopElement;
38 import org.onap.policy.clamp.controlloop.models.controlloop.concepts.ControlLoopOrderedState;
39 import org.onap.policy.clamp.controlloop.models.controlloop.concepts.ControlLoopState;
40 import org.onap.policy.clamp.controlloop.models.messages.dmaap.participant.ParticipantUpdate;
41 import org.onap.policy.clamp.controlloop.participant.dcae.httpclient.ClampHttpClient;
42 import org.onap.policy.clamp.controlloop.participant.dcae.httpclient.ConsulDcaeHttpClient;
43 import org.onap.policy.clamp.controlloop.participant.dcae.main.parameters.CommonTestData;
44 import org.onap.policy.clamp.controlloop.participant.dcae.main.rest.TestListenerUtils;
45 import org.onap.policy.clamp.controlloop.participant.dcae.model.Loop;
46 import org.onap.policy.clamp.controlloop.participant.intermediary.api.ParticipantIntermediaryApi;
47 import org.onap.policy.clamp.controlloop.participant.intermediary.comm.ParticipantUpdateListener;
48 import org.onap.policy.clamp.controlloop.participant.intermediary.handler.ParticipantHandler;
49 import org.onap.policy.common.endpoints.event.comm.Topic.CommInfrastructure;
50 import org.onap.policy.common.utils.coder.Coder;
51 import org.onap.policy.common.utils.coder.CoderException;
52 import org.onap.policy.common.utils.coder.StandardCoder;
53 import org.onap.policy.models.base.PfModelException;
54 import org.onap.policy.models.tosca.authorative.concepts.ToscaServiceTemplate;
55 import org.springframework.beans.factory.annotation.Autowired;
56 import org.springframework.test.context.junit.jupiter.SpringExtension;
57
58 /**
59  * Class to perform unit test of {@link ControlLoopElementHandler}.
60  *
61  */
62 @ExtendWith(SpringExtension.class)
63 class ControlLoopElementHandlerTest {
64
65     private static final String LOOP = "pmsh_loop";
66     private static final String TEMPLATE = "LOOP_TEMPLATE_k8s_pmsh";
67     private static final String BLUEPRINT_DEPLOYED = "BLUEPRINT_DEPLOYED";
68     private static final String MICROSERVICE_INSTALLED_SUCCESSFULLY = "MICROSERVICE_INSTALLED_SUCCESSFULLY";
69
70     public static final Coder CODER = new StandardCoder();
71     private static final Object lockit = new Object();
72     private static final CommInfrastructure INFRA = CommInfrastructure.NOOP;
73     private static final String TOPIC = "my-topic";
74     private CommonTestData commonTestData = new CommonTestData();
75
76     @Autowired
77     private ParticipantHandler participantHandler;
78
79     /**
80      * Initial ParticipantUpdate message.
81      */
82     @BeforeAll
83     public void sendParticipantUpdate() {
84         ParticipantUpdate participantUpdateMsg = TestListenerUtils.createParticipantUpdateMsg();
85
86         synchronized (lockit) {
87             ParticipantUpdateListener participantUpdateListener = new ParticipantUpdateListener(participantHandler);
88             participantUpdateListener.onTopicEvent(INFRA, TOPIC, null, participantUpdateMsg);
89         }
90     }
91
92     @Test
93     void test_ControlLoopElementStateChange() {
94         ClampHttpClient clampClient = spy(mock(ClampHttpClient.class));
95         ConsulDcaeHttpClient consulClient = mock(ConsulDcaeHttpClient.class);
96         ControlLoopElementHandler controlLoopElementHandler =
97                 new ControlLoopElementHandler(clampClient, consulClient, commonTestData.getParticipantDcaeParameters());
98
99         when(clampClient.getstatus(eq(LOOP))).thenReturn(new Loop());
100
101         ParticipantIntermediaryApi intermediaryApi = mock(ParticipantIntermediaryApi.class);
102         controlLoopElementHandler.setIntermediaryApi(intermediaryApi);
103
104         UUID controlLoopElementId = UUID.randomUUID();
105         controlLoopElementHandler.controlLoopElementStateChange(controlLoopElementId, ControlLoopState.PASSIVE,
106                 ControlLoopOrderedState.UNINITIALISED);
107
108         verify(clampClient).undeploy(eq(LOOP));
109         controlLoopElementHandler.handleStatistics(controlLoopElementId);
110         assertThat(intermediaryApi.getControlLoopElement(controlLoopElementId)).isNull();
111     }
112
113     @Test
114     void testCreate_ControlLoopElementUpdate() throws PfModelException, JSONException, CoderException {
115         ClampHttpClient clampClient = spy(mock(ClampHttpClient.class));
116         Loop loopDeployed = CODER.convert(CommonTestData.createJsonStatus(BLUEPRINT_DEPLOYED), Loop.class);
117         when(clampClient.create(eq(LOOP), eq(TEMPLATE))).thenReturn(loopDeployed);
118         when(clampClient.deploy(eq(LOOP))).thenReturn(true);
119
120         Loop loopInstalled =
121                 CODER.convert(CommonTestData.createJsonStatus(MICROSERVICE_INSTALLED_SUCCESSFULLY), Loop.class);
122         when(clampClient.getstatus(eq(LOOP))).thenReturn(null, loopInstalled);
123
124         ConsulDcaeHttpClient consulClient = spy(mock(ConsulDcaeHttpClient.class));
125         when(consulClient.deploy(any(String.class), any(String.class))).thenReturn(true);
126
127         ControlLoopElementHandler controlLoopElementHandler =
128                 new ControlLoopElementHandler(clampClient, consulClient, commonTestData.getParticipantDcaeParameters());
129
130         ParticipantIntermediaryApi intermediaryApi = mock(ParticipantIntermediaryApi.class);
131         controlLoopElementHandler.setIntermediaryApi(intermediaryApi);
132
133         ControlLoopElement element = new ControlLoopElement();
134         element.setId(UUID.randomUUID());
135         element.setOrderedState(ControlLoopOrderedState.PASSIVE);
136
137         final ToscaServiceTemplate controlLoopDefinition = new ToscaServiceTemplate();
138         controlLoopElementHandler.controlLoopElementUpdate(element,
139             controlLoopDefinition.getToscaTopologyTemplate().getNodeTemplates()
140             .get("org.onap.domain.pmsh.PMSH_DCAEMicroservice"));
141
142         verify(clampClient).create(eq(LOOP), eq(TEMPLATE));
143         verify(consulClient).deploy(any(String.class), any(String.class));
144         verify(clampClient).deploy(eq(LOOP));
145     }
146
147     @Test
148     void test_ControlLoopElementUpdate() throws PfModelException, JSONException, CoderException {
149         ClampHttpClient clampClient = spy(mock(ClampHttpClient.class));
150         Loop loopDeployed = CODER.convert(CommonTestData.createJsonStatus(BLUEPRINT_DEPLOYED), Loop.class);
151         Loop loopInstalled =
152                 CODER.convert(CommonTestData.createJsonStatus(MICROSERVICE_INSTALLED_SUCCESSFULLY), Loop.class);
153         when(clampClient.getstatus(eq(LOOP))).thenReturn(loopDeployed, loopInstalled);
154         when(clampClient.deploy(eq(LOOP))).thenReturn(true);
155
156         ConsulDcaeHttpClient consulClient = spy(mock(ConsulDcaeHttpClient.class));
157         when(consulClient.deploy(any(String.class), any(String.class))).thenReturn(true);
158
159         ControlLoopElementHandler controlLoopElementHandler =
160                 new ControlLoopElementHandler(clampClient, consulClient, commonTestData.getParticipantDcaeParameters());
161
162         ParticipantIntermediaryApi intermediaryApi = mock(ParticipantIntermediaryApi.class);
163         controlLoopElementHandler.setIntermediaryApi(intermediaryApi);
164
165         ControlLoopElement element = new ControlLoopElement();
166         element.setId(UUID.randomUUID());
167         element.setOrderedState(ControlLoopOrderedState.PASSIVE);
168
169         ToscaServiceTemplate controlLoopDefinition = new ToscaServiceTemplate();
170         controlLoopElementHandler.controlLoopElementUpdate(element,
171                 controlLoopDefinition.getToscaTopologyTemplate().getNodeTemplates()
172                 .get("org.onap.domain.pmsh.PMSH_DCAEMicroservice"));
173
174         verify(clampClient, times(0)).create(eq(LOOP), eq(TEMPLATE));
175         verify(consulClient).deploy(any(String.class), any(String.class));
176         verify(clampClient).deploy(eq(LOOP));
177     }
178 }