import org.dbunit.database.DatabaseDataSourceConnection; //导入依赖的package包/类
@Override
public void beforeTestMethod(TestContext testContext) throws Exception {
    DataSetLocation dsLocation = testContext.getTestInstance().getClass().getAnnotation(DataSetLocation.class);
    if (dsLocation != null) {
        String dataSetResourcePath = dsLocation.value();
        Resource dataSetResource = testContext.getApplicationContext().getResource(dataSetResourcePath);

        if (dataSetResource.exists()) {
            IDataSet dataSet = new FlatXmlDataSetBuilder().build(dataSetResource.getInputStream());
            IDatabaseConnection dbConn = new DatabaseDataSourceConnection(
                    testContext.getApplicationContext().getBean(DataSource.class)
            );
            DatabaseOperation.CLEAN_INSERT.execute(dbConn, dataSet);

            LOG.info("Annotated test, using data set: {}", dataSetResourcePath);
        }
    }
}
 

import org.dbunit.database.DatabaseDataSourceConnection; //导入依赖的package包/类
@AfterClass
public static void tearDownDB() {
    System.out.println("Tearing down DB");
    try (ConfigurableApplicationContext applicationContext = new ClassPathXmlApplicationContext("context.xml")) {
        IDatabaseConnection connection = new DatabaseDataSourceConnection(
                (DataSource) applicationContext.getBean("dataSource"));

        try {
            DatabaseOperation.DELETE_ALL.execute(connection,
                    new FlatXmlDataSetBuilder().build(
                            applicationContext.getResource("/database_tear_down.xml").getInputStream()));
        } finally {
            connection.close();
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}
 

import org.dbunit.database.DatabaseDataSourceConnection; //导入依赖的package包/类
public static void setupDb() {
    try (ConfigurableApplicationContext applicationContext = new ClassPathXmlApplicationContext("context.xml")) {
        IDatabaseConnection connection = new DatabaseDataSourceConnection(
                (DataSource) applicationContext.getBean("dataSource"));

        try {
            DatabaseOperation.CLEAN_INSERT.execute(connection,
                    new FlatXmlDataSetBuilder().build(
                            applicationContext.getResource("/database_seed_minimal.xml").getInputStream()));
        } finally {
            connection.close();
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}
 

import org.dbunit.database.DatabaseDataSourceConnection; //导入依赖的package包/类
@SuppressWarnings("ProhibitedExceptionDeclared")
@Bean
public DatabaseDataSourceConnection testConnection(final DataSource dataSource) throws Exception {
    final DatabaseConfigBean databaseConfigBean = new DatabaseConfigBean();
    databaseConfigBean.setDatatypeFactory(new H2DataTypeFactory());
    databaseConfigBean.setCaseSensitiveTableNames(false);

    final DatabaseDataSourceConnectionFactoryBean databaseDataSourceConnectionFactoryBean = new DatabaseDataSourceConnectionFactoryBean();
    databaseDataSourceConnectionFactoryBean.setDataSource(dataSource);
    databaseDataSourceConnectionFactoryBean.setDatabaseConfig(databaseConfigBean);
    databaseDataSourceConnectionFactoryBean.setSchema("FIND");

    return databaseDataSourceConnectionFactoryBean.getObject();
}
 

import org.dbunit.database.DatabaseDataSourceConnection; //导入依赖的package包/类
public static void main(String[] args) throws Exception {
    DataSource dataSource = new DataSourceFactoryBean().getDataSource();
    File file = new File("src/main/resources/mycollab.dtd");
    IDatabaseConnection connection = new DatabaseDataSourceConnection(dataSource);
    connection.getConfig().setProperty(DatabaseConfig.FEATURE_QUALIFIED_TABLE_NAMES, true);
    // write DTD file
    FlatDtdDataSet.write(connection.createDataSet(), new FileOutputStream(file));
}
 

import org.dbunit.database.DatabaseDataSourceConnection; //导入依赖的package包/类
@BeforeClass
public static void setupDB() {
    System.out.println("Setting up DB");
    try (ConfigurableApplicationContext applicationContext = new ClassPathXmlApplicationContext("context.xml")) {
        IDatabaseConnection connection = new DatabaseDataSourceConnection(
                (DataSource) applicationContext.getBean("dataSource"));

        try {
            DatabaseOperation.CLEAN_INSERT.execute(connection,
                    new FlatXmlDataSetBuilder().build(
                            applicationContext.getResource("/database_seed.xml").getInputStream()));
        } finally {
            connection.close();
        }
    } catch (Exception e) {
        e.printStackTrace();
    }

    System.out.println("Setting up Connector");
    OsiamConnector.Builder oConBuilder = new OsiamConnector.Builder().
            setAuthServerEndpoint(AUTH_ENDPOINT_ADDRESS).
            setResourceServerEndpoint(RESOURCE_ENDPOINT_ADDRESS).
            setClientId(CLIENT_ID).
            setClientSecret(CLIENT_SECRET);
    osiamConnector = oConBuilder.build();

    System.out.println("Retrieving access token");
    accessToken = osiamConnector.retrieveAccessToken("marissa", "koala", Scope.ADMIN);
}
 

import org.dbunit.database.DatabaseDataSourceConnection; //导入依赖的package包/类
@BeforeStory
public void deleteAllData() throws SQLException, DatabaseUnitException {
  DatabaseDataSourceConnection dbConn = new DatabaseDataSourceConnection(dataSource);
  IDataSet dataSet = dbConn.createDataSet(new String[] { "ORGANIZATION", "APPLICATION_USER" });
  DatabaseOperation.DELETE_ALL.execute(dbConn, dataSet);
}