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

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 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.runtime.supervision;
  22
  23 import static org.mockito.ArgumentMatchers.any;
  24 import static org.mockito.Mockito.mock;
  25 import static org.mockito.Mockito.verify;
  26 import static org.mockito.Mockito.when;
  27
  28 import java.util.List;
  29 import java.util.Optional;
  30 import java.util.UUID;
  31 import org.junit.jupiter.api.Test;
  32 import org.onap.policy.clamp.acm.runtime.supervision.comm.ParticipantDeregisterAckPublisher;
  33 import org.onap.policy.clamp.acm.runtime.supervision.comm.ParticipantRegisterAckPublisher;
  34 import org.onap.policy.clamp.acm.runtime.util.CommonTestData;
  35 import org.onap.policy.clamp.models.acm.concepts.AutomationCompositionInfo;
  36 import org.onap.policy.clamp.models.acm.concepts.ParticipantState;
  37 import org.onap.policy.clamp.models.acm.concepts.ParticipantSupportedElementType;
  38 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantDeregister;
  39 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantRegister;
  40 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantStatus;
  41 import org.onap.policy.clamp.models.acm.persistence.provider.AutomationCompositionProvider;
  42 import org.onap.policy.clamp.models.acm.persistence.provider.ParticipantProvider;
  43
  44 class SupervisionParticipantHandlerTest {
  45     @Test
  46     void testHandleParticipantDeregister() {
  47         var participant = CommonTestData.createParticipant(CommonTestData.getParticipantId());
  48
  49         var participantProvider = mock(ParticipantProvider.class);
  50         when(participantProvider.findParticipant(CommonTestData.getParticipantId()))
  51                 .thenReturn(Optional.of(participant));
  52
  53         var participantDeregisterMessage = new ParticipantDeregister();
  54         participantDeregisterMessage.setMessageId(UUID.randomUUID());
  55         participantDeregisterMessage.setParticipantId(CommonTestData.getParticipantId());
  56         var participantDeregisterAckPublisher = mock(ParticipantDeregisterAckPublisher.class);
  57         var handler = new SupervisionParticipantHandler(participantProvider,
  58             mock(ParticipantRegisterAckPublisher.class), participantDeregisterAckPublisher,
  59             mock(AutomationCompositionProvider.class));
  60
  61         handler.handleParticipantMessage(participantDeregisterMessage);
  62
  63         verify(participantProvider).updateParticipant(any());
  64         verify(participantProvider).resetParticipantAcElementState(any());
  65         verify(participantDeregisterAckPublisher).send(participantDeregisterMessage.getMessageId());
  66     }
  67
  68     @Test
  69     void testHandleParticipantRegister() {
  70         var participantRegisterMessage = new ParticipantRegister();
  71         participantRegisterMessage.setMessageId(UUID.randomUUID());
  72         participantRegisterMessage.setParticipantId(CommonTestData.getParticipantId());
  73         var supportedElementType = new ParticipantSupportedElementType();
  74         supportedElementType.setTypeName("Type");
  75         supportedElementType.setTypeVersion("1.0.0");
  76         participantRegisterMessage.setParticipantSupportedElementType(List.of(supportedElementType));
  77
  78         var participantProvider = mock(ParticipantProvider.class);
  79         var participantRegisterAckPublisher = mock(ParticipantRegisterAckPublisher.class);
  80         var handler = new SupervisionParticipantHandler(participantProvider, participantRegisterAckPublisher,
  81                 mock(ParticipantDeregisterAckPublisher.class),
  82                 mock(AutomationCompositionProvider.class));
  83         handler.handleParticipantMessage(participantRegisterMessage);
  84
  85         verify(participantProvider).saveParticipant(any());
  86         verify(participantRegisterAckPublisher).send(participantRegisterMessage.getMessageId(),
  87                 CommonTestData.getParticipantId());
  88     }
  89
  90     @Test
  91     void testHandleParticipantStatus() {
  92         var participantStatusMessage = new ParticipantStatus();
  93         participantStatusMessage.setParticipantId(CommonTestData.getParticipantId());
  94         participantStatusMessage.setState(ParticipantState.ON_LINE);
  95         var supportedElementType = new ParticipantSupportedElementType();
  96         supportedElementType.setTypeName("Type");
  97         supportedElementType.setTypeVersion("1.0.0");
  98         participantStatusMessage.setParticipantSupportedElementType(List.of(supportedElementType));
  99         participantStatusMessage.setAutomationCompositionInfoList(List.of(new AutomationCompositionInfo()));
 100
 101         var participantProvider = mock(ParticipantProvider.class);
 102         var handler = new SupervisionParticipantHandler(participantProvider,
 103             mock(ParticipantRegisterAckPublisher.class), mock(ParticipantDeregisterAckPublisher.class),
 104             mock(AutomationCompositionProvider.class));
 105         var participant = CommonTestData.createParticipant(CommonTestData.getParticipantId());
 106         when(participantProvider.findParticipant(CommonTestData.getParticipantId()))
 107                 .thenReturn(Optional.of(participant));
 108         handler.handleParticipantMessage(participantStatusMessage);
 109
 110         verify(participantProvider).updateParticipant(any());
 111     }
 112 }