我有一个带有延迟加载的站点集合的用户模型:
@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 不会在测试期间对我的关联进行延迟加载?
最佳答案
问题在于,在同一个事务中,您使用站点创建一个用户,然后检索该用户。会发生什么情况是,persist() 调用将用户及其站点放入 session 缓存中,而查询,无论它做什么,都会返回已经在缓存中的用户。
所以你有以下解决方案:
- 在第一个事务中保存用户及其站点,然后在第二个事务中调用您的 DAO 方法并使用
Hibernate.isInitialized()
测试查询是否加载了用户集合。请注意,调用retrievedUser.getSites().containsAll(sites)
将触发站点的延迟加载,因此不是测试查询是否急切获取集合的正确方法。 - 保存用户及其站点,清除 session ,然后调用您的 DAO 方法,然后使用
Hibernate.isInitialized()
测试用户集合是否由查询加载。< - 使测试成为非事务性的,而 DAO 成为事务性的,这样 DAO 就会返回分离的实体。如果 DAO 没有急切地获取集合,那么从集合中获取元素将抛出异常。
关于java - JUnit hibernate : testing lazy associations fetched,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/19057802/