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

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2021-2024 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.junit.jupiter.api.Assertions.assertDoesNotThrow;
  24 import static org.junit.jupiter.api.Assertions.assertEquals;
  25 import static org.mockito.ArgumentMatchers.any;
  26 import static org.mockito.ArgumentMatchers.anyInt;
  27 import static org.mockito.Mockito.clearInvocations;
  28 import static org.mockito.Mockito.mock;
  29 import static org.mockito.Mockito.times;
  30 import static org.mockito.Mockito.verify;
  31 import static org.mockito.Mockito.when;
  32
  33 import java.util.HashMap;
  34 import java.util.List;
  35 import java.util.Map;
  36 import java.util.UUID;
  37 import org.junit.jupiter.api.Test;
  38 import org.onap.policy.clamp.acm.participant.intermediary.comm.ParticipantMessagePublisher;
  39 import org.onap.policy.clamp.acm.participant.intermediary.main.parameters.CommonTestData;
  40 import org.onap.policy.clamp.models.acm.concepts.AcElementDeploy;
  41 import org.onap.policy.clamp.models.acm.concepts.AutomationComposition;
  42 import org.onap.policy.clamp.models.acm.concepts.AutomationCompositionElementDefinition;
  43 import org.onap.policy.clamp.models.acm.concepts.DeployState;
  44 import org.onap.policy.clamp.models.acm.concepts.ParticipantDeploy;
  45 import org.onap.policy.clamp.models.acm.messages.kafka.participant.AutomationCompositionDeploy;
  46 import org.onap.policy.clamp.models.acm.messages.kafka.participant.AutomationCompositionDeployAck;
  47 import org.onap.policy.clamp.models.acm.messages.kafka.participant.AutomationCompositionMigration;
  48 import org.onap.policy.clamp.models.acm.messages.kafka.participant.AutomationCompositionStateChange;
  49 import org.onap.policy.clamp.models.acm.messages.kafka.participant.PropertiesUpdate;
  50 import org.onap.policy.clamp.models.acm.messages.rest.instantiation.DeployOrder;
  51 import org.onap.policy.clamp.models.acm.messages.rest.instantiation.LockOrder;
  52 import org.onap.policy.models.tosca.authorative.concepts.ToscaConceptIdentifier;
  53
  54 class AutomationCompositionHandlerTest {
  55
  56     @Test
  57     void handleAutomationCompositionStateChangeNullTest() {
  58         var participantMessagePublisher = mock(ParticipantMessagePublisher.class);
  59         var cacheProvider = mock(CacheProvider.class);
  60         var ach =
  61                 new AutomationCompositionHandler(cacheProvider, participantMessagePublisher, mock(ThreadHandler.class));
  62
  63         var automationCompositionStateChange = new AutomationCompositionStateChange();
  64         assertDoesNotThrow(() -> ach.handleAutomationCompositionStateChange(automationCompositionStateChange));
  65
  66         automationCompositionStateChange.setAutomationCompositionId(UUID.randomUUID());
  67         automationCompositionStateChange.setDeployOrderedState(DeployOrder.DELETE);
  68         assertDoesNotThrow(() -> ach.handleAutomationCompositionStateChange(automationCompositionStateChange));
  69         verify(participantMessagePublisher).sendAutomationCompositionAck(any(AutomationCompositionDeployAck.class));
  70
  71         var automationComposition = CommonTestData.getTestAutomationCompositionMap().values().iterator().next();
  72         automationCompositionStateChange.setAutomationCompositionId(automationComposition.getInstanceId());
  73         when(cacheProvider.getAutomationComposition(automationComposition.getInstanceId()))
  74                 .thenReturn(automationComposition);
  75         automationCompositionStateChange.setDeployOrderedState(DeployOrder.UPDATE);
  76         assertDoesNotThrow(() -> ach.handleAutomationCompositionStateChange(automationCompositionStateChange));
  77     }
  78
  79     @Test
  80     void handleAutomationCompositionStateChangeUndeployTest() {
  81         var automationComposition = CommonTestData.getTestAutomationCompositionMap().values().iterator().next();
  82         var cacheProvider = mock(CacheProvider.class);
  83         when(cacheProvider.getAutomationComposition(automationComposition.getInstanceId()))
  84                 .thenReturn(automationComposition);
  85         when(cacheProvider.getCommonProperties(any(UUID.class), any(UUID.class))).thenReturn(Map.of());
  86
  87         var participantMessagePublisher = mock(ParticipantMessagePublisher.class);
  88         var listener = mock(ThreadHandler.class);
  89         var ach = new AutomationCompositionHandler(cacheProvider, participantMessagePublisher, listener);
  90         Map<ToscaConceptIdentifier, AutomationCompositionElementDefinition> map = new HashMap<>();
  91         for (var element : automationComposition.getElements().values()) {
  92             map.put(element.getDefinition(), new AutomationCompositionElementDefinition());
  93         }
  94         when(cacheProvider.getAcElementsDefinitions())
  95             .thenReturn(Map.of(automationComposition.getCompositionId(), map));
  96         var automationCompositionStateChange = CommonTestData.getStateChange(CommonTestData.getParticipantId(),
  97             automationComposition.getInstanceId(), DeployOrder.UNDEPLOY, LockOrder.NONE);
  98
  99         ach.handleAutomationCompositionStateChange(automationCompositionStateChange);
 100         verify(listener, times(automationComposition.getElements().size())).undeploy(any(), any(), any());
 101         for (var element : automationComposition.getElements().values()) {
 102             assertEquals(DeployState.UNDEPLOYING, element.getDeployState());
 103         }
 104
 105         clearInvocations(listener);
 106         automationCompositionStateChange.setStartPhase(2);
 107         ach.handleAutomationCompositionStateChange(automationCompositionStateChange);
 108         verify(listener, times(0)).undeploy(any(), any(), any());
 109     }
 110
 111     @Test
 112     void handleAutomationCompositionStateChangeDeleteTest() {
 113         var automationComposition = CommonTestData.getTestAutomationCompositionMap().values().iterator().next();
 114         var cacheProvider = mock(CacheProvider.class);
 115         when(cacheProvider.getAutomationComposition(automationComposition.getInstanceId()))
 116                 .thenReturn(automationComposition);
 117         when(cacheProvider.getCommonProperties(any(UUID.class), any(UUID.class))).thenReturn(Map.of());
 118
 119         var participantMessagePublisher = mock(ParticipantMessagePublisher.class);
 120         var listener = mock(ThreadHandler.class);
 121         var ach = new AutomationCompositionHandler(cacheProvider, participantMessagePublisher, listener);
 122         Map<ToscaConceptIdentifier, AutomationCompositionElementDefinition> map = new HashMap<>();
 123         for (var element : automationComposition.getElements().values()) {
 124             map.put(element.getDefinition(), new AutomationCompositionElementDefinition());
 125         }
 126         when(cacheProvider.getAcElementsDefinitions())
 127             .thenReturn(Map.of(automationComposition.getCompositionId(), map));
 128         var automationCompositionStateChange = CommonTestData.getStateChange(CommonTestData.getParticipantId(),
 129             automationComposition.getInstanceId(), DeployOrder.DELETE, LockOrder.NONE);
 130         ach.handleAutomationCompositionStateChange(automationCompositionStateChange);
 131         verify(listener, times(automationComposition.getElements().size())).delete(any(), any(), any());
 132         for (var element : automationComposition.getElements().values()) {
 133             assertEquals(DeployState.DELETING, element.getDeployState());
 134         }
 135
 136         clearInvocations(listener);
 137         automationCompositionStateChange.setStartPhase(2);
 138         ach.handleAutomationCompositionStateChange(automationCompositionStateChange);
 139         verify(listener, times(0)).delete(any(), any(), any());
 140     }
 141
 142     @Test
 143     void handleAcPropertyUpdateTest() {
 144         var cacheProvider = mock(CacheProvider.class);
 145         var listener = mock(ThreadHandler.class);
 146         var participantMessagePublisher = mock(ParticipantMessagePublisher.class);
 147         var ach = new AutomationCompositionHandler(cacheProvider, participantMessagePublisher, listener);
 148
 149         var updateMsg = new PropertiesUpdate();
 150         assertDoesNotThrow(() -> ach.handleAcPropertyUpdate(updateMsg));
 151
 152         updateMsg.setParticipantId(CommonTestData.getParticipantId());
 153         when(cacheProvider.getParticipantId()).thenReturn(CommonTestData.getParticipantId());
 154         var participantDeploy = new ParticipantDeploy();
 155         participantDeploy.setParticipantId(CommonTestData.getParticipantId());
 156         updateMsg.getParticipantUpdatesList().add(participantDeploy);
 157
 158         var automationComposition = CommonTestData.getTestAutomationCompositionMap().values().iterator().next();
 159         updateMsg.setAutomationCompositionId(automationComposition.getInstanceId());
 160         when(cacheProvider.getAutomationComposition(automationComposition.getInstanceId()))
 161                 .thenReturn(automationComposition);
 162         var acElementDeploy = new AcElementDeploy();
 163         acElementDeploy.setProperties(Map.of());
 164         acElementDeploy.setId(automationComposition.getElements().values().iterator().next().getId());
 165         participantDeploy.getAcElementList().add(acElementDeploy);
 166
 167         Map<ToscaConceptIdentifier, AutomationCompositionElementDefinition> map = new HashMap<>();
 168         for (var element : automationComposition.getElements().values()) {
 169             map.put(element.getDefinition(), new AutomationCompositionElementDefinition());
 170         }
 171         when(cacheProvider.getAcElementsDefinitions())
 172             .thenReturn(Map.of(automationComposition.getCompositionId(), map));
 173         ach.handleAcPropertyUpdate(updateMsg);
 174         verify(listener).update(any(), any(), any(), any());
 175     }
 176
 177     @Test
 178     void handleAutomationCompositionDeployTest() {
 179         var cacheProvider = mock(CacheProvider.class);
 180         var listener = mock(ThreadHandler.class);
 181         var participantMessagePublisher = mock(ParticipantMessagePublisher.class);
 182         var ach = new AutomationCompositionHandler(cacheProvider, participantMessagePublisher, listener);
 183
 184         var deployMsg = new AutomationCompositionDeploy();
 185         assertDoesNotThrow(() -> ach.handleAutomationCompositionDeploy(deployMsg));
 186
 187         deployMsg.setParticipantId(CommonTestData.getParticipantId());
 188         when(cacheProvider.getParticipantId()).thenReturn(CommonTestData.getParticipantId());
 189         var participantDeploy = new ParticipantDeploy();
 190         participantDeploy.setParticipantId(CommonTestData.getParticipantId());
 191         deployMsg.getParticipantUpdatesList().add(participantDeploy);
 192
 193         var automationComposition = CommonTestData.getTestAutomationCompositionMap().values().iterator().next();
 194         deployMsg.setAutomationCompositionId(automationComposition.getInstanceId());
 195         when(cacheProvider.getAutomationComposition(automationComposition.getInstanceId()))
 196                 .thenReturn(automationComposition);
 197         Map<ToscaConceptIdentifier, AutomationCompositionElementDefinition> map = new HashMap<>();
 198         for (var element : automationComposition.getElements().values()) {
 199             var acElementDeploy = new AcElementDeploy();
 200             acElementDeploy.setProperties(Map.of());
 201             acElementDeploy.setId(element.getId());
 202             participantDeploy.getAcElementList().add(acElementDeploy);
 203             map.put(element.getDefinition(), new AutomationCompositionElementDefinition());
 204         }
 205         when(cacheProvider.getAcElementsDefinitions())
 206             .thenReturn(Map.of(automationComposition.getCompositionId(), map));
 207
 208         ach.handleAutomationCompositionDeploy(deployMsg);
 209         verify(listener, times(automationComposition.getElements().size())).deploy(any(), any(), any());
 210
 211         clearInvocations(listener);
 212         deployMsg.setStartPhase(2);
 213         deployMsg.setFirstStartPhase(false);
 214         ach.handleAutomationCompositionDeploy(deployMsg);
 215         verify(listener, times(0)).deploy(any(), any(), any());
 216     }
 217
 218     @Test
 219     void handleMigrationNullTest() {
 220         var ach = new AutomationCompositionHandler(
 221                 mock(CacheProvider.class), mock(ParticipantMessagePublisher.class), mock(ThreadHandler.class));
 222         var migrationMsg = new AutomationCompositionMigration();
 223         migrationMsg.setStage(0);
 224         assertDoesNotThrow(() -> ach.handleAutomationCompositionMigration(migrationMsg));
 225         migrationMsg.setAutomationCompositionId(UUID.randomUUID());
 226         migrationMsg.setCompositionTargetId(UUID.randomUUID());
 227         assertDoesNotThrow(() -> ach.handleAutomationCompositionMigration(migrationMsg));
 228     }
 229
 230     @Test
 231     void handleAutomationCompositionMigrationTest() {
 232         var automationComposition = CommonTestData.getTestAutomationCompositionMap().values().iterator().next();
 233         automationComposition.setCompositionId(UUID.randomUUID());
 234         automationComposition.setInstanceId(UUID.randomUUID());
 235         automationComposition.setCompositionTargetId(UUID.randomUUID());
 236         var definitions =
 237                 CommonTestData.createAutomationCompositionElementDefinitionList(automationComposition);
 238         var participantDeploy =
 239                 CommonTestData.createparticipantDeploy(CommonTestData.getParticipantId(), automationComposition);
 240
 241         var cacheProvider = createCacheProvider(participantDeploy, automationComposition.getCompositionId(),
 242                 automationComposition.getInstanceId(), definitions,
 243                 automationComposition.getCompositionTargetId(), definitions);
 244
 245         testMigration(cacheProvider, automationComposition, 0, automationComposition.getElements().size());
 246     }
 247
 248     @Test
 249     void handleMigrationAddRemoveTest() {
 250         var automationComposition = CommonTestData.getTestAutomationCompositionMap().values().iterator().next();
 251         automationComposition.setCompositionId(UUID.randomUUID());
 252         automationComposition.setInstanceId(UUID.randomUUID());
 253
 254         var acMigrate = new AutomationComposition(automationComposition);
 255         acMigrate.setCompositionTargetId(UUID.randomUUID());
 256
 257         // replacing first element with new one
 258         var element = acMigrate.getElements().values().iterator().next();
 259         element.setDefinition(new ToscaConceptIdentifier("policy.clamp.new.element", "1.0.0"));
 260         element.setId(UUID.randomUUID());
 261
 262         var migrateDefinitions =
 263                 CommonTestData.createAutomationCompositionElementDefinitionList(acMigrate);
 264
 265         var participantDeploy =
 266                 CommonTestData.createparticipantDeploy(CommonTestData.getParticipantId(), automationComposition);
 267         var definitions =
 268                 CommonTestData.createAutomationCompositionElementDefinitionList(automationComposition);
 269         var cacheProvider = createCacheProvider(participantDeploy, automationComposition.getCompositionId(),
 270                 automationComposition.getInstanceId(), definitions,
 271                 acMigrate.getCompositionTargetId(), migrateDefinitions);
 272
 273         testMigration(cacheProvider, acMigrate, 0, acMigrate.getElements().size() + 1);
 274     }
 275
 276     @Test
 277     void handleAcMigrationStageTest() {
 278         var automationComposition = CommonTestData.getTestAutomationCompositionMap().values().iterator().next();
 279         automationComposition.setCompositionId(UUID.randomUUID());
 280         automationComposition.setInstanceId(UUID.randomUUID());
 281
 282         var acMigrate = new AutomationComposition(automationComposition);
 283         acMigrate.setCompositionTargetId(UUID.randomUUID());
 284
 285         // replacing first element with new one
 286         var element = acMigrate.getElements().values().iterator().next();
 287         element.setDefinition(new ToscaConceptIdentifier("policy.clamp.new.element", "1.2.4"));
 288         element.setId(UUID.randomUUID());
 289
 290         // replacing definition version
 291         acMigrate.getElements().values().forEach(el -> el.setDefinition(
 292                 new ToscaConceptIdentifier(el.getDefinition().getName(), "1.2.4")));
 293
 294         var migrateDefinitions =
 295                 CommonTestData.createAutomationCompositionElementDefinitionList(acMigrate);
 296
 297         migrateDefinitions.forEach(el -> el.getAutomationCompositionElementToscaNodeTemplate()
 298                 .setProperties(Map.of("stage", List.of(0, 1))));
 299
 300         var participantDeploy =
 301                 CommonTestData.createparticipantDeploy(CommonTestData.getParticipantId(), automationComposition);
 302         var definitions =
 303                 CommonTestData.createAutomationCompositionElementDefinitionList(automationComposition);
 304         var cacheProvider = createCacheProvider(participantDeploy, automationComposition.getCompositionId(),
 305                 automationComposition.getInstanceId(), definitions,
 306                 acMigrate.getCompositionTargetId(), migrateDefinitions);
 307
 308         // scenario 1,2
 309         migrateDefinitions.forEach(el -> el.getAutomationCompositionElementToscaNodeTemplate()
 310                 .setProperties(Map.of("stage", List.of(1, 2))));
 311
 312         // expected the element deleted
 313         testMigration(cacheProvider, acMigrate, 0, 1);
 314
 315         // expected 4 elements from stage 1
 316         testMigration(cacheProvider, acMigrate, 1, 4);
 317
 318         // scenario 0,2
 319         cacheProvider = createCacheProvider(participantDeploy, automationComposition.getCompositionId(),
 320                 automationComposition.getInstanceId(), definitions,
 321                 acMigrate.getCompositionTargetId(), migrateDefinitions);
 322
 323         migrateDefinitions.forEach(el -> el.getAutomationCompositionElementToscaNodeTemplate()
 324                 .setProperties(Map.of("stage", List.of(0, 2))));
 325
 326         // expected the element deleted + 4 elements from stage 0
 327         testMigration(cacheProvider, acMigrate, 0, 5);
 328
 329         // expected 0 elements
 330         testMigration(cacheProvider, acMigrate, 1, 0);
 331     }
 332
 333     private CacheProvider createCacheProvider(ParticipantDeploy participantDeploy,
 334             UUID compositionId, UUID instanceId, List<AutomationCompositionElementDefinition> definitions,
 335             UUID compositionTargetId, List<AutomationCompositionElementDefinition> migrateDefinitions) {
 336         var cacheProvider = new CacheProvider(CommonTestData.getParticipantParameters());
 337         cacheProvider.addElementDefinition(compositionId, definitions);
 338         cacheProvider.initializeAutomationComposition(compositionId, instanceId, participantDeploy);
 339         cacheProvider.addElementDefinition(compositionTargetId, migrateDefinitions);
 340         return cacheProvider;
 341     }
 342
 343     private void testMigration(CacheProvider cacheProvider, AutomationComposition acMigrate,
 344             int stage, int expectedMigrated) {
 345         var migrationMsg = new AutomationCompositionMigration();
 346         migrationMsg.setStage(stage);
 347         migrationMsg.setCompositionId(acMigrate.getCompositionId());
 348         migrationMsg.setAutomationCompositionId(acMigrate.getInstanceId());
 349         migrationMsg.setCompositionTargetId(acMigrate.getCompositionTargetId());
 350         var participantMigrate = CommonTestData.createparticipantDeploy(cacheProvider.getParticipantId(), acMigrate);
 351         migrationMsg.setParticipantUpdatesList(List.of(participantMigrate));
 352         var listener = mock(ThreadHandler.class);
 353
 354         clearInvocations();
 355         var ach = new AutomationCompositionHandler(cacheProvider, mock(ParticipantMessagePublisher.class), listener);
 356         ach.handleAutomationCompositionMigration(migrationMsg);
 357         verify(listener, times(expectedMigrated)).migrate(any(), any(), any(), any(), any(), anyInt());
 358     }
 359 }