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

JUnit5/Mockito:优雅测试内部依赖与异常处理的实践

胖萱君_8041

胖萱君_8041

发布时间:2025-12-01 14:32:37

|

549人浏览过

|

来源于php中文网

原创

JUnit5/Mockito:优雅测试内部依赖与异常处理的实践

本文旨在解决使用junit5和mockito测试java服务中内部依赖(如guava loadingcache)抛出并处理的异常(特别是catch块覆盖率)的挑战。核心策略是采用依赖注入模式,通过引入工厂接口或自定义包装器类来解耦服务与缓存的直接创建,从而使缓存实例在测试中可被模拟。文章将详细介绍两种实现方式及其对应的测试代码,确保`executionexception`的`catch`块能够被有效覆盖,提升代码测试的完整性。

在开发高健壮性的Java应用时,我们经常会遇到需要处理内部组件抛出的异常场景。然而,当这些组件在服务类内部直接实例化时,对其异常处理逻辑进行单元测试,特别是覆盖catch块,会变得极具挑战。本教程将以一个具体的案例为例,探讨如何利用JUnit5和Mockito,通过引入依赖注入(Dependency Injection)模式,有效测试一个void方法中由LoadingCache引起的ExecutionException的catch块。

问题剖析:内部依赖的测试难题

考虑以下MyService类中的unlockFailed方法:

public class MyService {
    private LoadingCache<String, Integer> attemptsCache;

    public MyService() {
       super();
       // 缓存直接在构造函数中实例化
       attemptsCache = CacheBuilder.newBuilder()
           .expireAfterWrite(1, TimeUnits.HOURS)
           .build(new CacheLoader<String, Integer>() {
              @Override
              public Integer load(String key) throws Exception {
                  return 0;
              }
       });
    }

    public void unlockFailed(String key) {
        int attempts = 0;
        try {
            attempts = attemptsCache.get(key); // 这里的get方法可能抛出ExecutionException
        } catch (ExecutionException e) {
            attempts = 0; // 目标:覆盖这个catch块
        }
        attempts++;
        attemptsCache.put(key, attempts);
    }
}

在上述代码中,MyService内部直接创建并持有一个LoadingCache实例。unlockFailed方法尝试从缓存中获取值,并包含一个try-catch块来处理可能发生的ExecutionException。传统的单元测试尝试通过Mockito.doThrow().when(attemptsCache).get(USER);来模拟异常,但由于attemptsCache是在MyService内部创建的真实对象,而非注入的模拟对象,因此这种模拟无法生效,导致catch块始终无法被覆盖。

问题的核心在于MyService与LoadingCache的紧密耦合。为了实现可测试性,我们需要解耦它们,使得在测试时可以注入一个模拟的LoadingCache。

解决方案:依赖注入策略

依赖注入是解决此类问题的关键。通过将LoadingCache的创建或实例本身作为依赖项注入到MyService中,我们就可以在测试环境中提供一个由Mockito创建的模拟对象。下面介绍两种常用的依赖注入实现方式。

方法一:注入缓存工厂(Cache Factory)

这种方法通过引入一个工厂接口来抽象LoadingCache的创建过程。MyService不再直接创建缓存,而是通过工厂获取。

1. 定义缓存工厂接口和实现类

// CacheFactory.java
public interface CacheFactory {
  LoadingCache<String, Integer> createCache();
}

// ExpiringCacheFactory.java (生产环境实现)
public class ExpiringCacheFactory implements CacheFactory {
  @Override
  public LoadingCache<String, Integer> createCache() {
      return CacheBuilder.newBuilder()
           .expireAfterWrite(1, TimeUnit.HOURS)
           .build(new CacheLoader<String, Integer>() {
              @Override
              public Integer load(String key) throws Exception {
                  return 0;
              }
       });
  }
}

2. 修改 MyService 构造函数

MyService现在通过构造函数接收CacheFactory实例。

// MyService.java (修改后)
public class MyService {
    private LoadingCache<String, Integer> attemptsCache;

    public MyService(final CacheFactory cacheFactory) { // 注入CacheFactory
       super();
       this.attemptsCache = cacheFactory.createCache(); // 通过工厂创建缓存
    }

    // unlockFailed 方法保持不变
    public void unlockFailed(String key) {
        int attempts = 0;
        try {
            attempts = attemptsCache.get(key);
        } catch (ExecutionException e) {
            attempts = 0;
        }
        attempts++;
        attemptsCache.put(key, attempts);
    }
}

3. 生产环境中的使用

final MyService myService = new MyService(new ExpiringCacheFactory());

4. 单元测试代码

在测试中,我们可以提供一个匿名CacheFactory实现,它返回一个由Mockito模拟的LoadingCache。

Google Colab
Google Colab

在Google Colab中运行Python与机器学习笔记本,支持可复现运行环境、数据管道、调试工作流及实验管理。

