动态数据源单元测试技巧分享

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《动态数据源方法单元测试技巧》，聊聊，我们一起来看看吧！

本文介绍如何使用 Mockito 3.4.0+ 的 `mockStatic` 功能，安全地模拟静态依赖（如 `Archicon.getResource()`），从而隔离 `searchQuery` 等依赖运行时动态数据源的数据库操作方法，实现可重复、快速、无副作用的单元测试。

在企业级 Java 应用中，常见将数据源通过 JNDI 或配置中心（如 Archicon.getResource("jdbc/xxx")）动态获取。这类设计提升了部署灵活性，却给单元测试带来挑战——直接调用 searchQuery() 会触发真实连接逻辑，导致测试失败、变慢甚至污染环境。

核心问题在于：searchQuery() 内部调用了静态工具类 Archicon.getResource()，而该方法返回的 DataSource 在测试环境中为 null，引发 NullPointerException。传统 @Mock 无法拦截静态方法调用，因此必须借助 Mockito 的 静态方法模拟能力（需 Mockito ≥ 3.4.0）。

✅ 正确做法：使用 Mockito.mockStatic() 模拟静态资源获取

以下是一个完整、可运行的测试示例，重点修复了原测试中的关键错误（如 @Mock ValidationData 导致实例未初始化、when().thenReturn() 误用于未 mock 的对象等）：

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.mockito.MockedStatic;

import javax.sql.DataSource;
import java.sql.*;
import java.util.List;

import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;

@ExtendWith(MockitoExtension.class)
public class AwpSapNewRepositoryTest {

    @Test
    public void searchQuery_shouldReturnResultsFromMockedDataSource() throws Exception {
        // Given: 模拟 DataSource 及其行为
        DataSource mockDs = mock(DataSource.class);
        Connection mockConn = mock(Connection.class);
        PreparedStatement mockStmt = mock(PreparedStatement.class);
        ResultSet mockRs = mock(ResultSet.class);

        // 链式模拟：getResource → getConnection → prepareStatement → executeQuery → ResultSet.next()
        try (MockedStatic<Archicon> archiconMock = mockStatic(Archicon.class)) {
            archiconMock.when(() -> Archicon.getResource(any())).thenReturn(mockDs);

            when(mockDs.getConnection()).thenReturn(mockConn);
            when(mockConn.prepareStatement(any(String.class))).thenReturn(mockStmt);
            when(mockStmt.executeQuery()).thenReturn(mockRs);

            // 模拟 ResultSet 返回一行数据
            when(mockRs.next()).thenReturn(true).thenReturn(false); // 第一次 true，第二次 false 结束循环
            when(mockRs.getString(1)).thenReturn("Name, 99"); // 假设 getValues(rs) 读取第1列

            // When: 执行被测方法
            ValidationData validator = new ValidationData();
            List<String> result = validator.searchQuery("11");

            // Then: 断言结果
            assertNotNull(result);
            assertEquals(1, result.size());
            assertEquals("Name, 99", result.get(0));
        }
    }
}

⚠️ 关键注意事项

• 不要 @Mock ValidationData：ValidationData 是被测对象（SUT），应 new ValidationData() 实例化；@Mock 它会导致所有方法返回 null，失去测试意义。
• mockStatic 必须用 try-with-resources：确保作用域内模拟生效且自动还原，避免影响其他测试。
• 完整模拟 JDBC 链路：DataSource → Connection → PreparedStatement → ResultSet 各环节均需 when(...).thenReturn(...) 显式定义行为，否则默认返回 null。
• getValues(rs) 方法需可测试：确保该私有/包级方法能从 ResultSet 中正确提取字符串（建议抽取为独立可测方法或使用 @VisibleForTesting 注解）。
• 避免硬编码路径：Archicon.setConfigPath(...) 属于集成测试范畴，单元测试中无需设置；静态模拟已完全接管资源加载逻辑。

✅ 进阶建议：提升可测性（重构推荐）

若长期维护，建议对 ValidationData 做轻量重构以降低测试耦合度：

// 将数据源获取逻辑抽离为可注入的策略
public class ValidationData {
    private final ArchiconResourceProvider resourceProvider; // 构造注入

    public ValidationData(ArchiconResourceProvider provider) {
        this.resourceProvider = provider;
    }

    private void createConnection(String key) {
        DataSource ds = resourceProvider.getResource("jdbc/" + key);
        conn = ds.getConnection();
    }
}

此时测试只需 mock 接口，无需 mockStatic，更简洁、更稳定。

总之，利用 Mockito.mockStatic() 是解决“动态数据源 + 静态工具类”测试难题的现代标准方案。只要严格遵循模拟链路完整性与作用域控制，即可高效、可靠地覆盖数据库访问逻辑。

本篇关于《动态数据源单元测试技巧分享》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于文章的相关知识，请关注golang学习网公众号！