main/src/test/java/org/onap/policy/pdpx/main/rest/TestXacmlPdpRestServer.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * Copyright (C) 2019, 2021-2022 AT&T Intellectual Property. All rights reserved.
   4  * Modifications Copyright (C) 2023 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.pdpx.main.rest;
  23
  24 import static org.junit.Assert.assertEquals;
  25
  26 import jakarta.ws.rs.client.Invocation;
  27 import java.util.HashMap;
  28 import java.util.Map;
  29 import org.junit.Test;
  30 import org.onap.policy.common.endpoints.report.HealthCheckReport;
  31 import org.onap.policy.pdpx.main.CommonRest;
  32 import org.onap.policy.pdpx.main.XacmlState;
  33 import org.onap.policy.pdpx.main.rest.model.StatisticsReport;
  34 import org.slf4j.Logger;
  35 import org.slf4j.LoggerFactory;
  36
  37 /**
  38  * Class to perform unit test of {@link XacmlPdpRestServer}.
  39  *
  40  */
  41 public class TestXacmlPdpRestServer extends CommonRest {
  42
  43     private static final Logger LOGGER = LoggerFactory.getLogger(TestXacmlPdpRestServer.class);
  44     private static final String NOT_ALIVE = "not alive";
  45     private static final String ALIVE = "alive";
  46     private static final String SELF = "self";
  47     private static final String NAME = XacmlState.PDP_NAME;
  48     private static final String HEALTHCHECK_ENDPOINT = "healthcheck";
  49     private static final String STATISTICS_ENDPOINT = "statistics";
  50
  51     private int nupdates = 0;
  52
  53     @Test
  54     public void testHealthCheckSuccess() throws Exception {
  55         LOGGER.info("***************************** Running testHealthCheckSuccess *****************************");
  56         final Invocation.Builder invocationBuilder = sendHttpsRequest(HEALTHCHECK_ENDPOINT);
  57         final HealthCheckReport report = invocationBuilder.get(HealthCheckReport.class);
  58         LOGGER.info("test1HealthCheckSuccess health report {}", report);
  59         validateHealthCheckReport(NAME, SELF, true, 200, ALIVE, report);
  60     }
  61
  62     @Test
  63     public void testHealthCheckFailure() throws Exception {
  64         LOGGER.info("***************************** Running testHealthCheckFailure *****************************");
  65
  66         markActivatorDead();
  67
  68         final Invocation.Builder invocationBuilder = sendHttpsRequest(HEALTHCHECK_ENDPOINT);
  69         final HealthCheckReport report = invocationBuilder.get(HealthCheckReport.class);
  70         LOGGER.info("testHealthCheckFailure health report {}", report);
  71         validateHealthCheckReport(NAME, SELF, false, 500, NOT_ALIVE, report);
  72     }
  73
  74     @Test
  75     public void testHttpsHealthCheckSuccess() throws Exception {
  76         LOGGER.info("***************************** Running testHttpsHealthCheckSuccess *****************************");
  77         final Invocation.Builder invocationBuilder = sendHttpsRequest(HEALTHCHECK_ENDPOINT);
  78         final HealthCheckReport report = invocationBuilder.get(HealthCheckReport.class);
  79         LOGGER.info("testHttpsHealthCheckSuccess health report {}", report);
  80         validateHealthCheckReport(NAME, SELF, true, 200, ALIVE, report);
  81     }
  82
  83     @Test
  84     public void testStatistics_200() throws Exception {
  85         LOGGER.info("***************************** Running testStatistics_200 *****************************");
  86         Invocation.Builder invocationBuilder = sendHttpsRequest(STATISTICS_ENDPOINT);
  87         StatisticsReport report = invocationBuilder.get(StatisticsReport.class);
  88         LOGGER.info("testStatistics_200 health report {}", report);
  89         validateStatisticsReport(report, 0, 200, new HashMap<>());
  90         updateXacmlPdpStatistics();
  91         invocationBuilder = sendHttpsRequest(STATISTICS_ENDPOINT);
  92         report = invocationBuilder.get(StatisticsReport.class);
  93         LOGGER.info("testStatistics_200 health report {}", report);
  94         validateStatisticsReport(report, 1, 200, returnStatisticsMap());
  95     }
  96
  97     @Test
  98     public void testStatistics_500() throws Exception {
  99         LOGGER.info("***************************** Running testStatistics_500 *****************************");
 100
 101         markActivatorDead();
 102
 103         final Invocation.Builder invocationBuilder = sendHttpsRequest(STATISTICS_ENDPOINT);
 104         final StatisticsReport report = invocationBuilder.get(StatisticsReport.class);
 105         LOGGER.info("testStatistics_500 health report {}", report);
 106         validateStatisticsReport(report, 0, 500, new HashMap<>());
 107     }
 108
 109     @Test
 110     public void testHttpsStatistic() throws Exception {
 111         LOGGER.info("***************************** Running testHttpsStatistic *****************************");
 112         final Invocation.Builder invocationBuilder = sendHttpsRequest(STATISTICS_ENDPOINT);
 113         final StatisticsReport report = invocationBuilder.get(StatisticsReport.class);
 114         LOGGER.info("testHttpsStatistic health report {}", report);
 115         validateStatisticsReport(report, 0, 200, new HashMap<>());
 116     }
 117
 118     private Map<String, Map<String, Integer>> returnStatisticsMap() {
 119         Map<String, Integer> testAppMetrics1 = new HashMap<>();
 120         Map<String, Integer> testAppMetrics2 = new HashMap<>();
 121         testAppMetrics1.put("permit_decisions_count", 1);
 122         testAppMetrics1.put("deny_decisions_count", 1);
 123         testAppMetrics2.put("indeterminant_decisions_count", 1);
 124         testAppMetrics2.put("not_applicable_decisions_count", 1);
 125         Map<String, Map<String, Integer>> statisticsMap = new HashMap<>();
 126         statisticsMap.put("testApp1", testAppMetrics1);
 127         statisticsMap.put("testApp2", testAppMetrics2);
 128         return statisticsMap;
 129     }
 130
 131     private void updateXacmlPdpStatistics() {
 132         XacmlPdpStatisticsManager stats = XacmlPdpStatisticsManager.getCurrent();
 133         ++nupdates;
 134         stats.setTotalPolicyCount(nupdates);
 135         stats.setTotalPolicyTypesCount(nupdates);
 136         stats.updatePermitDecisionsCount("testApp1");
 137         stats.updateDenyDecisionsCount("testApp1");
 138         stats.updateIndeterminantDecisionsCount("testApp2");
 139         stats.updateNotApplicableDecisionsCount("testApp2");
 140     }
 141
 142     private void validateStatisticsReport(final StatisticsReport report, final int count,
 143                                           final int code, final Map<String, Map<String, Integer>> decisionsMap) {
 144         assertEquals(code, report.getCode());
 145         assertEquals(count, report.getTotalPoliciesCount());
 146         assertEquals(count, report.getTotalPolicyTypesCount());
 147         assertEquals(count, report.getPermitDecisionsCount());
 148         assertEquals(count, report.getDenyDecisionsCount());
 149         assertEquals(count, report.getIndeterminantDecisionsCount());
 150         assertEquals(count, report.getNotApplicableDecisionsCount());
 151         assertEquals(decisionsMap, report.getApplicationMetrics());
 152     }
 153
 154     private void validateHealthCheckReport(final String name, final String url, final boolean healthy, final int code,
 155                     final String message, final HealthCheckReport report) {
 156         assertEquals(name, report.getName());
 157         assertEquals(url, report.getUrl());
 158         assertEquals(healthy, report.isHealthy());
 159         assertEquals(code, report.getCode());
 160         assertEquals(message, report.getMessage());
 161     }
 162 }