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.ArgumentMatchers.anyBoolean;
  25 import static org.mockito.ArgumentMatchers.anyInt;
  26 import static org.mockito.Mockito.mock;
  27 import static org.mockito.Mockito.verify;
  28 import static org.mockito.Mockito.when;
  29 import static org.onap.policy.clamp.acm.runtime.util.CommonTestData.TOSCA_SERVICE_TEMPLATE_YAML;
  30
  31 import java.util.Map;
  32 import java.util.Optional;
  33 import java.util.UUID;
  34 import org.junit.jupiter.api.Test;
  35 import org.onap.policy.clamp.acm.runtime.instantiation.InstantiationUtils;
  36 import org.onap.policy.clamp.acm.runtime.supervision.comm.AutomationCompositionDeployPublisher;
  37 import org.onap.policy.clamp.acm.runtime.supervision.comm.AutomationCompositionStateChangePublisher;
  38 import org.onap.policy.clamp.acm.runtime.util.CommonTestData;
  39 import org.onap.policy.clamp.models.acm.concepts.AcElementDeployAck;
  40 import org.onap.policy.clamp.models.acm.concepts.AcTypeState;
  41 import org.onap.policy.clamp.models.acm.concepts.AutomationComposition;
  42 import org.onap.policy.clamp.models.acm.concepts.DeployState;
  43 import org.onap.policy.clamp.models.acm.concepts.LockState;
  44 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.AutomationCompositionDeployAck;
  45 import org.onap.policy.clamp.models.acm.messages.dmaap.participant.ParticipantMessageType;
  46 import org.onap.policy.clamp.models.acm.persistence.provider.AutomationCompositionProvider;
  47
  48 class SupervisionAcHandlerTest {
  49     private static final String AC_INSTANTIATION_CREATE_JSON = "src/test/resources/rest/acm/AutomationComposition.json";
  50     private static final UUID IDENTIFIER = UUID.randomUUID();
  51
  52     @Test
  53     void testHandleAutomationCompositionStateChangeAckMessage() {
  54         var automationCompositionProvider = mock(AutomationCompositionProvider.class);
  55         var automationComposition =
  56                 InstantiationUtils.getAutomationCompositionFromResource(AC_INSTANTIATION_CREATE_JSON, "Crud");
  57         when(automationCompositionProvider.findAutomationComposition(IDENTIFIER))
  58                 .thenReturn(Optional.of(automationComposition));
  59
  60         var handler = new SupervisionAcHandler(automationCompositionProvider,
  61                 mock(AutomationCompositionDeployPublisher.class),
  62                 mock(AutomationCompositionStateChangePublisher.class));
  63
  64         var automationCompositionAckMessage =
  65                 new AutomationCompositionDeployAck(ParticipantMessageType.AUTOMATION_COMPOSITION_STATECHANGE_ACK);
  66         for (var elementEntry : automationComposition.getElements().entrySet()) {
  67             var acElementDeployAck =
  68                     new AcElementDeployAck(DeployState.DEPLOYED, LockState.UNLOCKED, "", "", Map.of(), true, "");
  69             automationCompositionAckMessage.getAutomationCompositionResultMap().put(elementEntry.getKey(),
  70                     acElementDeployAck);
  71         }
  72         automationCompositionAckMessage.setAutomationCompositionId(IDENTIFIER);
  73
  74         handler.handleAutomationCompositionStateChangeAckMessage(automationCompositionAckMessage);
  75
  76         verify(automationCompositionProvider).updateAutomationComposition(any(AutomationComposition.class));
  77     }
  78
  79     @Test
  80     void testHandleAutomationCompositionUpdateAckMessage() {
  81         var automationCompositionProvider = mock(AutomationCompositionProvider.class);
  82         var automationComposition =
  83                 InstantiationUtils.getAutomationCompositionFromResource(AC_INSTANTIATION_CREATE_JSON, "Crud");
  84         when(automationCompositionProvider.findAutomationComposition(IDENTIFIER))
  85                 .thenReturn(Optional.of(automationComposition));
  86
  87         var automationCompositionAckMessage =
  88                 new AutomationCompositionDeployAck(ParticipantMessageType.AUTOMATION_COMPOSITION_DEPLOY_ACK);
  89         for (var elementEntry : automationComposition.getElements().entrySet()) {
  90             var acElementDeployAck =
  91                     new AcElementDeployAck(DeployState.DEPLOYED, LockState.LOCKED, "", "", Map.of(), true, "");
  92             automationCompositionAckMessage
  93                     .setAutomationCompositionResultMap(Map.of(elementEntry.getKey(), acElementDeployAck));
  94         }
  95         automationCompositionAckMessage.setParticipantId(CommonTestData.getParticipantId());
  96         automationCompositionAckMessage.setAutomationCompositionId(IDENTIFIER);
  97
  98         var handler = new SupervisionAcHandler(automationCompositionProvider,
  99                 mock(AutomationCompositionDeployPublisher.class),
 100                 mock(AutomationCompositionStateChangePublisher.class));
 101
 102         handler.handleAutomationCompositionUpdateAckMessage(automationCompositionAckMessage);
 103
 104         verify(automationCompositionProvider).updateAutomationComposition(any(AutomationComposition.class));
 105     }
 106
 107     @Test
 108     void testUndeploy() {
 109         var automationCompositionProvider = mock(AutomationCompositionProvider.class);
 110         var acStateChangePublisher = mock(AutomationCompositionStateChangePublisher.class);
 111         var handler = new SupervisionAcHandler(automationCompositionProvider,
 112                 mock(AutomationCompositionDeployPublisher.class),
 113                 acStateChangePublisher);
 114         var serviceTemplate = InstantiationUtils.getToscaServiceTemplate(TOSCA_SERVICE_TEMPLATE_YAML);
 115         var acDefinition = CommonTestData.createAcDefinition(serviceTemplate, AcTypeState.PRIMED);
 116         var automationComposition =
 117                 InstantiationUtils.getAutomationCompositionFromResource(AC_INSTANTIATION_CREATE_JSON, "Undeploy");
 118         handler.undeploy(automationComposition, acDefinition);
 119
 120         verify(automationCompositionProvider).updateAutomationComposition(any(AutomationComposition.class));
 121         verify(acStateChangePublisher).send(any(AutomationComposition.class), anyInt(), anyBoolean());
 122     }
 123
 124     @Test
 125     void testUnlock() {
 126         var automationCompositionProvider = mock(AutomationCompositionProvider.class);
 127         var acStateChangePublisher = mock(AutomationCompositionStateChangePublisher.class);
 128         var handler = new SupervisionAcHandler(automationCompositionProvider,
 129                 mock(AutomationCompositionDeployPublisher.class),
 130                 acStateChangePublisher);
 131         var serviceTemplate = InstantiationUtils.getToscaServiceTemplate(TOSCA_SERVICE_TEMPLATE_YAML);
 132         var acDefinition = CommonTestData.createAcDefinition(serviceTemplate, AcTypeState.PRIMED);
 133         var automationComposition =
 134                 InstantiationUtils.getAutomationCompositionFromResource(AC_INSTANTIATION_CREATE_JSON, "UnLock");
 135         handler.unlock(automationComposition, acDefinition);
 136
 137         verify(automationCompositionProvider).updateAutomationComposition(any(AutomationComposition.class));
 138         verify(acStateChangePublisher).send(any(AutomationComposition.class), anyInt(), anyBoolean());
 139     }
 140
 141     @Test
 142     void testLock() {
 143         var automationCompositionProvider = mock(AutomationCompositionProvider.class);
 144         var acStateChangePublisher = mock(AutomationCompositionStateChangePublisher.class);
 145         var handler = new SupervisionAcHandler(automationCompositionProvider,
 146                 mock(AutomationCompositionDeployPublisher.class),
 147                 acStateChangePublisher);
 148         var serviceTemplate = InstantiationUtils.getToscaServiceTemplate(TOSCA_SERVICE_TEMPLATE_YAML);
 149         var acDefinition = CommonTestData.createAcDefinition(serviceTemplate, AcTypeState.PRIMED);
 150         var automationComposition =
 151                 InstantiationUtils.getAutomationCompositionFromResource(AC_INSTANTIATION_CREATE_JSON, "Lock");
 152         handler.lock(automationComposition, acDefinition);
 153
 154         verify(automationCompositionProvider).updateAutomationComposition(any(AutomationComposition.class));
 155         verify(acStateChangePublisher).send(any(AutomationComposition.class), anyInt(), anyBoolean());
 156     }
 157 }