
本文介绍如何使用 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 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 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() 是解决“动态数据源 + 静态工具类”测试难题的现代标准方案。只要严格遵循模拟链路完整性与作用域控制,即可高效、可靠地覆盖数据库访问逻辑。










