use of org.drools.workbench.models.datamodel.rule.IAction in project drools-wb by kiegroup.
the class GuidedDecisionTableGeneratorListenerTest method testMissingTemplateKeyValues_NonStringFields.
@Test
public // https://issues.jboss.org/browse/GUVNOR-2030
void testMissingTemplateKeyValues_NonStringFields() {
final ConversionResult result = new ConversionResult();
final List<DataListener> listeners = new ArrayList<DataListener>();
addModelField("org.test.Client", "this", "org.test.Client", DataType.TYPE_THIS);
addModelField("org.test.Client", "monthlyTransactions", Integer.class.getName(), DataType.TYPE_NUMERIC_INTEGER);
final GuidedDecisionTableGeneratorListener listener = new GuidedDecisionTableGeneratorListener(result, dmo);
listeners.add(listener);
// Convert
final ExcelParser parser = new ExcelParser(listeners);
final InputStream is = this.getClass().getResourceAsStream("GUVNOR-2030 (DecisionTable).xls");
try {
parser.parseFile(is);
} finally {
try {
is.close();
} catch (IOException ioe) {
fail(ioe.getMessage());
}
}
// Check conversion results
assertEquals(0, result.getMessages().size());
// Check basics
final List<GuidedDecisionTable52> dtables = listener.getGuidedDecisionTables();
assertNotNull(dtables);
assertEquals(1, dtables.size());
GuidedDecisionTable52 dtable = dtables.get(0);
assertEquals("Steps", dtable.getTableName());
assertEquals(GuidedDecisionTable52.TableFormat.EXTENDED_ENTRY, dtable.getTableFormat());
// Check expanded columns
List<BaseColumn> columns = dtable.getExpandedColumns();
assertNotNull(columns);
assertEquals(5, columns.size());
assertTrue(columns.get(0) instanceof RowNumberCol52);
assertTrue(columns.get(1) instanceof DescriptionCol52);
assertTrue(columns.get(2) instanceof BRLConditionVariableColumn);
assertTrue(columns.get(3) instanceof BRLConditionVariableColumn);
assertTrue(columns.get(4) instanceof BRLActionVariableColumn);
// Check individual condition columns
assertEquals(1, dtable.getConditions().size());
assertTrue(dtable.getConditions().get(0) instanceof BRLConditionColumn);
// Column 1
BRLConditionColumn conditionCol0 = ((BRLConditionColumn) dtable.getConditions().get(0));
assertEquals("Converted from ['From', 'To']", conditionCol0.getHeader());
assertEquals(2, conditionCol0.getChildColumns().size());
List<IPattern> conditionCol0definition = conditionCol0.getDefinition();
assertEquals(1, conditionCol0definition.size());
assertTrue(conditionCol0definition.get(0) instanceof FactPattern);
FactPattern conditionCol0fp = (FactPattern) conditionCol0definition.get(0);
assertEquals("Client", conditionCol0fp.getFactType());
assertEquals(2, conditionCol0fp.getNumberOfConstraints());
assertTrue(conditionCol0fp.getConstraint(0) instanceof SingleFieldConstraint);
final SingleFieldConstraint conditionCol0fpsfc0 = (SingleFieldConstraint) conditionCol0fp.getConstraint(0);
assertEquals("monthlyTransactions", conditionCol0fpsfc0.getFieldName());
assertEquals(">=", conditionCol0fpsfc0.getOperator());
assertEquals("param1", conditionCol0fpsfc0.getValue());
assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol0fpsfc0.getConstraintValueType());
assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0fpsfc0.getFieldType());
assertTrue(conditionCol0fp.getConstraint(1) instanceof SingleFieldConstraint);
final SingleFieldConstraint conditionCol0fpsfc1 = (SingleFieldConstraint) conditionCol0fp.getConstraint(1);
assertEquals("monthlyTransactions", conditionCol0fpsfc1.getFieldName());
assertEquals("<=", conditionCol0fpsfc1.getOperator());
assertEquals("param2", conditionCol0fpsfc1.getValue());
assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol0fpsfc1.getConstraintValueType());
assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0fpsfc1.getFieldType());
// Column 1 - Variable 1
BRLConditionVariableColumn conditionCol0param0 = conditionCol0.getChildColumns().get(0);
assertEquals("param1", conditionCol0param0.getVarName());
assertEquals("From", conditionCol0param0.getHeader());
assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0param0.getFieldType());
assertEquals("Client", conditionCol0param0.getFactType());
assertEquals("monthlyTransactions", conditionCol0param0.getFactField());
// Column 1 - Variable 2
BRLConditionVariableColumn conditionCol0param1 = conditionCol0.getChildColumns().get(1);
assertEquals("param2", conditionCol0param1.getVarName());
assertEquals("To", conditionCol0param1.getHeader());
assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0param1.getFieldType());
assertEquals("Client", conditionCol0param1.getFactType());
assertEquals("monthlyTransactions", conditionCol0param1.getFactField());
// Check individual action columns
assertEquals(1, dtable.getActionCols().size());
assertTrue(dtable.getActionCols().get(0) instanceof BRLActionColumn);
// Column 2
BRLActionColumn actionCol0 = ((BRLActionColumn) dtable.getActionCols().get(0));
assertEquals("Converted from ['Step']", actionCol0.getHeader());
assertEquals(1, actionCol0.getChildColumns().size());
List<IAction> actionCol0definition = actionCol0.getDefinition();
assertEquals(1, actionCol0definition.size());
assertTrue(actionCol0definition.get(0) instanceof FreeFormLine);
FreeFormLine actionCol0ffl = (FreeFormLine) actionCol0definition.get(0);
assertEquals("if (!$c.isPricingStepCustom() && ($c.getPricingStep() == null || $c.getPricingStep().compareTo(\"@{param3}\") < 0)) { modify($c) { setPricingStep(\"@{param3}\"); } };", actionCol0ffl.getText());
// Check data
assertEquals(1, dtable.getData().size());
assertTrue(isRowEquivalent(new Object[] { 1, "Created from row 12", 0, 100, "A1" }, dtable.getData().get(0)));
}
use of org.drools.workbench.models.datamodel.rule.IAction in project drools-wb by kiegroup.
the class GuidedDecisionTableGeneratorListenerTest method testMissingTemplateKeyValues_RHSInsertThenUpdate.
@Test
public // https://issues.jboss.org/browse/GUVNOR-2030
void testMissingTemplateKeyValues_RHSInsertThenUpdate() {
final ConversionResult result = new ConversionResult();
final List<DataListener> listeners = new ArrayList<DataListener>();
addModelField("org.test.Transaction", "this", "org.test.Transaction", DataType.TYPE_THIS);
addModelField("org.test.Transaction", "enabled", Boolean.class.getName(), DataType.TYPE_BOOLEAN);
final GuidedDecisionTableGeneratorListener listener = new GuidedDecisionTableGeneratorListener(result, dmo);
listeners.add(listener);
// Convert
final ExcelParser parser = new ExcelParser(listeners);
final InputStream is = this.getClass().getResourceAsStream("GUVNOR-2030 (RHS insert then update).xls");
try {
parser.parseFile(is);
} finally {
try {
is.close();
} catch (IOException ioe) {
fail(ioe.getMessage());
}
}
// Check conversion results
assertEquals(0, result.getMessages().size());
// Check basics
final List<GuidedDecisionTable52> dtables = listener.getGuidedDecisionTables();
assertNotNull(dtables);
assertEquals(1, dtables.size());
GuidedDecisionTable52 dtable = dtables.get(0);
assertEquals("asd", dtable.getTableName());
assertEquals(GuidedDecisionTable52.TableFormat.EXTENDED_ENTRY, dtable.getTableFormat());
// Check expanded columns
List<BaseColumn> columns = dtable.getExpandedColumns();
assertNotNull(columns);
assertEquals(6, columns.size());
assertTrue(columns.get(0) instanceof RowNumberCol52);
assertTrue(columns.get(1) instanceof DescriptionCol52);
assertTrue(columns.get(2) instanceof BRLConditionVariableColumn);
assertTrue(columns.get(3) instanceof BRLActionVariableColumn);
assertTrue(columns.get(4) instanceof BRLActionVariableColumn);
assertTrue(columns.get(5) instanceof BRLActionVariableColumn);
// Check individual condition columns
assertEquals(1, dtable.getConditions().size());
assertTrue(dtable.getConditions().get(0) instanceof BRLConditionColumn);
// Column 1
BRLConditionColumn conditionCol0 = ((BRLConditionColumn) dtable.getConditions().get(0));
assertEquals("Converted from ['Disabled']", conditionCol0.getHeader());
assertEquals(1, conditionCol0.getChildColumns().size());
List<IPattern> conditionCol0definition = conditionCol0.getDefinition();
assertEquals(1, conditionCol0definition.size());
assertTrue(conditionCol0definition.get(0) instanceof FactPattern);
FactPattern conditionCol0fp = (FactPattern) conditionCol0definition.get(0);
assertEquals("Transaction", conditionCol0fp.getFactType());
assertEquals(1, conditionCol0fp.getNumberOfConstraints());
assertTrue(conditionCol0fp.getConstraint(0) instanceof SingleFieldConstraint);
final SingleFieldConstraint conditionCol0fpsfc0 = (SingleFieldConstraint) conditionCol0fp.getConstraint(0);
assertEquals("enabled", conditionCol0fpsfc0.getFieldName());
assertEquals("==", conditionCol0fpsfc0.getOperator());
assertEquals("param1", conditionCol0fpsfc0.getValue());
assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol0fpsfc0.getConstraintValueType());
assertEquals(DataType.TYPE_BOOLEAN, conditionCol0fpsfc0.getFieldType());
// Column 1 - Variable 1
BRLConditionVariableColumn conditionCol0param0 = conditionCol0.getChildColumns().get(0);
assertEquals("param1", conditionCol0param0.getVarName());
assertEquals("Disabled", conditionCol0param0.getHeader());
assertEquals(DataType.TYPE_BOOLEAN, conditionCol0param0.getFieldType());
assertEquals("Transaction", conditionCol0param0.getFactType());
assertEquals("enabled", conditionCol0param0.getFactField());
// Check individual action columns
assertEquals(3, dtable.getActionCols().size());
assertTrue(dtable.getActionCols().get(0) instanceof BRLActionColumn);
// Column 2
BRLActionColumn actionCol0 = ((BRLActionColumn) dtable.getActionCols().get(0));
assertEquals("Converted from ['Enable']", actionCol0.getHeader());
assertEquals(1, actionCol0.getChildColumns().size());
List<IAction> actionCol0definition = actionCol0.getDefinition();
assertEquals(1, actionCol0definition.size());
assertTrue(actionCol0definition.get(0) instanceof ActionSetField);
ActionSetField actionCol0asf0 = (ActionSetField) actionCol0definition.get(0);
assertEquals("$t", actionCol0asf0.getVariable());
ActionFieldValue[] actionCol0asf0afvs = actionCol0asf0.getFieldValues();
assertEquals(1, actionCol0asf0afvs.length);
ActionFieldValue actionCol0asf0afv0 = actionCol0asf0afvs[0];
assertEquals("enabled", actionCol0asf0afv0.getField());
assertEquals("param2", actionCol0asf0afv0.getValue());
assertEquals(FieldNatureType.TYPE_TEMPLATE, actionCol0asf0afv0.getNature());
assertEquals(DataType.TYPE_BOOLEAN, actionCol0asf0afv0.getType());
// Column 3
BRLActionColumn actionCol1 = ((BRLActionColumn) dtable.getActionCols().get(1));
assertEquals("Converted from ['New']", actionCol1.getHeader());
assertEquals(1, actionCol1.getChildColumns().size());
List<IAction> actionCol1definition = actionCol1.getDefinition();
assertEquals(1, actionCol1definition.size());
assertTrue(actionCol1definition.get(0) instanceof ActionInsertFact);
ActionInsertFact actionCol1aif0 = (ActionInsertFact) actionCol1definition.get(0);
assertEquals("$t2", actionCol1aif0.getBoundName());
assertEquals("Transaction", actionCol1aif0.getFactType());
ActionFieldValue[] actionCol1aif0afvs = actionCol1aif0.getFieldValues();
assertEquals(0, actionCol1aif0afvs.length);
// Column 4
BRLActionColumn actionCol2 = ((BRLActionColumn) dtable.getActionCols().get(2));
assertEquals("Converted from ['Disable']", actionCol2.getHeader());
assertEquals(1, actionCol2.getChildColumns().size());
List<IAction> actionCol2definition = actionCol2.getDefinition();
assertEquals(1, actionCol2definition.size());
assertTrue(actionCol2definition.get(0) instanceof ActionSetField);
ActionSetField actionCol2asf0 = (ActionSetField) actionCol2definition.get(0);
assertEquals("$t2", actionCol2asf0.getVariable());
ActionFieldValue[] actionCol2asf0afvs = actionCol2asf0.getFieldValues();
assertEquals(1, actionCol2asf0afvs.length);
ActionFieldValue actionCol2asf0afv0 = actionCol2asf0afvs[0];
assertEquals("enabled", actionCol2asf0afv0.getField());
assertEquals("param3", actionCol2asf0afv0.getValue());
assertEquals(FieldNatureType.TYPE_TEMPLATE, actionCol2asf0afv0.getNature());
assertEquals(DataType.TYPE_BOOLEAN, actionCol2asf0afv0.getType());
// Check data
assertEquals(1, dtable.getData().size());
assertTrue(isRowEquivalent(new Object[] { 1, "asd", false, true, false, false }, dtable.getData().get(0)));
}
use of org.drools.workbench.models.datamodel.rule.IAction in project drools-wb by kiegroup.
the class GuidedDecisionTablePopulater method addIndirectSourceBuildersColumns.
private void addIndirectSourceBuildersColumns() {
final List<BRLVariableColumn> variableColumns = new ArrayList<BRLVariableColumn>();
for (GuidedDecisionTableSourceBuilder sb : sourceBuilders) {
if (sb instanceof GuidedDecisionTableSourceBuilderIndirect) {
for (BRLVariableColumn variableColumn : ((GuidedDecisionTableSourceBuilderIndirect) sb).getVariableColumns()) {
variableColumns.add(variableColumn);
}
}
}
// Convert the DRL to a RuleModel from which we construct BRLFragment columns
final StringBuilder rule = new StringBuilder();
if (!(dmo.getPackageName() == null || dmo.getPackageName().isEmpty())) {
rule.append("package ").append(dmo.getPackageName()).append("\n");
}
rule.append("rule 'temp' \n").append("when \n");
for (GuidedDecisionTableSourceBuilder sb : sourceBuilders) {
if (sb instanceof GuidedDecisionTableLHSBuilder) {
rule.append(sb.getResult());
}
}
rule.append("\nthen \n");
for (GuidedDecisionTableSourceBuilder sb : sourceBuilders) {
if (sb instanceof GuidedDecisionTableRHSBuilder) {
rule.append(sb.getResult());
}
}
rule.append("end");
final RuleModel rm = RuleModelDRLPersistenceImpl.getInstance().unmarshal(rule.toString(), Collections.emptyList(), dmo);
if (rm.lhs != null) {
for (IPattern pattern : rm.lhs) {
final BRLConditionColumn column = new BRLConditionColumn();
column.getDefinition().add(pattern);
dtable.getConditions().add(column);
final Map<InterpolationVariable, Integer> templateKeys = new HashMap<>();
final RuleModelVisitor rmv = new RuleModelVisitor(templateKeys);
rmv.visit(pattern);
final List<InterpolationVariable> ivs = new ArrayList<>(templateKeys.keySet());
for (BRLVariableColumn variableColumn : variableColumns) {
final Iterator<InterpolationVariable> ivsIts = ivs.iterator();
while (ivsIts.hasNext()) {
final InterpolationVariable iv = ivsIts.next();
if (iv.getVarName().equals(variableColumn.getVarName())) {
final BRLConditionVariableColumn source = (BRLConditionVariableColumn) variableColumn;
final BRLConditionVariableColumn target = makeBRLConditionVariableColumn(source, iv);
column.getChildColumns().add(target);
ivsIts.remove();
}
}
}
if (column.getChildColumns().isEmpty()) {
setZeroParameterConditionColumnHeader(column, variableColumns);
} else {
setCompositeColumnHeader(column);
}
}
}
if (rm.rhs != null) {
for (IAction action : rm.rhs) {
final BRLActionColumn column = new BRLActionColumn();
column.getDefinition().add(action);
dtable.getActionCols().add(column);
if (action instanceof ActionRetractFact) {
final ActionRetractFact arf = (ActionRetractFact) action;
createActionRetractFactChildColumns(arf, column, variableColumns);
} else {
createRHSChildColumns(rm, action, column, variableColumns);
}
if (column.getChildColumns().isEmpty()) {
setZeroParameterActionColumnHeader(column, variableColumns);
} else {
setCompositeColumnHeader(column);
}
}
}
}
use of org.drools.workbench.models.datamodel.rule.IAction in project drools-wb by kiegroup.
the class RuleModelCloneVisitorTest method testRuleModelCloning.
@Test
public void testRuleModelCloning() {
RuleModel model = new RuleModel();
// --------------------------------------------------------------------------------------------------------------
// LHS
// --------------------------------------------------------------------------------------------------------------
//
RuleAttribute attribute = new RuleAttribute("att1_name", "att1_val");
model.addAttribute(attribute);
RuleMetadata metadata = new RuleMetadata("meta1_name", "meta1_val");
model.addMetadata(metadata);
FactPattern f1 = buildFactPattern();
FactPattern f2 = new FactPattern();
f2.setBoundName("$f2");
f2.setFactType("Fact2");
f2.addConstraint(buildCompositeFieldConstraint());
CompositeFactPattern composite = new CompositeFactPattern();
composite.setType(CompositeFactPattern.COMPOSITE_TYPE_EXISTS);
composite.addFactPattern(f1);
composite.addFactPattern(f2);
model.addLhsItem(composite);
model.addLhsItem(buildFromAccumulateCompositeFactPattern());
model.addLhsItem(buildFromCollectCompositeFactPattern());
model.addLhsItem(buildFromCompositeFactPattern());
model.addLhsItem(buildFreeFormLine());
model.addLhsItem(buildDslSentence());
ActionGlobalCollectionAdd addGlobal = new ActionGlobalCollectionAdd();
addGlobal.setFactName("MyFact");
addGlobal.setGlobalName("glist");
model.addRhsItem(addGlobal);
// --------------------------------------------------------------------------------------------------------------
// RHS
// --------------------------------------------------------------------------------------------------------------
//
ActionInsertFact aif = new ActionInsertFact();
aif.setBoundName("$f");
aif.setFactType("FactType");
aif.addFieldValue(buildActionFieldValue());
aif.addFieldValue(buildActionFieldValue());
model.addRhsItem(aif);
ActionUpdateField auf = new ActionUpdateField();
auf.setVariable("$var");
auf.addFieldValue(buildActionFieldValue());
auf.addFieldValue(buildActionFieldValue());
model.addRhsItem(auf);
ActionSetField asf = new ActionSetField();
asf.setVariable("$var");
asf.addFieldValue(buildActionFieldValue());
asf.addFieldValue(buildActionFieldValue());
model.addRhsItem(asf);
ActionRetractFact arf = new ActionRetractFact();
arf.setVariableName("$f");
model.addRhsItem(arf);
ActionCallMethod callMethod1 = new ActionCallMethod();
callMethod1.setVariable("var1");
callMethod1.setMethodName("testMethod1 ");
callMethod1.setState(ActionCallMethod.TYPE_DEFINED);
callMethod1.addFieldValue(new ActionFieldFunction("field1", "value1", "type1"));
model.addRhsItem(callMethod1);
ActionCallMethod callMethod2 = new ActionCallMethod();
callMethod2.setVariable("var2");
callMethod2.setMethodName("testMethod2 ");
callMethod2.setState(ActionCallMethod.TYPE_UNDEFINED);
callMethod2.addFieldValue(new ActionFieldFunction("field2", "value2", "type2"));
model.addRhsItem(callMethod2);
model.addRhsItem(buildFreeFormLine());
// --------------------------------------------------------------------------------------------------------------
// Clone and verify
// --------------------------------------------------------------------------------------------------------------
//
RuleModel clone = new RuleModelCloneVisitor().visitRuleModel(model);
assertArrayEquals(model.attributes, clone.attributes);
int attIndex = 0;
for (RuleAttribute att : model.attributes) {
assertNotSame(att, clone.attributes[attIndex++]);
}
assertArrayEquals(model.metadataList, clone.metadataList);
int metIndex = 0;
for (RuleMetadata met : model.metadataList) {
assertNotSame(met, clone.metadataList[metIndex++]);
}
assertArrayEquals(model.lhs, clone.lhs);
int lhsIndex = 0;
for (IPattern pattern : model.lhs) {
assertNotSame(pattern, clone.lhs[lhsIndex++]);
}
assertArrayEquals(model.rhs, clone.rhs);
int rhsIndex = 0;
for (IAction action : model.rhs) {
assertNotSame(action, clone.rhs[rhsIndex++]);
}
}
use of org.drools.workbench.models.datamodel.rule.IAction in project drools-wb by kiegroup.
the class BRLActionColumnPluginTest method testSetupEditingColWhenColumnIsNotNew.
@Test
public void testSetupEditingColWhenColumnIsNotNew() throws Exception {
final BRLActionColumn originalColumn = mock(BRLActionColumn.class);
final BRLActionColumn editingColumn = mock(BRLActionColumn.class);
final String header = "header";
final ArrayList<IAction> definition = new ArrayList<>();
final ArrayList<BRLActionVariableColumn> childColumns = new ArrayList<>();
final boolean isHideColumn = false;
doReturn(false).when(plugin).isNewColumn();
doReturn(editingColumn).when(plugin).newBRLActionColumn();
doReturn(originalColumn).when(plugin).getOriginalColumnConfig52();
doReturn(header).when(originalColumn).getHeader();
doReturn(definition).when(originalColumn).getDefinition();
doReturn(childColumns).when(originalColumn).getChildColumns();
doReturn(isHideColumn).when(originalColumn).isHideColumn();
plugin.setupEditingCol();
verify(editingColumn).setHeader(header);
verify(editingColumn).setDefinition(definition);
verify(editingColumn).setChildColumns(childColumns);
verify(editingColumn).setHideColumn(isHideColumn);
assertEquals(editingColumn, plugin.editingCol());
}
Aggregations