@Test
public void testSimpleReturnValueConstraintEvaluator() throws Exception {
final InternalKnowledgePackage pkg = new KnowledgePackageImpl("pkg1");
ProcessDescr processDescr = new ProcessDescr();
processDescr.setClassName("Process1");
processDescr.setName("Process1");
WorkflowProcessImpl process = new WorkflowProcessImpl();
process.setName("Process1");
process.setPackageName("pkg1");
ReturnValueDescr descr = new ReturnValueDescr();
descr.setText("return value;");
KnowledgeBuilderImpl pkgBuilder = new KnowledgeBuilderImpl(pkg);
DialectCompiletimeRegistry dialectRegistry =
pkgBuilder.getPackageRegistry(pkg.getName()).getDialectCompiletimeRegistry();
JavaDialect javaDialect = (JavaDialect) dialectRegistry.getDialect("java");
ProcessBuildContext context =
new ProcessBuildContext(
pkgBuilder, pkg, process, processDescr, dialectRegistry, javaDialect);
pkgBuilder.addPackageFromDrl(new StringReader("package pkg1;\nglobal Boolean value;"));
ReturnValueConstraintEvaluator node = new ReturnValueConstraintEvaluator();
final JavaReturnValueEvaluatorBuilder builder = new JavaReturnValueEvaluatorBuilder();
builder.build(context, node, descr, null);
ProcessDialectRegistry.getDialect(JavaDialect.ID).addProcess(context);
javaDialect.compileAll();
assertEquals(0, javaDialect.getResults().size());
final KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
List<KnowledgePackage> packages = new ArrayList<KnowledgePackage>();
packages.add(pkgBuilder.getPackage());
kbase.addKnowledgePackages(packages);
final StatefulKnowledgeSession ksession = kbase.newStatefulKnowledgeSession();
ksession.setGlobal("value", true);
RuleFlowProcessInstance processInstance = new RuleFlowProcessInstance();
processInstance.setKnowledgeRuntime((InternalKnowledgeRuntime) ksession);
SplitInstance splitInstance = new SplitInstance();
splitInstance.setProcessInstance(processInstance);
assertTrue(node.evaluate(splitInstance, null, null));
ksession.setGlobal("value", false);
assertFalse(node.evaluate(splitInstance, null, null));
}
/**
* 加载规则package
*
* @param userRule
*/
private void loadPackage(UserRule userRule) {
if (kBase.getKiePackage("com.camel.drools.expert.sample." + userRule.getUserCode()) != null) {
return;
}
final KnowledgeBuilderImpl builder = new KnowledgeBuilderImpl();
File file =
new File(
KieSpringBasicService.class.getResource("/").getPath().toString()
+ drlDirPath
+ userRule.getUserCode()
+ ".drl");
try {
String drl = FileUtils.readFileToString(file);
builder.addPackageFromDrl(new StringReader(drl));
if (builder.hasErrors()) {
System.out.println("[ERROR: Parsing drl error: " + builder.getErrors());
}
} catch (DroolsParserException | IOException e) {
e.printStackTrace();
}
((InternalKnowledgeBase) kBase).addPackage(builder.getPackage());
}