Search in sources :

Example 56 with BRLConditionVariableColumn

use of org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn in project drools-wb by kiegroup.

the class BRLConditionColumnPluginTest method testCloneVariable.

@Test
public void testCloneVariable() throws Exception {
    final BRLConditionVariableColumn variable = makeVariable("variableName", "variableFieldType", "variableFactType", "variableFactField", "variableHeader", false, 999);
    final BRLConditionVariableColumn clone = plugin.cloneVariable(variable);
    assertEquals(variable.getVarName(), clone.getVarName());
    assertEquals(variable.getFieldType(), clone.getFieldType());
    assertEquals(variable.getFactType(), clone.getFactType());
    assertEquals(variable.getFactField(), clone.getFactField());
    assertEquals(variable.getHeader(), clone.getHeader());
    assertEquals(variable.isHideColumn(), clone.isHideColumn());
    assertEquals(variable.getWidth(), clone.getWidth());
    assertNotSame(variable, clone);
}
Also used : BRLConditionVariableColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn) Test(org.junit.Test)

Example 57 with BRLConditionVariableColumn

use of org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn in project drools-wb by kiegroup.

the class BRLConditionColumnDefinitionBuilderTest method setupBRLConditionColumn.

private void setupBRLConditionColumn() {
    final BRLConditionColumn brl = new BRLConditionColumn();
    final FactPattern fp1 = new FactPattern();
    fp1.setFactType("Person");
    final SingleFieldConstraint sfc1 = new SingleFieldConstraint();
    sfc1.setConstraintValueType(BaseSingleFieldConstraint.TYPE_TEMPLATE);
    sfc1.setFactType("Person");
    sfc1.setOperator("==");
    sfc1.setFieldName("name");
    sfc1.setValue("f1");
    fp1.addConstraint(sfc1);
    final FactPattern fp2 = new FactPattern();
    fp2.setFactType("Smurf");
    final SingleFieldConstraint sfc2 = new SingleFieldConstraint();
    sfc2.setConstraintValueType(BaseSingleFieldConstraint.TYPE_TEMPLATE);
    sfc2.setFactType("Smurf");
    sfc2.setOperator("==");
    sfc2.setFieldName("colour");
    sfc2.setValue("f2");
    fp2.addConstraint(sfc2);
    brl.getDefinition().add(fp1);
    brl.getDefinition().add(fp2);
    brl.getChildColumns().add(new BRLConditionVariableColumn("f1", DataType.TYPE_STRING));
    brl.getChildColumns().add(new BRLConditionVariableColumn("f2", DataType.TYPE_STRING));
    model.getConditions().add(brl);
    when(dmo.getFieldType(eq("Person"), eq("name"))).thenReturn(DataType.TYPE_STRING);
    when(dmo.getFieldType(eq("Smurf"), eq("colour"))).thenReturn(DataType.TYPE_STRING);
}
Also used : BRLConditionColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLConditionColumn) BaseSingleFieldConstraint(org.drools.workbench.models.datamodel.rule.BaseSingleFieldConstraint) SingleFieldConstraint(org.drools.workbench.models.datamodel.rule.SingleFieldConstraint) FactPattern(org.drools.workbench.models.datamodel.rule.FactPattern) BRLConditionVariableColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn)

Example 58 with BRLConditionVariableColumn

use of org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn in project drools-wb by kiegroup.

the class CellBuilder method build.

public void build() throws BuildException {
    if (baseColumn instanceof ActionCol52) {
        final Action action = builderFactory.getActionBuilder().with(rule).with(row).with(columnIndex).with((ActionCol52) baseColumn).build();
        rule.getActions().add(action);
    } else if (baseColumn instanceof ConditionCol52) {
        if (baseColumn instanceof BRLConditionVariableColumn) {
            final Condition condition = builderFactory.getBRLConditionBuilder().with((BRLConditionVariableColumn) baseColumn).with(getRealCellValue((BRLConditionVariableColumn) baseColumn, row.get(columnIndex))).with(columnIndex).build();
            rule.getConditions().add(condition);
        } else {
            final Condition condition = builderFactory.getFieldConditionsBuilder().with(resolvePattern(rule)).with((ConditionCol52) baseColumn).with(getRealCellValue((ConditionCol52) baseColumn, row.get(columnIndex))).with(columnIndex).build();
            rule.getConditions().add(condition);
        }
    }
}
Also used : Condition(org.drools.workbench.services.verifier.api.client.index.Condition) Action(org.drools.workbench.services.verifier.api.client.index.Action) ConditionCol52(org.drools.workbench.models.guided.dtable.shared.model.ConditionCol52) ActionCol52(org.drools.workbench.models.guided.dtable.shared.model.ActionCol52) BRLConditionVariableColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn)

