main/src/test/java/org/onap/policy/pap/main/service/PolicyStatusServiceTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2022-2023 Bell Canada. All rights reserved.
   4  *  Modifications Copyright (C) 2022 Nordix Foundation.
   5  * ================================================================================
   6  * Licensed under the Apache License, Version 2.0 (the "License");
   7  * you may not use this file except in compliance with the License.
   8  * You may obtain a copy of the License at
   9  *
  10  *      http://www.apache.org/licenses/LICENSE-2.0
  11  *
  12  * Unless required by applicable law or agreed to in writing, software
  13  * distributed under the License is distributed on an "AS IS" BASIS,
  14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15  * See the License for the specific language governing permissions and
  16  * limitations under the License.
  17  *
  18  * SPDX-License-Identifier: Apache-2.0
  19  * ============LICENSE_END=========================================================
  20  */
  21
  22 package org.onap.policy.pap.main.service;
  23
  24 import static org.assertj.core.api.Assertions.assertThat;
  25 import static org.assertj.core.api.Assertions.assertThatCode;
  26 import static org.assertj.core.api.Assertions.assertThatThrownBy;
  27
  28 import java.util.List;
  29 import org.junit.Before;
  30 import org.junit.Test;
  31 import org.onap.policy.models.base.PfModelRuntimeException;
  32 import org.onap.policy.models.pdp.concepts.PdpPolicyStatus;
  33 import org.onap.policy.models.tosca.authorative.concepts.ToscaConceptIdentifier;
  34 import org.onap.policy.models.tosca.authorative.concepts.ToscaConceptIdentifierOptVersion;
  35 import org.onap.policy.pap.main.rest.CommonPapRestServer;
  36 import org.springframework.beans.factory.annotation.Autowired;
  37 import org.springframework.test.context.ActiveProfiles;
  38
  39 @ActiveProfiles("test")
  40 public class PolicyStatusServiceTest extends CommonPapRestServer {
  41
  42     private static final String GROUP_A = "groupA";
  43     private static final String GROUP_B = "groupB";
  44     private static final ToscaConceptIdentifier MY_POLICY = new ToscaConceptIdentifier("MyPolicy", "1.2.3");
  45     private static final ToscaConceptIdentifier MY_POLICY2 = new ToscaConceptIdentifier("MyPolicyB", "2.3.4");
  46
  47     @Autowired
  48     private PolicyStatusService policyStatusService;
  49
  50     private PdpPolicyStatus.PdpPolicyStatusBuilder statusBuilder;
  51
  52     /**
  53      * Setup before tests.
  54      *
  55      * @throws Exception the exception
  56      */
  57     @Override
  58     @Before
  59     public void setUp() throws Exception {
  60         super.setUp();
  61         ToscaConceptIdentifier policyType = new ToscaConceptIdentifier("MyPolicyType", "1.2.4");
  62
  63         statusBuilder = PdpPolicyStatus.builder().deploy(true).pdpType("MyPdpType").policy(MY_POLICY)
  64             .policyType(policyType).state(PdpPolicyStatus.State.SUCCESS);
  65     }
  66
  67     @Test
  68     public void testGetAllPolicyStatus() {
  69         assertThat(policyStatusService.getAllPolicyStatus()).isEmpty();
  70
  71         var statusList = createStatusList();
  72         policyStatusService.cudPolicyStatus(statusList, null, null);
  73         assertThat(policyStatusService.getAllPolicyStatus()).hasSize(5);
  74         policyStatusService.cudPolicyStatus(null, null, statusList);
  75     }
  76
  77     @Test
  78     public void testGetAllPolicyStatusPfDaoToscaConceptIdentifierOptVersion() {
  79
  80         assertThatThrownBy(() -> {
  81             policyStatusService.getAllPolicyStatus(null);
  82         }).hasMessageContaining("policy").hasMessageContaining("null");
  83
  84         assertThat(policyStatusService.getAllPolicyStatus(new ToscaConceptIdentifierOptVersion("somePdp", null)))
  85             .isEmpty();
  86
  87         var statusList = createStatusList();
  88         policyStatusService.cudPolicyStatus(statusList, null, null);
  89
  90         assertThat(policyStatusService.getAllPolicyStatus(new ToscaConceptIdentifierOptVersion(MY_POLICY))).hasSize(2);
  91         assertThat(
  92             policyStatusService.getAllPolicyStatus(new ToscaConceptIdentifierOptVersion(MY_POLICY.getName(), null)))
  93                 .hasSize(3);
  94         policyStatusService.cudPolicyStatus(null, null, statusList);
  95     }
  96
  97     @Test
  98     public void testGetGroupPolicyStatus() {
  99
 100         assertThatThrownBy(() -> {
 101             policyStatusService.getGroupPolicyStatus(null);
 102         }).hasMessage("pdpGroup is marked non-null but is null");
 103
 104         assertThat(policyStatusService.getGroupPolicyStatus("PdpGroup0")).isEmpty();
 105
 106         var statusList = createStatusList();
 107         policyStatusService.cudPolicyStatus(statusList, null, null);
 108         assertThat(policyStatusService.getGroupPolicyStatus(GROUP_A)).hasSize(3);
 109         policyStatusService.cudPolicyStatus(null, null, statusList);
 110     }
 111
 112     @Test
 113     public void testCudPolicyStatus() {
 114         assertThatCode(() -> policyStatusService.cudPolicyStatus(null, null, null)).doesNotThrowAnyException();
 115
 116         assertThatThrownBy(() -> {
 117             policyStatusService.cudPolicyStatus(List.of(new PdpPolicyStatus()), null, null);
 118         }).isInstanceOf(PfModelRuntimeException.class);
 119         PdpPolicyStatus invalidStatus = statusBuilder.state(PdpPolicyStatus.State.WAITING).deploy(false).pdpGroup(null)
 120             .pdpId("pdp1").policy(MY_POLICY).build();
 121         assertThatThrownBy(() -> {
 122             policyStatusService.cudPolicyStatus(List.of(invalidStatus), null, null);
 123         }).isInstanceOf(PfModelRuntimeException.class)
 124             .hasMessageContaining("item \"pdpGroup\" value \"null\" INVALID, is null");
 125
 126         // Test create
 127         PdpPolicyStatus status = statusBuilder.state(PdpPolicyStatus.State.WAITING).deploy(false).pdpGroup(GROUP_A)
 128             .pdpId("pdp1").policy(MY_POLICY).build();
 129         policyStatusService.cudPolicyStatus(List.of(status), null, null);
 130         List<PdpPolicyStatus> createdStatusList = policyStatusService.getAllPolicyStatus();
 131         assertThat(createdStatusList).hasSize(1);
 132         assertThat(createdStatusList.get(0).getState()).isEqualTo(PdpPolicyStatus.State.WAITING);
 133
 134         // Test update
 135         status.setDeploy(true);
 136         status.setState(PdpPolicyStatus.State.SUCCESS);
 137         policyStatusService.cudPolicyStatus(null, List.of(status), null);
 138         createdStatusList = policyStatusService.getAllPolicyStatus();
 139         assertThat(createdStatusList).hasSize(1);
 140         assertThat(createdStatusList.get(0).getState()).isEqualTo(PdpPolicyStatus.State.SUCCESS);
 141
 142         // Test delete
 143         policyStatusService.cudPolicyStatus(null, null, List.of(status));
 144         assertThat(policyStatusService.getAllPolicyStatus()).isEmpty();
 145     }
 146
 147     private List<PdpPolicyStatus> createStatusList() {
 148         // same name, different version
 149         final ToscaConceptIdentifier policy3 = new ToscaConceptIdentifier(MY_POLICY.getName(), "10.20.30");
 150
 151         PdpPolicyStatus id1 = statusBuilder.pdpGroup(GROUP_A).pdpId("pdp1").policy(MY_POLICY).build();
 152         PdpPolicyStatus id2 = statusBuilder.pdpGroup(GROUP_A).pdpId("pdp2").policy(MY_POLICY2).build();
 153         PdpPolicyStatus id3 = statusBuilder.pdpGroup(GROUP_A).pdpId("pdp3").policy(policy3).build();
 154         PdpPolicyStatus id4 = statusBuilder.pdpGroup(GROUP_B).pdpId("pdp4").policy(MY_POLICY).build();
 155         PdpPolicyStatus id5 = statusBuilder.pdpGroup(GROUP_B).pdpId("pdp5").policy(MY_POLICY2).build();
 156         return List.of(id1, id2, id3, id4, id5);
 157     }
 158 }