org.dbunit.dataset.DataSetException Java Exaples

Source File: DataSetBuilderTest.java From database-rider with Apache License 2.0

6 votes

@Test
public void shouldAddEmptyTable() throws IOException, DataSetException {
    DataSetBuilder builder = new DataSetBuilder();
    IDataSet dataSet = builder.
            table("USER")
            .build();

    File datasetFile = Files.createTempFile("rider-empty-dataset", ".xml").toFile();
    FileOutputStream fos = new FileOutputStream(datasetFile);
    FlatXmlDataSet.write(dataSet, fos);
    assertThat(contentOf(datasetFile)).
            contains("<?xml version='1.0' encoding='UTF-8'?>"+NEW_LINE +
                    "<dataset>"+NEW_LINE +
                    "  <USER/>"+NEW_LINE +
                    "</dataset>");

}

Source File: DataSetBuilderExporterIt.java From database-rider with Apache License 2.0

6 votes

@Test
public void shouldExportJSONDataSetUsingColumnValuesSyntax() throws DataSetException {
    executor.createDataSet(new DataSetConfig("json/users.json"));
    IDataSet iDataSet = createDataSetFromDatabase();
    File outputDir = Paths.get("target/FromJSONBuilder.java").toAbsolutePath().toFile();
    new DataSetBuilderExporter().export(iDataSet, new BuilderExportConfig(BuilderType.COLUMNS_VALUES, outputDir));
    assertThat(contentOf(outputDir)).isEqualTo("DataSetBuilder builder = new DataSetBuilder();" + NEW_LINE +
            "IDataSet dataSet = builder" + NEW_LINE +
            "    .table(\"FOLLOWER\")" + NEW_LINE +
            "        .columns(\"ID\", \"USER_ID\", \"FOLLOWER_ID\")" + NEW_LINE +
            "        .values(1, 1, 2)" + NEW_LINE +
            "    .table(\"SEQUENCE\")" + NEW_LINE +
            "        .columns(\"SEQ_NAME\", \"SEQ_COUNT\")" + NEW_LINE +
            "        .values(\"SEQ_GEN\", 0)" + NEW_LINE +
            "    .table(\"TWEET\")" + NEW_LINE +
            "        .columns(\"ID\", \"CONTENT\", \"DATE\", \"LIKES\", \"TIMESTAMP\", \"USER_ID\")" + NEW_LINE +
            "        .values(\"abcdef12345\", \"dbunit rules json example\", \"" +
            "" + iDataSet.getTable("TWEET").getValue(0,"DATE") +
            "\", null, null, 1)" + NEW_LINE +
            "    .table(\"USER\")" + NEW_LINE +
            "        .columns(\"ID\", \"NAME\")" + NEW_LINE +
            "        .values(1, \"@realpestano\")" + NEW_LINE +
            "        .values(2, \"@dbunit\").build();");
}

Source File: JSONWriter.java From database-rider with Apache License 2.0

6 votes

@Override
public void row(Object[] values) throws DataSetException {
	rowCount++;
	try {
		out.write(FOUR_SPACES+"{"+NEW_LINE);
		String sb = createSetFromValues(values);
		out.write(sb);
		if(dataSet.getTable(metaData.getTableName()).getRowCount() != rowCount ){
			out.write(FOUR_SPACES+"},"+NEW_LINE);
		}else {
			out.write(FOUR_SPACES+"}"+NEW_LINE);
		}


	} catch (Exception e) {
		logger.warn("Could not write row.", e);
	}
}

Source File: DataSetBuilderExporterIt.java From database-rider with Apache License 2.0

6 votes

