如何在Spring Boot中快速测试Hibernate抓取策略注解效果？

问题背景 #

在使用IntelliJ作为IDE、Gradle作为构建工具的Spring Boot项目中，需要测试两个关联实体类的不同Hibernate注解效果。已知默认情况下@ManyToOne和@OneToOne为即时抓取（EAGER），需验证修改为延迟抓取（FetchType.LAZY）后的差异。通过浏览器调用控制器测试不仅耗时，还会受Jackson序列化响应的影响。

实体类定义 #

MechanicVisit #

@Entity
@Data
@Table(name="MechanicVisit")
public class MechanicVisit {

    @Id
    @GeneratedValue(strategy = IDENTITY)
    private Long id;

    @JsonManagedReference
    @ManyToOne(fetch = FetchType.LAZY, optional = false)
    private Car car;
}

Car #

@Entity
@Data
public class Car {
    @Id
    @GeneratedValue(strategy = IDENTITY)
    private Long id;

    @JsonBackReference
    @OneToMany(cascade = CascadeType.ALL, orphanRemoval = true, mappedBy = "car")
    private List<MechanicVisit> listMechanicVisit = new ArrayList<>();
    // 其他属性
}

自定义仓库方法 #

仓库中定义了手动获取MechanicVisit的方法（预期不会加载Car）：

@Query("SELECT m FROM MechanicVisit m WHERE m.id = :id")
Optional<MechanicVisit> getOneManually(@Param("id") Long id);

现有测试结构 #

当前已生成的测试类：

@SpringBootTest
class CarApplicationTests {

    @Test
    void contextLoads() {
    }

}

高效测试方案 #

无需频繁启停服务器，直接编写集成测试即可验证抓取策略效果，具体步骤如下：

1. 完善测试类依赖注入 #

在测试类中注入目标仓库和EntityManager，用于数据查询和加载状态检查：

@SpringBootTest
class MechanicVisitFetchStrategyTest {

    @Autowired
    private MechanicVisitRepository mechanicVisitRepo;

    @Autowired
    private EntityManager em;

    private Long testVisitId;
}

2. 预置测试数据 #

通过@BeforeEach初始化测试用的关联实体数据，并清空EntityManager缓存确保查询的是数据库最新实例：

@BeforeEach
void setUp() {
    // 初始化测试Car
    Car testCar = new Car();
    // 设置Car的其他必要属性
    em.persist(testCar);

    // 初始化关联的MechanicVisit
    MechanicVisit testVisit = new MechanicVisit();
    testVisit.setCar(testCar);
    em.persist(testVisit);
    testVisitId = testVisit.getId();

    em.flush();
    em.clear(); // 清空一级缓存，避免缓存干扰测试结果
}

3. 编写LAZY抓取验证测试 #

测试1：Session范围内的延迟加载

验证获取MechanicVisit时Car未被加载，访问Car属性才触发延迟加载：

@Test
void testLazyFetchWithinSession() {
    // 获取MechanicVisit实例
    Optional<MechanicVisit> visitOpt = mechanicVisitRepo.getOneManually(testVisitId);
    assertTrue(visitOpt.isPresent());
    MechanicVisit visit = visitOpt.get();

    // 检查Car是否未被加载
    PersistenceUnitUtil persistenceUtil = em.getEntityManagerFactory().getPersistenceUnitUtil();
    assertFalse(persistenceUtil.isLoaded(visit, "car"));

    // 访问Car属性，触发延迟加载
    Car car = visit.getCar();
    // 再次检查，确认Car已被加载
    assertTrue(persistenceUtil.isLoaded(visit, "car"));
    assertNotNull(car.getId());
}

测试2：Session关闭后的延迟加载

验证Session关闭后访问未加载的关联属性会抛出LazyInitializationException：

@Test
void testLazyFetchAfterSessionClosed() {
    MechanicVisit visit = mechanicVisitRepo.getOneManually(testVisitId).orElseThrow();
    
    // 清除EntityManager并关闭，模拟Session结束
    em.clear();
    em.close();

    // 访问未加载的Car属性，预期抛出异常
    assertThrows(LazyInitializationException.class, visit::getCar);
}

4. 对比EAGER抓取效果 #

若要验证默认的EAGER抓取，只需将@ManyToOne的fetch参数改为FetchType.EAGER，运行上述测试：

• persistenceUtil.isLoaded(visit, "car")会直接返回true
• Session关闭后访问Car属性也不会抛出异常

关键注意事项 #

• 使用PersistenceUnitUtil.isLoaded()判断关联实体是否加载，这是最可靠的方式，避免日志或SQL分析的误差
• 测试前务必清空EntityManager缓存（em.clear()），防止缓存中的已加载实体影响测试结果
• 若使用Lombok的@Data注解，注意其自动生成的toString()方法可能触发延迟加载，测试时避免调用，或自定义toString()排除关联属性

内容的提问来源于stack exchange，提问作者Ray Bond