Search in sources :

Example 16 with ConstraintNodeImpl

use of org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl in project drools by kiegroup.

the class GuidedDecisionTreeDRLPersistenceMarshallingTest method testValue_Byte.

@Test
public void testValue_Byte() throws Exception {
    final String expected = "rule \"test_0\"" + "when\n" + "  Person( byteField == 100 )\n" + "then\n" + "end";
    final GuidedDecisionTree model = new GuidedDecisionTree();
    model.setTreeName("test");
    final TypeNode type = new TypeNodeImpl("Person");
    final ConstraintNode c1 = new ConstraintNodeImpl("Person", "byteField", "==", new ByteValue(new Byte("100")));
    model.setRoot(type);
    type.addChild(c1);
    final String drl = GuidedDecisionTreeDRLPersistence.getInstance().marshal(model);
    assertEqualsIgnoreWhitespace(expected, drl);
}
Also used : ConstraintNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl) ByteValue(org.drools.workbench.models.guided.dtree.shared.model.values.impl.ByteValue) 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) Test(org.junit.Test)

Example 17 with ConstraintNodeImpl

use of org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl in project drools by kiegroup.

the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testMultipleRules_MultiplePatterns.

@Test
public void testMultipleRules_MultiplePatterns() throws Exception {
    final String drl = "rule \"test_0\"\n" + "when \n" + "  Person( name == \"Michael\" )\n" + "  Address( country == \"England\" )\n" + "then \n" + "end \n" + "rule \"test_1\"\n" + "when \n" + "  Person( name == \"Michael\" )\n" + "  Address( country == \"Norway\" )\n" + "then \n" + "end \n" + "rule \"test_2\"\n" + "when \n" + "  Person( name == \"Fred\" )\n" + "then \n" + "end \n";
    final GuidedDecisionTree expected = new GuidedDecisionTree();
    expected.setTreeName("test");
    final TypeNode type1 = new TypeNodeImpl("Person");
    final ConstraintNode c1a = new ConstraintNodeImpl("Person", "name", "==", new StringValue("Michael"));
    final ConstraintNode c1b = new ConstraintNodeImpl("Person", "name", "==", new StringValue("Fred"));
    final TypeNode type2 = new TypeNodeImpl("Address");
    final ConstraintNode c2a = new ConstraintNodeImpl("Address", "country", "==", new StringValue("England"));
    final ConstraintNode c2b = new ConstraintNodeImpl("Address", "country", "==", new StringValue("Norway"));
    expected.setRoot(type1);
    type1.addChild(c1a);
    type1.addChild(c1b);
    c1a.addChild(type2);
    type2.addChild(c2a);
    type2.addChild(c2b);
    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(2, _t1.getChildren().size());
    assertNotNull(_t1.getChildren().get(0));
    assertTrue(_t1.getChildren().get(0) instanceof ConstraintNode);
    final ConstraintNode _c1a = (ConstraintNode) _t1.getChildren().get(0);
    assertEquals(c1a.getClassName(), _c1a.getClassName());
    assertEquals(c1a.getFieldName(), _c1a.getFieldName());
    assertEquals(c1a.getOperator(), _c1a.getOperator());
    assertEquals(c1a.getValue().getValue().toString(), _c1a.getValue().getValue().toString());
    assertNotNull(_t1.getChildren().get(1));
    assertTrue(_t1.getChildren().get(1) instanceof ConstraintNode);
    final ConstraintNode _c1b = (ConstraintNode) _t1.getChildren().get(1);
    assertEquals(c1b.getClassName(), _c1b.getClassName());
    assertEquals(c1b.getFieldName(), _c1b.getFieldName());
    assertEquals(c1b.getOperator(), _c1b.getOperator());
    assertEquals(c1b.getValue().getValue().toString(), _c1b.getValue().getValue().toString());
    assertEquals(1, _c1a.getChildren().size());
    assertNotNull(_c1a.getChildren().get(0));
    assertTrue(_c1a.getChildren().get(0) instanceof TypeNode);
    final TypeNode _t2 = (TypeNode) _c1a.getChildren().get(0);
    assertEquals(type2.getClassName(), _t2.getClassName());
    assertFalse(_t2.isBound());
    assertEquals(2, _t2.getChildren().size());
    assertNotNull(_t2.getChildren().get(0));
    assertTrue(_t2.getChildren().get(0) instanceof ConstraintNode);
    final ConstraintNode _c2a = (ConstraintNode) _t2.getChildren().get(0);
    assertEquals(c2a.getClassName(), _c2a.getClassName());
    assertEquals(c2a.getFieldName(), _c2a.getFieldName());
    assertEquals(c2a.getOperator(), _c2a.getOperator());
    assertEquals(c2a.getValue().getValue().toString(), _c2a.getValue().getValue().toString());
    assertNotNull(_t2.getChildren().get(1));
    assertTrue(_t2.getChildren().get(1) instanceof ConstraintNode);
    final ConstraintNode _c2b = (ConstraintNode) _t2.getChildren().get(1);
    assertEquals(c2b.getClassName(), _c2b.getClassName());
    assertEquals(c2b.getFieldName(), _c2b.getFieldName());
    assertEquals(c2b.getOperator(), _c2b.getOperator());
    assertEquals(c2b.getValue().getValue().toString(), _c2b.getValue().getValue().toString());
    assertEquals(0, _c2a.getChildren().size());
    assertEquals(0, _c2b.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 18 with ConstraintNodeImpl

use of org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl in project drools by kiegroup.

the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testMultipleRules_2Rules_1Simple_1Complex.

@Test
public void testMultipleRules_2Rules_1Simple_1Complex() 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( name == \"Fred\", age == 41 )\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", "name", "==", new StringValue("Fred"));
    final ConstraintNode c3 = new ConstraintNodeImpl("Person", "age", "==", new IntegerValue(41));
    expected.setRoot(type);
    type.addChild(c1);
    type.addChild(c2);
    c2.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);
    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(2, _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());
    assertEquals(0, _c1.getChildren().size());
    assertEquals(1, _c2.getChildren().size());
    assertNotNull(_c2.getChildren().get(0));
    assertTrue(_c2.getChildren().get(0) instanceof ConstraintNode);
    final ConstraintNode _c3 = (ConstraintNode) _c2.getChildren().get(0);
    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, _c3.getChildren().size());
}
Also used : ConstraintNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl) BigInteger(java.math.BigInteger) ConstraintNode(org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode) TypeNodeImpl(org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.TypeNodeImpl) IntegerValue(org.drools.workbench.models.guided.dtree.shared.model.values.impl.IntegerValue) BigIntegerValue(org.drools.workbench.models.guided.dtree.shared.model.values.impl.BigIntegerValue) 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 19 with ConstraintNodeImpl