下载
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.mockito.InjectMocks;
import org.mockito.Spy;

import com.google.common.cache.LoadingCache;
import com.google.common.util.concurrent.ExecutionError;

import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;

import static org.mockito.Mockito.*;

public class MyServiceTest {

    private final String USER = "fakeuser";

    @DisplayName("unlockFailed 应该在 LoadingCache.get 抛出 ExecutionException 时处理异常")
    @Test
    public void unlockFailed_should_handle_ExecutionException_with_factory() throws ExecutionException {
        // Arrange
        // 1. 模拟 LoadingCache
        LoadingCache<String, Integer> attemptsCacheMock = Mockito.mock(LoadingCache.class);
        // 2. 配置模拟缓存,使其在调用get方法时抛出ExecutionException
        doThrow(new ExecutionException("Dummy ExecutionException", null)).when(attemptsCacheMock).get(USER);

        // 3. 创建 MyService 实例,注入一个返回模拟缓存的工厂
        // 这里使用 lambda 表达式简化匿名内部类
        MyService sut = new MyService(() -> attemptsCacheMock);

        // Act
        sut.unlockFailed(USER);

        // Assert
        // 验证 attemptsCache.get(USER) 方法被调用了一次
        verify(attemptsCacheMock, times(1)).get(USER);
        // 验证 attemptsCache.put(USER, 0) 方法被调用了一次 (因为catch块将attempts重置为0)
        verify(attemptsCacheMock, times(1)).put(USER, 1); // attempts = 0 + 1 = 1
        // 注意:这里不需要断言抛出异常,因为方法内部已经捕获并处理了
    }
}

通过这种方式,MyService在测试时使用的是一个完全受控的模拟LoadingCache,我们可以轻松地模拟get()方法抛出ExecutionException,从而覆盖到catch块的逻辑。

方法二:注入自定义缓存包装器(Custom Cache Wrapper)

如果LoadingCache的接口过于复杂,或者我们只想暴露其一部分方法,可以创建一个自定义接口来包装LoadingCache,并将其注入到MyService中。

1. 定义自定义缓存接口和实现类

// MyLoadingCache.java
public interface MyLoadingCache {
  // 暴露 LoadingCache 中需要用到的方法
  Integer get(String key) throws ExecutionException;
  void put(String key, Integer value);
}

// MyLoadingCacheImpl.java (生产环境实现)
public class MyLoadingCacheImpl implements MyLoadingCache {
  private LoadingCache<String, Integer> attemptsCache;

  public MyLoadingCacheImpl() {
      this.attemptsCache = CacheBuilder.newBuilder()
           .expireAfterWrite(1, TimeUnit.HOURS)
           .build(new CacheLoader<String, Integer>() {
              @Override
              public Integer load(String key) throws Exception {
                  return 0;
              }
       });
  }

  @Override
  public Integer get(String key) throws ExecutionException {
      return attemptsCache.get(key);
  }

  @Override
  public void put(String key, Integer value) {
      attemptsCache.put(key, value);
  }
}

2. 修改 MyService 构造函数

MyService现在直接接收MyLoadingCache接口实例。

// MyService.java (修改后)
public class MyService {
    private MyLoadingCache attemptsCache; // 使用自定义接口类型

    public MyService(final MyLoadingCache attemptsCache) { // 注入自定义缓存接口
       super();
       this.attemptsCache = attemptsCache;
    }

    // unlockFailed 方法保持不变
    public void unlockFailed(String key) {
        int attempts = 0;
        try {
            attempts = attemptsCache.get(key);
        } catch (ExecutionException e) {
            attempts = 0;
        }
        attempts++;
        attemptsCache.put(key, attempts);
    }
}

3. 生产环境中的使用

final MyService myService = new MyService(new MyLoadingCacheImpl());

4. 单元测试代码

测试时,我们可以直接模拟MyLoadingCache接口。

import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;

import java.util.concurrent.ExecutionException;

import static org.mockito.Mockito.*;

public class MyServiceTest {

    private final String USER = "fakeuser";

    @DisplayName("unlockFailed 应该在 MyLoadingCache.get 抛出 ExecutionException 时处理异常")
    @Test
    public void unlockFailed_should_handle_ExecutionException_with_wrapper() throws ExecutionException {
        // Arrange
        // 1. 模拟 MyLoadingCache 接口
        MyLoadingCache cacheMock = Mockito.mock(MyLoadingCache.class);
        // 2. 配置模拟缓存,使其在调用get方法时抛出ExecutionException
        doThrow(new ExecutionException("Dummy ExecutionException", null)).when(cacheMock).get(USER);

        // 3. 创建 MyService 实例,注入模拟缓存
        MyService sut = new MyService(cacheMock);

        // Act
        sut.unlockFailed(USER);

        // Assert
        // 验证 cacheMock.get(USER) 方法被调用了一次
        verify(cacheMock, times(1)).get(USER);
        // 验证 cacheMock.put(USER, 0) 方法被调用了一次
        verify(cacheMock, times(1)).put(USER, 1);
    }
}