Example 59 with BRLConditionVariableColumn

use of org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn in project drools-wb by kiegroup.

the class GuidedDecisionTableGeneratorListenerTest method checkConversionWithBigMaths.

private void checkConversionWithBigMaths(final ConversionResult result, final GuidedDecisionTableGeneratorListener listener, final Supplier<String> expectedDataType, final Supplier<Object> expectedDataValueRow0, final Supplier<Object> expectedDataValueRow1) {
    // 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("HelloWorld1", dtable.getTableName());
    assertEquals(GuidedDecisionTable52.TableFormat.EXTENDED_ENTRY, dtable.getTableFormat());
    // Check expanded columns
    List<BaseColumn> columns = dtable.getExpandedColumns();
    assertNotNull(columns);
    assertEquals(8, 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);
    assertTrue(columns.get(6) instanceof BRLActionVariableColumn);
    assertTrue(columns.get(7) instanceof BRLActionVariableColumn);
    // Check individual condition columns
    assertEquals(1, dtable.getConditions().size());
    assertTrue(dtable.getConditions().get(0) instanceof BRLConditionColumn);
    BRLConditionColumn conditionCol0 = ((BRLConditionColumn) dtable.getConditions().get(0));
    assertEquals("Converted from ['Status']", 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("Message", conditionCol0fp.getFactType());
    assertEquals(1, conditionCol0fp.getNumberOfConstraints());
    // Field Constraint 1
    assertTrue(conditionCol0fp.getConstraint(0) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol0fpsfc0 = (SingleFieldConstraint) conditionCol0fp.getConstraint(0);
    assertEquals("status", conditionCol0fpsfc0.getFieldName());
    assertEquals("==", conditionCol0fpsfc0.getOperator());
    assertEquals("param1", conditionCol0fpsfc0.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol0fpsfc0.getConstraintValueType());
    assertEquals(DataType.TYPE_STRING, conditionCol0fpsfc0.getFieldType());
    // Field Constraint 1 - Variable 1
    BRLConditionVariableColumn conditionCol0param0 = conditionCol0.getChildColumns().get(0);
    assertEquals("param1", conditionCol0param0.getVarName());
    assertEquals("Status", conditionCol0param0.getHeader());
    assertEquals(DataType.TYPE_STRING, conditionCol0param0.getFieldType());
    assertEquals("Message", conditionCol0param0.getFactType());
    assertEquals("status", conditionCol0param0.getFactField());
    // Check individual action columns
    assertEquals(3, dtable.getActionCols().size());
    // Action 1
    assertTrue(dtable.getActionCols().get(0) instanceof BRLActionColumn);
    BRLActionColumn actionCol0 = ((BRLActionColumn) dtable.getActionCols().get(0));
    assertEquals("Converted from ['Print out message?']", actionCol0.getHeader());
    assertEquals(1, actionCol0.getChildColumns().size());
    List<IAction> actionCol0definition = actionCol0.getDefinition();
    assertEquals(1, actionCol0definition.size());
    assertTrue(actionCol0definition.get(0) instanceof FreeFormLine);
    FreeFormLine actionCol0ffl0 = (FreeFormLine) actionCol0definition.get(0);
    assertEquals("System.out.println(m.getMessage());", actionCol0ffl0.getText());
    // Action 2
    BRLActionColumn actionCol1 = ((BRLActionColumn) dtable.getActionCols().get(1));
    assertEquals("Converted from ['Set message', 'Set status', 'Set rate']", actionCol1.getHeader());
    assertEquals(3, actionCol1.getChildColumns().size());
    List<IAction> actionCol1definition = actionCol1.getDefinition();
    assertEquals(1, actionCol1definition.size());
    assertTrue(actionCol1definition.get(0) instanceof ActionSetField);
    ActionSetField actionCol1asf0 = (ActionSetField) actionCol1definition.get(0);
    assertEquals("m", actionCol1asf0.getVariable());
    ActionFieldValue[] actionCol1asf0afvs = actionCol1asf0.getFieldValues();
    assertEquals(3, actionCol1asf0afvs.length);
    ActionFieldValue actionCol1asf0afv0 = actionCol1asf0afvs[0];
    assertEquals("message", actionCol1asf0afv0.getField());
    assertEquals("param2", actionCol1asf0afv0.getValue());
    assertEquals(FieldNatureType.TYPE_TEMPLATE, actionCol1asf0afv0.getNature());
    assertEquals(DataType.TYPE_STRING, actionCol1asf0afv0.getType());
    ActionFieldValue actionCol1asf0afv1 = actionCol1asf0afvs[1];
    assertEquals("status", actionCol1asf0afv1.getField());
    assertEquals("param3", actionCol1asf0afv1.getValue());
    assertEquals(FieldNatureType.TYPE_TEMPLATE, actionCol1asf0afv1.getNature());
    assertEquals(DataType.TYPE_STRING, actionCol1asf0afv1.getType());
    // Action 1 - Variable 1
    BRLActionVariableColumn actionCol0param0 = actionCol0.getChildColumns().get(0);
    assertEquals("", actionCol0param0.getVarName());
    assertEquals("Print out message?", actionCol0param0.getHeader());
    assertEquals(DataType.TYPE_BOOLEAN, actionCol0param0.getFieldType());
    assertNull(actionCol0param0.getFactType());
    assertNull(actionCol0param0.getFactField());
    assertTrue(dtable.getActionCols().get(0) instanceof BRLActionColumn);
    // Action 2 - Variable 1
    BRLActionVariableColumn actionCol1param0 = actionCol1.getChildColumns().get(0);
    assertEquals("param2", actionCol1param0.getVarName());
    assertEquals("Set message", actionCol1param0.getHeader());
    assertEquals(DataType.TYPE_STRING, actionCol1param0.getFieldType());
    assertEquals("Message", actionCol1param0.getFactType());
    assertEquals("message", actionCol1param0.getFactField());
    // Action 2 - Variable 2
    BRLActionVariableColumn actionCol1param1 = actionCol1.getChildColumns().get(1);
    assertEquals("param3", actionCol1param1.getVarName());
    assertEquals("Set status", actionCol1param1.getHeader());
    assertEquals(DataType.TYPE_STRING, actionCol1param1.getFieldType());
    assertEquals("Message", actionCol1param1.getFactType());
    assertEquals("status", actionCol1param1.getFactField());
    // Action 2 - Variable 3
    BRLActionVariableColumn actionCol1param2 = actionCol1.getChildColumns().get(2);
    assertEquals("param4", actionCol1param2.getVarName());
    assertEquals("Set rate", actionCol1param2.getHeader());
    assertEquals(expectedDataType.get(), actionCol1param2.getFieldType());
    assertEquals("Message", actionCol1param2.getFactType());
    assertEquals("rate", actionCol1param2.getFactField());
    // Check data
    assertEquals(2, dtable.getData().size());
    assertTrue(isRowEquivalent(new Object[] { 1, "Hello World", "Message.HELLO", false, "Goodbye cruel world", "Message.GOODBYE", expectedDataValueRow0.get(), "m" }, dtable.getData().get(0)));
    assertTrue(isRowEquivalent(new Object[] { 2, "Goodbye", "Message.GOODBYE", false, "", "", expectedDataValueRow1.get(), "m" }, dtable.getData().get(1)));
}
Also used : BRLConditionColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLConditionColumn) IPattern(org.drools.workbench.models.datamodel.rule.IPattern) GuidedDecisionTable52(org.drools.workbench.models.guided.dtable.shared.model.GuidedDecisionTable52) IAction(org.drools.workbench.models.datamodel.rule.IAction) CompositeFactPattern(org.drools.workbench.models.datamodel.rule.CompositeFactPattern) FactPattern(org.drools.workbench.models.datamodel.rule.FactPattern) BRLConditionVariableColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn) SingleFieldConstraint(org.drools.workbench.models.datamodel.rule.SingleFieldConstraint) BRLActionColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLActionColumn) FreeFormLine(org.drools.workbench.models.datamodel.rule.FreeFormLine) ActionSetField(org.drools.workbench.models.datamodel.rule.ActionSetField) DescriptionCol52(org.drools.workbench.models.guided.dtable.shared.model.DescriptionCol52) ActionFieldValue(org.drools.workbench.models.datamodel.rule.ActionFieldValue) BaseColumn(org.drools.workbench.models.guided.dtable.shared.model.BaseColumn) BRLActionVariableColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLActionVariableColumn) RowNumberCol52(org.drools.workbench.models.guided.dtable.shared.model.RowNumberCol52)

