How to use

setColumnTypes

method

in

org.dashbuilder.dataset.DataSetLookupConstraints

@Test
public void testEmptyConstraints() throws Exception {
  DataSetLookupConstraints constraints = new DataSetLookupConstraints()
      .setColumnTypes(null)
      .setGroupRequired(false);
  DataSetLookup lookup = constraints.newDataSetLookup(dataSet.getMetadata());
  assertThat(constraints.check(lookup)).isNull();
  DataSet result = dataSetManager.lookupDataSet(lookup);
  assertThat(result.getColumns().size()).isEqualTo(6);
  assertThat(result.getRowCount()).isEqualTo(50);
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  return new DisplayerConstraints(new DataSetLookupConstraints()
      .setGroupRequired(true)
      .setGroupColumn(true)
      .setMaxColumns(1)
      .setMinColumns(1)
      .setGroupsTitle("Categories")
      .setColumnsTitle("Values")
      .setColumnTypes(new ColumnType[] {
          ColumnType.LABEL}));
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
      .setGroupRequired(true)
      .setGroupColumn(true)
      .setMaxColumns(2)
      .setMinColumns(2)
      .setExtraColumnsAllowed(false)
      .setGroupsTitle(view.getGroupsTitle())
      .setColumnsTitle(view.getColumnsTitle())
      .setColumnTypes(new ColumnType[]{
          ColumnType.LABEL,
          ColumnType.NUMBER});
  return new CommonC3DisplayerConstants(lookupConstraints).create()
      .supportsAttribute(DisplayerAttributeDef.SUBTYPE);
}

@Test
public void testGroupWithOneCalculation() throws Exception {
  DataSetLookupConstraints constraints = new DataSetLookupConstraints()
      .setColumnTypes(new ColumnType[] {LABEL, NUMBER})
      .setGroupRequired(true);
  DataSetLookup lookup = constraints.newDataSetLookup(dataSet.getMetadata());
  assertThat(constraints.check(lookup)).isNull();
  DataSet result = dataSetManager.lookupDataSet(lookup);
  assertDataSetValues(result, dataSetFormatter, new String[][] {
      {"Barcelona", "21.00"},
      {"Madrid", "57.00"},
      {"Brno", "153.00"},
      {"Westford", "234.00"},
      {"Raleigh", "481.00"},
      {"London", "329.00"}
  }, 0);
}

@Test
public void testGroupWithLabels() throws Exception {
  DataSetLookupConstraints constraints = new DataSetLookupConstraints()
      .setColumnTypes(new ColumnType[] {LABEL, LABEL})
      .setGroupRequired(true);
  DataSetLookup lookup = constraints.newDataSetLookup(dataSet.getMetadata());
  assertThat(constraints.check(lookup)).isNull();
  DataSet result = dataSetManager.lookupDataSet(lookup);
  assertDataSetValues(result, dataSetFormatter, new String[][]{
      {"Barcelona", "Engineering"},
      {"Madrid", "Services"},
      {"Brno", "Support"},
      {"Westford", "Engineering"},
      {"Raleigh", "Management"},
      {"London", "Engineering"}
  }, 0);
}

