讲师中心 微信公众号
AI工具推荐 视频效率加速

如何用 Mockito 对调用私有方法的公有方法进行单元测试

夜磊君_1396

夜磊君_1396

发布时间:2026-06-08 10:26:03

|

364人浏览过

|

来源于php中文网

原创

如何用 Mockito 对调用私有方法的公有方法进行单元测试

junit + mockito 测试中,不应直接 mock 私有方法;正确做法是通过设计可测的公有接口、隔离外部依赖(如 entitymanager),并以行为驱动方式验证公有方法的整体输出。

junit + mockito 测试中,不应直接 mock 私有方法;正确做法是通过设计可测的公有接口、隔离外部依赖(如 entitymanager),并以行为驱动方式验证公有方法的整体输出。

在实际开发中,getUserDetails() 这类公有方法虽内部调用多个私有方法(如 getUserByNativeQuery() 和 getUserDetailsByNativeQuery()),但 Mockito 原生不支持直接 mock 私有方法,也不推荐借助反射或第三方库强行打桩——这会破坏封装性、增加测试脆弱性,并违背“测试行为而非实现”的单元测试原则。

✅ 正确策略是:将私有方法的逻辑视为公有方法的内部实现细节,通过 mock 其所依赖的协作对象(如 EntityManager)来间接控制私有方法的行为。

由于你的私有方法均通过 EntityManager#createNativeQuery() 执行数据库查询,关键在于:

  1. 将 EntityManager 作为可注入依赖(例如通过构造函数或 setter 注入);
  2. 在测试中使用 Mockito.mock(EntityManager.class) 创建模拟对象;
  3. 使用 when(...).thenReturn(...) 预设 createNativeQuery().getResultList() 等链式调用的返回值。

示例代码如下:

// 被测类(需支持依赖注入)
public class UserService {
    private final EntityManager entityManager;

    public UserService(EntityManager entityManager) {
        this.entityManager = entityManager;
    }

    public List<UserResponseDto> getUserDetails(UserRequestDto request) {
        UserDto userDto = getUserByNativeQuery(request);
        List<Object[]> listObj = getUserDetailsByNativeQuery(request);
        return setUserResponseDto(listObj);
    }

    private UserDto getUserByNativeQuery(UserRequestDto request) {
        Query query = entityManager.createNativeQuery("SELECT * FROM users WHERE id = ?");
        query.setParameter(1, request.getUserId());
        List<Object[]> results = query.getResultList();
        // ... 映射逻辑
        return new UserDto();
    }

    private List<Object[]> getUserDetailsByNativeQuery(UserRequestDto request) {
        Query query = entityManager.createNativeQuery("SELECT u.name, a.city FROM users u JOIN addresses a ON u.id = a.user_id WHERE u.id = ?");
        query.setParameter(1, request.getUserId());
        return query.getResultList(); // ← 此处将被 mock 控制
    }

    private List<UserResponseDto> setUserResponseDto(List<Object[]> data) {
        return data.stream()
                .map(row -> new UserResponseDto((String) row[0], (String) row[1]))
                .collect(Collectors.toList());
    }
}

对应测试类:

@Test
void getUserDetails_shouldReturnMappedResponse() {
    // Given
    UserRequestDto request = new UserRequestDto(123L);

    EntityManager emMock = mock(EntityManager.class);
    Query queryMock = mock(Query.class);

    // 模拟 getUserDetailsByNativeQuery 的查询结果
    List<Object[]> mockResults = Arrays.asList(
            new Object[]{"Alice", "Beijing"},
            new Object[]{"Bob", "Shanghai"}
    );
    when(emMock.createNativeQuery(anyString())).thenReturn(queryMock);
    when(queryMock.getResultList()).thenReturn(mockResults); // 关键:控制私有方法的输出

    UserService service = new UserService(emMock);

    // When
    List<UserResponseDto> result = service.getUserDetails(request);

    // Then
    assertThat(result).hasSize(2);
    assertThat(result.get(0).getName()).isEqualTo("Alice");
    assertThat(result.get(0).getCity()).isEqualTo("Beijing");
}

