2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017 Intel Corp. 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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
21 package org.onap.policy.template.demo;
23 import static org.junit.Assert.assertNotNull;
24 import static org.junit.Assert.assertTrue;
25 import static org.junit.Assert.fail;
27 import java.io.IOException;
28 import java.net.URLEncoder;
29 import java.time.Instant;
30 import java.util.HashMap;
31 import java.util.UUID;
33 import org.junit.AfterClass;
34 import org.junit.BeforeClass;
35 import org.junit.Test;
36 import org.kie.api.runtime.KieSession;
37 import org.kie.api.runtime.rule.FactHandle;
38 import org.onap.policy.controlloop.ControlLoopEventStatus;
39 import org.onap.policy.controlloop.ControlLoopLogger;
40 import org.onap.policy.controlloop.ControlLoopTargetType;
41 import org.onap.policy.controlloop.VirtualControlLoopEvent;
42 import org.onap.policy.controlloop.impl.ControlLoopLoggerStdOutImpl;
43 import org.onap.policy.controlloop.policy.ControlLoopPolicy;
44 import org.onap.policy.drools.http.server.HttpServletServer;
45 import org.onap.policy.drools.impl.PolicyEngineJUnitImpl;
46 import org.onap.policy.drools.system.PolicyEngine;
47 import org.onap.policy.vfc.util.Serialization;
48 import org.slf4j.Logger;
49 import org.slf4j.LoggerFactory;
52 public class VFCControlLoopTest {
54 private static final Logger log = LoggerFactory.getLogger(VFCControlLoopTest.class);
55 private KieSession kieSession;
56 private Util.Pair<ControlLoopPolicy, String> pair;
57 private PolicyEngineJUnitImpl engine;
60 /* Set environment properties */
67 public static void setUpSimulator() {
71 } catch (Exception e) {
77 public static void tearDownSimulator() {
78 HttpServletServer.factory.destroy();
82 public void testVolte() throws IOException {
84 final String yaml = "src/test/resources/yaml/policy_ControlLoop_VFC.yaml";
87 // Pull info from the yaml
89 final Util.Pair<ControlLoopPolicy, String> pair = Util.loadYaml(yaml);
91 assertNotNull(pair.a);
92 assertNotNull(pair.a.getControlLoop());
93 assertNotNull(pair.a.getControlLoop().getControlLoopName());
94 assertTrue(pair.a.getControlLoop().getControlLoopName().length() > 0);
95 final String closedLoopControlName = pair.a.getControlLoop().getControlLoopName();
98 * Start the kie session
101 kieSession = startSession("src/main/resources/ControlLoop_Template_xacml_guard.drl",
102 "src/test/resources/yaml/policy_ControlLoop_VFC.yaml",
103 "service=ServiceTest;resource=ResourceTest;type=operational",
105 "org.onap.closed_loop.ServiceTest:VNFS:1.0.0");
106 } catch (IOException e) {
108 log.debug("Could not create kieSession");
109 fail("Could not create kieSession");
114 // Insert our globals
116 final ControlLoopLogger logger = new ControlLoopLoggerStdOutImpl();
117 kieSession.setGlobal("Logger", logger);
118 final PolicyEngineJUnitImpl engine = new PolicyEngineJUnitImpl();
119 kieSession.setGlobal("Engine", engine);
122 // Initial fire of rules
124 kieSession.fireAllRules();
126 // Kick a thread that starts testing
128 new Thread(new Runnable() {
133 log.debug("\n************ Starting VoLTE Test *************\n");
136 // Generate an invalid DCAE Event with requestID=null
138 VirtualControlLoopEvent invalidEvent = new VirtualControlLoopEvent();
139 invalidEvent.closedLoopControlName = closedLoopControlName;
140 invalidEvent.requestID = null;
141 invalidEvent.closedLoopEventClient = "tca.instance00009";
142 invalidEvent.target_type = ControlLoopTargetType.VNF;
143 invalidEvent.target = "generic-vnf.vnf-id";
144 invalidEvent.from = "DCAE";
145 invalidEvent.closedLoopAlarmStart = Instant.now();
146 invalidEvent.AAI = new HashMap<String, String>();
147 invalidEvent.AAI.put("vserver.vserver-name", "vserver-name-16102016-aai3255-data-11-1");
148 invalidEvent.closedLoopEventStatus = ControlLoopEventStatus.ONSET;
150 log.debug("-------- Sending Invalid ONSET --------");
151 log.debug(Serialization.gsonPretty.toJson(invalidEvent));
154 // Insert invalid DCAE Event into memory
156 kieSession.insert(invalidEvent);
159 } catch (InterruptedException e) {
162 // Generate first DCAE ONSET Event
164 VirtualControlLoopEvent onsetEvent = new VirtualControlLoopEvent();
165 onsetEvent.closedLoopControlName = closedLoopControlName;
166 onsetEvent.requestID = UUID.randomUUID();
167 onsetEvent.closedLoopEventClient = "tca.instance00009";
168 onsetEvent.target_type = ControlLoopTargetType.VM;
169 onsetEvent.target = "VM_NAME";
170 onsetEvent.from = "DCAE";
171 onsetEvent.closedLoopAlarmStart = Instant.now();
172 onsetEvent.AAI = new HashMap<String, String>();
173 onsetEvent.AAI.put("vserver.vserver-name", "vserver-name-16102016-aai3255-data-11-1");
174 onsetEvent.AAI.put("vserver.vserver-id", "vserver-id-16102016-aai3255-data-11-1");
175 onsetEvent.AAI.put("generic-vnf.vnf-id", "vnf-id-16102016-aai3255-data-11-1");
176 onsetEvent.AAI.put("service-instance.service-instance-id", "service-instance-id-16102016-aai3255-data-11-1");
177 onsetEvent.AAI.put("vserver.is-closed-loop-disabled", "false");
178 onsetEvent.closedLoopEventStatus = ControlLoopEventStatus.ONSET;
180 log.debug("-------- Sending Valid ONSET --------");
181 log.debug(Serialization.gsonPretty.toJson(onsetEvent));
184 // Insert first DCAE ONSET Event into memory
186 kieSession.insert(onsetEvent);
188 // We have test for subsequent ONSET Events in testvFirewall()
189 // So no need to test it again here
193 } catch (InterruptedException e) {
196 // Test is finished, so stop the kieSession
203 // Start firing rules
205 kieSession.fireUntilHalt();
207 // Dump working memory
209 dumpFacts(kieSession);
212 // See if there is anything left in memory, there SHOULD only be
215 //assertEquals("There should only be 1 Fact left in memory.", 1, kieSession.getFactCount());
216 if (kieSession.getFactCount() != 1L) {
217 log.error("FACT count mismatch: 1 expected but there are {}", kieSession.getFactCount());
219 for (FactHandle handle : kieSession.getFactHandles()) {
220 Object fact = kieSession.getObject(handle);
221 // assertEquals("Non-Param Fact left in working memory", "org.onap.policy.controlloop.Params", fact.getClass().getName());
222 log.info("Working Memory FACT: {}", fact.getClass().getName());
227 public static void dumpFacts(KieSession kieSession) {
228 log.debug("Fact Count: " + kieSession.getFactCount());
229 for (FactHandle handle : kieSession.getFactHandles()) {
230 log.debug("FACT: " + handle);
235 * This method will start a kie session and instantiate
238 * @param droolsTemplate
241 * the yaml file containing the policies
246 * @param policyVersion
247 * version of the policy
248 * @return the kieSession to be used to insert facts
249 * @throws IOException
251 private KieSession startSession(String droolsTemplate,
255 String policyVersion) throws IOException {
258 * Load policies from yaml
260 pair = Util.loadYaml(yamlFile);
262 assertNotNull(pair.a);
263 assertNotNull(pair.a.getControlLoop());
264 assertNotNull(pair.a.getControlLoop().getControlLoopName());
265 assertTrue(pair.a.getControlLoop().getControlLoopName().length() > 0);
268 * Construct a kie session
270 final KieSession kieSession = Util.buildContainer(droolsTemplate,
271 pair.a.getControlLoop().getControlLoopName(),
275 URLEncoder.encode(pair.b, "UTF-8"));
278 * Retrieve the Policy Engine
280 engine = (PolicyEngineJUnitImpl) kieSession.getGlobal("Engine");
282 log.debug("============");
283 log.debug(URLEncoder.encode(pair.b, "UTF-8"));
284 log.debug("============");