@Test
public void shouldExportXMLDataSetUsingColumnValuesSyntax() throws DataSetException {
    IDataSet iDataSet = new FlatXmlDataSetBuilder().build(getDataSetStream("datasets/xml/users.xml"));
    File outputDir = Paths.get("target/FromXMlBuilder.java").toAbsolutePath().toFile();
    new DataSetBuilderExporter().export(iDataSet, new BuilderExportConfig(BuilderType.COLUMNS_VALUES, outputDir));
    assertThat(contentOf(outputDir)).isEqualTo("DataSetBuilder builder = new DataSetBuilder();" + NEW_LINE +
            "IDataSet dataSet = builder" + NEW_LINE +
            "    .table(\"USER\")" + NEW_LINE +
            "        .columns(\"id\", \"name\")" + NEW_LINE +
            "        .values(\"1\", \"@realpestano\")" + NEW_LINE +
            "        .values(\"2\", \"@dbunit\")" + NEW_LINE +
            "    .table(\"TWEET\")" + NEW_LINE +
            "        .columns(\"id\", \"content\", \"user_id\")" + NEW_LINE +
            "        .values(\"abcdef12345\", \"dbunit rules flat xml example\", \"1\")" + NEW_LINE +
            "    .table(\"FOLLOWER\")" + NEW_LINE +
            "        .columns(\"id\", \"user_id\", \"follower_id\")" + NEW_LINE +
            "        .values(\"1\", \"1\", \"2\").build();");
}

Source File: SqlDbFeatureExecutorTest.java From jpa-unit with Apache License 2.0

6 votes

@Test
public void testLoadDataSetsUsingAvailableFilePaths() throws DataSetException {
    // GIVEN

    // WHEN
    final List<IDataSet> dataSetList = featureExecutor.loadDataSets(Arrays.asList("test-data.json", "test-data.json"));

    // THEN
    assertNotNull(dataSetList);
    assertThat(dataSetList.size(), equalTo(2));

    final IDataSet ds1 = dataSetList.get(0);
    assertNotNull(ds1);
    assertThat(ds1.getTableNames().length, equalTo(2));

    final IDataSet ds2 = dataSetList.get(0);
    assertNotNull(ds2);
    assertThat(ds2.getTableNames().length, equalTo(2));

    assertThat(ds1, equalTo(ds2));
}

Source File: DataSetBuilderExporterIt.java From database-rider with Apache License 2.0

6 votes

@Test
public void shouldExportYamlDataSetUsingColumnValuesSyntax() throws DataSetException {
    executor.createDataSet(new DataSetConfig("yml/users.yml"));
    IDataSet iDataSet = createDataSetFromDatabase();
    File outputDir = Paths.get("target/FromYamlBuilder.java").toAbsolutePath().toFile();
    new DataSetBuilderExporter().export(iDataSet, new BuilderExportConfig(BuilderType.COLUMNS_VALUES, outputDir));
    assertThat(contentOf(outputDir)).isEqualTo("DataSetBuilder builder = new DataSetBuilder();" + NEW_LINE +
            "IDataSet dataSet = builder" + NEW_LINE +
            "    .table(\"FOLLOWER\")" + NEW_LINE +
            "        .columns(\"ID\", \"USER_ID\", \"FOLLOWER_ID\")" + NEW_LINE +
            "        .values(1, 1, 2)" + NEW_LINE +
            "    .table(\"SEQUENCE\")" + NEW_LINE +
            "        .columns(\"SEQ_NAME\", \"SEQ_COUNT\")" + NEW_LINE +
            "        .values(\"SEQ_GEN\", 0)" + NEW_LINE +
            "    .table(\"TWEET\")" + NEW_LINE +
            "        .columns(\"ID\", \"CONTENT\", \"DATE\", \"LIKES\", \"TIMESTAMP\", \"USER_ID\")" + NEW_LINE +
            "        .values(\"abcdef12345\", \"dbunit rules!\", \"" +
            "" + iDataSet.getTable("TWEET").getValue(0,"DATE") +
            "\", null, null, 1)" + NEW_LINE +
            "    .table(\"USER\")" + NEW_LINE +
            "        .columns(\"ID\", \"NAME\")" + NEW_LINE +
            "        .values(1, \"@realpestano\")" + NEW_LINE +
            "        .values(2, \"@dbunit\").build();");
}

