public String renderDRL() {
DRLOutput out = new DRLOutput();
out.writeLine(templateContainer.getHeader());
out.writeLine(generator.getDrl());
// System.err.println(out.getDRL());
return out.getDRL();
}
/** Check correct rendering of string-valued attribute */
@Test
public void testRuleAttributeRendering() {
makeRuleSet();
makeRuleTable();
makeRow(11, "C", "A", "G");
makeRow(12, "Foo", "Foo");
makeRow(13, "attr == $param", "x");
makeRow(15, "1", "show()", "foo bar");
makeRow(16, "2", "list()", "\"10\" group\"");
listener.finishSheet();
Package p = listener.getRuleSet();
DRLOutput dout = new DRLOutput();
p.renderDRL(dout);
String drl = dout.getDRL();
// System.out.println( drl );
assertTrue(drl.contains("agenda-group \"foo bar\""));
assertTrue(drl.contains("agenda-group \"10\\\" group\""));
}
@Test
public void testNumericDisabled() throws Exception {
// DROOLS-1378
final InputStream stream =
RuleWorksheetParseTest.class.getResourceAsStream("/data/NumericDisabled.xls");
final RuleSheetListener listener = getRuleSheetListener(stream);
final Package ruleset = listener.getRuleSet();
assertNotNull(ruleset);
DRLOutput dout = new DRLOutput();
ruleset.renderDRL(dout);
String drl = dout.getDRL();
System.out.println(drl);
// check rules
Rule rule = (Rule) ruleset.getRules().get(0);
Condition cond = (Condition) rule.getConditions().get(0);
assertEquals("Cheese(price == 6600)", cond.getSnippet());
}
@Test
public void testQuoteEscapingEnabled() throws Exception {
final InputStream stream =
RuleWorksheetParseTest.class.getResourceAsStream("/data/QuoteEscapeEnabledWorkbook.xls");
final RuleSheetListener listener = getRuleSheetListener(stream);
final Package ruleset = listener.getRuleSet();
assertNotNull(ruleset);
DRLOutput dout = new DRLOutput();
ruleset.renderDRL(dout);
String drl = dout.getDRL();
System.out.println(drl);
// check rules
Rule rule = (Rule) ruleset.getRules().get(0);
Condition cond = (Condition) rule.getConditions().get(0);
assertEquals(
"Foo(myObject.getColour().equals(red), myObject.size () > 12\\\")", cond.getSnippet());
}
/** Check correct rendering of package level attributes */
@Test
public void testPackageAttributeRendering() {
makeRuleSet();
makeAttribute("NO-LOOP", "true");
makeAttribute("agenda-group", "agroup");
makeRuleTable();
makeRow(11, "C", "A", "P");
makeRow(12, "foo:Foo", "foo");
makeRow(13, "attr == $param", "x($param)");
makeRow(15, "1", "1", "100");
listener.finishSheet();
Package p = listener.getRuleSet();
DRLOutput dout = new DRLOutput();
p.renderDRL(dout);
String drl = dout.getDRL();
// System.out.println( drl );
assertTrue(drl.contains("no-loop true"));
assertTrue(drl.contains("agenda-group \"agroup\""));
}
@Test
public void testMetadata() {
makeRuleSet();
makeRuleTable();
makeRow(11, "C", "A", "@", "@");
makeRow(12, "foo: Foo", "foo");
makeRow(13, "attr == $param", "goaway($param)", "Author($param)", "Version($1-$2)");
makeRow(15, "1", "1", "J.W.Goethe", "3,14");
makeRow(16, "2", "2", "", "");
listener.finishSheet();
Package p = listener.getRuleSet();
DRLOutput dout = new DRLOutput();
p.renderDRL(dout);
String drl = dout.getDRL();
assertTrue(drl.contains("@Author(J.W.Goethe)"));
assertTrue(drl.contains("@Version(3-14)"));
assertFalse(drl.contains("@Author()"));
assertFalse(drl.contains("@Version(-)"));
}
/** Must have a code snippet in an action. */
@Test
public void testMissingCodeSnippetAction() {
try {
makeRuleSet();
makeRuleTable();
makeRow(11, "C", "A");
makeRow(12, "foo: Foo", "Bar()");
makeRow(13, "attr == $param");
makeRow(15, "1", "1");
makeRow(16, "2", "2");
listener.finishSheet();
Package p = listener.getRuleSet();
DRLOutput dout = new DRLOutput();
p.renderDRL(dout);
String drl = dout.getDRL();
System.out.println(drl);
fail("should have failed");
} catch (DecisionTableParseException e) {
String badCell = RuleSheetParserUtil.rc2name(13, 2);
System.err.println(e.getMessage());
assertTrue(e.getMessage().contains(badCell));
}
}
/** Duplicate package level attribute */
@Test
public void testDuplicatePackageAttribute() {
try {
makeRuleSet();
makeAttribute("agenda-group", "agroup"); // B3, C3
makeAttribute("agenda-group", "bgroup"); // B3. B4
makeRuleTable();
makeRow(11, "C", "A", "P");
makeRow(12, "Foo", "Foo");
makeRow(13, "attr == $param", "x");
makeRow(15, "1", "show()", "10");
listener.finishSheet();
Package p = listener.getRuleSet();
DRLOutput dout = new DRLOutput();
p.renderDRL(dout);
String drl = dout.getDRL();
// System.out.println( drl );
fail("should have failed");
} catch (DecisionTableParseException e) {
System.err.println(e.getMessage());
assertTrue(e.getMessage().contains("C3, C4"));
}
}
public String emitDRL(PMML pmml) {
List<Rule> ruleList = createRuleList(pmml);
String pkgName =
ScorecardPMMLUtils.getExtensionValue(
pmml.getHeader().getExtensions(), PMMLExtensionNames.SCORECARD_PACKAGE);
org.drools.template.model.Package aPackage = new org.drools.template.model.Package(pkgName);
DRLOutput drlOutput = new DRLOutput();
for (Rule rule : ruleList) {
aPackage.addRule(rule);
}
addDeclaredTypes(pmml, aPackage);
addImports(pmml, aPackage);
addGlobals(pmml, aPackage);
internalEmitDRL(pmml, ruleList, aPackage);
aPackage.renderDRL(drlOutput);
String drl = drlOutput.getDRL();
return drl;
}
