use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionRetractNode in project drools by kiegroup.
the class GuidedDecisionTreeModelMarshallingVisitor method generateRuleDRL.
protected void generateRuleDRL(final List<Node> path) {
Node context = null;
final StringBuilder drl = new StringBuilder();
final boolean hasDateFieldValue = hasDateFieldValue(path);
this.varCounter = 0;
drl.append(generateRuleHeaderDRL());
drl.append(INDENTATION).append("when \n");
for (Node node : path) {
if (node instanceof TypeNode) {
generateTypeNodeDRL((TypeNode) node, context, drl);
} else if (node instanceof ConstraintNode) {
generateConstraintNodeDRL((ConstraintNode) node, context, drl);
} else if (node instanceof ActionRetractNode) {
generateActionRetractNodeDRL((ActionRetractNode) node, context, hasDateFieldValue, drl);
} else if (node instanceof ActionUpdateNode) {
generateActionUpdateNodeDRL((ActionUpdateNode) node, context, hasDateFieldValue, drl);
} else if (node instanceof ActionInsertNode) {
generateActionInsertNodeDRL((ActionInsertNode) node, context, hasDateFieldValue, drl);
}
context = node;
}
if (context == null) {
// No previous context to close
} else if (context instanceof ConstraintNode) {
drl.append(")\n").append("then \n").append("end\n");
} else if (context instanceof TypeNode) {
drl.append(")\n").append("then \n").append("end\n");
} else if (context instanceof ActionRetractNode) {
drl.append("end\n");
} else if (context instanceof ActionUpdateNode) {
drl.append("end\n");
} else if (context instanceof ActionInsertNode) {
drl.append("end\n");
}
ruleCount++;
rules.append(drl);
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionRetractNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceMarshallingTest method testSingleRule_ActionModifyActionRetract.
@Test
public void testSingleRule_ActionModifyActionRetract() throws Exception {
final String expected = "rule \"test_0\"" + "when\n" + " $p : Person( )\n" + "then\n" + " modify( $p ) {\n" + " setAge( 25 )\n" + " }\n" + " retract( $p );\n" + "end";
final GuidedDecisionTree model = new GuidedDecisionTree();
model.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
type.setBinding("$p");
model.setRoot(type);
final ActionUpdateNode action1 = new ActionUpdateNodeImpl(type);
action1.setModify(true);
action1.getFieldValues().add(new ActionFieldValueImpl("age", new IntegerValue(25)));
type.addChild(action1);
final ActionRetractNode action2 = new ActionRetractNodeImpl(type);
action1.addChild(action2);
final String drl = GuidedDecisionTreeDRLPersistence.getInstance().marshal(model);
assertEqualsIgnoreWhitespace(expected, drl);
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionRetractNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceMarshallingTest method testSingleRule_ActionRetract.
@Test
public void testSingleRule_ActionRetract() throws Exception {
final String expected = "rule \"test_0\"" + "when\n" + " $p : Person( )\n" + "then\n" + " retract( $p );\n" + "end";
final GuidedDecisionTree model = new GuidedDecisionTree();
model.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
type.setBinding("$p");
model.setRoot(type);
final ActionRetractNode action = new ActionRetractNodeImpl(type);
type.addChild(action);
final String drl = GuidedDecisionTreeDRLPersistence.getInstance().marshal(model);
assertEqualsIgnoreWhitespace(expected, drl);
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionRetractNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_ActionSetActionRetract.
@Test
public void testSingleRule_ActionSetActionRetract() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " $p : Person( )\n" + "then \n" + " $p.setAge( 25 );\n" + " retract( $p );\n" + "end";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
type.setBinding("$p");
expected.setRoot(type);
final ActionUpdateNode action1 = new ActionUpdateNodeImpl(type);
action1.setModify(false);
action1.getFieldValues().add(new ActionFieldValueImpl("age", new IntegerValue(25)));
type.addChild(action1);
final ActionRetractNode action2 = new ActionRetractNodeImpl(type);
action1.addChild(action2);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
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 ActionUpdateNode);
final ActionUpdateNode _action1 = (ActionUpdateNode) model.getRoot().getChildren().get(0);
assertEquals(action1.getBoundNode().getBinding(), _action1.getBoundNode().getBinding());
assertEquals(action1.isModify(), _action1.isModify());
assertEquals(action1.getFieldValues().size(), _action1.getFieldValues().size());
assertEquals(1, _action1.getFieldValues().size());
assertEquals(action1.getFieldValues().get(0).getFieldName(), _action1.getFieldValues().get(0).getFieldName());
assertEquals(action1.getFieldValues().get(0).getValue().getValue().toString(), _action1.getFieldValues().get(0).getValue().getValue().toString());
assertEquals(1, _action1.getChildren().size());
assertNotNull(_action1.getChildren().get(0));
assertTrue(_action1.getChildren().get(0) instanceof ActionRetractNode);
final ActionRetractNode _action2 = (ActionRetractNode) _action1.getChildren().get(0);
assertEquals(action2.getBoundNode().getBinding(), _action2.getBoundNode().getBinding());
}
use of org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionRetractNode in project drools by kiegroup.
the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_ActionRetract.
@Test
public void testSingleRule_ActionRetract() throws Exception {
final String drl = "rule \"test_0\"\n" + "when \n" + " $p : Person( )\n" + "then \n" + " retract( $p );\n" + "end";
final GuidedDecisionTree expected = new GuidedDecisionTree();
expected.setTreeName("test");
final TypeNode type = new TypeNodeImpl("Person");
type.setBinding("$p");
expected.setRoot(type);
final ActionRetractNode action = new ActionRetractNodeImpl(type);
type.addChild(action);
addModelField("Person", "this", "Person", DataType.TYPE_THIS);
addModelField("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());
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 ActionRetractNode);
final ActionRetractNode _a1 = (ActionRetractNode) model.getRoot().getChildren().get(0);
assertEquals(action.getBoundNode().getBinding(), _a1.getBoundNode().getBinding());
}
Aggregations