⚠️ 注意事项:

  • 不要试图 mock getUserByNativeQuery() 本身——它只是 EntityManager 调用的封装,mock EntityManager 即可覆盖其行为;
  • 若 setUserResponseDto() 含复杂逻辑,可单独提取为 package-private 方法并直接测试,提升可测性与可维护性;
  • 避免在生产代码中为测试而暴露私有方法(如改为 protected 或添加 @VisibleForTesting 注解),除非有明确重构价值;
  • 所有数据库交互必须被完全隔离,确保测试不依赖真实 DB,符合单元测试的快速、可靠、独立特性。

总结:真正的可测性源于良好的依赖抽象和职责分离,而非对私有方法的强行干预。聚焦于“输入 → 输出”契约,用 EntityManager 的 mock 控制整个调用链的数据流,即可高效、稳健地完成对 getUserDetails() 的全覆盖验证。

热门AI工具

更多
WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

Seko
Seko Hot

一款AI视频创作工具,主要用于商汤科技推出的创编一体的AI短视频创作Agent,适合需要提升相关任务效率的用户。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

相关专题

更多
软件测试常用工具
软件测试常用工具

软件测试常用工具有Selenium、JUnit、Appium、JMeter、LoadRunner、Postman、TestNG、LoadUI、SoapUI、Cucumber和Robot Framework等等。测试人员可以根据具体的测试需求和技术栈选择适合的工具,提高测试效率和准确性 。

4650

2023.10.13

java测试工具有哪些
java测试工具有哪些

java测试工具有JUnit、TestNG、Mockito、Selenium、Apache JMeter和Cucumber。php还给大家带来了java有关的教程,欢迎大家前来学习阅读,希望对大家能有所帮助。

3041

2023.10.23

Java 单元测试
Java 单元测试

本专题聚焦 Java 在软件测试与持续集成流程中的实战应用,系统讲解 JUnit 单元测试框架、Mock 数据、集成测试、代码覆盖率分析、Maven 测试配置、CI/CD 流水线搭建(Jenkins、GitHub Actions)等关键内容。通过实战案例(如企业级项目自动化测试、持续交付流程搭建),帮助学习者掌握 Java 项目质量保障与自动化交付的完整体系。

1086

2025.10.24

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

60

2026.10.16

OpenJDK怎么配置Java开发环境
OpenJDK怎么配置Java开发环境

怎样用OpenJDK搭建开发环境?本专题介绍版本选择、下载安装、环境变量配置和编译运行验证,帮助你理清JDK的作用及基本配置流程,并通过简单程序检查环境是否可用,为后续项目开发做好准备。

40

2026.10.16

OpenJDK怎么部署Java应用
OpenJDK怎么部署Java应用

Java项目开发完成后,怎样部署到目标环境并正常运行?本专题介绍OpenJDK应用部署流程,涵盖运行环境准备、版本兼容检查、应用打包、启动参数配置和日志查看,帮助你理清部署前后的操作顺序,减少环境差异造成的运行问题。

40

2026.10.16

Revel框架模板渲染与表单验证
Revel框架模板渲染与表单验证

本专题围绕Revel框架视图层开发,介绍模板语法、变量输出、布局复用、表单提交、参数绑定与验证规则,帮助开发者完成安全规范的页面交互。

20

2026.10.16

Revel框架MVC路由与控制器
Revel框架MVC路由与控制器

本专题聚焦Revel框架MVC核心机制,讲解路由定义、控制器Action、参数绑定、请求响应和反向路由,并结合常见场景梳理接口请求处理流程。

20

2026.10.16

Revel框架入门与安装配置
Revel框架入门与安装配置

本专题从Revel框架环境搭建讲起,系统介绍命令行工具安装、项目创建、目录结构、配置文件及热代码重载,帮助开发者快速完成首个Go Web应用。

20

2026.10.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
文心快码功能介绍
文心快码功能介绍

共0课时 | 0人学习

文心快码Agent使用手册
文心快码Agent使用手册

共0课时 | 0人学习

Fitten Code官方手册
Fitten Code官方手册

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn