use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_SingleConstraintJavaEnum.
@Test
public void testSingleRule_SingleConstraintJavaEnum() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " Person( name == Names.FRED )\n" + "then \n" + "end";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "name", "==", new EnumValue("Names.FRED"));
expected.setRoot(type);
type.addChild(c1);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
addModelField("Person", "name", String.class.getName(), DataType.TYPE_COMPARABLE);
addJavaEnumDefinition("Person", "name", new String[] { "Names.FRED=Names.FRED" });
final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
assertEquals(expected.getTreeName(), model.getTreeName());
assertNotNull(model.getRoot());
assertEquals(type.getClassName(), model.getRoot().getClassName());
assertFalse(model.getRoot().isBound());
assertEquals(1, model.getRoot().getChildren().size());
assertNotNull(model.getRoot().getChildren().get(0));
assertTrue(model.getRoot().getChildren().get(0) instanceof ConstraintNode);
final ConstraintNode _c1 = (ConstraintNode) model.getRoot().getChildren().get(0);
assertEquals(c1.getClassName(), _c1.getClassName());
assertEquals(c1.getFieldName(), _c1.getFieldName());
assertEquals(c1.getOperator(), _c1.getOperator());
assertNotNull(_c1.getValue());
assertTrue(_c1.getValue() instanceof EnumValue);
assertEquals(c1.getValue().getValue().toString(), _c1.getValue().getValue().toString());
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_MultiplePatterns.
@Test
public void testSingleRule_MultiplePatterns() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " Person( name == \"Michael\" )\n" + " Address( country == \"England\" )\n" + "then \n" + "end";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type1 = new TypeNodeImpl("Person");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "name", "==", new StringValue("Michael"));
final TypeNode type2 = new TypeNodeImpl("Address");
final ConstraintNode c2 = new ConstraintNodeImpl("Address", "country", "==", new StringValue("England"));
expected.setRoot(type1);
type1.addChild(c1);
c1.addChild(type2);
type2.addChild(c2);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
addModelField("Person", "name", String.class.getName(), DataType.TYPE_STRING);
addModelField("Address", "this", "Address", DataType.TYPE_THIS);
addModelField("Address", "country", String.class.getName(), DataType.TYPE_STRING);
final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
assertEquals(expected.getTreeName(), model.getTreeName());
assertNotNull(model.getRoot());
final TypeNode _t1 = model.getRoot();
assertEquals(type1.getClassName(), _t1.getClassName());
assertFalse(_t1.isBound());
assertEquals(1, _t1.getChildren().size());
assertNotNull(_t1.getChildren().get(0));
assertTrue(_t1.getChildren().get(0) instanceof ConstraintNode);
final ConstraintNode _c1 = (ConstraintNode) _t1.getChildren().get(0);
assertEquals(c1.getClassName(), _c1.getClassName());
assertEquals(c1.getFieldName(), _c1.getFieldName());
assertEquals(c1.getOperator(), _c1.getOperator());
assertEquals(c1.getValue().getValue().toString(), _c1.getValue().getValue().toString());
assertEquals(1, _c1.getChildren().size());
assertNotNull(_c1.getChildren().get(0));
assertTrue(_c1.getChildren().get(0) instanceof TypeNode);
final TypeNode _t2 = (TypeNode) _c1.getChildren().get(0);
assertEquals(type2.getClassName(), _t2.getClassName());
assertFalse(_t2.isBound());
assertEquals(1, _t2.getChildren().size());
assertNotNull(_t2.getChildren().get(0));
assertTrue(_t2.getChildren().get(0) instanceof ConstraintNode);
final ConstraintNode _c2 = (ConstraintNode) _t2.getChildren().get(0);
assertEquals(c2.getClassName(), _c2.getClassName());
assertEquals(c2.getFieldName(), _c2.getFieldName());
assertEquals(c2.getOperator(), _c2.getOperator());
assertEquals(c2.getValue().getValue().toString(), _c2.getValue().getValue().toString());
assertEquals(0, _c2.getChildren().size());
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testValue_Double.
@Test
public void testValue_Double() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " Person( doubleField == 1000.56 )\n" + "then \n" + "end";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "doubleField", "==", new DoubleValue(1000.56));
expected.setRoot(type);
type.addChild(c1);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
addModelField("Person", "doubleField", Double.class.getName(), DataType.TYPE_NUMERIC_DOUBLE);
final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
assertEquals(expected.getTreeName(), model.getTreeName());
assertNotNull(model.getRoot());
assertEquals(type.getClassName(), model.getRoot().getClassName());
assertFalse(model.getRoot().isBound());
assertEquals(1, model.getRoot().getChildren().size());
assertNotNull(model.getRoot().getChildren().get(0));
assertTrue(model.getRoot().getChildren().get(0) instanceof ConstraintNode);
final ConstraintNode _c1 = (ConstraintNode) model.getRoot().getChildren().get(0);
assertEquals(c1.getClassName(), _c1.getClassName());
assertEquals(c1.getFieldName(), _c1.getFieldName());
assertEquals(c1.getOperator(), _c1.getOperator());
assertEquals(c1.getValue().getValue().toString(), _c1.getValue().getValue().toString());
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_WithImport.
@Test
public void testSingleRule_WithImport() throws Exception {
final String drl = "package smurf; \n" + "import org.drools.workbench.models.guided.dtree.backend.Person; \n" + "rule \"test_0\"\n" + "when \n" + " Person( name == \"Michael\" )\n" + "then \n" + "end";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "name", "==", new StringValue("Michael"));
expected.setRoot(type);
type.addChild(c1);
addModelField("org.drools.workbench.models.guided.dtree.backend.Person", "this", "org.drools.workbench.models.guided.dtree.backend.Person", DataType.TYPE_THIS);
addModelField("org.drools.workbench.models.guided.dtree.backend.Person", "name", String.class.getName(), DataType.TYPE_STRING);
final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
assertEquals(expected.getTreeName(), model.getTreeName());
assertNotNull(model.getRoot());
assertEquals(type.getClassName(), model.getRoot().getClassName());
assertFalse(model.getRoot().isBound());
assertEquals(1, model.getRoot().getChildren().size());
assertNotNull(model.getRoot().getChildren().get(0));
assertTrue(model.getRoot().getChildren().get(0) instanceof ConstraintNode);
final ConstraintNode _c1 = (ConstraintNode) model.getRoot().getChildren().get(0);
assertEquals(c1.getClassName(), _c1.getClassName());
assertEquals(c1.getFieldName(), _c1.getFieldName());
assertEquals(c1.getOperator(), _c1.getOperator());
assertEquals(c1.getValue().getValue().toString(), _c1.getValue().getValue().toString());
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeModelMarshallingVisitor method generateActionInsertNodeDRL.
protected void generateActionInsertNodeDRL(final ActionInsertNode an, final Node context, final boolean hasDateFieldValue, final StringBuilder drl) {
if (context == null) {
// No previous context to close
} else if (context instanceof ConstraintNode) {
drl.append(")\n").append(INDENTATION).append("then \n");
if (hasDateFieldValue) {
drl.append(INDENTATION).append(INDENTATION).append("java.text.SimpleDateFormat sdf = new java.text.SimpleDateFormat(\"" + DateUtils.getDateFormatMask() + "\");\n");
}
} else if (context instanceof TypeNode) {
drl.append(")\n").append(INDENTATION).append("then \n");
if (hasDateFieldValue) {
drl.append(INDENTATION).append(INDENTATION).append("java.text.SimpleDateFormat sdf = new java.text.SimpleDateFormat(\"" + DateUtils.getDateFormatMask() + "\");\n");
}
}
final Iterator<ActionFieldValue> itr = an.getFieldValues().iterator();
if (!itr.hasNext()) {
return;
}
final String var = "$var" + (varCounter++);
drl.append(INDENTATION).append(INDENTATION).append(an.getClassName()).append(" ").append(var).append(" = new ").append(an.getClassName()).append("();\n");
while (itr.hasNext()) {
final ActionFieldValue afv = itr.next();
drl.append(INDENTATION).append(INDENTATION).append(var).append(".");
drl.append("set");
drl.append(Character.toUpperCase(afv.getFieldName().charAt(0)));
drl.append(afv.getFieldName().substring(1));
drl.append("( ").append(generateRHSValueDRL(afv.getValue())).append(" );\n");
}
if (an.isLogicalInsertion()) {
drl.append(INDENTATION).append(INDENTATION).append("insertLogical( ").append(var).append(" );\n");
} else {
drl.append(INDENTATION).append(INDENTATION).append("insert( ").append(var).append(" );\n");
}
}
Aggregations