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

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * Copyright (C) 2021-2025 OpenInfra Foundation Europe. All rights reserved.
   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.models.acm.persistence.provider;
  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.mockito.ArgumentMatchers.any;
  27 import static org.mockito.ArgumentMatchers.anyIterable;
  28 import static org.mockito.Mockito.mock;
  29 import static org.mockito.Mockito.verify;
  30 import static org.mockito.Mockito.when;
  31
  32 import java.util.ArrayList;
  33 import java.util.List;
  34 import java.util.Optional;
  35 import java.util.UUID;
  36 import org.junit.jupiter.api.BeforeEach;
  37 import org.junit.jupiter.api.Test;
  38 import org.mockito.Mockito;
  39 import org.onap.policy.clamp.models.acm.concepts.AutomationCompositions;
  40 import org.onap.policy.clamp.models.acm.concepts.DeployState;
  41 import org.onap.policy.clamp.models.acm.concepts.LockState;
  42 import org.onap.policy.clamp.models.acm.persistence.concepts.JpaAutomationComposition;
  43 import org.onap.policy.clamp.models.acm.persistence.concepts.JpaAutomationCompositionElement;
  44 import org.onap.policy.clamp.models.acm.persistence.repository.AutomationCompositionElementRepository;
  45 import org.onap.policy.clamp.models.acm.persistence.repository.AutomationCompositionRepository;
  46 import org.onap.policy.common.utils.coder.Coder;
  47 import org.onap.policy.common.utils.coder.StandardCoder;
  48 import org.onap.policy.common.utils.resources.ResourceUtils;
  49 import org.springframework.data.domain.Example;
  50 import org.springframework.data.domain.PageImpl;
  51 import org.springframework.data.domain.PageRequest;
  52 import org.springframework.data.domain.Pageable;
  53
  54 class AutomationCompositionProviderTest {
  55
  56     private static final String AC_IS_NULL = "automationComposition is marked non-null but is null";
  57     private static final String ACELEMENT_ID_IS_NULL = "elementId is marked non-null but is null";
  58
  59     private static final Coder CODER = new StandardCoder();
  60     private static final String AUTOMATION_COMPOSITION_JSON =
  61             "src/test/resources/providers/TestAutomationCompositions.json";
  62
  63     private AutomationCompositions inputAutomationCompositions;
  64     private List<JpaAutomationComposition> inputAutomationCompositionsJpa;
  65     private final String originalJson = ResourceUtils.getResourceAsString(AUTOMATION_COMPOSITION_JSON);
  66
  67     @BeforeEach
  68     void beforeSetupDao() throws Exception {
  69         inputAutomationCompositions = CODER.decode(originalJson, AutomationCompositions.class);
  70         inputAutomationCompositionsJpa =
  71                 ProviderUtils.getJpaAndValidateList(inputAutomationCompositions.getAutomationCompositionList(),
  72                         JpaAutomationComposition::new, "automation compositions");
  73     }
  74
  75     @Test
  76     void testAutomationCompositionCreate() {
  77         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
  78         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
  79                 mock(AutomationCompositionElementRepository.class));
  80
  81         when(automationCompositionRepository.save(any(JpaAutomationComposition.class)))
  82                 .thenReturn(inputAutomationCompositionsJpa.get(0));
  83         var inputAc = inputAutomationCompositions.getAutomationCompositionList().get(0);
  84
  85         var createdAutomationComposition = automationCompositionProvider.createAutomationComposition(inputAc);
  86         inputAc.setInstanceId(createdAutomationComposition.getInstanceId());
  87         inputAc.setLastMsg(createdAutomationComposition.getLastMsg());
  88         assertEquals(inputAc, createdAutomationComposition);
  89     }
  90
  91     @Test
  92     void testAutomationCompositionUpdate() {
  93         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
  94         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
  95                 mock(AutomationCompositionElementRepository.class));
  96
  97         assertThatThrownBy(() -> automationCompositionProvider.updateAutomationComposition(null))
  98                 .hasMessageMatching(AC_IS_NULL);
  99
 100         when(automationCompositionRepository.save(inputAutomationCompositionsJpa.get(0)))
 101                 .thenReturn(inputAutomationCompositionsJpa.get(0));
 102
 103         var createdAutomationComposition = automationCompositionProvider
 104                 .updateAutomationComposition(inputAutomationCompositions.getAutomationCompositionList().get(0));
 105
 106         assertEquals(inputAutomationCompositions.getAutomationCompositionList().get(0), createdAutomationComposition);
 107     }
 108
 109     @Test
 110     void testGetAutomationCompositionsWithNull() {
 111         var automationCompositionProvider = new AutomationCompositionProvider(
 112                 mock(AutomationCompositionRepository.class), mock(AutomationCompositionElementRepository.class));
 113
 114         assertThatThrownBy(() -> automationCompositionProvider
 115                 .getAutomationCompositions(UUID.randomUUID(), null, null, null))
 116                 .hasMessage("pageable is marked non-null but is null");
 117
 118         assertThatThrownBy(() -> automationCompositionProvider
 119                 .getAutomationCompositions(null, null, null, Pageable.unpaged()))
 120                 .hasMessage("compositionId is marked non-null but is null");
 121     }
 122
 123     @Test
 124     void testGetAutomationCompositions() {
 125         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
 126         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
 127                 mock(AutomationCompositionElementRepository.class));
 128
 129         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
 130         when(automationCompositionRepository
 131                 .findAll(Mockito.<Example<JpaAutomationComposition>>any(), any(Pageable.class)))
 132                 .thenReturn(new PageImpl<>(inputAutomationCompositionsJpa));
 133         var acList = automationCompositionProvider.getAutomationCompositions(UUID.randomUUID(),
 134                 automationComposition.getName(), automationComposition.getVersion(), Pageable.unpaged());
 135         assertThat(acList).hasSize(2);
 136
 137         acList = automationCompositionProvider.getAutomationCompositions(automationComposition.getCompositionId(), null,
 138                 null, Pageable.unpaged());
 139         assertThat(acList).hasSize(2);
 140
 141         when(automationCompositionRepository
 142             .findAll(Mockito.<Example<JpaAutomationComposition>>any(), Mockito.any(Pageable.class)))
 143             .thenReturn(new PageImpl<>(inputAutomationCompositionsJpa));
 144         acList = automationCompositionProvider.getAutomationCompositions(automationComposition.getCompositionId(), null,
 145             null, PageRequest.of(0, 10));
 146         assertThat(acList).hasSize(2);
 147     }
 148
 149     @Test
 150     void testGetAutomationComposition() {
 151         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
 152         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
 153                 mock(AutomationCompositionElementRepository.class));
 154
 155         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
 156         assertThatThrownBy(
 157                 () -> automationCompositionProvider.getAutomationComposition(automationComposition.getInstanceId()))
 158                         .hasMessageMatching("AutomationComposition not found");
 159
 160         when(automationCompositionRepository.findById(automationComposition.getInstanceId().toString()))
 161                 .thenReturn(Optional.of(inputAutomationCompositionsJpa.get(0)));
 162         var ac = automationCompositionProvider.getAutomationComposition(automationComposition.getInstanceId());
 163         assertEquals(automationComposition, ac);
 164     }
 165
 166     @Test
 167     void testFindAutomationComposition() {
 168         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
 169         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
 170                 mock(AutomationCompositionElementRepository.class));
 171
 172         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
 173         var acOpt = automationCompositionProvider.findAutomationComposition(automationComposition.getInstanceId());
 174         assertThat(acOpt).isEmpty();
 175
 176         acOpt = automationCompositionProvider.findAutomationComposition(automationComposition.getKey().asIdentifier());
 177         assertThat(acOpt).isEmpty();
 178
 179         when(automationCompositionRepository.findById(automationComposition.getInstanceId().toString()))
 180                 .thenReturn(Optional.of(inputAutomationCompositionsJpa.get(0)));
 181         acOpt = automationCompositionProvider.findAutomationComposition(automationComposition.getInstanceId());
 182         assertEquals(automationComposition, acOpt.get());
 183
 184         when(automationCompositionRepository.findOne(Mockito.<Example<JpaAutomationComposition>>any()))
 185                 .thenReturn(Optional.of(inputAutomationCompositionsJpa.get(0)));
 186         acOpt = automationCompositionProvider.findAutomationComposition(automationComposition.getKey().asIdentifier());
 187         assertEquals(automationComposition, acOpt.get());
 188     }
 189
 190     @Test
 191     void testGetAcInstancesByCompositionId() {
 192         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
 193         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
 194                 mock(AutomationCompositionElementRepository.class));
 195
 196         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
 197         when(automationCompositionRepository.findByCompositionId(automationComposition.getCompositionId().toString()))
 198                 .thenReturn(inputAutomationCompositionsJpa);
 199         var acList =
 200                 automationCompositionProvider.getAcInstancesByCompositionId(automationComposition.getCompositionId());
 201         assertEquals(inputAutomationCompositions.getAutomationCompositionList(), acList);
 202     }
 203
 204     @Test
 205     void testGetAcInstancesInTransition() {
 206         inputAutomationCompositions.getAutomationCompositionList().get(0).setDeployState(DeployState.DEPLOYING);
 207         inputAutomationCompositions.getAutomationCompositionList().get(1).setLockState(LockState.LOCKING);
 208         inputAutomationCompositionsJpa.get(0).setDeployState(DeployState.DEPLOYING);
 209         inputAutomationCompositionsJpa.get(1).setLockState(LockState.LOCKING);
 210
 211         List<JpaAutomationComposition> res1 = new ArrayList<>();
 212         res1.add(inputAutomationCompositionsJpa.get(0));
 213         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
 214         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
 215             mock(AutomationCompositionElementRepository.class));
 216         when(automationCompositionRepository.findByDeployStateIn(List.of(DeployState.DEPLOYING,
 217             DeployState.UNDEPLOYING, DeployState.DELETING, DeployState.UPDATING, DeployState.MIGRATING)))
 218             .thenReturn(res1);
 219         when(automationCompositionRepository.findByLockStateIn(List.of(LockState.LOCKING, LockState.UNLOCKING)))
 220             .thenReturn(List.of(inputAutomationCompositionsJpa.get(1)));
 221         var acList = automationCompositionProvider.getAcInstancesInTransition();
 222         assertThat(acList).hasSize(2)
 223                 .contains(inputAutomationCompositions.getAutomationCompositionList().get(0).getInstanceId())
 224                 .contains(inputAutomationCompositions.getAutomationCompositionList().get(1).getInstanceId());
 225     }
 226
 227     @Test
 228     void testDeleteAutomationComposition() {
 229         var automationCompositionRepository = mock(AutomationCompositionRepository.class);
 230         var automationCompositionProvider = new AutomationCompositionProvider(automationCompositionRepository,
 231                 mock(AutomationCompositionElementRepository.class));
 232
 233         assertThatThrownBy(() -> automationCompositionProvider.deleteAutomationComposition(UUID.randomUUID()))
 234                 .hasMessageMatching(".*.failed, automation composition does not exist");
 235
 236         var automationComposition = inputAutomationCompositions.getAutomationCompositionList().get(0);
 237         when(automationCompositionRepository.findById(automationComposition.getInstanceId().toString()))
 238                 .thenReturn(Optional.of(inputAutomationCompositionsJpa.get(0)));
 239
 240         var deletedAc =
 241                 automationCompositionProvider.deleteAutomationComposition(automationComposition.getInstanceId());
 242         assertEquals(automationComposition, deletedAc);
 243     }
 244
 245     @Test
 246     void testDeleteElementById() {
 247         var acElementRepository = mock(AutomationCompositionElementRepository.class);
 248         var automationCompositionProvider = new AutomationCompositionProvider(
 249                 mock(AutomationCompositionRepository.class), acElementRepository);
 250         assertThatThrownBy(() -> automationCompositionProvider.deleteAutomationCompositionElement(null))
 251                 .hasMessageMatching(ACELEMENT_ID_IS_NULL);
 252         var elementId = UUID.randomUUID();
 253         automationCompositionProvider.deleteAutomationCompositionElement(elementId);
 254         verify(acElementRepository).deleteById(elementId.toString());
 255     }
 256
 257     @Test
 258     void testValidateElementIds() {
 259         var acElementRepository = mock(AutomationCompositionElementRepository.class);
 260         var automationCompositionProvider = new AutomationCompositionProvider(
 261             mock(AutomationCompositionRepository.class), acElementRepository);
 262
 263         var ac = inputAutomationCompositions.getAutomationCompositionList().get(0);
 264
 265         var result = automationCompositionProvider.validateElementIds(ac);
 266         assertThat(result.isValid()).isTrue();
 267
 268         var jpaElement = new JpaAutomationCompositionElement(ac.getElements().values().iterator().next());
 269         when(acElementRepository.findAllById(anyIterable()))
 270             .thenReturn(List.of(jpaElement));
 271
 272         ac.setInstanceId(null);
 273         result = automationCompositionProvider.validateElementIds(ac);
 274         assertThat(result.isValid()).isFalse();
 275
 276         ac.setInstanceId(UUID.randomUUID());
 277         jpaElement.setInstanceId(UUID.randomUUID().toString());
 278         result = automationCompositionProvider.validateElementIds(ac);
 279         assertThat(result.isValid()).isFalse();
 280
 281         ac.setInstanceId(UUID.randomUUID());
 282         jpaElement.setInstanceId(ac.getInstanceId().toString());
 283         result = automationCompositionProvider.validateElementIds(ac);
 284         assertThat(result.isValid()).isTrue();
 285     }
 286 }