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

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2018 Ericsson. All rights reserved.
   4  *  Modifications Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
   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.distribution.forwarding.xacml.pdp.engine;
  23
  24 import static org.mockito.Matchers.anyObject;
  25 import static org.mockito.Matchers.argThat;
  26 import static org.mockito.Matchers.eq;
  27 import static org.mockito.Mockito.mock;
  28 import static org.mockito.Mockito.times;
  29 import static org.mockito.Mockito.verify;
  30 import static org.mockito.Mockito.when;
  31
  32 import java.util.ArrayList;
  33 import java.util.Collection;
  34 import java.util.HashMap;
  35
  36 import javax.ws.rs.client.Entity;
  37 import javax.ws.rs.core.Response;
  38
  39 import org.hamcrest.BaseMatcher;
  40 import org.hamcrest.Description;
  41 import org.junit.BeforeClass;
  42 import org.junit.Test;
  43 import org.onap.policy.api.PolicyParameters;
  44 import org.onap.policy.api.PushPolicyParameters;
  45 import org.onap.policy.common.endpoints.event.comm.bus.internal.BusTopicParams;
  46 import org.onap.policy.common.endpoints.http.client.HttpClient;
  47 import org.onap.policy.common.endpoints.http.client.HttpClientConfigException;
  48 import org.onap.policy.common.endpoints.http.client.HttpClientFactory;
  49 import org.onap.policy.common.parameters.ParameterGroup;
  50 import org.onap.policy.common.parameters.ParameterService;
  51 import org.onap.policy.distribution.forwarding.xacml.pdp.XacmlPdpPolicyForwarder;
  52 import org.onap.policy.distribution.forwarding.xacml.pdp.XacmlPdpPolicyForwarderParameterGroup.XacmlPdpPolicyForwarderParameterGroupBuilder;
  53 import org.onap.policy.models.tosca.authorative.concepts.ToscaEntity;
  54 import org.onap.policy.models.tosca.authorative.concepts.ToscaPolicy;
  55
  56 public class XacmlPdpPolicyForwarderTest {
  57
  58     private static final BusTopicParams BUS_TOPIC_PARAMS = BusTopicParams.builder().useHttps(false).hostname("myHost")
  59             .port(1234).userName("myUser").password("myPassword").managed(true).build();
  60     private static final String CLIENT_AUTH = "ClientAuth";
  61     private static final String CLIENT_AUTH_VALUE = "myClientAuth";
  62     private static final String PDP_GROUP_VALUE = "myPdpGroup";
  63     private HashMap<String, Object> headers = new HashMap<>();
  64     private BusTopicParamsMatcher matcher = new BusTopicParamsMatcher(BUS_TOPIC_PARAMS);
  65
  66     /**
  67      * Set up.
  68      */
  69     @BeforeClass
  70     public static void setUp() {
  71         final ParameterGroup parameterGroup = new XacmlPdpPolicyForwarderParameterGroupBuilder()
  72                 .setUseHttps(BUS_TOPIC_PARAMS.isUseHttps()).setHostname(BUS_TOPIC_PARAMS.getHostname())
  73                 .setPort(BUS_TOPIC_PARAMS.getPort()).setUserName(BUS_TOPIC_PARAMS.getUserName())
  74                 .setPassword(BUS_TOPIC_PARAMS.getPassword()).setClientAuth(CLIENT_AUTH_VALUE)
  75                 .setIsManaged(BUS_TOPIC_PARAMS.isManaged()).setPdpGroup(PDP_GROUP_VALUE).build();
  76         parameterGroup.setName("xacmlPdpConfiguration");
  77         ParameterService.register(parameterGroup);
  78     }
  79
  80     @Test
  81     public void testForwardPolicy() throws Exception {
  82
  83         final HttpClient httpClientMock = mock(HttpClient.class);
  84         headers.put(CLIENT_AUTH, CLIENT_AUTH_VALUE);
  85         when(httpClientMock.put(eq("createPolicy"), anyObject(), eq(headers))).thenReturn(Response.ok().build());
  86         when(httpClientMock.put(eq("pushPolicy"), anyObject(), eq(headers))).thenReturn(Response.ok().build());
  87
  88         final HttpClientFactory httpClientFactoryMock = mock(HttpClientFactory.class);
  89         when(httpClientFactoryMock.build(argThat(matcher))).thenReturn(httpClientMock);
  90
  91         final XacmlPdpPolicyForwarder forwarder = new MyXacmlPdpPolicyForwarder(httpClientFactoryMock);
  92         forwarder.configure("xacmlPdpConfiguration");
  93
  94         final Collection<ToscaEntity> policies = new ArrayList<>();
  95
  96         final ToscaPolicy policy1 = createPolicy(policies, "policy1", "optimization");
  97
  98         final ToscaEntity policy2 = new UnsupportedPolicy();
  99         policies.add(policy2);
 100
 101         final ToscaPolicy policy3 = createPolicy(policies, "policy3", "optimization");
 102
 103         forwarder.forward(policies);
 104
 105         verify(httpClientMock).put(eq("createPolicy"), argThat(new PolicyParametersEntityMatcher(policy1)),
 106                 eq(headers));
 107         verify(httpClientMock).put(eq("createPolicy"), argThat(new PolicyParametersEntityMatcher(policy3)),
 108                 eq(headers));
 109         verify(httpClientMock).put(eq("pushPolicy"), argThat(new PushPolicyParametersEntityMatcher(policy1)),
 110                 eq(headers));
 111         verify(httpClientMock).put(eq("pushPolicy"), argThat(new PushPolicyParametersEntityMatcher(policy3)),
 112                 eq(headers));
 113     }
 114
 115     @Test
 116     public void testForwardPolicy_CreateFailsPushNotInvoked() throws Exception {
 117
 118         final HttpClient httpClientMock = mock(HttpClient.class);
 119         headers.put(CLIENT_AUTH, CLIENT_AUTH_VALUE);
 120         when(httpClientMock.put(eq("createPolicy"), anyObject(), eq(headers))).thenReturn(Response.status(400).build());
 121         when(httpClientMock.put(eq("pushPolicy"), anyObject(), eq(headers))).thenReturn(Response.ok().build());
 122
 123         final HttpClientFactory httpClientFactoryMock = mock(HttpClientFactory.class);
 124         when(httpClientFactoryMock.build(argThat(matcher))).thenReturn(httpClientMock);
 125
 126         final XacmlPdpPolicyForwarder forwarder = new MyXacmlPdpPolicyForwarder(httpClientFactoryMock);
 127         forwarder.configure("xacmlPdpConfiguration");
 128
 129         final Collection<ToscaEntity> policies = new ArrayList<>();
 130         final ToscaPolicy policy = createPolicy(policies, "policy", "optimization");
 131         forwarder.forward(policies);
 132
 133         verify(httpClientMock).put(eq("createPolicy"), argThat(new PolicyParametersEntityMatcher(policy)), eq(headers));
 134         verify(httpClientMock, times(0)).put(eq("pushPolicy"), anyObject(), anyObject());
 135     }
 136
 137     @Test
 138     public void testForwardPolicy_PushFails() throws Exception {
 139
 140         final HttpClient httpClientMock = mock(HttpClient.class);
 141         headers.put(CLIENT_AUTH, CLIENT_AUTH_VALUE);
 142         when(httpClientMock.put(eq("createPolicy"), anyObject(), eq(headers))).thenReturn(Response.ok().build());
 143         when(httpClientMock.put(eq("pushPolicy"), anyObject(), eq(headers))).thenReturn(Response.status(400).build());
 144
 145         final HttpClientFactory httpClientFactoryMock = mock(HttpClientFactory.class);
 146         when(httpClientFactoryMock.build(argThat(matcher))).thenReturn(httpClientMock);
 147
 148         final XacmlPdpPolicyForwarder forwarder = new MyXacmlPdpPolicyForwarder(httpClientFactoryMock);
 149         forwarder.configure("xacmlPdpConfiguration");
 150
 151         final Collection<ToscaEntity> policies = new ArrayList<>();
 152         final ToscaPolicy policy = createPolicy(policies, "policy", "optimization");
 153         forwarder.forward(policies);
 154
 155         verify(httpClientMock).put(eq("createPolicy"), argThat(new PolicyParametersEntityMatcher(policy)), eq(headers));
 156         verify(httpClientMock).put(eq("pushPolicy"), argThat(new PushPolicyParametersEntityMatcher(policy)),
 157                 eq(headers));
 158     }
 159
 160     @Test
 161     public void testForwardPolicy_HttpClientInitFailureForPolicyCreate() throws Exception {
 162
 163         final HttpClient httpClientMock = mock(HttpClient.class);
 164         headers.put(CLIENT_AUTH, CLIENT_AUTH_VALUE);
 165         when(httpClientMock.put(eq("createPolicy"), anyObject(), eq(headers))).thenReturn(Response.ok().build());
 166         when(httpClientMock.put(eq("pushPolicy"), anyObject(), eq(headers))).thenReturn(Response.status(400).build());
 167
 168         final HttpClientFactory httpClientFactoryMock = mock(HttpClientFactory.class);
 169         when(httpClientFactoryMock.build(argThat(matcher))).thenThrow(new HttpClientConfigException());
 170
 171         final XacmlPdpPolicyForwarder forwarder = new MyXacmlPdpPolicyForwarder(httpClientFactoryMock);
 172         forwarder.configure("xacmlPdpConfiguration");
 173
 174         final Collection<ToscaEntity> policies = new ArrayList<>();
 175         final ToscaPolicy policy = createPolicy(policies, "policy", "optimization");
 176         forwarder.forward(policies);
 177
 178         verify(httpClientMock, times(0)).put(eq("createPolicy"), anyObject(), anyObject());
 179         verify(httpClientMock, times(0)).put(eq("pushPolicy"), anyObject(), anyObject());
 180     }
 181
 182     @Test
 183     public void testForwardPolicy_HttpClientInitFailureForPolicyPush() throws Exception {
 184
 185         final HttpClient httpClientMock = mock(HttpClient.class);
 186         headers.put(CLIENT_AUTH, CLIENT_AUTH_VALUE);
 187         when(httpClientMock.put(eq("createPolicy"), anyObject(), eq(headers))).thenReturn(Response.ok().build());
 188         when(httpClientMock.put(eq("pushPolicy"), anyObject(), eq(headers))).thenReturn(Response.status(400).build());
 189
 190         final HttpClientFactory httpClientFactoryMock = mock(HttpClientFactory.class);
 191         when(httpClientFactoryMock.build(argThat(matcher))).thenReturn(httpClientMock)
 192                 .thenThrow(new HttpClientConfigException());
 193
 194         final XacmlPdpPolicyForwarder forwarder = new MyXacmlPdpPolicyForwarder(httpClientFactoryMock);
 195         forwarder.configure("xacmlPdpConfiguration");
 196
 197         final Collection<ToscaEntity> policies = new ArrayList<>();
 198         final ToscaPolicy policy = createPolicy(policies, "policy", "optimization");
 199         forwarder.forward(policies);
 200
 201         verify(httpClientMock).put(eq("createPolicy"), argThat(new PolicyParametersEntityMatcher(policy)), eq(headers));
 202         verify(httpClientMock, times(0)).put(eq("pushPolicy"), anyObject(), anyObject());
 203     }
 204
 205     class BusTopicParamsMatcher extends BaseMatcher<BusTopicParams> {
 206
 207         private BusTopicParams busTopicParams;
 208
 209         BusTopicParamsMatcher(final BusTopicParams busTopicParams) {
 210             this.busTopicParams = busTopicParams;
 211         }
 212
 213         @Override
 214         public boolean matches(final Object arg0) {
 215             if (arg0 instanceof BusTopicParams) {
 216                 final BusTopicParams toCompareTo = (BusTopicParams) arg0;
 217                 return toCompareTo.isUseHttps() == busTopicParams.isUseHttps()
 218                         && toCompareTo.getHostname().equals(busTopicParams.getHostname())
 219                         && toCompareTo.getPort() == busTopicParams.getPort()
 220                         && toCompareTo.getUserName().equals(busTopicParams.getUserName())
 221                         && toCompareTo.getPassword().equals(busTopicParams.getPassword())
 222                         && toCompareTo.isManaged() == busTopicParams.isManaged();
 223             }
 224             return false;
 225         }
 226
 227         @Override
 228         public void describeTo(final Description arg0) {}
 229     }
 230
 231     class PolicyParametersEntityMatcher extends BaseMatcher<Entity<PolicyParameters>> {
 232
 233         private ToscaPolicy policy;
 234
 235         PolicyParametersEntityMatcher(final ToscaPolicy policy) {
 236             this.policy = policy;
 237         }
 238
 239         @SuppressWarnings("unchecked")
 240         @Override
 241         public boolean matches(final Object arg0) {
 242             if (arg0 instanceof Entity) {
 243                 final PolicyParameters toCompareTo = ((Entity<PolicyParameters>) arg0).getEntity();
 244                 return toCompareTo.getPolicyName().equals(policy.getName());
 245             }
 246             return false;
 247         }
 248
 249         @Override
 250         public void describeTo(final Description arg0) {}
 251     }
 252
 253     class PushPolicyParametersEntityMatcher extends BaseMatcher<Entity<PushPolicyParameters>> {
 254
 255         private ToscaPolicy policy;
 256
 257         PushPolicyParametersEntityMatcher(final ToscaPolicy policy) {
 258             this.policy = policy;
 259         }
 260
 261         @SuppressWarnings("unchecked")
 262         @Override
 263         public boolean matches(final Object arg0) {
 264             if (arg0 instanceof Entity) {
 265                 final PushPolicyParameters toCompareTo = ((Entity<PushPolicyParameters>) arg0).getEntity();
 266                 return toCompareTo.getPolicyName().equals(policy.getName())
 267                         && toCompareTo.getPolicyType().equals(policy.getType())
 268                         && toCompareTo.getPdpGroup().equals(PDP_GROUP_VALUE);
 269             }
 270             return false;
 271         }
 272
 273         @Override
 274         public void describeTo(final Description arg0) {}
 275     }
 276
 277     class UnsupportedPolicy extends ToscaEntity {
 278
 279         @Override
 280         public String getName() {
 281             return "unsupported";
 282         }
 283     }
 284
 285     private class MyXacmlPdpPolicyForwarder extends XacmlPdpPolicyForwarder {
 286         private HttpClientFactory factory;
 287
 288         public MyXacmlPdpPolicyForwarder(final HttpClientFactory httpClientFactory) {
 289             this.factory = httpClientFactory;
 290         }
 291
 292         @Override
 293         protected HttpClientFactory getHttpClientFactory() {
 294             return this.factory;
 295         }
 296     }
 297
 298     private ToscaPolicy createPolicy(final Collection<ToscaEntity> policies, final String name, final String type) {
 299         final ToscaPolicy policy1 = new ToscaPolicy();
 300         policy1.setName(name);
 301         policy1.setType(type);
 302         policies.add(policy1);
 303         return policy1;
 304     }
 305 }