Source File: DataSetComparator.java From jpa-unit with Apache License 2.0

6 votes

private List<String> extractColumnsToBeIgnored(final ITable expectedTableState, final ITable currentTableState)
        throws DataSetException {
    final List<String> columnsToIgnore = extractNotExpectedColumnNames(expectedTableState, currentTableState);
    final String tableName = expectedTableState.getTableMetaData().getTableName();

    columnsToIgnore.addAll(toExclude.getColumns(tableName));

    final List<String> nonExistingColumns = new ArrayList<>(columnsToIgnore);
    nonExistingColumns.removeAll(extractColumnNames(currentTableState.getTableMetaData().getColumns()));

    if (!nonExistingColumns.isEmpty()) {
        LOG.debug("Columns which are specified to be filtered out {} are not existing in the table {}",
                Arrays.toString(nonExistingColumns.toArray()), tableName);
    }
    return columnsToIgnore;
}

Source File: DataSetProviderIt.java From database-rider with Apache License 2.0

6 votes

@Override
public IDataSet provide() throws DataSetException {
    DataSetBuilder builder = new DataSetBuilder();
    ColumnSpec id = ColumnSpec.of("ID");
    ColumnSpec name = ColumnSpec.of("NAME");
    builder.table("USER")
            .row()
                .column("ID", 1)
                .column(name, "@realpestano")
            .table("USER")
            .row()
                .column(id, 2).column("NAME", "@dbunit")
           .table("TWEET")
           .row()
                .column("ID", "abcdef12345")
                .column("CONTENT", "dbunit rules!")
                .column("DATE", "[DAY,NOW]")
                .column("USER_ID", 9999)
        .build();

    return builder.build();
}

Source File: ScriptableTableTest.java From database-rider with Apache License 2.0

6 votes

@Test
public void testGetValue() throws DataSetException {
    // prepare
    when(iTable.getValue(0, "category")).thenReturn("POPE:01");
    when(manager.getEngineByName(anyString())).thenReturn(null);
    
    // test
    Object value = scriptableTable.getValue(0, "category");
    
    // assert/verify
    assertEquals("POPE:01", value);
    verify(ScriptableTable.log).warning("Could not find script engine by name 'POPE'");
    verify(manager).getEngineByName("POPE");
    verify(iTable).getValue(0, "category");
    verifyNoMoreInteractions(ScriptableTable.log, manager, iTable);
}

Source File: DataSetBuilderTest.java From database-rider with Apache License 2.0

6 votes

@Test
public void shouldGeneratFlatXmlDataSetWithEmptyUserTableWithColumnValuesSyntax() throws DataSetException, IOException {
    DataSetBuilder builder = new DataSetBuilder();
    builder
            .table("USER")
            .table("TWEET")
                .columns("ID","CONTENT","DATE")
                .values("abcdef12345", "dbunit rules!", "[DAY,NOW]")
            .table("FOLLOWER")
                .columns("ID", "USER_ID","FOLLOWER_ID")
                .values(1 , 1 , 2)
            .build();

    IDataSet dataSet = builder.build();

    File datasetFile = Files.createTempFile("rider-dataset", ".xml").toFile();
    FileOutputStream fos = new FileOutputStream(datasetFile);
    FlatXmlDataSet.write(dataSet, fos);
    assertThat(contentOf(datasetFile)).
            contains("<?xml version='1.0' encoding='UTF-8'?>"+NEW_LINE +
                    "<dataset>"+NEW_LINE +
                    "  <USER/>"+NEW_LINE +
                    "  <TWEET ID=\"abcdef12345\" CONTENT=\"dbunit rules!\" DATE=\"[DAY,NOW]\"/>"+NEW_LINE +
                    "  <FOLLOWER ID=\"1\" USER_ID=\"1\" FOLLOWER_ID=\"2\"/>"+NEW_LINE +
                    "</dataset>");
}