Example 60 with BRLConditionVariableColumn

use of org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn in project drools-wb by kiegroup.

the class GuidedDecisionTableGeneratorListenerTest method conversionWithRetract.

@Test
public // https://issues.jboss.org/browse/GUVNOR-2888
void conversionWithRetract() {
    final ConversionResult result = new ConversionResult();
    final List<DataListener> listeners = new ArrayList<>();
    addModelField("org.test.Employee", "this", "org.test.Employee", DataType.TYPE_THIS);
    addModelField("org.test.Employee", "yearsOfService", Integer.class.getName(), DataType.TYPE_NUMERIC_INTEGER);
    addModelField("org.test.Employee", "name", String.class.getName(), DataType.TYPE_STRING);
    addModelField("org.test.Employee", "bigAbsence", Boolean.class.getName(), DataType.TYPE_BOOLEAN);
    addModelField("org.test.Bonus", "this", "org.test.Bonus", DataType.TYPE_THIS);
    addModelField("org.test.Bonus", "amount", Double.class.getName(), DataType.TYPE_NUMERIC_DOUBLE);
    addModelField("org.test.Bonus", "employee", "org.test.Employee", DataType.TYPE_OBJECT);
    final GuidedDecisionTableGeneratorListener listener = new GuidedDecisionTableGeneratorListener(result, dmo);
    listeners.add(listener);
    // Convert
    final ExcelParser parser = new ExcelParser(listeners);
    final InputStream is = this.getClass().getResourceAsStream("GUVNOR-2888.xlsx");
    try {
        parser.parseFile(is);
    } finally {
        try {
            is.close();
        } catch (IOException ioe) {
            fail(ioe.getMessage());
        }
    }
    // Check conversion results
    assertEquals(1, result.getMessages().size());
    assertTrue(result.getMessages().get(0).getMessage().contains("I13"));
    // Check basics
    final List<GuidedDecisionTable52> dtables = listener.getGuidedDecisionTables();
    assertNotNull(dtables);
    assertEquals(1, dtables.size());
    GuidedDecisionTable52 dtable = dtables.get(0);
    assertEquals("bonus_program", dtable.getTableName());
    assertEquals(GuidedDecisionTable52.TableFormat.EXTENDED_ENTRY, dtable.getTableFormat());
    // Check expanded columns
    List<BaseColumn> columns = dtable.getExpandedColumns();
    assertNotNull(columns);
    assertEquals(11, 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 BRLConditionVariableColumn);
    assertTrue(columns.get(5) instanceof BRLConditionVariableColumn);
    assertTrue(columns.get(6) instanceof BRLConditionVariableColumn);
    assertTrue(columns.get(7) instanceof BRLConditionVariableColumn);
    assertTrue(columns.get(8) instanceof BRLActionVariableColumn);
    assertTrue(columns.get(9) instanceof BRLActionVariableColumn);
    assertTrue(columns.get(10) instanceof BRLActionVariableColumn);
    // Check individual condition columns
    assertEquals(3, dtable.getConditions().size());
    // Pattern 0
    assertTrue(dtable.getConditions().get(0) instanceof BRLConditionColumn);
    BRLConditionColumn conditionCol0 = ((BRLConditionColumn) dtable.getConditions().get(0));
    assertEquals("Converted from ['Served between', 'Served between', 'Served more', 'absence']", conditionCol0.getHeader());
    assertEquals(4, 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("Employee", conditionCol0fp.getFactType());
    assertEquals(4, conditionCol0fp.getNumberOfConstraints());
    // Pattern 0 - Field Constraint 0
    assertTrue(conditionCol0fp.getConstraint(0) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol0fpsfc0 = (SingleFieldConstraint) conditionCol0fp.getConstraint(0);
    assertEquals("yearsOfService", conditionCol0fpsfc0.getFieldName());
    assertEquals(">=", conditionCol0fpsfc0.getOperator());
    assertEquals("param1", conditionCol0fpsfc0.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol0fpsfc0.getConstraintValueType());
    assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0fpsfc0.getFieldType());
    // Pattern 0 - Field Constraint 0 - Variable 0
    BRLConditionVariableColumn conditionCol0param0 = conditionCol0.getChildColumns().get(0);
    assertEquals("param1", conditionCol0param0.getVarName());
    assertEquals("Served between", conditionCol0param0.getHeader());
    assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0param0.getFieldType());
    assertEquals("Employee", conditionCol0param0.getFactType());
    assertEquals("yearsOfService", conditionCol0param0.getFactField());
    // Pattern 0 - Field Constraint 1
    assertTrue(conditionCol0fp.getConstraint(1) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol0fpsfc1 = (SingleFieldConstraint) conditionCol0fp.getConstraint(1);
    assertEquals("yearsOfService", conditionCol0fpsfc1.getFieldName());
    assertEquals("<", conditionCol0fpsfc1.getOperator());
    assertEquals("param2", conditionCol0fpsfc1.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol0fpsfc1.getConstraintValueType());
    assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0fpsfc1.getFieldType());
    // Pattern 0 - Field Constraint 1 - Variable 0
    BRLConditionVariableColumn conditionCol0param1 = conditionCol0.getChildColumns().get(1);
    assertEquals("param2", conditionCol0param1.getVarName());
    assertEquals("Served between", conditionCol0param1.getHeader());
    assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0param1.getFieldType());
    assertEquals("Employee", conditionCol0param1.getFactType());
    assertEquals("yearsOfService", conditionCol0param1.getFactField());
    // Pattern 0 - Field Constraint 2
    assertTrue(conditionCol0fp.getConstraint(2) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol0fpsfc2 = (SingleFieldConstraint) conditionCol0fp.getConstraint(2);
    assertEquals("yearsOfService", conditionCol0fpsfc2.getFieldName());
    assertEquals(">", conditionCol0fpsfc2.getOperator());
    assertEquals("param3", conditionCol0fpsfc2.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol0fpsfc2.getConstraintValueType());
    assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0fpsfc2.getFieldType());
    // Pattern 0 - Field Constraint 2 - Variable 0
    BRLConditionVariableColumn conditionCol0param2 = conditionCol0.getChildColumns().get(2);
    assertEquals("param3", conditionCol0param2.getVarName());
    assertEquals("Served more", conditionCol0param2.getHeader());
    assertEquals(DataType.TYPE_NUMERIC_INTEGER, conditionCol0param2.getFieldType());
    assertEquals("Employee", conditionCol0param2.getFactType());
    assertEquals("yearsOfService", conditionCol0param2.getFactField());
    // Pattern 0 - Field Constraint 3
    assertTrue(conditionCol0fp.getConstraint(3) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol0fpsfc3 = (SingleFieldConstraint) conditionCol0fp.getConstraint(3);
    assertEquals("bigAbsence", conditionCol0fpsfc3.getFieldName());
    assertEquals("==", conditionCol0fpsfc3.getOperator());
    assertEquals("param4", conditionCol0fpsfc3.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol0fpsfc3.getConstraintValueType());
    assertEquals(DataType.TYPE_BOOLEAN, conditionCol0fpsfc3.getFieldType());
    // Pattern 0 - Field Constraint 3 - Variable 0
    BRLConditionVariableColumn conditionCol0param3 = conditionCol0.getChildColumns().get(3);
    assertEquals("param4", conditionCol0param3.getVarName());
    assertEquals("absence", conditionCol0param3.getHeader());
    assertEquals(DataType.TYPE_BOOLEAN, conditionCol0param3.getFieldType());
    assertEquals("Employee", conditionCol0param3.getFactType());
    assertEquals("bigAbsence", conditionCol0param3.getFactField());
    // Pattern 1
    assertTrue(dtable.getConditions().get(1) instanceof BRLConditionColumn);
    BRLConditionColumn conditionCol1 = ((BRLConditionColumn) dtable.getConditions().get(1));
    assertEquals("Converted from ['bonus for employee']", conditionCol1.getHeader());
    assertEquals(1, conditionCol1.getChildColumns().size());
    List<IPattern> conditionCol1definition = conditionCol1.getDefinition();
    assertEquals(1, conditionCol1definition.size());
    assertTrue(conditionCol1definition.get(0) instanceof FactPattern);
    FactPattern conditionCol1fp = (FactPattern) conditionCol1definition.get(0);
    assertEquals("Bonus", conditionCol1fp.getFactType());
    assertEquals(1, conditionCol1fp.getNumberOfConstraints());
    // Pattern 1 - Field Constraint 0
    assertTrue(conditionCol1fp.getConstraint(0) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol1fpsfc0 = (SingleFieldConstraint) conditionCol1fp.getConstraint(0);
    assertEquals("employee", conditionCol1fpsfc0.getFieldName());
    assertEquals("==", conditionCol1fpsfc0.getOperator());
    assertEquals("param5", conditionCol1fpsfc0.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol1fpsfc0.getConstraintValueType());
    assertEquals(DataType.TYPE_OBJECT, conditionCol1fpsfc0.getFieldType());
    // Pattern 1 - Field Constraint 0 - Variable 0
    BRLConditionVariableColumn conditionCol1param0 = conditionCol1.getChildColumns().get(0);
    assertEquals("param5", conditionCol1param0.getVarName());
    assertEquals("bonus for employee", conditionCol1param0.getHeader());
    assertEquals(DataType.TYPE_OBJECT, conditionCol1param0.getFieldType());
    assertEquals("Bonus", conditionCol1param0.getFactType());
    assertEquals("employee", conditionCol1param0.getFactField());
    // Pattern 2
    assertTrue(dtable.getConditions().get(2) instanceof BRLConditionColumn);
    BRLConditionColumn conditionCol2 = ((BRLConditionColumn) dtable.getConditions().get(2));
    assertEquals("Converted from ['no bonus for employee']", conditionCol2.getHeader());
    assertEquals(1, conditionCol2.getChildColumns().size());
    List<IPattern> conditionCol2definition = conditionCol2.getDefinition();
    assertEquals(1, conditionCol2definition.size());
    assertTrue(conditionCol2definition.get(0) instanceof CompositeFactPattern);
    CompositeFactPattern conditionCol2cfp = (CompositeFactPattern) conditionCol2definition.get(0);
    assertEquals(CompositeFactPattern.COMPOSITE_TYPE_NOT, conditionCol2cfp.getType());
    assertEquals(1, conditionCol2cfp.getPatterns().length);
    assertTrue(conditionCol2cfp.getPatterns()[0] instanceof FactPattern);
    FactPattern conditionCol2fp = (FactPattern) conditionCol2cfp.getPatterns()[0];
    assertEquals("Bonus", conditionCol2fp.getFactType());
    assertEquals(1, conditionCol2fp.getNumberOfConstraints());
    // Pattern 2 - Field Constraint 0
    assertTrue(conditionCol2fp.getConstraint(0) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol2fpsfc0 = (SingleFieldConstraint) conditionCol2fp.getConstraint(0);
    assertEquals("employee", conditionCol2fpsfc0.getFieldName());
    assertEquals("==", conditionCol2fpsfc0.getOperator());
    assertEquals("param6", conditionCol2fpsfc0.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol2fpsfc0.getConstraintValueType());
    assertEquals(DataType.TYPE_OBJECT, conditionCol2fpsfc0.getFieldType());
    // Pattern 2 - Field Constraint 0 - Variable 0
    BRLConditionVariableColumn conditionCol2param0 = conditionCol2.getChildColumns().get(0);
    assertEquals("param6", conditionCol2param0.getVarName());
    assertEquals("no bonus for employee", conditionCol2param0.getHeader());
    assertEquals(DataType.TYPE_OBJECT, conditionCol2param0.getFieldType());
    assertEquals("Bonus", conditionCol2param0.getFactType());
    assertEquals("employee", conditionCol2param0.getFactField());
    // Check individual action columns
    assertEquals(3, dtable.getActionCols().size());
    // Action 0
    assertTrue(dtable.getActionCols().get(0) instanceof BRLActionColumn);
    BRLActionColumn actionCol0 = ((BRLActionColumn) dtable.getActionCols().get(0));
    assertEquals("Converted from ['Inser bonus of amount']", actionCol0.getHeader());
    assertEquals(1, actionCol0.getChildColumns().size());
    List<IAction> actionCol0definition = actionCol0.getDefinition();
    assertEquals(1, actionCol0definition.size());
    assertTrue(actionCol0definition.get(0) instanceof FreeFormLine);
    FreeFormLine actionCol0ffl0 = (FreeFormLine) actionCol0definition.get(0);
    assertEquals("Bonus $b = new Bonus(); $b.setAmount((double)@{param7}); $b.setEmployee($e); insert($b);", actionCol0ffl0.getText());
    // Action 0 - Variable 0
    BRLActionVariableColumn actionCol0param0 = actionCol0.getChildColumns().get(0);
    assertEquals("param7", actionCol0param0.getVarName());
    assertEquals("Inser bonus of amount", actionCol0param0.getHeader());
    assertEquals(DataType.TYPE_OBJECT, actionCol0param0.getFieldType());
    // Action 1
    BRLActionColumn actionCol1 = ((BRLActionColumn) dtable.getActionCols().get(1));
    assertEquals("Converted from ['retract']", actionCol1.getHeader());
    assertEquals(1, actionCol1.getChildColumns().size());
    List<IAction> actionCol1definition = actionCol1.getDefinition();
    assertEquals(1, actionCol1definition.size());
    assertTrue(actionCol1definition.get(0) instanceof ActionRetractFact);
    ActionRetractFact actionCol1arf = (ActionRetractFact) actionCol1definition.get(0);
    assertEquals("$bonus", actionCol1arf.getVariableName());
    assertTrue(dtable.getActionCols().get(1) instanceof BRLActionColumn);
    // Action 1 - Variable 0
    BRLActionVariableColumn actionCol1param0 = actionCol1.getChildColumns().get(0);
    assertEquals("param8", actionCol1param0.getVarName());
    assertEquals("retract", actionCol1param0.getHeader());
    assertEquals(DataType.TYPE_BOOLEAN, actionCol1param0.getFieldType());
    // Action 2
    BRLActionColumn actionCol2 = ((BRLActionColumn) dtable.getActionCols().get(2));
    assertEquals("Converted from ['retract2']", actionCol2.getHeader());
    assertEquals(1, actionCol2.getChildColumns().size());
    List<IAction> actionCol2definition = actionCol2.getDefinition();
    assertEquals(1, actionCol2definition.size());
    assertTrue(actionCol2definition.get(0) instanceof ActionRetractFact);
    ActionRetractFact actionCol2arf = (ActionRetractFact) actionCol2definition.get(0);
    assertEquals("$b", actionCol2arf.getVariableName());
    assertTrue(dtable.getActionCols().get(2) instanceof BRLActionColumn);
    // Action 2 - Variable 1
    BRLActionVariableColumn actionCol2param0 = actionCol2.getChildColumns().get(0);
    assertEquals("", actionCol2param0.getVarName());
    assertEquals("retract2", actionCol2param0.getHeader());
    assertEquals(DataType.TYPE_BOOLEAN, actionCol2param0.getFieldType());
    // Check data
    assertEquals(4, dtable.getData().size());
    assertTrue(isRowEquivalent(new Object[] { 1, "Lowest bonus", 0, 2, null, false, "", "$e", "400", false, false }, dtable.getData().get(0)));
    assertTrue(isRowEquivalent(new Object[] { 2, "Middle bonus", 2, 6, null, false, "", "$e", "800", false, false }, dtable.getData().get(1)));
    assertTrue(isRowEquivalent(new Object[] { 3, "High bonus", null, null, 6, false, "", "", "1000", false, false }, dtable.getData().get(2)));
    assertTrue(isRowEquivalent(new Object[] { 4, "Big absence", null, null, null, true, "$e", "", "", true, true }, dtable.getData().get(3)));
}
Also used : BRLConditionColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLConditionColumn) IPattern(org.drools.workbench.models.datamodel.rule.IPattern) GuidedDecisionTable52(org.drools.workbench.models.guided.dtable.shared.model.GuidedDecisionTable52) ArrayList(java.util.ArrayList) CompositeFactPattern(org.drools.workbench.models.datamodel.rule.CompositeFactPattern) FactPattern(org.drools.workbench.models.datamodel.rule.FactPattern) CompositeFactPattern(org.drools.workbench.models.datamodel.rule.CompositeFactPattern) BRLConditionVariableColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn) SingleFieldConstraint(org.drools.workbench.models.datamodel.rule.SingleFieldConstraint) ActionRetractFact(org.drools.workbench.models.datamodel.rule.ActionRetractFact) BRLActionColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLActionColumn) DescriptionCol52(org.drools.workbench.models.guided.dtable.shared.model.DescriptionCol52) ExcelParser(org.drools.decisiontable.parser.xls.ExcelParser) BaseColumn(org.drools.workbench.models.guided.dtable.shared.model.BaseColumn) IAction(org.drools.workbench.models.datamodel.rule.IAction) InputStream(java.io.InputStream) IOException(java.io.IOException) BigInteger(java.math.BigInteger) ConversionResult(org.drools.workbench.models.guided.dtable.shared.conversion.ConversionResult) FreeFormLine(org.drools.workbench.models.datamodel.rule.FreeFormLine) DataListener(org.drools.template.parser.DataListener) BRLActionVariableColumn(org.drools.workbench.models.guided.dtable.shared.model.BRLActionVariableColumn) RowNumberCol52(org.drools.workbench.models.guided.dtable.shared.model.RowNumberCol52) Test(org.junit.Test)

