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

夜磊君_1396

夜磊君_1396

2026-06-08

364人浏览

原创

如何用 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() 等链式调用的返回值。

示例代码如下:

Icons8 Background Remover
Icons8 Background Remover

一款由Icons8提供的在线AI背景移除工具,可自动分离图片主体与背景,方便快速制作透明图片和设计素材。

下载
// 被测类(需支持依赖注入)
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());
    }
}</object></userresponsedto></object></object></object></userresponsedto>

对应测试类:

@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");
}</userresponsedto></object>

⚠️ 注意事项:

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

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

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

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

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

2023.10.13

4650

8

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

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

2023.10.23

3041

10

Java 单元测试
Java 单元测试

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

2025.10.24

1086

26

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

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

2026.10.16

60

10

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

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

2026.10.16

40

10

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

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

2026.10.16

40

10

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

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

2026.10.16

20

15

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

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

2026.10.16

20

15

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

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

2026.10.16

20

15

热门下载

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

精品课程

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

共0课时 | 0人学习

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

共0课时 | 0人学习

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

共0课时 | 0人学习