@Test
public void testGroupMultipleColumns() throws Exception {
  DataSetLookupConstraints constraints = new DataSetLookupConstraints()
      .setColumnTypes(new ColumnType[] {LABEL, NUMBER, NUMBER, NUMBER})
      .setGroupRequired(true);
  DataSetLookup lookup = constraints.newDataSetLookup(dataSet.getMetadata());
  assertThat(constraints.check(lookup)).isNull();
  DataSet result = dataSetManager.lookupDataSet(lookup);
  assertDataSetValues(result, dataSetFormatter, new String[][] {
       {"Barcelona", "21.00", "2,913.14", "21.00"},
       {"Madrid", "57.00", "2,453.36", "57.00"},
       {"Brno", "153.00", "4,659.24", "153.00"},
       {"Westford", "234.00", "3,594.97", "234.00"},
       {"Raleigh", "481.00", "4,970.78", "481.00"},
       {"London", "329.00", "4,139.77", "329.00"}
   }, 0);
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
      .setGroupRequired(true)
      .setGroupColumn(true)
      .setMaxColumns(10)
      .setMinColumns(2)
      .setExtraColumnsAllowed(true)
      .setExtraColumnsType( ColumnType.NUMBER)
      .setGroupsTitle(view.getGroupsTitle())
      .setColumnsTitle(view.getColumnsTitle())
      .setColumnTypes(new ColumnType[] {
          ColumnType.LABEL,
          ColumnType.NUMBER});
  return new CommonC3DisplayerConstants(lookupConstraints).create()
          .supportsAttribute(DisplayerAttributeDef.SUBTYPE)
          .supportsAttribute(DisplayerAttributeGroupDef.AXIS_GROUP);
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
      .setGroupRequired(true)
      .setGroupColumn(true)
      .setMaxGroups(1)
      .setMinColumns(2)
      .setMaxColumns(10)
      .setExtraColumnsAllowed(true)
      .setExtraColumnsType(ColumnType.NUMBER)
      .setGroupsTitle(view.getGroupsTitle())
      .setColumnsTitle(view.getColumnsTitle())
      .setColumnTypes(new ColumnType[] {
          ColumnType.LABEL,
          ColumnType.NUMBER});
  return new CommonC3DisplayerConstants(lookupConstraints).create()
        .supportsAttribute(DisplayerAttributeDef.SUBTYPE)
        .supportsAttribute(DisplayerAttributeGroupDef.AXIS_GROUP);
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
      .setGroupRequired(true)
      .setGroupColumn(true)
      .setMaxGroups(1)
      .setMinColumns(2)
      .setMaxColumns(10)
      .setExtraColumnsAllowed(true)
      .setExtraColumnsType(ColumnType.NUMBER)
      .setGroupsTitle(view.getGroupsTitle())
      .setColumnsTitle(view.getColumnsTitle())
      .setColumnTypes(new ColumnType[] {
          ColumnType.LABEL,
          ColumnType.NUMBER});
  return new CommonC3DisplayerConstants(lookupConstraints).create()
      .supportsAttribute(DisplayerAttributeDef.SUBTYPE)
      .supportsAttribute(DisplayerAttributeGroupDef.AXIS_GROUP);
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
      .setGroupRequired(true)
      .setGroupColumn(true)
      .setMaxColumns(-1)
      .setMinColumns(1)
      .setExtraColumnsAllowed(true)
      .setGroupsTitle(view.getGroupsTitle())
      .setColumnsTitle(view.getColumnsTitle())
      .setColumnTypes(new ColumnType[] {
          ColumnType.LABEL});
  return new DisplayerConstraints(lookupConstraints)
      .supportsAttribute( DisplayerAttributeDef.TYPE )
      .supportsAttribute( DisplayerAttributeGroupDef.COLUMNS_GROUP )
      .supportsAttribute( DisplayerAttributeGroupDef.FILTER_GROUP )
      .supportsAttribute( DisplayerAttributeGroupDef.REFRESH_GROUP )
      .supportsAttribute( DisplayerAttributeGroupDef.GENERAL_GROUP );
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
      .setGroupRequired(true)
      .setGroupColumn(true)
      .setMaxColumns(-1)
      .setMinColumns(1)
      .setExtraColumnsAllowed(true)
      .setGroupsTitle(view.getGroupsTitle())
      .setColumnsTitle(view.getColumnsTitle())
      .setColumnTypes(new ColumnType[] {
          ColumnType.LABEL});
  return new DisplayerConstraints(lookupConstraints)
      .supportsAttribute( DisplayerAttributeDef.TYPE )
      .supportsAttribute( DisplayerAttributeGroupDef.COLUMNS_GROUP )
      .supportsAttribute( DisplayerAttributeGroupDef.FILTER_GROUP )
      .supportsAttribute( DisplayerAttributeGroupDef.REFRESH_GROUP )
      .supportsAttribute( DisplayerAttributeGroupDef.GENERAL_GROUP );
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
      .setGroupRequired(true)
      .setGroupColumn(true)
      .setMaxColumns(5)
      .setMinColumns(5)
      .setExtraColumnsAllowed(false)
      .setGroupsTitle(view.getGroupsTitle())
      .setColumnsTitle(view.getColumnsTitle())
      .setColumnTitle(1, "X")
      .setColumnTitle(2, "Y")
      .setColumnTitle(3, "Bubble Size")
      .setColumnTypes(new ColumnType[] {
          ColumnType.LABEL,
          ColumnType.NUMBER,
          ColumnType.NUMBER,
          ColumnType.LABEL,
          ColumnType.NUMBER});
  return new CommonC3DisplayerConstants(lookupConstraints).create()
        .supportsAttribute(DisplayerAttributeGroupDef.AXIS_GROUP);
}

