Java IDatabaseConnection.createDataSet方法代码示例

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
@Override
protected DbFeature<IDatabaseConnection> createVerifyDataAfterFeature(final ExpectedDataSets expectedDataSets) {
    return (final IDatabaseConnection connection) -> {
        try {
            final IDataSet currentDataSet = connection.createDataSet();
            final IDataSet expectedDataSet = mergeDataSets(loadDataSets(Arrays.asList(expectedDataSets.value())));

            final DataSetComparator dataSetComparator = new DataSetComparator(expectedDataSets.orderBy(),
                    expectedDataSets.excludeColumns(), expectedDataSets.strict(), getColumnFilter(expectedDataSets));

            final AssertionErrorCollector errorCollector = new AssertionErrorCollector();
            dataSetComparator.compare(currentDataSet, expectedDataSet, errorCollector);

            errorCollector.report();
        } catch (final SQLException | DatabaseUnitException e) {
            throw new DbFeatureException("Could not execute DB contents verification feature", e);
        }
    };
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
public void compareTables(String expectedDatasetFileName, String... tableNames) throws Exception {
	JdbcDatabaseTester jdbc = new JdbcDatabaseTester("org.hsqldb.jdbcDriver", "jdbc:hsqldb:file:./db/example-db;ifexists=true;shutdown=true;", "sa", "");
	IDatabaseConnection connection = jdbc.getConnection();
	
	DatabaseConfig config = connection.getConfig();
	config.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new HsqldbDataTypeFactory());
	
	IDataSet databaseDataSet = connection.createDataSet();
	IDataSet expectedDataSet = new FlatXmlDataFileLoader().load(expectedDatasetFileName);
	
	for (String tableName : tableNames) {
		ITable actualTable = databaseDataSet.getTable(tableName);
		ITable expectedTable = expectedDataSet.getTable(tableName);
		Assertion.assertEquals(expectedTable, actualTable);
	}
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
@Test
public void doRecreate() throws DatabaseUnitException, SQLException, FileNotFoundException {
	IDatabaseConnection dconnection = new DatabaseConnection(dataSource.getConnection());
	IDataSet dataSet = dconnection.createDataSet();
	File file = new File("formbuilder-dataset.dtd");
	
	Assert.assertTrue("DTD file can not be deleted.", !file.exists() || file.delete());
	
	Writer out = new OutputStreamWriter(new FileOutputStream(file));
	FlatDtdWriter datasetWriter = new FlatDtdWriter(out);
	datasetWriter.setContentModel(FlatDtdWriter.CHOICE);
	// You could also use the sequence model which is the default
	// datasetWriter.setContentModel(FlatDtdWriter.SEQUENCE);
	datasetWriter.write(dataSet);
	
	Assert.assertTrue("DTD file did not created.", file.exists());
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
public static void main(String[] args) throws Exception
{
    Class driverClass = Class.forName("org.postgresql.Driver");
    Connection jdbcConnection = DriverManager.getConnection(
            "jdbc:postgresql://localhost:5432/FormBuilder?autoReconnect=true", "fbdev", "fbdev");
    IDatabaseConnection connection = new DatabaseConnection(jdbcConnection);

    // partial database export
    /*QueryDataSet partialDataSet = new QueryDataSet(connection);
    partialDataSet.addTable("FOO", "SELECT * FROM TABLE WHERE COL='VALUE'");
    partialDataSet.addTable("BAR");
    FlatXmlDataSet.write(partialDataSet, new FileOutputStream("partial.xml"));*/

    // full database export
    IDataSet fullDataSet = connection.createDataSet();
    FlatXmlDataSet.write(fullDataSet, new FileOutputStream("skips_on_table.xml"));
    
    // dependent tables database export: export table X and all tables that
    // have a PK which is a FK on X, in the right order for insertion
    /*String[] depTableNames = 
      TablesDependencyHelper.getAllDependentTables( connection, "X" );
    IDataSet depDataset = connection.createDataSet( depTableNames );
    FlatXmlDataSet.write(depDataSet, new FileOutputStream("dependents.xml"));*/          
    
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
/**
 * Generates the actual content of the DTD file as an in-memory string.
 *
 * @return the DTD content, not null
 */
protected String generateDtdContent() {
    Connection conn = null;
    try {
        conn = sqlHandler.getDataSource().getConnection();
        IDatabaseConnection dbUnitDatabaseConnection = new DatabaseConnection(conn, defaultDbSupport.getSchemaName());

        StringWriter stringWriter = new StringWriter();

        FlatDtdWriter datasetWriter = new FlatDtdWriter(stringWriter);
        datasetWriter.setContentModel(FlatDtdWriter.CHOICE);

        // create a dataset for the database content
        // filter out all system table names
        Set<String> tableNames = defaultDbSupport.getTableNames();
        IDataSet actualDataSet = dbUnitDatabaseConnection.createDataSet();
        IDataSet filteredActualDataSet = new FilteredDataSet(new IncludeTableFilter(tableNames.toArray(new String[0])), actualDataSet);

        datasetWriter.write(filteredActualDataSet);
        return stringWriter.toString();

    } catch (Exception e) {
        throw new UnitilsException("Error generating content for DTD file.", e);
    } finally {
        DbUtils.closeQuietly(conn);
    }
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
@After
@Transactional
public void dumpDatabase() throws SQLException, DatabaseUnitException, FileNotFoundException, IOException {
    // full database export
    Connection jdbcConnection = dataSource.getConnection();
    IDatabaseConnection connection = new DatabaseConnection(jdbcConnection);
    IDataSet fullDataSet = connection.createDataSet();
    FlatXmlDataSet.write(fullDataSet, new FileOutputStream("./target/QueryKeyRepositoryTest-dbunit.xml"));
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
@After
@Transactional
public void dumpDatabase() throws SQLException, DatabaseUnitException, FileNotFoundException, IOException {
    // full database export
    Connection jdbcConnection = dataSource.getConnection();
    IDatabaseConnection connection = new DatabaseConnection(jdbcConnection);
    IDataSet fullDataSet = connection.createDataSet();
    FlatXmlDataSet.write(fullDataSet, new FileOutputStream(
        "./target/TermFrequencyResultsDataManagerTest-dbunit.xml"));
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
/**
 * 从数据库中读取数据写入到文件中，支持excel和xml格式，以文件后缀名区分
 * @param databaseConnection  使用的IDatabaseConnection数据连接
 * @param destFilePath 写入文件的路径
 */
public static void writeToFileFromDataBase(IDatabaseConnection databaseConnection,String destFilePath){
	IDataSet dataSet=null;
	
	try {
		dataSet=databaseConnection.createDataSet();
	} catch (Exception e) {
		log.error("未获取到测试集");
		log.error(e.getMessage());
		return;
	}
	writeToFileFromDataSet(dataSet,destFilePath);
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
protected void compareTables(String expectedDatasetFileName, String... tableNames) throws Exception {
	IDatabaseConnection connection = jdt.getConnection();
	
	DatabaseConfig config = connection.getConfig();
	config.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new HsqldbDataTypeFactory());
	
	IDataSet databaseDataSet = connection.createDataSet();
	IDataSet expectedDataSet = new FlatXmlDataFileLoader().load(expectedDatasetFileName);
	for (String tableName : tableNames) {
		ITable actualTable = databaseDataSet.getTable(tableName);
		ITable expectedTable = expectedDataSet.getTable(tableName);
		Assertion.assertEquals(expectedTable, actualTable);
	}
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
public static void main(String[] args) throws Exception {
    Config config = new Config();
    Class.forName(config.getDbDriver());
    Connection jdbcConnection = DriverManager.getConnection(config.getDbUrl(), config.getDbUser(), config.getDbPass());
    IDatabaseConnection connection = new DatabaseConnection(jdbcConnection, "public");

    // dtd
    FlatDtdDataSet.write(connection.createDataSet(), new FileOutputStream("src/test/java/it/pronetics/acl/conf/dbunit/database-schema.dtd"));

    // export
    IDataSet fullDataSet = connection.createDataSet();
    FlatXmlDataSet.write(fullDataSet, new FileOutputStream("src/test/java/opennms/opennms/acl/conf/dbunit/full.xml"));
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
/**
 * Performs the extraction. If no tables or queries are specified, data from
 * entire database will be extracted.
 * Otherwise, a partial extraction will be performed.
 * 
 * @throws Exception
 */
public void extract() throws Exception {
    Connection conn = null;
    try {
        conn = dataSource.getConnection();
        logger.info("Beginning extraction from '" + conn.toString() + "'.");
        IDatabaseConnection connection = new DatabaseConnection(conn, schema);
        configConnection((DatabaseConnection) connection);
        if (tableList != null || queryList != null) {
            // partial database export
            QueryDataSet partialDataSet = new QueryDataSet(connection);
            addTables(partialDataSet);
            addQueries(partialDataSet);
            FlatXmlDataSet.write(partialDataSet, new FileOutputStream(dataSetName));
        } else {
            // full database export
            IDataSet fullDataSet = connection.createDataSet();
            FlatXmlDataSet.write(fullDataSet, new FileOutputStream(dataSetName));

            // dependent tables database export: export table X and all
            // tables that
            // have a PK which is a FK on X, in the right order for
            // insertion
            String[] depTableNames = TablesDependencyHelper.getAllDependentTables(connection, "research");
            IDataSet depDataset = connection.createDataSet(depTableNames);

            FlatXmlWriter datasetWriter = new FlatXmlWriter(new FileOutputStream("target/dependents.xml"));
            datasetWriter.write(depDataset);

        }
    } finally {
        if (conn != null) {
            conn.close();
        }
    }
    logger.info("Completed extraction to '" + dataSetName + "'.");
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
public IDataSet createDataSet() throws GoalExecutionException {
	IDatabaseConnection connection = richDriverConfig.createConnection();
	try {
		String[] dependentTables = exportDataSet.getDependentTables();
		IDataSet dataSet = null;
		if (dependentTables == null) {
			dataSet = connection.createDataSet();
		} else {
			String[] depTableNames = TablesDependencyHelper
					.getAllDependentTables(connection, dependentTables);
			dataSet = connection.createDataSet(depTableNames);
		}

		String[] tableNames = dataSet.getTableNames();
		List<String> filteredTableNameList = new ArrayList<String>(
				Arrays.asList(tableNames));

		String[] excludeTables = exportDataSet.getExcludeTables();
		filteredTableNameList.removeAll(Arrays.asList(excludeTables));

		String[] filteredTableNames = (String[]) filteredTableNameList
				.toArray(new String[filteredTableNameList.size()]);

		FilteredDataSet filteredDataSet = new FilteredDataSet(
				filteredTableNames, dataSet);
		return filteredDataSet;
	} catch (Exception e) {
		try {
			connection.close();
		} catch (SQLException e1) {
		}
		throw new GoalExecutionException("Unable to create data set", e);
	}
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
@Test
public void safelyCsvExportAndImportStandardColumnTables()
        throws IOException, DatabaseUnitException, SQLException {
    IDatabaseConnection dbunit_conn = new DatabaseConnection(conn);

    // cache(save) expected data snap-shot.
    IDataSet expectedDataSet = new CachedDataSet(
            dbunit_conn.createDataSet());

    // export only t1, t2 table to CSV
    QueryDataSet exportDataSet = new QueryDataSet(dbunit_conn);
    exportDataSet.addTable("t1");
    exportDataSet.addTable("t2");
    CsvDataSetWriter.write(exportDataSet, tmpDir);

    // after export, insert new record to t3 (not exported)
    new T3("new label").insertMe(conn);

    // clear & insert from exported CSV
    IDataSet csvDataSet = new CsvDataSet(tmpDir);
    DatabaseOperation.CLEAN_INSERT.execute(dbunit_conn, csvDataSet);

    IDataSet actualDataSet = dbunit_conn.createDataSet();
    Assertion.assertEquals(expectedDataSet.getTable("t1"),
            actualDataSet.getTable("t1"));
    Assertion.assertEquals(expectedDataSet.getTable("t2"),
            actualDataSet.getTable("t2"));
    // this produce assertion error : table record will be 3.
    // Assertion.assertEquals(expectedDataSet.getTable("t3"),
    // actualDataSet.getTable("t3"));
    assertEquals(3, actualDataSet.getTable("t3").getRowCount());
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
@Test
public void customExportAndImportForControlCodeStrings()
        throws IOException, DatabaseUnitException, SQLException {
    IDatabaseConnection dbunit_conn = new DatabaseConnection(conn);

    // cache(save) expected data snap-shot.
    IDataSet expectedDataSet = new CachedDataSet(
            dbunit_conn.createDataSet());

    // H2DB can store and load binary string to varchar column safely.
    Set<T4> a = new T4().findAll(conn);
    for (T4 a2 : a) {
        assertEquals(a2.stringField, UnsignedByte.create0x00to0xFFString());
    }

    // export t4 using base64 binary safely csv dataset writer.
    QueryDataSet exportDataSet = new QueryDataSet(dbunit_conn);
    exportDataSet.addTable("t4");
    CsvBase64BinarySafeDataSetWriter.write(exportDataSet, tmpDir);

    // clear & insert from exported CSV using base64 binary safely csv data producer.
    IDataSet csvDataSet = new CsvBase64BinarySafeDataSet(tmpDir);
    DatabaseOperation.CLEAN_INSERT.execute(dbunit_conn, csvDataSet);

    IDataSet actualDataSet = dbunit_conn.createDataSet();
    Assertion.assertEquals(expectedDataSet.getTable("t4"),
            actualDataSet.getTable("t4"));
}
 

import org.dbunit.database.IDatabaseConnection; //导入方法依赖的package包/类
public static void main(String[] args) throws Exception {
    String driver = System.getProperty("CliDbUnitCsvExportDemo.driver",
            "org.h2.Driver");
    Class.forName(driver);
    String url = System.getProperty("CliDbUnitCsvExportDemo.url",
            "jdbc:h2:mem:CliDbUnitCsvExportDemo");
    String dbUser = System.getProperty("CliDbUnitCsvExportDemo.dbUser",
            "sa");
    String dbPassword = System.getProperty(
            "CliDbUnitCsvExportDemo.dbPassword", "");
    Connection conn = DriverManager.getConnection(url, dbUser, dbPassword);

    CliDbUnitCsvExportDemo demo = new CliDbUnitCsvExportDemo();
    demo.setup(conn);

    SimpleDateFormat sdf1 = new SimpleDateFormat("yyyy-MM-dd_HH-mm-ss");
    Calendar c = Calendar.getInstance();
    File outDir = new File(sdf1.format(c.getTime()));
    outDir.mkdir();

    IDatabaseConnection dbunit_conn = new DatabaseConnection(conn);
    IDataSet dataSet = dbunit_conn.createDataSet();

    CsvBase64BinarySafeDataSetWriter.write(dataSet, outDir);

    conn.close();
}