Source File: KylinTestBase.java From kylin-on-parquet-v2 with Apache License 2.0

6 votes

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());
}

Source File: HackedDbUnitAssert.java From kylin-on-parquet-v2 with Apache License 2.0

6 votes

private void compareDataContains(ITable expectedTable, ITable actualTable, ComparisonColumn[] comparisonCols, FailureHandler failureHandler) throws DataSetException {
    logger.debug("compareData(expectedTable={}, actualTable={}, " + "comparisonCols={}, failureHandler={}) - start", new Object[] { expectedTable, actualTable, comparisonCols, failureHandler });

    if (expectedTable == null) {
        throw new NullPointerException("The parameter 'expectedTable' must not be null");
    }
    if (actualTable == null) {
        throw new NullPointerException("The parameter 'actualTable' must not be null");
    }
    if (comparisonCols == null) {
        throw new NullPointerException("The parameter 'comparisonCols' must not be null");
    }
    if (failureHandler == null) {
        throw new NullPointerException("The parameter 'failureHandler' must not be null");
    }

    for (int index = 0; index < actualTable.getRowCount(); index++) {
        if (!findRowInExpectedTable(expectedTable, actualTable, comparisonCols, failureHandler, index)) {
            throw new IllegalStateException();
        }
    }

}

Source File: DataSetProviderIt.java From database-rider with Apache License 2.0

5 votes

@Override
public IDataSet provide() throws DataSetException {
    DataSetBuilder builder = new DataSetBuilder();
    builder.table("TWEET")
            .row()
                .column("ID", "abcdef12345")
                .column("CONTENT", "dbrider rules!")
                .column("DATE", "[DAY,NOW]");
    return builder.build();
}

Source File: DataSetProviderIt.java From database-rider with Apache License 2.0

5 votes

@Override
public IDataSet provide() throws DataSetException {
    DataSetBuilder builder = new DataSetBuilder();
    builder.table("user")
            .row()
                .column("id", 2)
                .column("name", "@dbrider");
    return builder.build();
}

Source File: DataSetComparator.java From jpa-unit with Apache License 2.0

5 votes

private List<String> extractNotExpectedColumnNames(final ITable expectedTable, final ITable currentTable) throws DataSetException {
    final Set<String> actualColumnNames = new HashSet<>();
    final Set<String> expectedColumnNames = new HashSet<>();

    if (currentTable != null) {
        actualColumnNames.addAll(extractColumnNames(currentTable.getTableMetaData().getColumns()));
    }

    if (expectedTable != null) {
        expectedColumnNames.addAll(extractColumnNames(expectedTable.getTableMetaData().getColumns()));
    }

    actualColumnNames.removeAll(expectedColumnNames);
    return new ArrayList<>(actualColumnNames);
}

Source File: JsonDataSetProducer.java From jpa-unit with Apache License 2.0

5 votes

@SuppressWarnings("unchecked")
@Override
protected Map<String, List<Map<String, String>>> loadDataSet() throws DataSetException {
    try {
        return gson.fromJson(new InputStreamReader(input), Map.class);
    } catch (final Exception e) {
        throw new DataSetException("Error parsing json data set", e);
    }
}

Source File: DataSetProducer.java From jpa-unit with Apache License 2.0

5 votes

@Override
public void produce() throws DataSetException {

    consumer.startDataSet();

    final Map<String, List<Map<String, String>>> dataset = loadDataSet();

    for (final Map.Entry<String, List<Map<String, String>>> entry : dataset.entrySet()) {
        // an entry represents a table
        final List<Map<String, String>> rows = entry.getValue();
        // each row represents a record in a table
        final Collection<String> columnNames = extractColumnNames(rows);
        final ITableMetaData tableMetaData = new DefaultTableMetaData(entry.getKey(), createColumns(columnNames));

        consumer.startTable(tableMetaData);

        for (final Map<String, String> row : rows) {
            final List<String> values = new ArrayList<>();
            for (final Column column : tableMetaData.getColumns()) {
                final Object rawValue = row.get(column.getColumnName());
                final String value = rawValue == null ? null : String.valueOf(rawValue);
                values.add(value);
            }
            consumer.row(values.toArray());
        }

        consumer.endTable();

    }

    consumer.endDataSet();
}

