aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--policy-core/src/main/java/org/onap/policy/drools/core/DroolsExecutor.java47
-rw-r--r--policy-core/src/test/java/org/onap/policy/drools/core/DroolsContainerTest.java41
2 files changed, 75 insertions, 13 deletions
diff --git a/policy-core/src/main/java/org/onap/policy/drools/core/DroolsExecutor.java b/policy-core/src/main/java/org/onap/policy/drools/core/DroolsExecutor.java
new file mode 100644
index 00000000..538970d6
--- /dev/null
+++ b/policy-core/src/main/java/org/onap/policy/drools/core/DroolsExecutor.java
@@ -0,0 +1,47 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP
+ * ================================================================================
+ * Copyright (C) 2020 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.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.onap.policy.drools.core;
+
+import java.util.concurrent.Executor;
+import org.drools.core.WorkingMemory;
+
+/**
+ * An executor that executes its tasks by inserting {@link DroolsRunnable} objects into
+ * the working memory of a running session.
+ */
+public class DroolsExecutor implements Executor {
+ private final WorkingMemory workingMemory;
+
+ /**
+ * Constructs the object.
+ *
+ * @param workingMemory where tasks should be injected
+ */
+ public DroolsExecutor(WorkingMemory workingMemory) {
+ this.workingMemory = workingMemory;
+ }
+
+ @Override
+ public void execute(Runnable command) {
+ DroolsRunnable runnable = command::run;
+ workingMemory.insert(runnable);
+ }
+}
diff --git a/policy-core/src/test/java/org/onap/policy/drools/core/DroolsContainerTest.java b/policy-core/src/test/java/org/onap/policy/drools/core/DroolsContainerTest.java
index 2f45f01e..928af4de 100644
--- a/policy-core/src/test/java/org/onap/policy/drools/core/DroolsContainerTest.java
+++ b/policy-core/src/test/java/org/onap/policy/drools/core/DroolsContainerTest.java
@@ -2,14 +2,14 @@
* ============LICENSE_START=======================================================
* policy-core
* ================================================================================
- * Copyright (C) 2017-2019 AT&T Intellectual Property. All rights reserved.
+ * Copyright (C) 2017-2020 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.
* You may obtain a copy of the License at
- *
+ *
* http://www.apache.org/licenses/LICENSE-2.0
- *
+ *
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -24,12 +24,16 @@ import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.Collection;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;
+import org.drools.core.WorkingMemory;
import org.junit.BeforeClass;
import org.junit.Test;
import org.onap.policy.drools.util.KieUtils;
@@ -41,7 +45,7 @@ import org.onap.policy.drools.util.KieUtils;
* PolicySessionFeatureAPI
*/
public class DroolsContainerTest {
-
+
private static final long TIMEOUT_SEC = 5;
/**
@@ -159,24 +163,35 @@ public class DroolsContainerTest {
session.getFullName());
// the updated rules should now multiply 3 * 8 * 2, and return 48
-
+
result = new LinkedBlockingQueue<>();
container.insert("session1", Arrays.asList(3, 8, 2));
container.insert("session1", result);
assertEquals(48, result.poll(TIMEOUT_SEC, TimeUnit.SECONDS).intValue());
- // verify that default KiePackages have been added by testing
- // that 'DroolsRunnable' is functioning
+ /*
+ * verify that default KiePackages have been added by testing that
+ * 'DroolsRunnable' is functioning. Also verifies that DroolsExecutor works by
+ * using it to inject the runnable, though we have to mock the WorkingMemory
+ * object since I don't know how to obtain it from the kieSession
+ */
+
+ WorkingMemory wm = mock(WorkingMemory.class);
+ when(wm.insert(any())).thenAnswer(ans -> {
+ Object object = ans.getArgument(0);
+ return container.getPolicySession("session1").getKieSession().insert(object);
+ });
+
+ DroolsExecutor executor = new DroolsExecutor(wm);
final LinkedBlockingQueue<String> lbq = new LinkedBlockingQueue<>();
- container.insert("session1", new DroolsRunnable() {
- @Override
- public void run() {
- lbq.add("DroolsRunnable String");
- }
- });
+ executor.execute(() -> lbq.add("DroolsRunnable String"));
+ executor.execute(() -> lbq.add("another DroolsRunnable String"));
+
assertEquals("DroolsRunnable String", lbq.poll(TIMEOUT_SEC, TimeUnit.SECONDS));
+ assertEquals("another DroolsRunnable String", lbq.poll(TIMEOUT_SEC, TimeUnit.SECONDS));
+
} finally {
container.shutdown();
assertFalse(container.isAlive());