gerrit.onap Code Review - policy/clamp.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2021-2023 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.acm.participant.intermediary.handler;
  22
  23 import static org.assertj.core.api.Assertions.assertThat;
  24 import static org.assertj.core.api.Assertions.assertThatThrownBy;
  25 import static org.junit.jupiter.api.Assertions.assertEquals;
  26 import static org.junit.jupiter.api.Assertions.assertFalse;
  27 import static org.junit.jupiter.api.Assertions.assertTrue;
  28 import static org.mockito.ArgumentMatchers.any;
  29 import static org.mockito.Mockito.mock;
  30 import static org.mockito.Mockito.verify;
  31 import static org.mockito.Mockito.when;
  32
  33 import java.time.Instant;
  34 import java.util.List;
  35 import java.util.UUID;
  36 import org.junit.jupiter.api.Test;
  37 import org.onap.policy.clamp.acm.participant.intermediary.comm.ParticipantMessagePublisher;
  38 import org.onap.policy.clamp.acm.participant.intermediary.main.parameters.CommonTestData;
  39 import org.onap.policy.clamp.models.acm.concepts.ParticipantDefinition;
  40 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantAckMessage;
  41 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantDeregister;
  42 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantMessage;
  43 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantMessageType;
  44 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantPrime;
  45 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantRegisterAck;
  46 import org.onap.policy.common.utils.coder.CoderException;
  47 import org.onap.policy.models.base.PfModelException;
  48
  49 class ParticipantHandlerTest {
  50
  51     private final CommonTestData commonTestData = new CommonTestData();
  52
  53     @Test
  54     void handleUpdateTest() {
  55         var parameters = CommonTestData.getParticipantParameters();
  56         var automationCompositionHander = commonTestData.getMockAutomationCompositionHandler();
  57         var publisher = new ParticipantMessagePublisher();
  58         var emptyParticipantHandler =
  59                 new ParticipantHandler(parameters, publisher, automationCompositionHander);
  60         var participantPrimeMsg = new ParticipantPrime();
  61
  62         assertThatThrownBy(() ->
  63                 emptyParticipantHandler.handleParticipantPrime(participantPrimeMsg))
  64                 .isInstanceOf(RuntimeException.class);
  65
  66         var participantHandler = commonTestData.getMockParticipantHandler();
  67
  68         var participantId = CommonTestData.getParticipantId();
  69         participantPrimeMsg.setAutomationCompositionId(CommonTestData.AC_ID_1);
  70         participantPrimeMsg.setCompositionId(CommonTestData.AC_ID_1);
  71         participantPrimeMsg.setParticipantId(participantId);
  72         participantPrimeMsg.setMessageId(UUID.randomUUID());
  73         participantPrimeMsg.setTimestamp(Instant.ofEpochMilli(3000));
  74
  75         var heartbeatF = participantHandler.makeHeartbeat(false);
  76         assertEquals(participantId, heartbeatF.getParticipantId());
  77         assertThat(heartbeatF.getAutomationCompositionInfoList()).isEmpty();
  78
  79         participantHandler.handleParticipantPrime(participantPrimeMsg);
  80
  81         var heartbeatT = participantHandler.makeHeartbeat(true);
  82         assertEquals(participantId, heartbeatT.getParticipantId());
  83         assertThat(heartbeatT.getParticipantDefinitionUpdates()).isNotEmpty();
  84         assertEquals(participantId, heartbeatT.getParticipantDefinitionUpdates().get(0).getParticipantId());
  85
  86         var pum = setListParticipantDefinition(participantPrimeMsg);
  87         participantHandler.handleParticipantPrime(pum);
  88         var heartbeatTAfterUpdate = participantHandler.makeHeartbeat(true);
  89         assertEquals(participantId, heartbeatTAfterUpdate.getParticipantId());
  90     }
  91
  92     private ParticipantPrime setListParticipantDefinition(ParticipantPrime participantPrimeMsg) {
  93         var def = new ParticipantDefinition();
  94         def.setParticipantId(CommonTestData.getParticipantId());
  95         participantPrimeMsg.setParticipantDefinitionUpdates(List.of(def));
  96         return participantPrimeMsg;
  97     }
  98
  99     @Test
 100     void checkAppliesTo() {
 101         var participantHandler = commonTestData.getMockParticipantHandler();
 102         var participantAckMsg =
 103                 new ParticipantAckMessage(ParticipantMessageType.AUTOMATION_COMPOSITION_DEPLOY);
 104         assertTrue(participantHandler.appliesTo(participantAckMsg));
 105
 106         var participantMsg =
 107                 new ParticipantMessage(ParticipantMessageType.PARTICIPANT_STATUS);
 108         assertTrue(participantHandler.appliesTo(participantMsg));
 109
 110         var randomId = UUID.randomUUID();
 111         participantMsg.setParticipantId(randomId);
 112         assertFalse(participantHandler.appliesTo(participantMsg));
 113
 114     }
 115
 116     @Test
 117     void getAutomationCompositionInfoListTest() throws CoderException {
 118         var participantHandler = commonTestData.getParticipantHandlerAutomationCompositions();
 119         participantHandler.sendHeartbeat();
 120         assertEquals(CommonTestData.AC_ID_1, participantHandler.makeHeartbeat(false)
 121                 .getAutomationCompositionInfoList()
 122                 .get(0)
 123                 .getAutomationCompositionId());
 124
 125     }
 126
 127     @Test
 128     void testHandleParticipantRegisterAck() {
 129         var parameters = CommonTestData.getParticipantParameters();
 130         var automationCompositionHandler = commonTestData.getMockAutomationCompositionHandler();
 131         var publisher = mock(ParticipantMessagePublisher.class);
 132         var participantHandler = new ParticipantHandler(parameters, publisher, automationCompositionHandler);
 133
 134         participantHandler.handleParticipantRegisterAck(new ParticipantRegisterAck());
 135         verify(publisher).sendParticipantStatus(any());
 136     }
 137
 138     @Test
 139     void testSendParticipantDeregister() throws PfModelException {
 140         var commonTestData = new CommonTestData();
 141         var automationCompositionMap = commonTestData.getTestAutomationCompositionMap();
 142         var automationCompositionHandler = mock(AutomationCompositionHandler.class);
 143         var listener = mock(DummyAcElementListener.class);
 144
 145         when(automationCompositionHandler.getListeners()).thenReturn(List.of(listener));
 146         automationCompositionMap.values().iterator().next().getElements().values().iterator().next()
 147             .setParticipantId(CommonTestData.getParticipantId());
 148         when(automationCompositionHandler.getAutomationCompositionMap()).thenReturn(automationCompositionMap);
 149
 150         var publisher = mock(ParticipantMessagePublisher.class);
 151         var parameters = CommonTestData.getParticipantParameters();
 152         var participantHandler = new ParticipantHandler(parameters, publisher, automationCompositionHandler);
 153
 154         participantHandler.sendParticipantDeregister();
 155         verify(publisher).sendParticipantDeregister(any(ParticipantDeregister.class));
 156         verify(listener).undeploy(any(), any());
 157     }
 158 }