Source File: ScriptableTable.java From database-rider with Apache License 2.0

5 votes

@Override
public Object getValue(int row, String column) throws DataSetException {
    Object value = delegate.getValue(row, column);
    if (value != null && scriptEnginePattern.matcher(value.toString()).matches()) {
        ScriptEngine engine = getScriptEngine(value.toString().trim());
        if (engine != null) {
            try {
                return getScriptResult(value.toString(), engine);
            } catch (Exception e) {
                log.log(Level.WARNING,String.format("Could not evaluate script expression for table '%s', column '%s'. The original value will be used.", getTableMetaData().getTableName(), column),e);
            }
        }
    }
    return value;
}

Source File: AbstractDbUnitTestCase.java From wetech-cms with MIT License

5 votes

protected void backupCustomTable(String[] tname) throws DataSetException, IOException {
	QueryDataSet ds = new QueryDataSet(dbunitCon);
	for(String str:tname) {
		ds.addTable(str);
	}
	writeBackupFile(ds);
}

Source File: DataSetComparator.java From jpa-unit with Apache License 2.0

5 votes

private List<String> defineColumnsForSorting(final IDataSet currentDataSet, final IDataSet expectedDataSet, final String tableName)
        throws DataSetException {

    final List<String> additionalColumns = additionalColumnsForSorting(expectedDataSet.getTable(tableName),
            currentDataSet.getTable(tableName));

    final List<String> result = new ArrayList<>();
    result.addAll(orderBy.getColumns(tableName));
    result.addAll(additionalColumns);
    return result;
}

Source File: DataSetBuilderExporterIt.java From database-rider with Apache License 2.0

5 votes

@Test
public void shouldExportJSONDataSetUsingDefaultSyntax() throws DataSetException {
    executor.createDataSet(new DataSetConfig("json/users.json"));
    IDataSet iDataSet = createDataSetFromDatabase();
    File outputDir = Paths.get("target/FromJSONDefaultBuilder.java").toAbsolutePath().toFile();
    new DataSetBuilderExporter().export(iDataSet, new BuilderExportConfig(BuilderType.DEFAULT, outputDir));
    assertThat(contentOf(outputDir)).isEqualTo("DataSetBuilder builder = new DataSetBuilder();" + NEW_LINE +
            "IDataSet dataSet = builder" + NEW_LINE +
            "    .table(\"FOLLOWER\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"ID\", 1)" + NEW_LINE +
            "        .column(\"USER_ID\", 1)" + NEW_LINE +
            "        .column(\"FOLLOWER_ID\", 2)" + NEW_LINE +
            "    .table(\"SEQUENCE\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"SEQ_NAME\", \"SEQ_GEN\")" + NEW_LINE +
            "        .column(\"SEQ_COUNT\", 0)" + NEW_LINE +
            "    .table(\"TWEET\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"ID\", \"abcdef12345\")" + NEW_LINE +
            "        .column(\"CONTENT\", \"dbunit rules json example\")" + NEW_LINE +
            "        .column(\"DATE\", \"" + iDataSet.getTable("TWEET").getValue(0,"DATE") + "\")" + NEW_LINE +
            "        .column(\"LIKES\", null)" + NEW_LINE +
            "        .column(\"TIMESTAMP\", null)" + NEW_LINE +
            "        .column(\"USER_ID\", 1)" + NEW_LINE +
            "    .table(\"USER\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"ID\", 1)" + NEW_LINE +
            "        .column(\"NAME\", \"@realpestano\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"ID\", 2)" + NEW_LINE +
            "        .column(\"NAME\", \"@dbunit\").build();");
}

