protected static void printResult(ITable resultTable) throws DataSetException { StringBuilder sb = new StringBuilder(); int columnCount = resultTable.getTableMetaData().getColumns().length; String[] columns = new String[columnCount]; for (int i = 0; i < columnCount; i++) { sb.append(resultTable.getTableMetaData().getColumns()[i].getColumnName()); sb.append("-"); sb.append(resultTable.getTableMetaData().getColumns()[i].getDataType()); sb.append("\t"); columns[i] = resultTable.getTableMetaData().getColumns()[i].getColumnName(); } sb.append("\n"); for (int i = 0; i < resultTable.getRowCount(); i++) { for (int j = 0; j < columns.length; j++) { sb.append(resultTable.getValue(i, columns[j])); sb.append("\t"); } sb.append("\n"); } System.out.println(sb.toString()); }
protected ITable executeDynamicQuery(IDatabaseConnection dbConn, String queryName, String sql, List<String> parameters, boolean needSort) throws Exception { // change join type to match current setting sql = changeJoinType(sql, joinType); PreparedStatement prepStat = dbConn.getConnection().prepareStatement(sql); for (int j = 1; j <= parameters.size(); ++j) { prepStat.setString(j, parameters.get(j - 1).trim()); } ITable queryTable = dbConn.createTable(resultTableName + queryName, prepStat); String[] columnNames = new String[queryTable.getTableMetaData().getColumns().length]; for (int i = 0; i < columnNames.length; i++) { columnNames[i] = queryTable.getTableMetaData().getColumns()[i].getColumnName(); } if (needSort) { queryTable = new SortedTable(queryTable, columnNames); } // printResult(queryTable); return queryTable; }
public static Collection<String> extractColumnNames(final Column[] columns) { final List<String> names = new ArrayList<String>(columns.length); for (Column column : columns) { names.add(column.getColumnName().toLowerCase()); } return names; }
public static Collection<String> extractColumnNames(final Column[] columns) { final List<String> names = new ArrayList<String>(columns.length); for (Column column : columns) { names.add(column.getColumnName().toLowerCase()); } return names; }
public static Collection<String> extractColumnNames(final Column[] columns) { final List<String> names = new ArrayList<String>(columns.length); for (Column column : columns) { names.add(column.getColumnName().toLowerCase()); } return names; }
public static Collection<String> extractColumnNames(final Column[] columns) { final List<String> names = new ArrayList<String>(columns.length); for (Column column : columns) { names.add(column.getColumnName().toLowerCase()); } return names; }
protected ITable executeQuery(IDatabaseConnection dbConn, String queryName, String sql, boolean needSort) throws Exception { // change join type to match current setting sql = changeJoinType(sql, joinType); ITable queryTable = dbConn.createQueryTable(resultTableName + queryName, sql); String[] columnNames = new String[queryTable.getTableMetaData().getColumns().length]; for (int i = 0; i < columnNames.length; i++) { columnNames[i] = queryTable.getTableMetaData().getColumns()[i].getColumnName(); } if (needSort) { queryTable = new SortedTable(queryTable, columnNames); } // printResult(queryTable); return queryTable; }
public void addColumn(Column column) { if (!CollectionUtils.exists(columnNames, new FindColumnPredicate(column.getColumnName()))) { boolean added = columns.add(column); if (added) { columnNames.add(column.getColumnName()); } } }
protected Object getValue(Column column) { return getValue(column.getColumnName()); }
private String toKey(Column column) { return policy.toKey(column.getColumnName()); }
protected Set<String> getColumnsToIgnore(ITableMetaData expectedMetaData, ITableMetaData actualMetaData) throws DataSetException { Column[] notSpecifiedInExpected = Columns.getColumnDiff(expectedMetaData, actualMetaData).getActual(); Set<String> result = new LinkedHashSet<String>(); for (Column column : notSpecifiedInExpected) { result.add(column.getColumnName()); } return result; }
protected Set<String> getColumnsToIgnore(ITableMetaData expectedMetaData, ITableMetaData actualMetaData) throws DataSetException { Column[] notSpecifiedInExpected = Columns.getColumnDiff(expectedMetaData, actualMetaData).getActual(); Set<String> result = new LinkedHashSet<String>(); for (Column column : notSpecifiedInExpected) { result.add(column.getColumnName()); } return result; }
/** * Gets the primary key column names for the given DbUnit table. * * @param dbUnitTable The DbUnit table, not null * @return The pk column names, empty if none found */ protected List<String> getPrimaryKeyColumnNames(ITable dbUnitTable) throws DataSetException { List<String> result = new ArrayList<String>(); for (org.dbunit.dataset.Column column : dbUnitTable.getTableMetaData().getPrimaryKeys()) { result.add(column.getColumnName()); } return result; }
private ITable filterTableByPrimaryKey(ITable expectedDataSet, ITable actualDataSet) throws DataSetException { //PrimaryKeyFilteredTableWrapper uses Set.contains to check PK values, so types should be the same in both datasets DataType dataType = actualDataSet.getTableMetaData().getPrimaryKeys()[0].getDataType(); String pkName = actualDataSet.getTableMetaData().getPrimaryKeys()[0].getColumnName(); Set<Object> pkSet = new HashSet<>(); for (int i = 0; i < expectedDataSet.getRowCount(); i++) { pkSet.add(dataType.typeCast(expectedDataSet.getValue(i, pkName))); } return new PrimaryKeyFilteredTableWrapper(actualDataSet, pkSet); }
@Override public void produce() throws DataSetException { consumer.startDataSet(); @SuppressWarnings("unchecked") final List<Table> tables = createTables((Map<String, List<Map<String, String>>>) createYamlReader().load(input)); for (Table table : tables) { ITableMetaData tableMetaData = createTableMetaData(table); consumer.startTable(tableMetaData); for (Row row : table.getRows()) { List<String> values = new ArrayList<String>(); for (Column column : tableMetaData.getColumns()) { values.add(row.valueOf(column.getColumnName())); } consumer.row(values.toArray()); } consumer.endTable(); } consumer.endDataSet(); }
@Override public void produce() throws DataSetException { consumer.startDataSet(); @SuppressWarnings("unchecked") final List<Table> tables = createTables((Map<String, List<Map<String, String>>>) createYamlReader().load(input)); for (Table table : tables) { ITableMetaData tableMetaData = createTableMetaData(table); consumer.startTable(tableMetaData); for (Row row : table.getRows()) { List<String> values = new ArrayList<String>(); for (Column column : tableMetaData.getColumns()) { values.add(row.valueOf(column.getColumnName())); } consumer.row(values.toArray()); } consumer.endTable(); } consumer.endDataSet(); }
@Override public void produce() throws DataSetException { consumer.startDataSet(); @SuppressWarnings("unchecked") final List<Table> tables = createTables((Map<String, List<Map<String, String>>>) createYamlReader().load(input)); for (Table table : tables) { ITableMetaData tableMetaData = createTableMetaData(table); consumer.startTable(tableMetaData); for (Row row : table.getRows()) { List<String> values = new ArrayList<String>(); for (Column column : tableMetaData.getColumns()) { values.add(row.valueOf(column.getColumnName())); } consumer.row(values.toArray()); } consumer.endTable(); } consumer.endDataSet(); }
@Override public void produce() throws DataSetException { consumer.startDataSet(); Map<String, List<Map<String, String>>> dataset = loadDataSet(); final List<Table> tables = createTables(dataset); for (Table table : tables) { ITableMetaData tableMetaData = createTableMetaData(table); consumer.startTable(tableMetaData); for (Row row : table.getRows()) { List<String> values = new ArrayList<String>(); for (Column column : tableMetaData.getColumns()) { values.add(row.valueOf(column.getColumnName())); } consumer.row(values.toArray()); } consumer.endTable(); } consumer.endDataSet(); }
@Override public void produce() throws DataSetException { consumer.startDataSet(); Map<String, List<Map<String, String>>> dataset = loadDataSet(); final List<Table> tables = createTables(dataset); for (Table table : tables) { ITableMetaData tableMetaData = createTableMetaData(table); consumer.startTable(tableMetaData); for (Row row : table.getRows()) { List<String> values = new ArrayList<String>(); for (Column column : tableMetaData.getColumns()) { values.add(row.valueOf(column.getColumnName())); } consumer.row(values.toArray()); } consumer.endTable(); } consumer.endDataSet(); }
@Override public void produce() throws DataSetException { consumer.startDataSet(); Map<String, List<Map<String, String>>> dataset = loadDataSet(); final List<Table> tables = createTables(dataset); for (Table table : tables) { ITableMetaData tableMetaData = createTableMetaData(table); consumer.startTable(tableMetaData); for (Row row : table.getRows()) { List<String> values = new ArrayList<String>(); for (Column column : tableMetaData.getColumns()) { values.add(row.valueOf(column.getColumnName())); } consumer.row(values.toArray()); } consumer.endTable(); } consumer.endDataSet(); }