use of org.drools.workbench.models.datamodel.rule.ActionFieldValue in project drools by kiegroup.
the class GuidedDTDRLPersistenceTest method testRHSWithBRLColumn_ParseToRuleModel.
@Test
public // This test checks a Decision Table involving BRL columns is correctly converted into a RuleModel
void testRHSWithBRLColumn_ParseToRuleModel() {
GuidedDecisionTable52 dtable = new GuidedDecisionTable52();
GuidedDTDRLPersistence p = new GuidedDTDRLPersistence();
// All three rows are entered, some columns with optional data
String[][] data = new String[][] { new String[] { "1", "desc", "Gargamel", "Pupa", "50" }, new String[] { "2", "desc", "Gargamel", "", "50" }, new String[] { "3", "desc", "Gargamel", "Pupa", "" } };
// Simple (mandatory) columns
dtable.setRowNumberCol(new RowNumberCol52());
dtable.setDescriptionCol(new DescriptionCol52());
// Simple Action
ActionInsertFactCol52 a1 = new ActionInsertFactCol52();
a1.setBoundName("$b");
a1.setFactType("Baddie");
a1.setFactField("name");
a1.setType(DataType.TYPE_STRING);
dtable.getActionCols().add(a1);
// BRL Column
BRLActionColumn brl1 = new BRLActionColumn();
// BRL Column definition
List<IAction> brl1Definition = new ArrayList<IAction>();
ActionInsertFact brl1DefinitionAction1 = new ActionInsertFact("Smurf");
ActionFieldValue brl1DefinitionAction1FieldValue1 = new ActionFieldValue("name", "$name", DataType.TYPE_STRING);
brl1DefinitionAction1FieldValue1.setNature(BaseSingleFieldConstraint.TYPE_TEMPLATE);
brl1DefinitionAction1.addFieldValue(brl1DefinitionAction1FieldValue1);
ActionFieldValue brl1DefinitionAction1FieldValue2 = new ActionFieldValue("age", "$age", DataType.TYPE_NUMERIC_INTEGER);
brl1DefinitionAction1FieldValue2.setNature(BaseSingleFieldConstraint.TYPE_TEMPLATE);
brl1DefinitionAction1.addFieldValue(brl1DefinitionAction1FieldValue2);
brl1Definition.add(brl1DefinitionAction1);
brl1.setDefinition(brl1Definition);
// Setup BRL column bindings
BRLActionVariableColumn brl1Variable1 = new BRLActionVariableColumn("$name", DataType.TYPE_STRING, "Person", "name");
brl1.getChildColumns().add(brl1Variable1);
BRLActionVariableColumn brl1Variable2 = new BRLActionVariableColumn("$age", DataType.TYPE_NUMERIC_INTEGER, "Person", "age");
brl1.getChildColumns().add(brl1Variable2);
dtable.getActionCols().add(brl1);
// Now to test conversion
RuleModel rm = new RuleModel();
List<BaseColumn> allColumns = dtable.getExpandedColumns();
List<ActionCol52> allActions = dtable.getActionCols();
// Row 0
List<DTCellValue52> dtRowData0 = DataUtilities.makeDataRowList(data[0]);
TemplateDataProvider rowDataProvider0 = new GuidedDTTemplateDataProvider(allColumns, dtRowData0);
p.doActions(allColumns, allActions, rowDataProvider0, dtRowData0, rm);
assertEquals(2, rm.rhs.length);
assertEquals("Baddie", ((ActionInsertFact) rm.rhs[0]).getFactType());
assertEquals("Smurf", ((ActionInsertFact) rm.rhs[1]).getFactType());
// examine the first action
ActionInsertFact result0Action1 = (ActionInsertFact) rm.rhs[0];
assertEquals(1, result0Action1.getFieldValues().length);
ActionFieldValue result0Action1FieldValue1 = (ActionFieldValue) result0Action1.getFieldValues()[0];
assertEquals(DataType.TYPE_STRING, result0Action1FieldValue1.getType());
assertEquals("name", result0Action1FieldValue1.getField());
assertEquals("Gargamel", result0Action1FieldValue1.getValue());
// examine the second action
ActionInsertFact result0Action2 = (ActionInsertFact) rm.rhs[1];
assertEquals(2, result0Action2.getFieldValues().length);
ActionFieldValue result0Action2FieldValue1 = (ActionFieldValue) result0Action2.getFieldValues()[0];
assertEquals(DataType.TYPE_STRING, result0Action2FieldValue1.getType());
assertEquals("name", result0Action2FieldValue1.getField());
assertEquals("$name", result0Action2FieldValue1.getValue());
ActionFieldValue result0Action2FieldValue2 = (ActionFieldValue) result0Action2.getFieldValues()[1];
assertEquals(DataType.TYPE_NUMERIC_INTEGER, result0Action2FieldValue2.getType());
assertEquals("age", result0Action2FieldValue2.getField());
assertEquals("$age", result0Action2FieldValue2.getValue());
// Row 1
List<DTCellValue52> dtRowData1 = DataUtilities.makeDataRowList(data[1]);
TemplateDataProvider rowDataProvider1 = new GuidedDTTemplateDataProvider(allColumns, dtRowData1);
p.doActions(allColumns, allActions, rowDataProvider1, dtRowData1, rm);
assertEquals(2, rm.rhs.length);
assertEquals("Baddie", ((ActionInsertFact) rm.rhs[0]).getFactType());
assertEquals("Smurf", ((ActionInsertFact) rm.rhs[1]).getFactType());
// examine the first action
ActionInsertFact result1Action1 = (ActionInsertFact) rm.rhs[0];
assertEquals(1, result1Action1.getFieldValues().length);
ActionFieldValue result1Action1FieldValue1 = (ActionFieldValue) result1Action1.getFieldValues()[0];
assertEquals(DataType.TYPE_STRING, result1Action1FieldValue1.getType());
assertEquals("name", result1Action1FieldValue1.getField());
assertEquals("Gargamel", result1Action1FieldValue1.getValue());
// examine the second action
ActionInsertFact result1Action2 = (ActionInsertFact) rm.rhs[1];
assertEquals(2, result1Action2.getFieldValues().length);
ActionFieldValue result1Action2FieldValue1 = (ActionFieldValue) result1Action2.getFieldValues()[0];
assertEquals(DataType.TYPE_STRING, result1Action2FieldValue1.getType());
assertEquals("name", result1Action2FieldValue1.getField());
assertEquals("$name", result1Action2FieldValue1.getValue());
ActionFieldValue result1Action2FieldValue2 = (ActionFieldValue) result1Action2.getFieldValues()[1];
assertEquals(DataType.TYPE_NUMERIC_INTEGER, result1Action2FieldValue2.getType());
assertEquals("age", result1Action2FieldValue2.getField());
assertEquals("$age", result1Action2FieldValue2.getValue());
// Row 2
List<DTCellValue52> dtRowData2 = DataUtilities.makeDataRowList(data[2]);
TemplateDataProvider rowDataProvider2 = new GuidedDTTemplateDataProvider(allColumns, dtRowData2);
p.doActions(allColumns, allActions, rowDataProvider2, dtRowData2, rm);
assertEquals(2, rm.rhs.length);
assertEquals("Baddie", ((ActionInsertFact) rm.rhs[0]).getFactType());
assertEquals("Smurf", ((ActionInsertFact) rm.rhs[1]).getFactType());
// examine the first action
ActionInsertFact result2Action1 = (ActionInsertFact) rm.rhs[0];
assertEquals(1, result2Action1.getFieldValues().length);
ActionFieldValue result2Action1FieldValue1 = (ActionFieldValue) result2Action1.getFieldValues()[0];
assertEquals(DataType.TYPE_STRING, result2Action1FieldValue1.getType());
assertEquals("name", result2Action1FieldValue1.getField());
assertEquals("Gargamel", result2Action1FieldValue1.getValue());
// examine the second action
ActionInsertFact result2Action2 = (ActionInsertFact) rm.rhs[1];
assertEquals(2, result2Action2.getFieldValues().length);
ActionFieldValue result2Action2FieldValue1 = (ActionFieldValue) result2Action2.getFieldValues()[0];
assertEquals(DataType.TYPE_STRING, result2Action2FieldValue1.getType());
assertEquals("name", result2Action2FieldValue1.getField());
assertEquals("$name", result2Action2FieldValue1.getValue());
ActionFieldValue result3Action2FieldValue2 = (ActionFieldValue) result2Action2.getFieldValues()[1];
assertEquals(DataType.TYPE_NUMERIC_INTEGER, result3Action2FieldValue2.getType());
assertEquals("age", result3Action2FieldValue2.getField());
assertEquals("$age", result3Action2FieldValue2.getValue());
}
use of org.drools.workbench.models.datamodel.rule.ActionFieldValue in project drools by kiegroup.
the class RuleTemplateModelDRLPersistenceTest method testActionModifyTwoFieldsFirstTemplateSecondTemplate3.
@Test
public void testActionModifyTwoFieldsFirstTemplateSecondTemplate3() {
TemplateModel m = new TemplateModel();
m.name = "r1";
FactPattern fp = new FactPattern("Person");
fp.setBoundName("$p");
m.addLhsItem(fp);
ActionUpdateField auf1 = new ActionUpdateField("$p");
ActionFieldValue afv0 = new ActionFieldValue();
afv0.setNature(FieldNatureType.TYPE_TEMPLATE);
afv0.setField("field1");
afv0.setValue("$f1");
auf1.addFieldValue(afv0);
ActionFieldValue afv1 = new ActionFieldValue();
afv1.setNature(FieldNatureType.TYPE_TEMPLATE);
afv1.setField("field2");
afv1.setValue("$f2");
auf1.addFieldValue(afv1);
m.addRhsItem(auf1);
String expected = "rule \"r1_0\"\n" + "dialect \"mvel\"\n" + "when\n" + " $p : Person()\n" + "then\n" + " modify( $p ) {\n" + " setField2(\"bar\")\n" + " }\n" + "end";
m.addRow(new String[] { null, "bar" });
checkMarshall(expected, m);
}
use of org.drools.workbench.models.datamodel.rule.ActionFieldValue in project drools by kiegroup.
the class RuleTemplateModelDRLPersistenceTest method testActionInsertFactZeroValues.
@Test
public void testActionInsertFactZeroValues() {
TemplateModel m = new TemplateModel();
m.name = "r1";
FactPattern fp = new FactPattern("Person");
fp.setBoundName("$p");
m.addLhsItem(fp);
ActionInsertFact aif = new ActionInsertFact("Present");
aif.setBoundName("f0");
ActionFieldValue afv0 = new ActionFieldValue();
afv0.setNature(FieldNatureType.TYPE_TEMPLATE);
afv0.setField("field1");
afv0.setValue("$f1");
aif.addFieldValue(afv0);
ActionFieldValue afv1 = new ActionFieldValue();
afv1.setNature(FieldNatureType.TYPE_TEMPLATE);
afv1.setField("field2");
afv1.setValue("$f2");
aif.addFieldValue(afv1);
m.addRhsItem(aif);
String expected = "rule \"r1_0\"\n" + "dialect \"mvel\"\n" + "when\n" + "$p : Person()\n" + "then\n" + "Present f0 = new Present();\n" + "insert(f0);\n" + "end";
m.addRow(new String[] { null, null });
checkMarshall(expected, m);
}
use of org.drools.workbench.models.datamodel.rule.ActionFieldValue in project drools by kiegroup.
the class RuleTemplateModelDRLPersistenceTest method testActionModifyTwoFieldsFirstLiteralSecondTemplate1.
@Test
public void testActionModifyTwoFieldsFirstLiteralSecondTemplate1() {
TemplateModel m = new TemplateModel();
m.name = "r1";
FactPattern fp = new FactPattern("Person");
fp.setBoundName("$p");
m.addLhsItem(fp);
ActionUpdateField auf1 = new ActionUpdateField("$p");
ActionFieldValue afv0 = new ActionFieldValue();
afv0.setNature(FieldNatureType.TYPE_LITERAL);
afv0.setField("field1");
afv0.setValue("foo");
auf1.addFieldValue(afv0);
ActionFieldValue afv1 = new ActionFieldValue();
afv1.setNature(FieldNatureType.TYPE_TEMPLATE);
afv1.setField("field2");
afv1.setValue("$f2");
auf1.addFieldValue(afv1);
m.addRhsItem(auf1);
String expected = "rule \"r1_0\"\n" + "dialect \"mvel\"\n" + "when\n" + " $p : Person()\n" + "then\n" + " modify( $p ) {\n" + " setField1(\"foo\"),\n" + " setField2(\"bar\")\n" + " }\n" + "end";
m.addRow(new String[] { "bar" });
checkMarshall(expected, m);
}
use of org.drools.workbench.models.datamodel.rule.ActionFieldValue in project drools by kiegroup.
the class RuleTemplateModelDRLPersistenceTest method testActionUpdateFactFirstValue.
@Test
public void testActionUpdateFactFirstValue() {
TemplateModel m = new TemplateModel();
m.name = "r1";
FactPattern fp = new FactPattern("Person");
fp.setBoundName("$p");
m.addLhsItem(fp);
ActionSetField asf = new ActionSetField("$p");
ActionFieldValue afv0 = new ActionFieldValue();
afv0.setNature(FieldNatureType.TYPE_TEMPLATE);
afv0.setField("field1");
afv0.setValue("$f1");
asf.addFieldValue(afv0);
ActionFieldValue afv1 = new ActionFieldValue();
afv1.setNature(FieldNatureType.TYPE_TEMPLATE);
afv1.setField("field2");
afv1.setValue("$f2");
asf.addFieldValue(afv1);
m.addRhsItem(asf);
String expected = "rule \"r1_0\"\n" + "dialect \"mvel\"\n" + "when\n" + "$p : Person()\n" + "then\n" + "$p.setField1(\"foo\");\n" + "end";
m.addRow(new String[] { "foo", null });
checkMarshall(expected, m);
}
Aggregations