@Test
public void testColumnTypes() throws Exception {
DataSet result =
dataSetManager.lookupDataSet(
DataSetFactory.newDataSetLookupBuilder()
.dataset(EXPENSE_REPORTS)
.column(COLUMN_CITY)
.column(COLUMN_AMOUNT)
.column(COLUMN_DATE)
.buildLookup());
assertThat(result.getColumnByIndex(0).getColumnType()).isEqualTo(ColumnType.LABEL);
assertThat(result.getColumnByIndex(1).getColumnType()).isEqualTo(ColumnType.NUMBER);
assertThat(result.getColumnByIndex(2).getColumnType()).isEqualTo(ColumnType.DATE);
assertThat(String.class.isAssignableFrom(result.getValueAt(0, 0).getClass())).isTrue();
assertThat(Double.class.isAssignableFrom(result.getValueAt(0, 1).getClass())).isTrue();
assertThat(
java.util.Date.class.equals(result.getValueAt(0, 2).getClass())
|| java.sql.Date.class.equals(result.getValueAt(0, 2).getClass())
|| java.sql.Timestamp.class.equals(result.getValueAt(0, 2).getClass()))
.isTrue();
}