use of org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl in project drools by kiegroup.

the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_SingleConstraintNoOperatorNoValue.

@Test
public void testSingleRule_SingleConstraintNoOperatorNoValue() throws Exception {
    final String drl = "rule \"test_0\"\n" + "when \n" + "  Person( name )\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");
    expected.setRoot(type);
    type.addChild(c1);
    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());
    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());
    assertNull(_c1.getOperator());
    assertNull(_c1.getValue());
}
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) Test(org.junit.Test)

Example 20 with ConstraintNodeImpl

use of org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl in project drools by kiegroup.

the class GuidedDecisionTreeDRLPersistenceUnmarshallingTest method testSingleRule_SingleConstraintNotNullOperator.

@Test
public void testSingleRule_SingleConstraintNotNullOperator() throws Exception {
    final String drl = "rule \"test_0\"\n" + "when \n" + "  Person( name != null )\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", "!= null", null);
    expected.setRoot(type);
    type.addChild(c1);
    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());
    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());
    assertNull(_c1.getValue());
}
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) Test(org.junit.Test)

Aggregations

ConstraintNode (org.drools.workbench.models.guided.dtree.shared.model.nodes.ConstraintNode)56 ConstraintNodeImpl (org.drools.workbench.models.guided.dtree.shared.model.nodes.impl.ConstraintNodeImpl)56 TypeNode (org.drools.workbench.models.guided.dtree.shared.model.nodes.TypeNode)54 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)5 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 ActionRetractNode (org.drools.workbench.models.guided.dtree.shared.model.nodes.ActionRetractNode)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 DoubleValue (org.drools.workbench.models.guided.dtree.shared.model.values.impl.DoubleValue)3 EnumValue (org.drools.workbench.models.guided.dtree.shared.model.values.impl.EnumValue)3