替代方案:直接注入 LoadingCache 接口

在某些情况下,如果LoadingCache的创建逻辑不复杂,或者我们接受直接注入Guava的LoadingCache接口,也可以选择直接将LoadingCache<String, Integer>作为依赖注入。

// MyService.java (直接注入LoadingCache)
public class MyService(final LoadingCache<String, Integer> attemptsCache) {
   super();
   this.attemptsCache = attemptsCache;
}

这种方式最直接,但需要确保LoadingCache的实际创建逻辑(如CacheLoader的定义)可以被外部管理或在生产环境中通过其他方式(如Spring的@Bean)提供。

注意事项与最佳实践

  • 依赖注入的优势: 依赖注入是实现高内聚低耦合、提高代码可测试性和可维护性的基石。它使得组件可以独立开发和测试,易于替换和扩展。
  • 选择合适的注入方式:
    • 如果缓存的创建逻辑复杂,且需要多种配置(例如不同的过期策略),工厂模式是更好的选择。
    • 如果只想暴露缓存的一部分功能,或者需要对缓存的行为进行更细粒度的控制,自定义包装器会更合适。
    • 如果缓存的创建相对简单,且直接注入第三方接口没有问题,直接注入接口是最简洁的方式。
  • Mockito的doThrow().when(): 这是模拟void方法或在when()语句中无法直接调用方法时抛出异常的标准方式。对于有返回值的方法,也可以使用when().thenThrow()。
  • 测试覆盖率: 采用上述方法后,unlockFailed方法中的catch (ExecutionException e)块将能够被成功触发和覆盖,从而提高代码的测试完整性。

总结

通过引入依赖注入模式,无论是通过缓存工厂还是自定义缓存包装器,我们都成功地将MyService与LoadingCache的具体实现解耦。这不仅解决了在单元测试中模拟内部依赖并覆盖异常处理块的难题,更重要的是,它提升了代码的设计质量和可维护性。在未来的开发中,应优先考虑使用依赖注入来管理类之间的协作关系,以构建更健壮、更易于测试的应用程序。

热门AI工具

更多
DeepSeek

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

WorkBuddy

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

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

二狗PPT
二狗PPT Hot

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

豆包大模型

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

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

相关专题

更多
spring框架介绍
spring框架介绍

本专题整合了spring框架相关内容,想了解更多详细内容,请阅读专题下面的文章。

2111

2025.08.06

Java Spring Security 与认证授权
Java Spring Security 与认证授权

本专题系统讲解 Java Spring Security 框架在认证与授权中的应用,涵盖用户身份验证、权限控制、JWT与OAuth2实现、跨站请求伪造(CSRF)防护、会话管理与安全漏洞防范。通过实际项目案例,帮助学习者掌握如何 使用 Spring Security 实现高安全性认证与授权机制,提升 Web 应用的安全性与用户数据保护。

397

2026.01.26

guava包作用
guava包作用

guava是一个java库,增强了java标准库,提供更有效率和易于使用的集合、实用程序、缓存和并发工具。想了解更多guava的相关内容,可以阅读本专题下面的文章。

4231

2024.05.29

string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

5099

2023.08.02

javascriptvoid(o)怎么解决
javascriptvoid(o)怎么解决

javascriptvoid(o)的解决办法:1、检查语法错误;2、确保正确的执行环境;3、检查其他代码的冲突;4、使用事件委托;5、使用其他绑定方式;6、检查外部资源等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

616

2023.11.23

java中void的含义
java中void的含义

本专题整合了Java中void的相关内容,阅读专题下面的文章了解更多详细内容。

331

2025.11.27

硬盘接口类型介绍
硬盘接口类型介绍

硬盘接口类型有IDE、SATA、SCSI、Fibre Channel、USB、eSATA、mSATA、PCIe等等。详细介绍:1、IDE接口是一种并行接口,主要用于连接硬盘和光驱等设备,它主要有两种类型:ATA和ATAPI,IDE接口已经逐渐被SATA接口;2、SATA接口是一种串行接口,相较于IDE接口,它具有更高的传输速度、更低的功耗和更小的体积;3、SCSI接口等等。

3028

2023.10.19

PHP接口编写教程
PHP接口编写教程

本专题整合了PHP接口编写教程,阅读专题下面的文章了解更多详细内容。

4229

2025.10.17

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

0

2026.09.23

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
dev.java 官方:Learn Java
dev.java 官方:Learn Java

共0课时 | 0人学习

Java JDBC数据库连接官方教程
Java JDBC数据库连接官方教程

共0课时 | 0人学习

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

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