东软单元测试中如何模拟数据库交互？

Question

在东软的单元测试实践中，如何有效模拟数据库交互是一个常见难题。当Service层方法涉及复杂的JPA或MyBatis操作时，直接连接真实数据库会导致测试效率低下且难以覆盖异常场景。常见的问题是：使用@SpringBootTest加载全容器虽能运行数据库相关逻辑，但违背了单元测试快速、隔离的原则；而仅用@MockBean模拟Repository时，又难以验证SQL逻辑或级联操作的正确性。因此，开发者常面临如何在不依赖真实数据库的前提下，精准模拟数据访问行为并验证方法调用流程的困境。

Answer 1

东软单元测试实践中数据库交互模拟的深度解析

1. 问题背景与挑战分析

在东软的软件研发体系中，随着微服务架构和Spring Boot的广泛应用，Service层承担了越来越多的业务逻辑处理任务。当这些方法依赖JPA或MyBatis进行复杂的数据访问时，如何高效地开展单元测试成为一大痛点。

常见的测试策略存在明显缺陷：

• @SpringBootTest：加载完整上下文容器，虽能运行真实SQL，但启动时间长、资源消耗大，违背单元测试“快速、独立”的原则；
• @MockBean：可隔离外部依赖，但仅模拟方法调用，无法验证SQL执行路径、实体映射或级联行为；
• 真实数据库集成测试：难以构造边界异常（如唯一键冲突、连接超时），且数据状态难维护。

2. 分层应对策略：从浅层模拟到深层验证

为解决上述困境，需构建多层次的测试策略体系：

测试层级	技术手段	适用场景	优点	局限性
纯Mock层	@MockBean + Mockito	逻辑分支覆盖	速度快，完全隔离	无法验证SQL语义
内存数据库层	H2 + @DataJpaTest	验证JPA注解、级联操作	接近真实SQL执行	语法差异（如分页）
SQL模拟层	MyBatis: MockSqlSession	验证动态SQL生成	可断言SQL文本	配置复杂度高
轻量容器层	Testcontainers	端到端集成验证	环境一致性高	资源开销较大

3. 典型解决方案详解

针对不同ORM框架，推荐如下实践方案：

3.1 基于H2的JPA行为验证

使用@DataJpaTest注解加载最小化JPA上下文，并配合H2内存数据库：

@DataJpaTest
class UserServiceTest {

    @Autowired
    private UserRepository userRepository;

    @Test
    void shouldCascadeDeleteWhenUserRemoved() {
        User user = new User("john");
        user.addOrder(new Order("book"));
        userRepository.save(user);

        userRepository.deleteById(user.getId());

        assertThat(userRepository.findById(user.getId())).isEmpty();
        // 可进一步验证order表是否被级联清除
    }
}
    

3.2 MyBatis动态SQL断言方案

通过拦截Executor行为，捕获实际生成的SQL语句：

@Test
void shouldGenerateCorrectDynamicSQL() {
    try (SqlSession session = sqlSessionFactory.openSession()) {
        BoundSql boundSql = session.getConfiguration()
            .getMappedStatement("com.example.UserMapper.selectByCondition")
            .getBoundSql(params);
        
        String sql = boundSql.getSql().toLowerCase();
        assertThat(sql).contains("where");
        assertThat(sql).contains("status = ?");
    }
}
    

4. 高级模式：混合测试架构设计

在大型项目中，建议采用组合式测试策略，依据方法特性选择最优路径：

graph TD
            A[Service Method] --> B{Contains Complex SQL?}
            B -- No --> C[Use @MockBean for Repository]
            B -- Yes --> D{Is Cascading Critical?}
            D -- Yes --> E[Use @DataJpaTest with H2]
            D -- No --> F[Use Mocked SqlSession for MyBatis]
            E --> G[Assert DB State After Execution]
            F --> H[Assert Generated SQL Structure]
            C --> I[Verify Business Logic Flow]

5. 异常场景覆盖技巧

为了提升测试完整性，应主动模拟数据库异常：

• 使用Mockito.doThrow()模拟DataAccessException；
• 通过H2设置延迟或锁表模拟超时；
• 利用AOP在测试环境中注入故障点；
• 结合Resilience4j测试重试机制对数据库失败的响应；
• 使用嵌入式MongoDB或PostgreSQL Docker实例测试特定方言行为；
• 通过自定义EntityListener触发预设异常；
• 在TransactionTemplate中包装异常传播路径；
• 利用Spring Profiles切换数据源实现异常注入；
• 结合JUnit Jupiter的@Nested组织异常测试用例树；
• 使用AssertJ SoftAssertions批量验证多字段状态。