@Test
public void testEmptyConstraints() throws Exception {
  DataSetLookupConstraints constraints = new DataSetLookupConstraints()
      .setColumnTypes(null)
      .setGroupRequired(false);
  DataSetLookup lookup = constraints.newDataSetLookup(dataSet.getMetadata());
  assertThat(constraints.check(lookup)).isNull();
  DataSet result = dataSetManager.lookupDataSet(lookup);
  assertThat(result.getColumns().size()).isEqualTo(6);
  assertThat(result.getRowCount()).isEqualTo(50);
}

@Test
public void testGroupMultipleColumns() throws Exception {
  DataSetLookupConstraints constraints = new DataSetLookupConstraints()
      .setColumnTypes(new ColumnType[] {LABEL, NUMBER, NUMBER, NUMBER})
      .setGroupRequired(true);
  DataSetLookup lookup = constraints.newDataSetLookup(dataSet.getMetadata());
  assertThat(constraints.check(lookup)).isNull();
  DataSet result = dataSetManager.lookupDataSet(lookup);
  assertDataSetValues(result, dataSetFormatter, new String[][] {
       {"Barcelona", "21.00", "2,913.14", "21.00"},
       {"Madrid", "57.00", "2,453.36", "57.00"},
       {"Brno", "153.00", "4,659.24", "153.00"},
       {"Westford", "234.00", "3,594.97", "234.00"},
       {"Raleigh", "481.00", "4,970.78", "481.00"},
       {"London", "329.00", "4,139.77", "329.00"}
   }, 0);
}

@Test
public void testGroupWithOneCalculation() throws Exception {
  DataSetLookupConstraints constraints = new DataSetLookupConstraints()
      .setColumnTypes(new ColumnType[] {LABEL, NUMBER})
      .setGroupRequired(true);
  DataSetLookup lookup = constraints.newDataSetLookup(dataSet.getMetadata());
  assertThat(constraints.check(lookup)).isNull();
  DataSet result = dataSetManager.lookupDataSet(lookup);
  assertDataSetValues(result, dataSetFormatter, new String[][] {
      {"Barcelona", "21.00"},
      {"Madrid", "57.00"},
      {"Brno", "153.00"},
      {"Westford", "234.00"},
      {"Raleigh", "481.00"},
      {"London", "329.00"}
  }, 0);
}