Source File: AbstractDbUnitTestCase.java From wetech-cms with MIT License

5 votes

protected void backupCustomTable(String[] tname) throws DataSetException, IOException {
	QueryDataSet ds = new QueryDataSet(dbunitCon);
	for(String str:tname) {
		ds.addTable(str);
	}
	writeBackupFile(ds);
}

Source File: AbstractDbUnitTestCase.java From wetech-cms with MIT License

5 votes

protected IDataSet createDateSet(String tname) throws DataSetException, FileNotFoundException, IOException {
	InputStream is = AbstractDbUnitTestCase
				.class
				.getClassLoader().getResourceAsStream(tname+".xml");
	Assert.assertNotNull("dbunit的基本数据文件不存在",is);
	//通过dtd和传入的文件创建测试的IDataSet
	return new FlatXmlDataSet(new FlatXmlProducer(new InputSource(is),new FlatDtdDataSet(new FileReader(dtdFile))));
}

Source File: DataSetComparator.java From jpa-unit with Apache License 2.0

5 votes

private List<String> additionalColumnsForSorting(final ITable expectedTableState, final ITable currentTableState)
        throws DataSetException {
    final List<String> columnsForSorting = new ArrayList<>();
    final Set<String> allColumns = new HashSet<>(extractColumnNames(expectedTableState.getTableMetaData().getColumns()));
    final Set<String> columnsToIgnore = new HashSet<>(extractColumnsToBeIgnored(expectedTableState, currentTableState));
    for (final String column : allColumns) {
        if (!columnsToIgnore.contains(column)) {
            columnsForSorting.add(column);
        }
    }

    return columnsForSorting;
}

Source File: DataSetLoaderProviderTest.java From jpa-unit with Apache License 2.0

5 votes

@Test(expected = NullPointerException.class)
public void testXlsLoaderLoadUsingNullFileName() throws IOException, DataSetException {
    // WHEN
    final DataSetLoader<IDataSet> loader = LOADER_PROVIDER.xlsLoader();

    // THEN
    assertThat(loader, notNullValue());

    // WHEN
    loader.load(null);

    // THEN
    // IOException is thrown
}

Source File: DataSetExecutorImplTest.java From database-rider with Apache License 2.0

5 votes

@Test(expected = RuntimeException.class)
public void shouldFailLoadingDatasetFromHttp() throws IOException, DataSetException {
    DataSetExecutorImpl dse = DataSetExecutorImpl.instance(new ConnectionHolder() {
        private static final long serialVersionUID = 1L;

        @Override
        public Connection getConnection() {
            return null;
        }
    });
    dse.loadDataSet("http://somewhere.de/test.csv");
}

Source File: DataSetBuilderTest.java From database-rider with Apache License 2.0

5 votes

@Test
public void shouldGenerateDataSetUsingMetaModel() throws DataSetException, IOException {

    EntityManagerProvider.instance("contactPU");
    DataSetBuilder builder = new DataSetBuilder();
    Calendar calendar = Calendar.getInstance();
    calendar.add(Calendar.YEAR, 1);
    Date date = new Date();

    IDataSet dataSet = builder.table("CONTACT")
            .row()
                .column(Contact_.id, 1)
                .column(Contact_.name, "dbrider")
                .column(Contact_.date,date )
                .column(Contact_.calendar, calendar)
            .build();


    File datasetFile = Files.createTempFile("rider-dataset", ".yml").toFile();
    FileOutputStream fos = new FileOutputStream(datasetFile);
    new YMLWriter(fos).write(dataSet);


    assertThat(contentOf(datasetFile)).
            contains("CONTACT:" + NEW_LINE +
                    "  - ID: \"1\"" + NEW_LINE +
                    "    NAME: \"dbrider\"" + NEW_LINE +
                    "    DATE: \"" + DateUtils.format(date) +"\"" + NEW_LINE +
                    "    CALENDAR: \"" + DateUtils.format(calendar.getTime()) +"\"" + NEW_LINE +
                    "");
}

