java - JUnit hibernate : testing lazy associations fetched

Question

我有一个带有延迟加载的站点集合的用户模型:

@Entity
public class User {

    @Id
    @GeneratedValue
    private Integer id;

    @OneToMany(fetch = FetchType.LAZY)
    private Set<Site> sites;

    ...

以及我的 UserDao 中的 HQL 查询，它通过 id 加载用户，并获取关联的站点:

import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Query;
import org.springframework.stereotype.Repository;

@Repository
public class userDaoImpl implements UserDao() {

    @Inject
    private SessionFactory sessionFactory;

    public Session getCurrentSession() {
        return sessionFactory.getCurrentSession();
    }

    public User findByIdFetchSites(final Integer id) {
        String queryString = "SELECT user FROM User user LEFT JOIN FETCH user.sites WHERE user.id = :id";
        Query qry = getCurrentSession().createQuery(queryString).setInteger("id", id);
        return qry.uniqueResult();
    }

    ...

当我运行此 Dao 方法时，我想测试网站是否随用户一起获取。我有一个针对 HSQLDB 运行的 JUnit 测试类，如下所示:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration({ "classpath:/test-context.xml" })
@DirtiesContext(classMode=ClassMode.AFTER_EACH_TEST_METHOD)
@Transactional
public class UserDaoImplTest {

@Inject private UserDaoImpl userDaoImpl;

@Test
public void retrievesUserFetchesSites() {
    Set<Site> sites = new HashSet<Site>();
    ...

    User user = new User();
    user.setSites(sites);
    userDaoImpl.saveOrUpdate(user); 

    User retrievedUser = userDaoImpl.findByIdFetchSites(1);

    assertTrue(retrievedUser.getSites().containsAll(sites));
}

其中@Transactional是spring事务注解(org.springframework.transaction.annotation.Transactional)。即使我针对一个简单的 findUserById DAO 方法运行测试，该测试也通过了，该方法没有显式获取站点:

public User findById(final Integer id) {
    String queryString = "SELECT user FROM User user WHERE user.id = :id";
    Query qry = getCurrentSession().createQuery(queryString).setInteger("id", id);
    return qry.uniqueResult();
}

我不完全确定为什么在断言之前获取站点(我无法在日志的选择语句中看到它)，但我想我想做的是在断言之前关闭当前 session 发生；像这样的东西:

@Test
public void retrievesUserFetchesSites() {
    Set<Site> sites = new HashSet<Site>();
    ...

    User user = new User();
    user.setSites(sites);
    userDaoImpl.saveOrUpdate(user); 

    User retrievedUser = userDaoImpl.findByIdFetchSites(1);

    userDaoImpl.getCurrentSession().close();

    assertTrue(retrievedUser.getSites().containsAll(sites));
}

但是，当我尝试这样做时，出现了以下异常:

org.springframework.transaction.TransactionSystemException: Could not roll back Hibernate transaction; nested exception is org.hibernate.TransactionException: rollback failed
at org.springframework.orm.hibernate4.HibernateTransactionManager.doRollback

我认为这是由于@DirtiesContext(每次测试前我需要一个干净的数据库)。我如何确保 Hibernate 不会在测试期间对我的关联进行延迟加载？

Answer 1

问题在于，在同一个事务中，您使用站点创建一个用户，然后检索该用户。会发生什么情况是，persist() 调用将用户及其站点放入 session 缓存中，而查询，无论它做什么，都会返回已经在缓存中的用户。

所以你有以下解决方案:

• 在第一个事务中保存用户及其站点，然后在第二个事务中调用您的 DAO 方法并使用 Hibernate.isInitialized() 测试查询是否加载了用户集合。请注意，调用 retrievedUser.getSites().containsAll(sites) 将触发站点的延迟加载，因此不是测试查询是否急切获取集合的正确方法。
• 保存用户及其站点，清除 session ，然后调用您的 DAO 方法，然后使用 Hibernate.isInitialized() 测试用户集合是否由查询加载。<
• 使测试成为非事务性的，而 DAO 成为事务性的，这样 DAO 就会返回分离的实体。如果 DAO 没有急切地获取集合，那么从集合中获取元素将抛出异常。