4dfb8a9fd21553c8a340b12037129ed3d98e8c83
[policy/clamp.git] /
1 /*-
2  * ============LICENSE_START=======================================================
3  *  Copyright (C) 2021 Nordix Foundation.
4  *  Modifications Copyright (C) 2021 AT&T Intellectual Property. All rights reserved.
5  * ================================================================================
6  * Licensed under the Apache License, Version 2.0 (the "License");
7  * you may not use this file except in compliance with the License.
8  * You may obtain a copy of the License at
9  *
10  *      http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing, software
13  * distributed under the License is distributed on an "AS IS" BASIS,
14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15  * See the License for the specific language governing permissions and
16  * limitations under the License.
17  *
18  * SPDX-License-Identifier: Apache-2.0
19  * ============LICENSE_END=========================================================
20  */
21
22 package org.onap.policy.clamp.controlloop.runtime.supervision.comm;
23
24 import static org.assertj.core.api.Assertions.assertThatCode;
25 import static org.assertj.core.api.Assertions.assertThatThrownBy;
26 import static org.mockito.ArgumentMatchers.anyString;
27 import static org.mockito.Mockito.mock;
28 import static org.mockito.Mockito.verify;
29
30 import java.time.Instant;
31 import java.util.Collections;
32 import java.util.List;
33 import java.util.UUID;
34 import org.junit.jupiter.api.AfterAll;
35 import org.junit.jupiter.api.BeforeAll;
36 import org.junit.jupiter.api.Test;
37 import org.mockito.Mockito;
38 import org.onap.policy.clamp.controlloop.models.controlloop.concepts.ControlLoop;
39 import org.onap.policy.clamp.controlloop.models.controlloop.concepts.ControlLoopState;
40 import org.onap.policy.clamp.controlloop.models.controlloop.persistence.provider.ClElementStatisticsProvider;
41 import org.onap.policy.clamp.controlloop.models.controlloop.persistence.provider.ControlLoopProvider;
42 import org.onap.policy.clamp.controlloop.models.controlloop.persistence.provider.ParticipantProvider;
43 import org.onap.policy.clamp.controlloop.models.controlloop.persistence.provider.ParticipantStatisticsProvider;
44 import org.onap.policy.clamp.controlloop.models.messages.dmaap.participant.ParticipantDeregister;
45 import org.onap.policy.clamp.controlloop.models.messages.dmaap.participant.ParticipantDeregisterAck;
46 import org.onap.policy.clamp.controlloop.models.messages.dmaap.participant.ParticipantRegisterAck;
47 import org.onap.policy.clamp.controlloop.models.messages.dmaap.participant.ParticipantUpdateAck;
48 import org.onap.policy.clamp.controlloop.runtime.main.parameters.ClRuntimeParameterGroup;
49 import org.onap.policy.clamp.controlloop.runtime.monitoring.MonitoringProvider;
50 import org.onap.policy.clamp.controlloop.runtime.supervision.SupervisionHandler;
51 import org.onap.policy.clamp.controlloop.runtime.util.CommonTestData;
52 import org.onap.policy.clamp.controlloop.runtime.util.rest.CommonRestController;
53 import org.onap.policy.common.endpoints.event.comm.Topic.CommInfrastructure;
54 import org.onap.policy.common.endpoints.event.comm.TopicSink;
55 import org.onap.policy.models.base.PfModelException;
56 import org.onap.policy.models.provider.PolicyModelsProvider;
57 import org.onap.policy.models.tosca.authorative.concepts.ToscaConceptIdentifier;
58
59 class SupervisionMessagesTest extends CommonRestController {
60
61     private static final String NOT_ACTIVE = "Not Active!";
62     private static final Object lockit = new Object();
63     private static final CommInfrastructure INFRA = CommInfrastructure.NOOP;
64     private static final String TOPIC = "my-topic";
65     private static SupervisionHandler supervisionHandler;
66     private static ControlLoopProvider clProvider;
67     private static PolicyModelsProvider modelsProvider;
68
69     /**
70      * setup Db Provider Parameters.
71      *
72      * @throws PfModelException if an error occurs
73      */
74     @BeforeAll
75     public static void setupDbProviderParameters() throws PfModelException {
76         ClRuntimeParameterGroup controlLoopParameters = CommonTestData.geParameterGroup("instantproviderdb");
77
78         modelsProvider = CommonTestData.getPolicyModelsProvider(controlLoopParameters.getDatabaseProviderParameters());
79         clProvider = new ControlLoopProvider(controlLoopParameters.getDatabaseProviderParameters());
80
81         var participantStatisticsProvider =
82                 new ParticipantStatisticsProvider(controlLoopParameters.getDatabaseProviderParameters());
83         var clElementStatisticsProvider =
84                 new ClElementStatisticsProvider(controlLoopParameters.getDatabaseProviderParameters());
85         var monitoringProvider =
86                 new MonitoringProvider(participantStatisticsProvider, clElementStatisticsProvider, clProvider);
87         var participantProvider = new ParticipantProvider(controlLoopParameters.getDatabaseProviderParameters());
88         var modelsProvider = Mockito.mock(PolicyModelsProvider.class);
89         var controlLoopUpdatePublisher = Mockito.mock(ControlLoopUpdatePublisher.class);
90         var controlLoopStateChangePublisher = Mockito.mock(ControlLoopStateChangePublisher.class);
91         var participantRegisterAckPublisher = Mockito.mock(ParticipantRegisterAckPublisher.class);
92         var participantDeregisterAckPublisher = Mockito.mock(ParticipantDeregisterAckPublisher.class);
93         var participantUpdatePublisher = Mockito.mock(ParticipantUpdatePublisher.class);
94         supervisionHandler = new SupervisionHandler(clProvider, participantProvider, monitoringProvider, modelsProvider,
95                 controlLoopUpdatePublisher, controlLoopStateChangePublisher, participantRegisterAckPublisher,
96                 participantDeregisterAckPublisher, participantUpdatePublisher);
97     }
98
99     @AfterAll
100     public static void closeDbProvider() throws PfModelException {
101         clProvider.close();
102         modelsProvider.close();
103     }
104
105     @Test
106     void testSendParticipantRegisterAck() throws Exception {
107         final ParticipantRegisterAck participantRegisterAckMsg = new ParticipantRegisterAck();
108         participantRegisterAckMsg.setMessage("ParticipantRegisterAck message");
109         participantRegisterAckMsg.setResponseTo(UUID.randomUUID());
110         participantRegisterAckMsg.setResult(true);
111
112         synchronized (lockit) {
113             ParticipantRegisterAckPublisher clRegisterAckPublisher = new ParticipantRegisterAckPublisher();
114             clRegisterAckPublisher.active(Collections.singletonList(Mockito.mock(TopicSink.class)));
115             assertThatCode(() -> clRegisterAckPublisher.send(participantRegisterAckMsg)).doesNotThrowAnyException();
116         }
117     }
118
119     @Test
120     void testReceiveParticipantDeregister() throws Exception {
121         final ParticipantDeregister participantDeregisterMsg = new ParticipantDeregister();
122         participantDeregisterMsg.setParticipantId(getParticipantId());
123         participantDeregisterMsg.setTimestamp(Instant.now());
124         participantDeregisterMsg.setParticipantType(getParticipantType());
125
126         synchronized (lockit) {
127             ParticipantDeregisterListener participantDeregisterListener =
128                     new ParticipantDeregisterListener(supervisionHandler);
129             assertThatCode(
130                     () -> participantDeregisterListener.onTopicEvent(INFRA, TOPIC, null, participantDeregisterMsg))
131                             .doesNotThrowAnyException();
132         }
133     }
134
135     @Test
136     void testSendParticipantDeregisterAck() throws Exception {
137         final ParticipantDeregisterAck participantDeregisterAckMsg = new ParticipantDeregisterAck();
138         participantDeregisterAckMsg.setMessage("ParticipantDeregisterAck message");
139         participantDeregisterAckMsg.setResponseTo(UUID.randomUUID());
140         participantDeregisterAckMsg.setResult(true);
141
142         synchronized (lockit) {
143             ParticipantDeregisterAckPublisher clDeregisterAckPublisher = new ParticipantDeregisterAckPublisher();
144             clDeregisterAckPublisher.active(Collections.singletonList(Mockito.mock(TopicSink.class)));
145             assertThatCode(() -> clDeregisterAckPublisher.send(participantDeregisterAckMsg)).doesNotThrowAnyException();
146         }
147     }
148
149     @Test
150     void testReceiveParticipantUpdateAckMessage() throws Exception {
151         final ParticipantUpdateAck participantUpdateAckMsg = new ParticipantUpdateAck();
152         participantUpdateAckMsg.setMessage("ParticipantUpdateAck message");
153         participantUpdateAckMsg.setResponseTo(UUID.randomUUID());
154         participantUpdateAckMsg.setResult(true);
155         participantUpdateAckMsg.setParticipantId(getParticipantId());
156         participantUpdateAckMsg.setParticipantType(getParticipantType());
157
158         synchronized (lockit) {
159             ParticipantUpdateAckListener participantUpdateAckListener =
160                     new ParticipantUpdateAckListener(supervisionHandler);
161             assertThatCode(() -> participantUpdateAckListener.onTopicEvent(INFRA, TOPIC, null, participantUpdateAckMsg))
162                     .doesNotThrowAnyException();
163         }
164     }
165
166     @Test
167     void testSendControlLoopStateChangePublisherNotActive() {
168         var publisher = new ControlLoopStateChangePublisher();
169         assertThatThrownBy(() -> publisher.send(getControlLoop(), 0)).hasMessage(NOT_ACTIVE);
170     }
171
172     @Test
173     void testSendControlLoopStateChangePublisher() {
174         var publisher = new ControlLoopStateChangePublisher();
175         var topicSink = mock(TopicSink.class);
176         publisher.active(List.of(topicSink));
177         publisher.send(getControlLoop(), 0);
178         verify(topicSink).send(anyString());
179     }
180
181     private ControlLoop getControlLoop() {
182         var controlLoop = new ControlLoop();
183         controlLoop.setName("NAME");
184         controlLoop.setVersion("0.0.1");
185         controlLoop.setState(ControlLoopState.UNINITIALISED);
186         return controlLoop;
187     }
188
189     private ToscaConceptIdentifier getParticipantId() {
190         return new ToscaConceptIdentifier("org.onap.PM_Policy", "1.0.0");
191     }
192
193     private ToscaConceptIdentifier getParticipantType() {
194         return new ToscaConceptIdentifier("org.onap.policy.controlloop.PolicyControlLoopParticipant", "2.3.1");
195     }
196 }