Search in sources :

Example 71 with BRLActionColumn

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

the class GuidedDecisionTableGeneratorListenerTest method testMultipleRuleTables.

@Test
public void testMultipleRuleTables() {
    final ConversionResult result = new ConversionResult();
    final List<DataListener> listeners = new ArrayList<DataListener>();
    final GuidedDecisionTableGeneratorListener listener = new GuidedDecisionTableGeneratorListener(result, dmo);
    listeners.add(listener);
    // Convert
    final ExcelParser parser = new ExcelParser(listeners);
    final InputStream is = this.getClass().getResourceAsStream("MultipleRuleTables.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
    List<GuidedDecisionTable52> dtables = listener.getGuidedDecisionTables();
    assertNotNull(dtables);
    assertEquals(2, dtables.size());
    GuidedDecisionTable52 dtable0 = dtables.get(0);
    assertEquals("Table1", dtable0.getTableName());
    assertEquals(GuidedDecisionTable52.TableFormat.EXTENDED_ENTRY, dtable0.getTableFormat());
    GuidedDecisionTable52 dtable1 = dtables.get(1);
    assertEquals("Table2", dtable1.getTableName());
    assertEquals(GuidedDecisionTable52.TableFormat.EXTENDED_ENTRY, dtable1.getTableFormat());
    // Check expanded columns
    List<BaseColumn> columns0 = dtable0.getExpandedColumns();
    assertNotNull(columns0);
    assertEquals(5, columns0.size());
    assertTrue(columns0.get(0) instanceof RowNumberCol52);
    assertTrue(columns0.get(1) instanceof DescriptionCol52);
    assertTrue(columns0.get(2) instanceof AttributeCol52);
    assertTrue(columns0.get(3) instanceof BRLConditionVariableColumn);
    assertTrue(columns0.get(4) instanceof BRLActionVariableColumn);
    AttributeCol52 attrCol0_2 = ((AttributeCol52) columns0.get(2));
    assertEquals(GuidedDecisionTable52.AGENDA_GROUP_ATTR, attrCol0_2.getAttribute());
    // Check individual condition columns
    assertEquals(1, dtable0.getConditions().size());
    assertTrue(dtable0.getConditions().get(0) instanceof BRLConditionColumn);
    // Column 1
    BRLConditionColumn conditionCol0_0 = ((BRLConditionColumn) dtable0.getConditions().get(0));
    assertEquals("Converted from ['Person's name']", conditionCol0_0.getHeader());
    assertEquals(1, conditionCol0_0.getChildColumns().size());
    List<IPattern> conditionCol0_0definition = conditionCol0_0.getDefinition();
    assertEquals(1, conditionCol0_0definition.size());
    assertTrue(conditionCol0_0definition.get(0) instanceof FactPattern);
    FactPattern conditionCol0_0fp = (FactPattern) conditionCol0_0definition.get(0);
    assertEquals("Person", conditionCol0_0fp.getFactType());
    assertEquals(1, conditionCol0_0fp.getNumberOfConstraints());
    assertTrue(conditionCol0_0fp.getConstraint(0) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol0_0fpsfc0 = (SingleFieldConstraint) conditionCol0_0fp.getConstraint(0);
    assertEquals("name", conditionCol0_0fpsfc0.getFieldName());
    assertEquals("==", conditionCol0_0fpsfc0.getOperator());
    assertEquals("param1", conditionCol0_0fpsfc0.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol0_0fpsfc0.getConstraintValueType());
    // Column 1 - Variable 1
    BRLConditionVariableColumn conditionCol0_0param0 = conditionCol0_0.getChildColumns().get(0);
    assertEquals("param1", conditionCol0_0param0.getVarName());
    assertEquals("Person's name", conditionCol0_0param0.getHeader());
    assertEquals(DataType.TYPE_STRING, conditionCol0_0param0.getFieldType());
    assertEquals("Person", conditionCol0_0param0.getFactType());
    assertEquals("name", conditionCol0_0param0.getFactField());
    // Column 2
    BRLActionColumn actionCol0_0 = ((BRLActionColumn) dtable0.getActionCols().get(0));
    assertEquals("Converted from ['Salutation']", actionCol0_0.getHeader());
    assertEquals(1, actionCol0_0.getChildColumns().size());
    List<IAction> actionCol0_0definition = actionCol0_0.getDefinition();
    assertEquals(1, actionCol0_0definition.size());
    assertTrue(actionCol0_0definition.get(0) instanceof FreeFormLine);
    FreeFormLine actionCol0_0ffl = (FreeFormLine) actionCol0_0definition.get(0);
    assertEquals("System.out.println(\"@{param2}\");", actionCol0_0ffl.getText());
    // Column 1 - Variable 1
    BRLActionVariableColumn actionCol0_0param0 = actionCol0_0.getChildColumns().get(0);
    assertEquals("param2", actionCol0_0param0.getVarName());
    assertEquals("Salutation", actionCol0_0param0.getHeader());
    assertEquals(DataType.TYPE_OBJECT, actionCol0_0param0.getFieldType());
    assertNull(actionCol0_0param0.getFactType());
    assertNull(actionCol0_0param0.getFactField());
    // Check data
    assertEquals(2, dtable0.getData().size());
    assertTrue(isRowEquivalent(new Object[] { 1, "Created from row 7", "AG1", "John", "Hello Sir" }, dtable0.getData().get(0)));
    assertTrue(isRowEquivalent(new Object[] { 2, "Row 2", "AG2", "Jane", "Hello Madam" }, dtable0.getData().get(1)));
    // Check expanded columns
    List<BaseColumn> columns1 = dtable1.getExpandedColumns();
    assertNotNull(columns1);
    assertEquals(4, columns1.size());
    assertTrue(columns1.get(0) instanceof RowNumberCol52);
    assertTrue(columns1.get(1) instanceof DescriptionCol52);
    assertTrue(columns1.get(2) instanceof BRLConditionVariableColumn);
    assertTrue(columns1.get(3) instanceof BRLConditionVariableColumn);
    // Check individual condition columns
    assertEquals(1, dtable0.getConditions().size());
    assertTrue(dtable0.getConditions().get(0) instanceof BRLConditionColumn);
    // Column 1
    BRLConditionColumn conditionCol1_0 = ((BRLConditionColumn) dtable1.getConditions().get(0));
    assertEquals("Converted from ['Person’s name', 'Person’s age']", conditionCol1_0.getHeader());
    assertEquals(2, conditionCol1_0.getChildColumns().size());
    List<IPattern> conditionCol1_0definition = conditionCol1_0.getDefinition();
    assertEquals(1, conditionCol1_0definition.size());
    assertTrue(conditionCol1_0definition.get(0) instanceof FactPattern);
    FactPattern conditionCol1_0fp = (FactPattern) conditionCol1_0definition.get(0);
    assertEquals("Person", conditionCol1_0fp.getFactType());
    assertEquals(2, conditionCol1_0fp.getNumberOfConstraints());
    assertTrue(conditionCol1_0fp.getConstraint(0) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol1_0fpsfc0 = (SingleFieldConstraint) conditionCol1_0fp.getConstraint(0);
    assertEquals("name", conditionCol1_0fpsfc0.getFieldName());
    assertEquals("==", conditionCol1_0fpsfc0.getOperator());
    assertEquals("param1", conditionCol1_0fpsfc0.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol1_0fpsfc0.getConstraintValueType());
    assertTrue(conditionCol1_0fp.getConstraint(1) instanceof SingleFieldConstraint);
    final SingleFieldConstraint conditionCol1_0fpsfc1 = (SingleFieldConstraint) conditionCol1_0fp.getConstraint(1);
    assertEquals("age", conditionCol1_0fpsfc1.getFieldName());
    assertEquals("==", conditionCol1_0fpsfc1.getOperator());
    assertEquals("param2", conditionCol1_0fpsfc1.getValue());
    assertEquals(SingleFieldConstraint.TYPE_TEMPLATE, conditionCol1_0fpsfc1.getConstraintValueType());
    // Column 1 - Variable 1
    BRLConditionVariableColumn conditionCol1_0param0 = conditionCol1_0.getChildColumns().get(0);
    assertEquals("param1", conditionCol1_0param0.getVarName());
    assertEquals("Person’s name", conditionCol1_0param0.getHeader());
    assertEquals(DataType.TYPE_STRING, conditionCol1_0param0.getFieldType());
    assertEquals("Person", conditionCol1_0param0.getFactType());
    assertEquals("name", conditionCol1_0param0.getFactField());
    // Column 1 - Variable 2
    BRLConditionVariableColumn conditionCol1_0param1 = conditionCol1_0.getChildColumns().get(1);
    assertEquals("param2", conditionCol1_0param1.getVarName());
    assertEquals("Person’s age", conditionCol1_0param1.getHeader());
    assertEquals(DataType.TYPE_STRING, conditionCol1_0param1.getFieldType());
    assertEquals("Person", conditionCol1_0param1.getFactType());
    assertEquals("age", conditionCol1_0param1.getFactField());
    // Check data
    assertEquals(2, dtable1.getData().size());
    assertTrue(isRowEquivalent(new Object[] { 1, "Row 1", "John", "25" }, dtable1.getData().get(0)));
    assertTrue(isRowEquivalent(new Object[] { 2, "Created from row 16", "Jane", "29" }, dtable1.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) ArrayList(java.util.ArrayList) 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) AttributeCol52(org.drools.workbench.models.guided.dtable.shared.model.AttributeCol52) SingleFieldConstraint(org.drools.workbench.models.datamodel.rule.SingleFieldConstraint) 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) 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

BRLActionColumn (org.drools.workbench.models.guided.dtable.shared.model.BRLActionColumn)71 BRLActionVariableColumn (org.drools.workbench.models.guided.dtable.shared.model.BRLActionVariableColumn)51 Test (org.junit.Test)51 GuidedDecisionTable52 (org.drools.workbench.models.guided.dtable.shared.model.GuidedDecisionTable52)31 ActionFieldValue (org.drools.workbench.models.datamodel.rule.ActionFieldValue)24 ArrayList (java.util.ArrayList)21 IAction (org.drools.workbench.models.datamodel.rule.IAction)21 DescriptionCol52 (org.drools.workbench.models.guided.dtable.shared.model.DescriptionCol52)19 RowNumberCol52 (org.drools.workbench.models.guided.dtable.shared.model.RowNumberCol52)19 BaseColumn (org.drools.workbench.models.guided.dtable.shared.model.BaseColumn)18 Pattern52 (org.drools.workbench.models.guided.dtable.shared.model.Pattern52)17 SingleFieldConstraint (org.drools.workbench.models.datamodel.rule.SingleFieldConstraint)16 BRLConditionColumn (org.drools.workbench.models.guided.dtable.shared.model.BRLConditionColumn)16 ActionInsertFactCol52 (org.drools.workbench.models.guided.dtable.shared.model.ActionInsertFactCol52)15 BRLConditionVariableColumn (org.drools.workbench.models.guided.dtable.shared.model.BRLConditionVariableColumn)15 ActionInsertFact (org.drools.workbench.models.datamodel.rule.ActionInsertFact)13 IPattern (org.drools.workbench.models.datamodel.rule.IPattern)13 ConditionCol52 (org.drools.workbench.models.guided.dtable.shared.model.ConditionCol52)13 FactPattern (org.drools.workbench.models.datamodel.rule.FactPattern)11 IOException (java.io.IOException)9