@Test
public void testGroupWithLabels() throws Exception {
  DataSetLookupConstraints constraints = new DataSetLookupConstraints()
      .setColumnTypes(new ColumnType[] {LABEL, LABEL})
      .setGroupRequired(true);
  DataSetLookup lookup = constraints.newDataSetLookup(dataSet.getMetadata());
  assertThat(constraints.check(lookup)).isNull();
  DataSet result = dataSetManager.lookupDataSet(lookup);
  assertDataSetValues(result, dataSetFormatter, new String[][]{
      {"Barcelona", "Engineering"},
      {"Madrid", "Services"},
      {"Brno", "Support"},
      {"Westford", "Engineering"},
      {"Raleigh", "Management"},
      {"London", "Engineering"}
  }, 0);
}

  @Override
  public DisplayerConstraints createDisplayerConstraints() {
    DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
        .setGroupRequired(true)
        .setGroupColumn(true)
        .setMaxColumns(2)
        .setMinColumns(2)
        .setExtraColumnsAllowed(false)
        .setGroupsTitle(getView().getGroupsTitle())
        .setColumnsTitle(getView().getColumnsTitle())
        .setColumnTypes(new ColumnType[] {
            ColumnType.LABEL,
            ColumnType.NUMBER});

    return new DisplayerConstraints(lookupConstraints)
        .supportsAttribute(DisplayerAttributeDef.TYPE)
        .supportsAttribute(DisplayerAttributeDef.RENDERER)
        .supportsAttribute(DisplayerAttributeGroupDef.COLUMNS_GROUP)
        .supportsAttribute(DisplayerAttributeGroupDef.FILTER_GROUP)
        .supportsAttribute(DisplayerAttributeGroupDef.REFRESH_GROUP)
        .supportsAttribute(DisplayerAttributeGroupDef.GENERAL_GROUP)
        .supportsAttribute(DisplayerAttributeGroupDef.CHART_GROUP );
  }
}

  @Override
  public DisplayerConstraints createDisplayerConstraints() {
    DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
        .setGroupRequired(true)
        .setGroupColumn(true)
        .setMaxColumns(2)
        .setMinColumns(2)
        .setExtraColumnsAllowed(false)
        .setGroupsTitle(getView().getGroupsTitle())
        .setColumnsTitle(getView().getColumnsTitle())
        .setColumnTypes(new ColumnType[] {
            ColumnType.LABEL,
            ColumnType.NUMBER});

    return new DisplayerConstraints(lookupConstraints)
        .supportsAttribute(DisplayerAttributeDef.TYPE)
        .supportsAttribute(DisplayerAttributeDef.RENDERER)
        .supportsAttribute(DisplayerAttributeGroupDef.COLUMNS_GROUP)
        .supportsAttribute(DisplayerAttributeGroupDef.FILTER_GROUP)
        .supportsAttribute(DisplayerAttributeGroupDef.REFRESH_GROUP)
        .supportsAttribute(DisplayerAttributeGroupDef.GENERAL_GROUP)
        .supportsAttribute(DisplayerAttributeGroupDef.CHART_GROUP );
  }
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
      .setGroupAllowed(false)
      .setMaxColumns(1)
      .setMinColumns(1)
      .setFunctionRequired(true)
      .setExtraColumnsAllowed(false)
      .setColumnsTitle(view.getColumnsTitle())
      .setColumnTypes(new ColumnType[] {
          ColumnType.NUMBER});
  return new DisplayerConstraints(lookupConstraints)
      .supportsAttribute(DisplayerAttributeDef.TYPE)
      .supportsAttribute(DisplayerAttributeDef.RENDERER)
      .supportsAttribute(DisplayerAttributeGroupDef.COLUMNS_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.FILTER_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.REFRESH_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.GENERAL_GROUP)
      .supportsAttribute(DisplayerAttributeDef.CHART_WIDTH)
      .supportsAttribute(DisplayerAttributeDef.CHART_HEIGHT)
      .supportsAttribute(DisplayerAttributeDef.CHART_BGCOLOR)
      .supportsAttribute(DisplayerAttributeGroupDef.CHART_MARGIN_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.METER_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.HTML_GROUP);
}

@Override
public DisplayerConstraints createDisplayerConstraints() {
  DataSetLookupConstraints lookupConstraints = new DataSetLookupConstraints()
      .setGroupAllowed(false)
      .setMaxColumns(1)
      .setMinColumns(1)
      .setFunctionRequired(true)
      .setExtraColumnsAllowed(false)
      .setColumnsTitle(view.getColumnsTitle())
      .setColumnTypes(new ColumnType[] {
          ColumnType.NUMBER});
  return new DisplayerConstraints(lookupConstraints)
      .supportsAttribute(DisplayerAttributeDef.TYPE)
      .supportsAttribute(DisplayerAttributeDef.RENDERER)
      .supportsAttribute(DisplayerAttributeGroupDef.COLUMNS_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.FILTER_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.REFRESH_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.GENERAL_GROUP)
      .supportsAttribute(DisplayerAttributeDef.CHART_WIDTH)
      .supportsAttribute(DisplayerAttributeDef.CHART_HEIGHT)
      .supportsAttribute(DisplayerAttributeDef.CHART_BGCOLOR)
      .supportsAttribute(DisplayerAttributeGroupDef.CHART_MARGIN_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.METER_GROUP)
      .supportsAttribute(DisplayerAttributeGroupDef.HTML_GROUP);
}