DBUnit,Hibernate / JPA,DAO和flush

时间:2012-12-28 09:22:02

标签: spring hibernate jpa dbunit

目标

为了测试DAO的create方法,我创建一个实例,将其插入数据库,刷新实体管理器以更新数据库,然后使用dbunit比较使用数据集的表。


代码

这是代码,它使用Spring测试,DBUnit和JPA(通过Hibernate):

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {
    "/WEB-INF/applicationContext-database.xml"})
public class MyEntityTest extends AbstractTransactionalJUnit4SpringContextTests {

    @PersistenceContext
    protected EntityManager em;

    @Autowired
    MyEntityDAO myEntityDAO;

    @Test
    public void createTest() {
        // create the entity
        MyEntity record = new MyEntity();
        record.setData("test");    
        myEntityDAO.insertNew(record);

        // flush to update the database
        em.flush();         

        // get actual dataset from the connection
        Session session = em.unwrap(Session.class);
        Connection conn = SessionFactoryUtils.getDataSource(
            session.getSessionFactory()).getConnection();
        DatabaseConnection connection = new DatabaseConnection(conn);
        ITable actualTable = connection.createTable("MY_ENTITY");

        // get expected dataset
        IDataSet expectedDataSet = new FlatXmlDataSetBuilder().build(getClass().getResource("dataset.xml"));
        ITable expectedTable = expectedDataSet.getTable("MY_ENTITY");

        // compare the dataset
        Assertion.assertEquals(expectedTable, actualTable);
    }
}

问题

此代码永远不会结束,在此命令期间似乎冻结(无限循环?):

ITable actualTable = connection.createTable("MY_ENTITY");

但是如果我评论em.flush()块,则测试结束(没有冻结或无限循环)。在这种情况下,测试失败,因为插入后数据库尚未更新。


问题

如何在调用dataset.getTable()时使用类似方法(将数据集与dbunit进行比较)而不冻结来测试DAO的create方法?

1 个答案:

答案 0 :(得分:1)

我找到了解决方案。问题是由连接引起的。

如果我更换:

Session session = em.unwrap(Session.class);
Connection conn = SessionFactoryUtils.getDataSource(
        session.getSessionFactory()).getConnection();

通过

DataSource ds = (DataSource) applicationContext.getBean("dataSource");
Connection conn = DataSourceUtils.getConnection(ds);

一切运行良好......

我不明白为什么,所以如果您有任何线索可以帮助我理解,请在评论中告诉我。