import org.dbunit.DataSourceDatabaseTester; //导入依赖的package包/类
private void loadDatasource(DBUnitDataSet annotation, TestContext testContext) throws Exception{
	DataSource dataSource = testContext.getApplicationContext().getBean(DataSource.class);
	IDatabaseTester databaseTester = new DataSourceDatabaseTester(dataSource, "PUBLIC");	
	
	//Ignore referential integrity so DBUnit can correctly delete rows related to the data set
	dataSource.getConnection().createStatement().execute("SET DATABASE REFERENTIAL INTEGRITY FALSE;");

	for(String sourceFile : annotation.value()){
		InputStream input = this.getClass().getResourceAsStream(sourceFile);
		FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder();
		builder.setColumnSensing(true);
		IDataSet ds = builder.build(input);
		databaseTester.setDataSet(ds);
		databaseTester.onSetup();			
	}
	
	//Add Referential constraints back in, now that the dbunit setup is complete
	dataSource.getConnection().createStatement().execute("SET DATABASE REFERENTIAL INTEGRITY TRUE;");
}
 

import org.dbunit.DataSourceDatabaseTester; //导入依赖的package包/类
private void initialize(DataSource dataSource) throws AmbiguousTableNameException {
    databaseTester = new DataSourceDatabaseTester(dataSource);
    databaseTester.setSetUpOperation(DatabaseOperation.CLEAN_INSERT);
    databaseTester.setTearDownOperation(DatabaseOperation.DELETE_ALL);
    DefaultDataSet dataSet = new DefaultDataSet();
    dataSet.addTable(new DefaultTable("agents"));

    dataSet.addTable(new DefaultTable("pipelines"));
    dataSet.addTable(new DefaultTable("pipelinestates"));
    dataSet.addTable(new DefaultTable("materials"));
    dataSet.addTable(new DefaultTable("modifications"));
    dataSet.addTable(new DefaultTable("pipelineMaterialRevisions"));
    dataSet.addTable(new DefaultTable("modifiedFiles"));

    dataSet.addTable(new DefaultTable("notificationfilters"));
    dataSet.addTable(new DefaultTable("users"));
    dataSet.addTable(new DefaultTable("artifactPropertiesGenerator"));
    dataSet.addTable(new DefaultTable("stages"));
    dataSet.addTable(new DefaultTable("pipelineLabelCounts"));
    dataSet.addTable(new DefaultTable("environmentVariables"));
    dataSet.addTable(new DefaultTable("properties"));
    dataSet.addTable(new DefaultTable("artifactPlans"));
    dataSet.addTable(new DefaultTable("buildStateTransitions"));
    dataSet.addTable(new DefaultTable("resources"));
    dataSet.addTable(new DefaultTable("builds"));

    dataSet.addTable(new DefaultTable("oauthclients"));
    dataSet.addTable(new DefaultTable("oauthauthorizations"));
    dataSet.addTable(new DefaultTable("oauthtokens"));

    dataSet.addTable(new DefaultTable("stageArtifactCleanupProhibited"));
    dataSet.addTable(new DefaultTable("serverBackups"));
    dataSet.addTable(new DefaultTable("jobAgentMetadata"));

    databaseTester.setDataSet(dataSet);
}
 

import org.dbunit.DataSourceDatabaseTester; //导入依赖的package包/类
@Before
public void setUp() throws Exception {
	dbTester = new DataSourceDatabaseTester(businessDataSource);
	FlatXmlDataSetBuilder aBuilder = new FlatXmlDataSetBuilder();
	FileInputStream aDTDStream = new FileInputStream("src/test/resources/db/dataset.dtd");
	FileInputStream aDataStream = new FileInputStream("src/test/resources/db/dataset.xml");
	aBuilder.setMetaDataSetFromDtd(aDTDStream);
	IDataSet dataSet = aBuilder.build(aDataStream);
	aDTDStream.close();
	aDataStream.close();
	dbTester.setDataSet(dataSet);
	dbTester.onSetup();
}
 

import org.dbunit.DataSourceDatabaseTester; //导入依赖的package包/类
@Bean
IDatabaseTester databaseTester() {
    return new DataSourceDatabaseTester(dataSource);
}
 

import org.dbunit.DataSourceDatabaseTester; //导入依赖的package包/类
@Bean
public DataSourceDatabaseTester databaseTester() {
	return new DataSourceDatabaseTester(dataSource());
}
 

import org.dbunit.DataSourceDatabaseTester; //导入依赖的package包/类
/**
 * Creates a {@link DataSourceDatabaseTesterRule}.
 *
 * <p>This constructor assumes that the supplied {@link
 * DataSourceDatabaseTester} is completely configured and this
 * {@link DataSourceDatabaseTesterRule} will therefore perform no
 * further configuration.  Please see in particular the {@link
 * #getDataSet(Description)} method, which will consequently have no
 * effect.</p>
 *
 * @param tester the {@link DataSourceDatabaseTester} to which most
 * operations will delegate; must not be {@code null}
 *
 * @exception IllegalArgumentException if {@code tester} is {@code
 * null}
 */
public DataSourceDatabaseTesterRule(final DataSourceDatabaseTester tester) {
  super();
  if (tester == null) {
    throw new IllegalArgumentException("tester", new NullPointerException("tester"));
  }
  this.tester = tester;
}