use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testMultipleRules_3Rules.
@Test
public void testMultipleRules_3Rules() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " Person( name == \"Michael\" )\n" + "then \n" + "end \n" + "rule \"test_1\"\n" + "when \n" + " Person( age == 41 )\n" + "then \n" + "end \n" + "rule \"test_2\"\n" + "when \n" + " Person( gender == \"Male\" )\n" + "then \n" + "end \n";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "name", "==", new StringValue("Michael"));
final ConstraintNode c2 = new ConstraintNodeImpl("Person", "age", "==", new IntegerValue(41));
final ConstraintNode c3 = new ConstraintNodeImpl("Person", "gender", "==", new StringValue("Male"));
expected.setRoot(type);
type.addChild(c1);
type.addChild(c2);
type.addChild(c3);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
addModelField("Person", "name", String.class.getName(), DataType.TYPE_STRING);
addModelField("Person", "age", Integer.class.getName(), DataType.TYPE_NUMERIC_INTEGER);
addModelField("Person", "gender", 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(type.getClassName(), _t1.getClassName());
assertFalse(_t1.isBound());
assertEquals(3, _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());
assertNotNull(_t1.getChildren().get(1));
assertTrue(_t1.getChildren().get(1) instanceof ConstraintNode);
final ConstraintNode _c2 = (ConstraintNode) _t1.getChildren().get(1);
assertEquals(c2.getClassName(), _c2.getClassName());
assertEquals(c2.getFieldName(), _c2.getFieldName());
assertEquals(c2.getOperator(), _c2.getOperator());
assertEquals(c2.getValue().getValue().toString(), _c2.getValue().getValue().toString());
assertNotNull(_t1.getChildren().get(2));
assertTrue(_t1.getChildren().get(2) instanceof ConstraintNode);
final ConstraintNode _c3 = (ConstraintNode) _t1.getChildren().get(2);
assertEquals(c3.getClassName(), _c3.getClassName());
assertEquals(c3.getFieldName(), _c3.getFieldName());
assertEquals(c3.getOperator(), _c3.getOperator());
assertEquals(c3.getValue().getValue().toString(), _c3.getValue().getValue().toString());
assertEquals(0, _c1.getChildren().size());
assertEquals(0, _c2.getChildren().size());
assertEquals(0, _c3.getChildren().size());
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_MultipleConstraints.
@Test
public void testSingleRule_MultipleConstraints() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " Person( name == \"Michael\", age == 41 )\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"));
final ConstraintNode c2 = new ConstraintNodeImpl("Person", "age", "==", new IntegerValue(41));
expected.setRoot(type);
type.addChild(c1);
c1.addChild(c2);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
addModelField("Person", "name", String.class.getName(), DataType.TYPE_STRING);
addModelField("Person", "age", Integer.class.getName(), DataType.TYPE_NUMERIC_INTEGER);
final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
assertEquals(expected.getTreeName(), model.getTreeName());
assertNotNull(model.getRoot());
final TypeNode _t1 = model.getRoot();
assertEquals(type.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 ConstraintNode);
final ConstraintNode _c2 = (ConstraintNode) _c1.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_String.
@Test
public void testValue_String() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " Person( stringField == \"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", "stringField", "==", new StringValue("Michael"));
expected.setRoot(type);
type.addChild(c1);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
addModelField("Person", "stringField", 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 GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_ActionSetWithConstraint.
@Test
public void testSingleRule_ActionSetWithConstraint() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " $p : Person( $n : name )\n" + "then \n" + " $p.setAge( 25 );\n" + "end";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
type.setBinding("$p");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "name");
c1.setBinding("$n");
expected.setRoot(type);
type.addChild(c1);
final ActionUpdateNode action = new ActionUpdateNodeImpl(type);
action.setModify(false);
action.getFieldValues().add(new ActionFieldValueImpl("age", new IntegerValue(25)));
c1.addChild(action);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
addModelField("Person", "name", String.class.getName(), DataType.TYPE_STRING);
addModelField("Person", "age", Integer.class.getName(), DataType.TYPE_NUMERIC_INTEGER);
final GuidedDecisionTree model = getAndTestUnmarshalledModel(drl, "test", 0);
assertEquals(expected.getTreeName(), model.getTreeName());
assertNotNull(model.getRoot());
assertEquals(type.getClassName(), model.getRoot().getClassName());
assertTrue(model.getRoot().isBound());
assertEquals(type.getBinding(), model.getRoot().getBinding());
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());
assertNull(_c1.getOperator());
assertNull(_c1.getValue());
assertTrue(_c1.isBound());
assertEquals(c1.getBinding(), _c1.getBinding());
assertEquals(1, _c1.getChildren().size());
assertNotNull(_c1.getChildren().get(0));
assertTrue(_c1.getChildren().get(0) instanceof ActionUpdateNode);
final ActionUpdateNode _action = (ActionUpdateNode) _c1.getChildren().get(0);
assertEquals(action.getBoundNode().getBinding(), _action.getBoundNode().getBinding());
assertEquals(action.isModify(), _action.isModify());
assertEquals(action.getFieldValues().size(), _action.getFieldValues().size());
assertEquals(1, _action.getFieldValues().size());
assertEquals(action.getFieldValues().get(0).getFieldName(), _action.getFieldValues().get(0).getFieldName());
assertEquals(action.getFieldValues().get(0).getValue().getValue().toString(), _action.getFieldValues().get(0).getValue().getValue().toString());
assertEquals(0, _action.getChildren().size());
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceMarshallingTest method testValue_Long.
@Test
public void testValue_Long() throws Exception {
final String expected = "rule \"test_0\"" + "when\n" + " Person( longField == 1000000 )\n" + "then\n" + "end";
final GuidedDecisionTree model = new GuidedDecisionTree();
model.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
final ConstraintNode c1 = new ConstraintNodeImpl("Person", "longField", "==", new LongValue(1000000L));
model.setRoot(type);
type.addChild(c1);
final String drl = GuidedDecisionTreeDRLPersistence.getInstance().marshal(model);
assertEqualsIgnoreWhitespace(expected, drl);
}
Aggregations