Source File: DataSetBuilderExporterIt.java From database-rider with Apache License 2.0

5 votes

@Test
public void shouldExportYamlDataSetUsingDefaultSyntax() throws DataSetException {
    executor.createDataSet(new DataSetConfig("yml/users.yml"));
    IDataSet iDataSet = createDataSetFromDatabase();
    File outputDir = Paths.get("target/FromYamlDefaultBuilder.java").toAbsolutePath().toFile();
    new DataSetBuilderExporter().export(iDataSet, new BuilderExportConfig(BuilderType.DEFAULT, outputDir));
    assertThat(contentOf(outputDir)).isEqualTo("DataSetBuilder builder = new DataSetBuilder();" + NEW_LINE +
            "IDataSet dataSet = builder" + NEW_LINE +
            "    .table(\"FOLLOWER\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"ID\", 1)" + NEW_LINE +
            "        .column(\"USER_ID\", 1)" + NEW_LINE +
            "        .column(\"FOLLOWER_ID\", 2)" + NEW_LINE +
            "    .table(\"SEQUENCE\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"SEQ_NAME\", \"SEQ_GEN\")" + NEW_LINE +
            "        .column(\"SEQ_COUNT\", 0)" + NEW_LINE +
            "    .table(\"TWEET\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"ID\", \"abcdef12345\")" + NEW_LINE +
            "        .column(\"CONTENT\", \"dbunit rules!\")" + NEW_LINE +
            "        .column(\"DATE\", \"" +
            "" + iDataSet.getTable("TWEET").getValue(0,"DATE") +
            "\")" + NEW_LINE +
            "        .column(\"LIKES\", null)" + NEW_LINE +
            "        .column(\"TIMESTAMP\", null)" + NEW_LINE +
            "        .column(\"USER_ID\", 1)" + NEW_LINE +
            "    .table(\"USER\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"ID\", 1)" + NEW_LINE +
            "        .column(\"NAME\", \"@realpestano\")" + NEW_LINE +
            "    .row()" + NEW_LINE +
            "        .column(\"ID\", 2)" + NEW_LINE +
            "        .column(\"NAME\", \"@dbunit\").build();");
}

Source File: DataSetLoaderProviderTest.java From jpa-unit with Apache License 2.0

5 votes

@Test(expected = NullPointerException.class)
public void testXmlLoaderLoadUsingNullFileName() throws IOException, DataSetException {
    // WHEN
    final DataSetLoader<IDataSet> loader = LOADER_PROVIDER.xmlLoader();

    // THEN
    assertThat(loader, notNullValue());

    // WHEN
    loader.load(null);

    // THEN
    // IOException is thrown
}

Source File: DataSetBuilderTest.java From database-rider with Apache License 2.0

5 votes

@Test
public void shouldGenerateDataSetcolumnDateColumn() throws DataSetException, IOException {
    DataSetBuilder builder = new DataSetBuilder();
    Calendar calendar = Calendar.getInstance();
    calendar.add(Calendar.YEAR, 1);
    Date date = new Date();

    IDataSet dataSet = builder.table("USER")
            .row()
                .column("ID", 1)
                .column("DATE", date)
                .column("CALENDAR", calendar)
            .build();


    File datasetFile = Files.createTempFile("rider-dataset", ".yml").toFile();
    FileOutputStream fos = new FileOutputStream(datasetFile);
    new YMLWriter(fos).write(dataSet);


    assertThat(contentOf(datasetFile)).
            contains("USER:" + NEW_LINE +
                    "  - ID: \"1\"" + NEW_LINE +
                    "    DATE: \"" + DateUtils.format(date) +"\"" + NEW_LINE +
                    "    CALENDAR: \"" + DateUtils.format(calendar.getTime()) +"\"" + NEW_LINE +
                    "");
}

org.dbunit.dataset.DataSetException Java Examples