gerrit.onap Code Review - policy/drools-applications.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP
   4  * ================================================================================
   5  * Copyright (C) 2018-2021 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.drools.apps.controlloop.feature.trans;
  22
  23 import static org.awaitility.Awaitility.await;
  24 import static org.junit.Assert.assertEquals;
  25 import static org.junit.Assert.assertFalse;
  26 import static org.junit.Assert.assertNotEquals;
  27 import static org.junit.Assert.assertNotNull;
  28 import static org.junit.Assert.assertNull;
  29 import static org.junit.Assert.assertTrue;
  30
  31 import java.nio.file.Path;
  32 import java.util.UUID;
  33 import java.util.concurrent.TimeUnit;
  34 import org.junit.AfterClass;
  35 import org.junit.Before;
  36 import org.junit.BeforeClass;
  37 import org.junit.Test;
  38 import org.onap.policy.common.endpoints.event.comm.Topic.CommInfrastructure;
  39 import org.onap.policy.common.utils.resources.ResourceUtils;
  40 import org.onap.policy.controlloop.ControlLoopNotificationType;
  41 import org.onap.policy.controlloop.VirtualControlLoopNotification;
  42 import org.onap.policy.controlloop.util.Serialization;
  43 import org.onap.policy.drools.persistence.SystemPersistenceConstants;
  44 import org.onap.policy.drools.system.PolicyController;
  45 import org.onap.policy.drools.system.PolicyEngineConstants;
  46
  47 /**
  48  * ControlLoopMetrics Tests.
  49  */
  50 public class ControlLoopMetricsFeatureTest {
  51
  52     private static final String POLICY_CL_MGT = "POLICY-CL-MGT";
  53     private static final Path configPath = SystemPersistenceConstants.getManager().getConfigurationPath();
  54     private static PolicyController testController;
  55
  56     private static void resetStats() {
  57         PolicyEngineConstants.getManager().getStats().getGroupStat().setAverageExecutionTime(0d);
  58         PolicyEngineConstants.getManager().getStats().getGroupStat().setLastExecutionTime(0L);
  59         PolicyEngineConstants.getManager().getStats().getGroupStat().setLastStart(0L);
  60         PolicyEngineConstants.getManager().getStats().getGroupStat().setAverageExecutionTime(0d);
  61         PolicyEngineConstants.getManager().getStats().getGroupStat().setPolicyExecutedCount(0L);
  62         PolicyEngineConstants.getManager().getStats().getGroupStat().setPolicyExecutedFailCount(0L);
  63         PolicyEngineConstants.getManager().getStats().getGroupStat().setPolicyExecutedSuccessCount(0L);
  64         PolicyEngineConstants.getManager().getStats().getGroupStat().setTotalElapsedTime(0d);
  65     }
  66
  67     /**
  68      * Setup method.
  69      */
  70     @BeforeClass
  71     public static void setUp() {
  72         SystemPersistenceConstants.getManager().setConfigurationDir("src/test/resources");
  73         testController = PolicyEngineConstants.getManager().createPolicyController("metrics",
  74                 SystemPersistenceConstants.getManager().getControllerProperties("metrics"));
  75     }
  76
  77     @AfterClass
  78     public static void tearDown() {
  79         SystemPersistenceConstants.getManager().setConfigurationDir(configPath.toString());
  80         resetStats();
  81     }
  82
  83     @Before
  84     public void beforeTest() {
  85         resetStats();
  86     }
  87
  88     @Test
  89     public void testCacheDefaults() {
  90         assertEquals(3, ControlLoopMetricsManager.getManager().getCacheSize());
  91         assertEquals(2, ControlLoopMetricsManager.getManager().getTransactionTimeout());
  92         assertEquals(0, ControlLoopMetricsManager.getManager().getCacheOccupancy());
  93     }
  94
  95     @Test
  96     public void testInvalidNotifications() {
  97         ControlLoopMetricsFeature feature = new ControlLoopMetricsFeature();
  98         VirtualControlLoopNotification notification = new VirtualControlLoopNotification();
  99         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, notification);
 100         this.testCacheDefaults();
 101
 102         UUID requestId = UUID.randomUUID();
 103         notification.setRequestId(requestId);
 104
 105         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, notification);
 106         assertNull(ControlLoopMetricsManager.getManager().getTransaction(requestId));
 107         this.testCacheDefaults();
 108     }
 109
 110     @Test
 111     public void testValidActiveNotification() {
 112         ControlLoopMetricsFeature feature = new ControlLoopMetricsFeature();
 113         VirtualControlLoopNotification notification = new VirtualControlLoopNotification();
 114         UUID requestId = UUID.randomUUID();
 115         notification.setRequestId(requestId);
 116         notification.setNotification(ControlLoopNotificationType.ACTIVE);
 117
 118         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, notification);
 119         assertNotNull(ControlLoopMetricsManager.getManager().getTransaction(requestId));
 120         assertTrue(ControlLoopMetricsManager.getManager().getTransaction(requestId).getFrom()
 121                         .contains(testController.getName()));
 122         assertNotNull(ControlLoopMetricsManager.getManager().getTransaction(requestId).getNotificationTime());
 123         assertEquals(1, ControlLoopMetricsManager.getManager().getCacheOccupancy());
 124
 125         /* wait for the entries to expire */
 126         await().atMost(ControlLoopMetricsManager.getManager().getTransactionTimeout() + 1, TimeUnit.SECONDS)
 127                         .until(() -> ControlLoopMetricsManager.getManager().getTransaction(requestId) == null);
 128
 129         this.testCacheDefaults();
 130     }
 131
 132     @Test
 133     public void testReset() {
 134         VirtualControlLoopNotification notification = this.generateNotification();
 135         new ControlLoopMetricsFeature().beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT,
 136                 notification);
 137
 138         assertNotNull(ControlLoopMetricsManager.getManager().getTransaction(notification.getRequestId()));
 139
 140         ControlLoopMetricsManager.getManager().resetCache(ControlLoopMetricsManager.getManager().getCacheSize(),
 141                 ControlLoopMetricsManager.getManager().getTransactionTimeout());
 142         assertNull(ControlLoopMetricsManager.getManager().getTransaction(notification.getRequestId()));
 143         this.testCacheDefaults();
 144     }
 145
 146     @Test
 147     public void testRemoveTransaction() {
 148         VirtualControlLoopNotification notification = this.generateNotification();
 149         assertNull(ControlLoopMetricsManager.getManager().getTransaction(notification.getRequestId()));
 150         ControlLoopMetricsManager.getManager().removeTransaction(notification.getRequestId());
 151
 152         ControlLoopMetricsManager.getManager().transactionEvent(testController, notification);
 153         assertNotNull(ControlLoopMetricsManager.getManager().getTransaction(notification.getRequestId()));
 154         ControlLoopMetricsManager.getManager().removeTransaction(notification.getRequestId());
 155         assertNull(ControlLoopMetricsManager.getManager().getTransaction(notification.getRequestId()));
 156     }
 157
 158     @Test
 159     public void testEviction() {
 160         ControlLoopMetricsFeature feature = new ControlLoopMetricsFeature();
 161         for (int i = 0; i < ControlLoopMetricsManager.getManager().getCacheSize(); i++) {
 162             VirtualControlLoopNotification notification = generateNotification();
 163             feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, notification);
 164             assertNotNull(ControlLoopMetricsManager.getManager().getTransaction(notification.getRequestId()));
 165         }
 166
 167         assertEquals(ControlLoopMetricsManager.getManager().getCacheOccupancy(),
 168                         ControlLoopMetricsManager.getManager().getCacheOccupancy());
 169
 170         VirtualControlLoopNotification overflowNotification = generateNotification();
 171         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, overflowNotification);
 172         assertEquals(ControlLoopMetricsManager.getManager().getCacheOccupancy(),
 173                         ControlLoopMetricsManager.getManager().getCacheOccupancy());
 174         assertNotNull(ControlLoopMetricsManager.getManager().getTransaction(overflowNotification.getRequestId()));
 175         assertEquals(ControlLoopMetricsManager.getManager().getTransactionIds().size(),
 176                 ControlLoopMetricsManager.getManager().getCacheSize());
 177         assertEquals(ControlLoopMetricsManager.getManager().getCacheOccupancy(),
 178                 ControlLoopMetricsManager.getManager().getCacheSize());
 179         assertFalse(ControlLoopMetricsManager.getManager().getTransactionIds().isEmpty());
 180         assertFalse(ControlLoopMetricsManager.getManager().getTransactions().isEmpty());
 181
 182         /* wait for the entries to expire */
 183         await().atMost(ControlLoopMetricsManager.getManager().getTransactionTimeout() + 1, TimeUnit.SECONDS)
 184                         .until(() -> ControlLoopMetricsManager.getManager().getTransactions().isEmpty());
 185
 186         ControlLoopMetricsManager.getManager().refresh();
 187         assertEquals(ControlLoopMetricsManager.getManager().getTransactionIds().size(),
 188                 ControlLoopMetricsManager.getManager().getCacheOccupancy());
 189         assertNotEquals(ControlLoopMetricsManager.getManager().getCacheOccupancy(),
 190                 ControlLoopMetricsManager.getManager().getCacheSize());
 191         assertTrue(ControlLoopMetricsManager.getManager().getTransactionIds().isEmpty());
 192         assertTrue(ControlLoopMetricsManager.getManager().getTransactions().isEmpty());
 193
 194         this.testCacheDefaults();
 195     }
 196
 197     private VirtualControlLoopNotification generateNotification() {
 198         VirtualControlLoopNotification notification = new VirtualControlLoopNotification();
 199         UUID requestId = UUID.randomUUID();
 200         notification.setRequestId(requestId);
 201         notification.setNotification(ControlLoopNotificationType.ACTIVE);
 202         return notification;
 203     }
 204
 205     @Test
 206     public void getSequenceNumber() {
 207         ControlLoopMetricsFeature feature = new ControlLoopMetricsFeature();
 208         assertEquals(ControlLoopMetricsFeature.FEATURE_SEQUENCE_PRIORITY, feature.getSequenceNumber());
 209     }
 210
 211     @Test
 212     public void testSuccessControlLoop() {
 213         ControlLoopMetricsFeature feature = new ControlLoopMetricsFeature();
 214
 215         String activeNotification = ResourceUtils.getResourceAsString("policy-cl-mgt-active.json");
 216         VirtualControlLoopNotification active =
 217                 Serialization.gsonPretty.fromJson(activeNotification, VirtualControlLoopNotification.class);
 218         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, active);
 219         assertEquals(1, ControlLoopMetricsManager.getManager().getTransactionIds().size());
 220
 221         String opStartNotification = ResourceUtils.getResourceAsString("policy-cl-mgt-operation.json");
 222         VirtualControlLoopNotification opStart =
 223                 Serialization.gsonPretty.fromJson(opStartNotification, VirtualControlLoopNotification.class);
 224         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, opStart);
 225         assertEquals(1, ControlLoopMetricsManager.getManager().getTransactionIds().size());
 226
 227         String permitNotification = ResourceUtils.getResourceAsString("policy-cl-mgt-permit.json");
 228         VirtualControlLoopNotification permit =
 229                 Serialization.gsonPretty.fromJson(permitNotification, VirtualControlLoopNotification.class);
 230         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, permit);
 231         assertEquals(1, ControlLoopMetricsManager.getManager().getTransactionIds().size());
 232
 233         String restartNotification = ResourceUtils.getResourceAsString("policy-cl-mgt-restart.json");
 234         VirtualControlLoopNotification restart =
 235                 Serialization.gsonPretty.fromJson(restartNotification, VirtualControlLoopNotification.class);
 236         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, restart);
 237         assertEquals(1, ControlLoopMetricsManager.getManager().getTransactionIds().size());
 238
 239         String restartSuccessNotification =
 240                 ResourceUtils.getResourceAsString("policy-cl-mgt-restart-success.json");
 241         VirtualControlLoopNotification restartSuccess =
 242                 Serialization.gsonPretty.fromJson(restartSuccessNotification, VirtualControlLoopNotification.class);
 243         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, restartSuccess);
 244         assertEquals(1, ControlLoopMetricsManager.getManager().getTransactionIds().size());
 245
 246         String finalSuccessNotification =
 247                 ResourceUtils.getResourceAsString("policy-cl-mgt-final-success.json");
 248         VirtualControlLoopNotification finalSuccess =
 249                 Serialization.gsonPretty.fromJson(finalSuccessNotification, VirtualControlLoopNotification.class);
 250         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, finalSuccess);
 251         assertEquals(0, ControlLoopMetricsManager.getManager().getTransactionIds().size());
 252         assertEquals(1,
 253                 PolicyEngineConstants.getManager().getStats().getGroupStat().getPolicyExecutedSuccessCount());
 254         assertEquals(0,
 255                 PolicyEngineConstants.getManager().getStats().getGroupStat().getPolicyExecutedFailCount());
 256         assertEquals(1, PolicyEngineConstants.getManager().getStats().getGroupStat().getPolicyExecutedCount());
 257         assertEquals(1587409937684L,
 258                 PolicyEngineConstants.getManager().getStats().getGroupStat().getLastExecutionTime());
 259         assertEquals(461d,
 260                 PolicyEngineConstants.getManager().getStats().getGroupStat().getAverageExecutionTime(), 0.0d);
 261         assertEquals(1587409937223L,
 262                 PolicyEngineConstants.getManager().getStats().getGroupStat().getLastStart());
 263         assertEquals(461d,
 264                 PolicyEngineConstants.getManager().getStats().getGroupStat().getTotalElapsedTime(), 0.0d);
 265     }
 266
 267     @Test
 268     public void testUntrackedNotifications() throws InterruptedException {
 269         ControlLoopMetricsFeature feature = new ControlLoopMetricsFeature();
 270
 271         String finalSuccessNotification =
 272                 ResourceUtils.getResourceAsString("policy-cl-mgt-final-success.json");
 273         VirtualControlLoopNotification finalSuccess =
 274                 Serialization.gsonPretty.fromJson(finalSuccessNotification, VirtualControlLoopNotification.class);
 275         finalSuccess.setRequestId(UUID.randomUUID());
 276         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, finalSuccess);
 277         assertEquals(0, ControlLoopMetricsManager.getManager().getTransactionIds().size());
 278
 279         String opStartNotification =
 280                 ResourceUtils.getResourceAsString("policy-cl-mgt-operation.json");
 281         VirtualControlLoopNotification opStart =
 282                 Serialization.gsonPretty.fromJson(opStartNotification, VirtualControlLoopNotification.class);
 283         feature.beforeDeliver(testController, CommInfrastructure.DMAAP, POLICY_CL_MGT, opStart);
 284         assertEquals(1, ControlLoopMetricsManager.getManager().getTransactionIds().size());
 285
 286         Thread.sleep((ControlLoopMetricsManager.getManager().getTransactionTimeout() + 1) * 1000L);  // NOSONAR
 287         assertEquals(0, ControlLoopMetricsManager.getManager().getTransactionIds().size());
 288     }
 289
 290 }