main/src/test/java/org/onap/policy/pdpx/main/comm/XacmlPdpUpdatePublisherTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * Copyright (C) 2019, 2021 AT&T Intellectual Property. 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.pdpx.main.comm;
  22
  23 import static org.junit.Assert.assertEquals;
  24 import static org.mockito.ArgumentMatchers.any;
  25 import static org.mockito.ArgumentMatchers.startsWith;
  26 import static org.mockito.Mockito.doThrow;
  27 import static org.mockito.Mockito.never;
  28 import static org.mockito.Mockito.verify;
  29 import static org.mockito.Mockito.when;
  30
  31 import java.util.Arrays;
  32 import java.util.HashMap;
  33 import java.util.List;
  34 import java.util.Map;
  35 import org.junit.Before;
  36 import org.junit.Test;
  37 import org.junit.runner.RunWith;
  38 import org.mockito.Mock;
  39 import org.mockito.junit.MockitoJUnitRunner;
  40 import org.onap.policy.common.endpoints.event.comm.client.TopicSinkClient;
  41 import org.onap.policy.models.pdp.concepts.PdpStatus;
  42 import org.onap.policy.models.pdp.concepts.PdpUpdate;
  43 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicy;
  44 import org.onap.policy.pdp.xacml.application.common.XacmlApplicationException;
  45 import org.onap.policy.pdp.xacml.application.common.XacmlApplicationServiceProvider;
  46 import org.onap.policy.pdpx.main.XacmlState;
  47 import org.onap.policy.pdpx.main.rest.XacmlPdpApplicationManager;
  48 import org.onap.policy.pdpx.main.rest.XacmlPdpStatisticsManager;
  49
  50
  51 /**
  52  * Initializes objects, including the publisher.
  53  */
  54 @RunWith(MockitoJUnitRunner.class)
  55 public class XacmlPdpUpdatePublisherTest {
  56
  57     private static final int NEW_COUNT = 4;
  58
  59     @Mock
  60     private TopicSinkClient client;
  61
  62     @Mock
  63     private XacmlState state;
  64
  65     @Mock
  66     private PdpStatus status;
  67
  68     @Mock
  69     private XacmlPdpApplicationManager appmgr;
  70
  71     @Mock
  72     private ToscaPolicy deployed1;
  73
  74     @Mock
  75     private ToscaPolicy deployed2;
  76
  77     @Mock
  78     private ToscaPolicy deployed3;
  79
  80     @Mock
  81     private ToscaPolicy deployed4;
  82
  83     @Mock
  84     private ToscaPolicy added1;
  85
  86     @Mock
  87     private ToscaPolicy added2;
  88
  89     @Mock
  90     private ToscaPolicy failPolicy1;
  91
  92     @Mock
  93     private ToscaPolicy failPolicy2;
  94
  95     @Mock
  96     private PdpUpdate update;
  97
  98     @Mock
  99     private PdpUpdate failurePdpUpdate;
 100
 101     private XacmlPdpUpdatePublisher publisher;
 102
 103
 104     /**
 105      * Initializes objects, including the publisher.
 106      */
 107     @Before
 108     public void setUp() {
 109         Map<ToscaPolicy, XacmlApplicationServiceProvider> deployedPolicies = new HashMap<>();
 110         deployedPolicies.put(deployed1, null);
 111         deployedPolicies.put(deployed2, null);
 112         deployedPolicies.put(deployed3, null);
 113         deployedPolicies.put(deployed4, null);
 114         when(appmgr.getToscaPolicies()).thenReturn(deployedPolicies);
 115
 116         // update includes two overlaps
 117         List<ToscaPolicy> updatePolicies = Arrays.asList(added1, deployed2, deployed3, added2);
 118         when(update.getPolicies()).thenReturn(updatePolicies);
 119
 120         List<ToscaPolicy> failureUpdatePolicies = Arrays.asList(added1, deployed2, deployed3, failPolicy1, failPolicy2);
 121         when(failurePdpUpdate.getPolicies()).thenReturn(failureUpdatePolicies);
 122
 123         when(appmgr.getPolicyCount()).thenReturn(NEW_COUNT);
 124
 125         when(state.updateInternalState(any(), any())).thenReturn(status);
 126
 127         when(client.send(any())).thenReturn(true);
 128
 129         publisher = new XacmlPdpUpdatePublisher(client, state, appmgr);
 130     }
 131
 132     @Test
 133     public void testHandlePdpUpdate() throws XacmlApplicationException {
 134         XacmlPdpStatisticsManager statmgr = new XacmlPdpStatisticsManager();
 135         XacmlPdpStatisticsManager.setCurrent(statmgr);
 136
 137         publisher.handlePdpUpdate(update);
 138
 139         // two removed
 140         verify(appmgr).removeUndeployedPolicy(deployed1);
 141         verify(appmgr).removeUndeployedPolicy(deployed4);
 142
 143         // two added
 144         verify(appmgr).loadDeployedPolicy(added1);
 145         verify(appmgr).loadDeployedPolicy(added2);
 146
 147         // two untouched
 148         verify(appmgr, never()).removeUndeployedPolicy(deployed2);
 149         verify(appmgr, never()).removeUndeployedPolicy(deployed3);
 150         verify(appmgr, never()).loadDeployedPolicy(deployed2);
 151         verify(appmgr, never()).loadDeployedPolicy(deployed3);
 152
 153         assertEquals(NEW_COUNT, statmgr.getTotalPoliciesCount());
 154
 155         verify(client).send(status);
 156     }
 157
 158     @Test
 159     public void testHandlePdpUpdate_LoadPolicyFailed() throws XacmlApplicationException {
 160         // Set loadPolicy to fail
 161         doThrow(new XacmlApplicationException()).when(appmgr).loadDeployedPolicy(failPolicy1);
 162         doThrow(new XacmlApplicationException()).when(appmgr).loadDeployedPolicy(failPolicy2);
 163
 164         publisher.handlePdpUpdate(failurePdpUpdate);
 165
 166         // two removed
 167         verify(appmgr).removeUndeployedPolicy(deployed1);
 168         verify(appmgr).removeUndeployedPolicy(deployed4);
 169
 170         verify(failurePdpUpdate).setPolicies(any());
 171
 172         verify(state).updateInternalState(any(), startsWith("Failed to load policy"));
 173         verify(client).send(status);
 174     }
 175
 176     @Test
 177     public void testHandlePdpUpdate_NullPolicies() throws XacmlApplicationException {
 178         when(update.getPolicies()).thenReturn(null);
 179
 180         publisher.handlePdpUpdate(update);
 181
 182         // all removed
 183         verify(appmgr).removeUndeployedPolicy(deployed1);
 184         verify(appmgr).removeUndeployedPolicy(deployed2);
 185         verify(appmgr).removeUndeployedPolicy(deployed3);
 186         verify(appmgr).removeUndeployedPolicy(deployed4);
 187
 188         // none added
 189         verify(appmgr, never()).loadDeployedPolicy(any());
 190
 191         verify(client).send(status);
 192     }
 193
 194     @Test
 195     public void testHandlePdpUpdate_NullStats() {
 196         XacmlPdpStatisticsManager.setCurrent(null);
 197
 198         // should work without throwing an exception
 199         publisher.handlePdpUpdate(update);
 200
 201         verify(client).send(status);
 202     }
 203
 204     @Test
 205     public void testHandlePdpUpdate_SendFail() {
 206
 207         when(client.send(any())).thenReturn(false);
 208
 209         publisher.handlePdpUpdate(update);
 210
 211         verify(client).send(status);
 212     }
 213
 214 }