hibernateintegrationtest.java

来自「网上拍卖系统」· Java 代码 · 共 97 行

package auction.test;

import org.hibernate.impl.SessionFactoryImpl;
import auction.persistence.HibernateUtil;
import org.testng.annotations.*;
import org.dbunit.dataset.*;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.dbunit.operation.DatabaseOperation;
import org.dbunit.database.*;

import java.io.*;
import java.util.*;
import java.sql.Connection;
import java.net.MalformedURLException;

public abstract class HibernateIntegrationTest {

    protected String dataSetLocation;
    protected List<DatabaseOperation> beforeTestOperations
            = new ArrayList<DatabaseOperation>();
    protected List<DatabaseOperation> afterTestOperations
            = new ArrayList<DatabaseOperation>();

    private ReplacementDataSet dataSet;

    @BeforeClass(groups = "integration-hibernate")
    public void prepareDataSet() throws Exception {

        // Check if subclass has prepared everything
        prepareSettings();
        if (dataSetLocation == null)
            throw new RuntimeException(
                "Test subclass needs to prepare a dataset location"
            );

        // Load the base dataset file
        try {

            InputStream input =
                    Thread.currentThread().getContextClassLoader()
                            .getResourceAsStream(dataSetLocation);

            dataSet = new ReplacementDataSet(
                            new FlatXmlDataSet(input)
                          );
            dataSet.addReplacementObject("[NULL]", null);

        } catch (DataSetException ex) {
            throw new RuntimeException(
                    "Couldn't load DBUnit dataset: " + dataSetLocation, ex
            );
        }
    }

    @BeforeMethod(groups = "integration-hibernate")
    public void beforeTestMethod() throws Exception {
        prepareSettings();
        for (DatabaseOperation op : beforeTestOperations ) {
            op.execute(getConnection(), dataSet);
        }
    }

    @AfterMethod(groups = "integration-hibernate")
    public void afterTestMethod() throws Exception {
        for (DatabaseOperation op : afterTestOperations ) {
            op.execute(getConnection(), dataSet);
        }
    }

    // Subclasses can/have to override the following methods

    protected IDatabaseConnection getConnection() throws Exception {

        // Get a JDBC connection from Hibernate
        Connection con =
            ((SessionFactoryImpl)HibernateUtil.getSessionFactory()).getSettings()
                    .getConnectionProvider().getConnection();

        // Disable foreign key constraint checking
        // This really depends on the DBMS product... here for HSQL DB
        con.prepareStatement("set referential_integrity FALSE")
            .execute();

        IDatabaseConnection dbUnitCon =
            new DatabaseConnection( con);

        // TODO: Remove this once DBUnit works with the latest HSQL DB
        DatabaseConfig config = dbUnitCon.getConfig();
        config.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new FixDBUnit());

        return dbUnitCon;
    }

    protected abstract void prepareSettings();

}