Search in sources :

Example 11 with ConstraintNode

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());
}
Also used : ConstraintNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl) ConstraintNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode) TypeNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.TypeNodeImpl) EnumValue(org.drools.workbench.models.guided.dtree.shared.model.values.impl.EnumValue) GuidedDecisionTree(org.drools.workbench.models.guided.dtree.shared.model.GuidedDecisionTree) TypeNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode) Test(org.junit.Test)

Example 12 with ConstraintNode

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());
}
Also used : ConstraintNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl) ConstraintNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode) TypeNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.TypeNodeImpl) GuidedDecisionTree(org.drools.workbench.models.guided.dtree.shared.model.GuidedDecisionTree) TypeNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode) StringValue(org.drools.workbench.models.guided.dtree.shared.model.values.impl.StringValue) Test(org.junit.Test)

Example 13 with ConstraintNode

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());
}
Also used : ConstraintNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl) ConstraintNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode) DoubleValue(org.drools.workbench.models.guided.dtree.shared.model.values.impl.DoubleValue) TypeNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.TypeNodeImpl) GuidedDecisionTree(org.drools.workbench.models.guided.dtree.shared.model.GuidedDecisionTree) TypeNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode) Test(org.junit.Test)

Example 14 with ConstraintNode

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());
}
Also used : ConstraintNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl) ConstraintNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode) TypeNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.TypeNodeImpl) GuidedDecisionTree(org.drools.workbench.models.guided.dtree.shared.model.GuidedDecisionTree) TypeNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode) StringValue(org.drools.workbench.models.guided.dtree.shared.model.values.impl.StringValue) Test(org.junit.Test)

Example 15 with ConstraintNode

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");
    }
}
Also used : ConstraintNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode) ActionFieldValue(org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionFieldValue) TypeNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode)

Aggregations

ConstraintNode (org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode)60 TypeNode (org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode)58 ConstraintNodeImpl (org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl)56 TypeNodeImpl (org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.TypeNodeImpl)53 GuidedDecisionTree (org.drools.workbench.models.guided.dtree.shared.model.GuidedDecisionTree)52 Test (org.junit.Test)52 StringValue (org.drools.workbench.models.guided.dtree.shared.model.values.impl.StringValue)20 BigIntegerValue (org.drools.workbench.models.guided.dtree.shared.model.values.impl.BigIntegerValue)17 IntegerValue (org.drools.workbench.models.guided.dtree.shared.model.values.impl.IntegerValue)15 BigInteger (java.math.BigInteger)9 ActionUpdateNode (org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionUpdateNode)8 ActionRetractNode (org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionRetractNode)6 ActionInsertNode (org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionInsertNode)4 ActionFieldValueImpl (org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ActionFieldValueImpl)4 ActionUpdateNodeImpl (org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ActionUpdateNodeImpl)4 Node (org.drools.workbench.models.guided.dtree.shared.model.nodes.Node)3 BigDecimalValue (org.drools.workbench.models.guided.dtree.shared.model.values.impl.BigDecimalValue)3 BooleanValue (org.drools.workbench.models.guided.dtree.shared.model.values.impl.BooleanValue)3 ByteValue (org.drools.workbench.models.guided.dtree.shared.model.values.impl.ByteValue)3 DateValue (org.drools.workbench.models.guided.dtree.shared.model.values.impl.DateValue)3