Aggregations

BRLConditionVariableColumn (org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn)69 BRLConditionColumn (org.drools.workbench.models.guided.dtable.shared.model.BRLConditionColumn)53 Test (org.junit.Test)39 ArrayList (java.util.ArrayList)28 FactPattern (org.drools.workbench.models.datamodel.rule.FactPattern)24 IPattern (org.drools.workbench.models.datamodel.rule.IPattern)23 SingleFieldConstraint (org.drools.workbench.models.datamodel.rule.SingleFieldConstraint)23 DescriptionCol52 (org.drools.workbench.models.guided.dtable.shared.model.DescriptionCol52)23 GuidedDecisionTable52 (org.drools.workbench.models.guided.dtable.shared.model.GuidedDecisionTable52)23 RowNumberCol52 (org.drools.workbench.models.guided.dtable.shared.model.RowNumberCol52)23 BaseColumn (org.drools.workbench.models.guided.dtable.shared.model.BaseColumn)18 BRLActionColumn (org.drools.workbench.models.guided.dtable.shared.model.BRLActionColumn)15 BRLActionVariableColumn (org.drools.workbench.models.guided.dtable.shared.model.BRLActionVariableColumn)15 IOException (java.io.IOException)11 InputStream (java.io.InputStream)11 ExcelParser (org.drools.decisiontable.parser.xls.ExcelParser)11 DataListener (org.drools.template.parser.DataListener)11 BaseSingleFieldConstraint (org.drools.workbench.models.datamodel.rule.BaseSingleFieldConstraint)11 ConversionResult (org.drools.workbench.models.guided.dtable.shared.conversion.ConversionResult)11 ConditionCol52 (org.drools.workbench.models.guided.dtable.shared.model.ConditionCol52)11