Spring紧耦合遗留系统测试选型咨询：单元与轻量集成测试抉择

方案1：纯单元测试中手动触发@PrePersist回调 #

既然Mock仓库时JPA不会自动执行@PrePersist方法，最直接的解决方式是在测试代码里手动触发该回调逻辑：

• 如果实体类的@PrePersist方法是公开的（比如public void prePersist()），在调用Mock仓库的save()方法前直接调用该方法：

@Test
void testBusinessObjectPersist() {
    BusinessObject bo = new BusinessObject();
    // 手动触发@PrePersist逻辑
    bo.prePersist();
    when(mockRepository.save(bo)).thenReturn(bo);
    
    // 执行业务逻辑并断言
    service.saveBusinessObject(bo);
    verify(mockRepository).save(bo);
    assertNotNull(bo.getCreatedTime()); // 假设@PrePersist设置了创建时间
}

• 如果@PrePersist方法是私有/受保护的，用反射调用：

@Test
void testPrivatePrePersist() throws Exception {
    BusinessObject bo = new BusinessObject();
    Method prePersistMethod = BusinessObject.class.getDeclaredMethod("prePersist");
    prePersistMethod.setAccessible(true);
    prePersistMethod.invoke(bo);
    
    when(mockRepository.save(bo)).thenReturn(bo);
    // 后续测试逻辑
}

这种方案完全不需要加载Spring或JPA上下文，测试速度极快，适合覆盖纯业务逻辑。

方案2：用@DataJpaTest优化轻量集成测试 #

你之前引入JPA后测试慢且上下文无法复用，大概率是没用到Spring Boot专门的JPA测试注解。@DataJpaTest会加载仅JPA相关的最小上下文，默认使用内存数据库（H2），并且Spring Boot会自动缓存测试上下文（只要测试类的配置、依赖不变），后续测试会复用上下文，大幅缩短执行时间：

@DataJpaTest
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE) // 可选：如果要用真实数据库，默认是内存库
class BusinessObjectRepositoryTest {
    @Autowired
    private BusinessObjectRepository repository;

    @Test
    void testPrePersistTriggered() {
        BusinessObject bo = new BusinessObject();
        BusinessObject savedBo = repository.save(bo);
        
        assertNotNull(savedBo.getCreatedTime()); // 验证@PrePersist执行
    }
}

注意：@DataJpaTest默认会回滚事务，避免污染测试数据；如果需要提交事务，可以加@Transactional(propagation = Propagation.NOT_SUPPORTED)或者@Rollback(false)。

方案3：自定义Mock仓库并注入回调逻辑 #

如果想保留Mock仓库的灵活性，同时确保@PrePersist执行，可以用Mockito自定义save()方法的行为，在保存前触发回调：

@ExtendWith(MockitoExtension.class)
class TestService1Test {
    @Mock
    private BusinessObjectRepository mockRepository;
    @InjectMocks
    private TestService1 service;

    @BeforeEach
    void setup() {
        // 自定义save方法的行为：先执行@PrePersist，再返回实体
        when(mockRepository.save(any(BusinessObject.class))).thenAnswer(invocation -> {
            BusinessObject bo = invocation.getArgument(0);
            // 触发@PrePersist逻辑，这里根据实际方法调用
            bo.prePersist(); // 如果是私有方法用反射调用
            return bo;
        });
    }

    @Test
    void testServiceSave() {
        BusinessObject bo = new BusinessObject();
        service.save(bo);
        
        verify(mockRepository).save(bo);
        assertNotNull(bo.getCreatedTime());
    }
}

这种方案兼顾了Mock的速度和JPA回调的完整性，适合需要Mock其他依赖但又要验证实体生命周期逻辑的场景。

方案4：混合测试策略（推荐遗留代码库） #

针对遗留代码，不需要追求100%测试覆盖，优先覆盖核心业务路径：

• 对纯业务逻辑层：用纯单元测试（方案1或3），快速验证逻辑正确性，避免加载上下文。
• 对JPA交互核心路径：用@DataJpaTest（方案2），确保实体生命周期、仓库方法的正确性，利用上下文缓存减少重复加载的耗时。
• 对涉及多服务协作的场景：用@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE)加载轻量Spring上下文，避免启动Web容器，同时复用上下文缓存。

内容的提问来源于stack exchange，提问作者Van Teo Le