diff options
Diffstat (limited to 'policy-management/src/test')
-rw-r--r-- | policy-management/src/test/java/org/onap/policy/drools/system/PolicyEngineManagerTest.java | 41 |
1 files changed, 36 insertions, 5 deletions
diff --git a/policy-management/src/test/java/org/onap/policy/drools/system/PolicyEngineManagerTest.java b/policy-management/src/test/java/org/onap/policy/drools/system/PolicyEngineManagerTest.java index 4be25d1a..82c28695 100644 --- a/policy-management/src/test/java/org/onap/policy/drools/system/PolicyEngineManagerTest.java +++ b/policy-management/src/test/java/org/onap/policy/drools/system/PolicyEngineManagerTest.java @@ -2,7 +2,7 @@ * ============LICENSE_START======================================================= * ONAP * ================================================================================ - * Copyright (C) 2018-2021 AT&T Intellectual Property. All rights reserved. + * Copyright (C) 2018-2022 AT&T Intellectual Property. All rights reserved. * ================================================================================ * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -40,6 +40,8 @@ import static org.mockito.Mockito.times; import static org.mockito.Mockito.verify; import static org.mockito.Mockito.when; +import io.prometheus.client.CollectorRegistry; +import io.prometheus.client.Summary; import java.util.Arrays; import java.util.Collections; import java.util.List; @@ -47,6 +49,7 @@ import java.util.Properties; import java.util.concurrent.ScheduledExecutorService; import java.util.function.BiConsumer; import java.util.function.Consumer; +import org.junit.After; import org.junit.Before; import org.junit.Test; import org.onap.policy.common.endpoints.event.comm.Topic.CommInfrastructure; @@ -74,6 +77,7 @@ import org.onap.policy.drools.protocol.configuration.PdpdConfiguration; import org.onap.policy.drools.stats.PolicyStatsManager; import org.onap.policy.drools.system.internal.SimpleLockManager; import org.onap.policy.drools.system.internal.SimpleLockProperties; +import org.onap.policy.models.pdp.enums.PdpResponseStatus; public class PolicyEngineManagerTest { private static final String EXPECTED = "expected exception"; @@ -90,6 +94,8 @@ public class PolicyEngineManagerTest { private static final String MESSAGE = "my-message"; private static final String MY_OWNER = "my-owner"; private static final String MY_RESOURCE = "my-resource"; + private static final String POLICY = "policy"; + private static final String CONTROLLOOP = "controlloop"; private static final Object MY_EVENT = new Object(); @@ -153,7 +159,7 @@ public class PolicyEngineManagerTest { */ @Before public void setUp() throws Exception { - + CollectorRegistry.defaultRegistry.clear(); properties = new Properties(); prov1 = mock(PolicyEngineFeatureApi.class); prov2 = mock(PolicyEngineFeatureApi.class); @@ -317,6 +323,11 @@ public class PolicyEngineManagerTest { mgr = new PolicyEngineManagerImpl(); } + @After + public void tearDown() { + CollectorRegistry.defaultRegistry.clear(); + } + @Test public void testSerialize() { mgr.configure(properties); @@ -1360,14 +1371,34 @@ public class PolicyEngineManagerTest { @Test public void testTransaction() { - mgr.metric("foo", "bar", new Metric()); + mgr.metric(CONTROLLER1, POLICY, new Metric()); assertEquals(0, mgr.getStats().getGroupStat().getPolicyExecutedCount()); assertEquals(0, mgr.getStats().getSubgroupStats().size()); - mgr.transaction("foo", "bar", new Metric()); + Metric metric = new Metric(); + mgr.transaction(CONTROLLER1, CONTROLLOOP, metric); assertEquals(1, mgr.getStats().getGroupStat().getPolicyExecutedCount()); assertEquals(1, mgr.getStats().getSubgroupStats().size()); - assertEquals(1, mgr.getStats().getSubgroupStats().get("bar").getPolicyExecutedFailCount()); + assertEquals(1, mgr.getStats().getSubgroupStats().get(CONTROLLOOP).getPolicyExecutedFailCount()); + + Summary.Child.Value summary = + PolicyEngineManagerImpl.transLatencySecsSummary + .labels(CONTROLLER1, CONTROLLOOP, POLICY, PdpResponseStatus.FAIL.name()).get(); + + assertEquals(0, summary.count, 0.0); + assertEquals(0, summary.sum, 0.0); + + metric.setServiceInstanceId(POLICY); + metric.setElapsedTime(5000L); + metric.setSuccess(false); + mgr.transaction(CONTROLLER1, CONTROLLOOP, metric); + + summary = + PolicyEngineManagerImpl.transLatencySecsSummary + .labels(CONTROLLER1, CONTROLLOOP, POLICY, PdpResponseStatus.FAIL.name()).get(); + + assertEquals(1, summary.count, 0.0); + assertEquals(5, summary.sum, 0.0); } @Test |