Spring Boot 整合 Mockito：提升Java单元测试的高效实践

引言

在Java开发领域，Spring Boot因其便捷的配置和强大的功能而受到广泛欢迎，而Mockito作为一款成熟的单元测试模拟框架，则在提高测试质量、确保代码模块间解耦方面扮演着至关重要的角色。本文将详细介绍如何在Spring Boot项目中整合Mockito，以及Mockito的概念、功能点、优势及实际应用案例。

一、Mockito概念

Mockito是一个面向Java开发者的模拟框架，它的核心目标是**通过创建和配置模拟对象**（Mock Objects）来替代真实依赖项，以便在单元测试中有效地隔离被测代码。在Spring Boot应用程序中，Mockito可用于模拟DAOs、Services、Repositories以及其他依赖服务，使得测试仅针对单一的业务逻辑进行验证，而无需启动数据库、网络请求等实际资源。

为什么写单元测试？

验证功能正确性：

单元测试允许开发者针对代码的最小可测试单元（如类、方法）逐一验证它们是否按预期工作，确保每个独立组件的功能正确无误。

隔离问题定位：

当系统出现问题时，单元测试能快速定位具体哪个模块出现了故障，避免因多个模块相互影响而导致的诊断困难。

支持持续集成/持续部署（CI/CD）：

在CI/CD流水线中，单元测试作为构建过程的一部分，确保每次提交的新代码都不会破坏现有的功能。

促进重构和演化：

编写了充分的单元测试后，重构代码时就有了安全网，可以放心地修改内部结构而不必担心会影响到现有功能。

设计指导：

TDD（测试驱动开发）提倡先编写单元测试，这有助于推动设计出更易于测试的代码，即模块化程度更高、依赖关系更清晰的设计。

文档作用：

单元测试实际上是另一种形式的文档，它展示了代码如何被预期使用，以及不同输入下产生的输出，是活生生的、可执行的契约。

单元测试的优点

尽早发现问题：

开发阶段就能发现潜在的缺陷，而不是等到集成测试或生产环境中才显现，节省了后期修正的成本。

提升代码质量：

通过全面覆盖边界条件、异常情况和其他关键场景，促使开发人员考虑更多的边缘用例，从而提高代码的健壮性。

可维护性：

有了良好的单元测试覆盖，未来的开发人员更容易理解代码行为，并有信心在修改代码时不会无意中破坏既有功能。

依赖管理：

使用像Mockito这样的框架可以模拟和隔离依赖项，使得测试关注于单个单元本身的行为，不受外部因素的影响。

迭代速度：

单元测试使得开发周期更快，因为开发人员可以迅速验证他们的更改是否有效，无需每次修改后都进行全面的手动回归测试。

信心保障：

经过单元测试的代码提供了额外的信心，尤其是在大型项目中，确保每个模块的质量，有助于形成稳定的软件整体。

**一种测试手段，更是提升代码质量、支持敏捷开发和维护软件长期稳定性的有效工具。

二、Mockito功能点

Mock对象创建：使用Mockito的mock()函数可以轻松创建模拟对象，例如，对于一个UserMapper接口：


UserMapper userMapper = Mockito.mock(UserMapper.class);

方法行为设置：可以通过when()方法定义模拟对象的方法调用时的预期行为，例如设置返回值或抛出异常：

// 准备测试数据和模拟行为
 when(userMapper.findUserByUsernameAndPassword(testLoginReq.getUsername(), testLoginReq.getPassword())).thenReturn(null);

// 执行测试方法并验证期望的异常被抛出
Exception exception = assertThrows(RuntimeException.class, () -> userService.login(testLoginReq));

验证方法调用：使用verify()函数来确保模拟对象的方法已经被正确调用：

// Verify that the method was called with the correct parameters
verify(userMapper).findUserByUsernameAndPassword(testLoginReq.getUsername(), testLoginReq.getPassword());

参数匹配器：提供了一系列参数匹配器，如any(), eq(), argThat()等，方便在验证时不需明确指定参数值：

verify(userMapper).findByEmail(argThat(email -> email.endsWith("@example.com")));

Spies： Mockito还支持创建Spy对象，它允许对已有真实对象进行部分模拟，同时保留原有对象的功能：


UserService realUserService = new UserService();
UserServiceImpl userServiceSpy = Mockito.spy(UserServiceImpl);

三、Mockito优势

隔离性：通过模拟依赖项，避免了测试之间不必要的耦合，提高了单元测试的准确性。
简洁性：Mockito API设计简洁明了，使得编写和维护测试代码变得容易。
深度控制：能够精细控制模拟对象的行为，包括方法调用的顺序、次数和异常处理等。
文档作用：通过模拟的交互，反映了被测试代码对外部依赖的使用方式，起到一定的文档作用。

四、Spring Boot整合Mockito案例

添加POM依赖


<parent>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-parent</artifactId>
    <version>2.4.2</version>
    <relativePath/><!-- lookup parent from repository -->
</parent>


<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-test</artifactId>
    <scope>test</scope>
</dependency>

业务方法

@Service
@Slf4j(topic = "UserServiceImpl")
public class UserServiceImpl implements UserService {

    @Resource
    private UserMapper userMapper;


    @Override
    public LoginUserResp login(LoginUserReq loginReq) {
        log.info("loginReq:{}", loginReq);
        User user = userMapper.findUserByUsernameAndPassword(loginReq.getUsername(), loginReq.getPassword());
        if (Objects.isNull(user)) {
            throw new RuntimeException("用户名或密码错误");
        }
        LoginUserResp loginUserResp = new LoginUserResp();
        loginUserResp.setId(0L);
        loginUserResp.setUsername(user.getUsername());
        loginUserResp.setNickName(user.getNickname());
        loginUserResp.setToken("token");
        loginUserResp.setPhone("phone");
        loginUserResp.setUserType(0);
        return loginUserResp;
    }


    @Override
    public Boolean createUser(UserAddReq userAddReq) {
        log.info("userAddReq:{}", userAddReq);
        String email = userAddReq.getEmail();
        if (Objects.isNull(email)) {
            throw new RuntimeException("邮箱不能为空");
        }
        if (!email.contains("@example.com")) {
            throw new RuntimeException("邮箱格式不正确");
        }
        userMapper.insert(userAddReq);
        return Boolean.TRUE;
    }



}

UserServiceImplTest 测试类

假设我们正在测试一个UserService类，它依赖于UserMapper。在Spring Boot测试中，可以利用@Mock注解来自动创建并替换Spring容器中的Mock对象：


@ExtendWith(MockitoExtension.class)
public class UserServiceImplTest {

    @Mock
    private UserMapper userMapper;

    @InjectMocks
    private UserServiceImpl userService;

    private User testUser;
    private LoginUserReq testLoginReq;
    private LoginUserResp expectedLoginResp;

    private UserAddReq validUserAddReq;
    private UserAddReq invalidEmailUserAddReq;
    private UserAddReq nullEmailUserAddReq;

    @BeforeEach
    public void setUp() {
        testUser = new User();
        testUser.setId(1L);
        testUser.setUsername("testUser");
        testUser.setNickname("TestNick");

        testLoginReq = new LoginUserReq();
        testLoginReq.setUsername("testUser");
        testLoginReq.setPassword("password");

        expectedLoginResp = new LoginUserResp();
        expectedLoginResp.setId(testUser.getId());
        expectedLoginResp.setUsername(testUser.getUsername());
        expectedLoginResp.setNickName(testUser.getNickname());
        expectedLoginResp.setToken("token");
        expectedLoginResp.setPhone("phone");
        expectedLoginResp.setUserType(0);


        validUserAddReq = new UserAddReq();
        validUserAddReq.setUsername("testUser");
        validUserAddReq.setPassword("testPass");
        validUserAddReq.setEmail("test@example.com");

        invalidEmailUserAddReq = new UserAddReq();
        invalidEmailUserAddReq.setUsername("testUser");
        invalidEmailUserAddReq.setPassword("testPass");
        invalidEmailUserAddReq.setEmail("test@example");

        nullEmailUserAddReq = new UserAddReq();
        nullEmailUserAddReq.setUsername("testUser");
        nullEmailUserAddReq.setPassword("testPass");
        nullEmailUserAddReq.setEmail(null);
    }

    /**
     * 测试使用有效的凭据进行登录时，应成功登录。
     *
     * Arrange 配置测试环境：
     * 设置当使用测试请求中的用户名和密码调用 userMapper.findUserByUsernameAndPassword 方法时，
     * 返回预设的测试用户对象。
     *
     * Act 执行动作：
     * 使用测试登录请求调用 userService.login 方法，获取实际的登录响应。
     *
     * Assert 断言结果：
     * 验证实际的登录响应不为空，并且其各个字段（用户名、昵称、令牌、电话、用户类型）与预期的登录响应相匹配。
     *
     * Verify 验证调用：
     * 验证 userMapper.findUserByUsernameAndPassword 方法确实被使用了正确的参数（测试请求中的用户名和密码）调用。
     */
    @Test
    public void whenValidCredentials_thenSuccessfulLogin() {
        // Arrange
        when(userMapper.findUserByUsernameAndPassword(testLoginReq.getUsername(), testLoginReq.getPassword())).thenReturn(testUser);

        // Act
        LoginUserResp actualLoginResp = userService.login(testLoginReq);

        // Assert
        assertNotNull(actualLoginResp);
        assertEquals(expectedLoginResp.getUsername(), actualLoginResp.getUsername());
        assertEquals(expectedLoginResp.getNickName(), actualLoginResp.getNickName());
        assertEquals(expectedLoginResp.getToken(), actualLoginResp.getToken());

        // Verify that the method was called with the correct parameters
        verify(userMapper).findUserByUsernameAndPassword(testLoginReq.getUsername(), testLoginReq.getPassword());
    }


    /**
     * 测试登录服务时，使用无效的用户名和密码应该导致登录失败。
     * 这个测试用例验证当提供的用户名和密码不匹配任何已知用户时，login方法是否抛出运行时异常。
     */
    @Test
    public void whenInvalidCredentials_thenLoginFailure() {
        // 准备测试数据和模拟行为
        when(userMapper.findUserByUsernameAndPassword(testLoginReq.getUsername(), testLoginReq.getPassword())).thenReturn(null);

        // 执行测试方法并验证期望的异常被抛出
        Exception exception = assertThrows(RuntimeException.class, () -> userService.login(testLoginReq));

        // 验证抛出的异常消息是否匹配预期
        assertEquals("用户名或密码错误", exception.getMessage());

        // 验证userMapper的findUserByUsernameAndPassword方法是否被正确调用
        verify(userMapper).findUserByUsernameAndPassword(testLoginReq.getUsername(), testLoginReq.getPassword());
    }


    /**
     * 测试创建用户功能。
     * 当提供的用户信息有效时，应该成功保存用户信息并返回true。
     */
    @Test
    public void createUser_WithValidUser_ShouldPersistAndReturnTrue() {
        // 准备测试环境
        when(userMapper.insert(any(UserAddReq.class))).thenReturn(1);

        // 执行测试动作
        Boolean result = userService.createUser(validUserAddReq);

        // 验证测试结果
        assertTrue(result);
        verify(userMapper).insert(validUserAddReq);
    }



}

五、异常处理与断言
在Mockito中，可以模拟方法抛出异常，并在测试中捕获和验证：

/**
     * 测试创建用户时使用无效邮箱地址应该抛出异常的情况。
     * 该测试方法不会返回任何值，它的目的是验证当提供一个无效的邮箱地址时，
     * {@link userService.createUser(UserAddReq)} 方法是否会抛出预期的 {@link RuntimeException} 异常。
     * 
     * @param none 该测试方法不接受任何参数。
     * @return void 该测试方法没有返回值。
     * @throws RuntimeException 如果提供的用户添加请求中的邮箱地址无效，该方法将抛出异常。
     */
@Test
public void createUser_WithInvalidEmail_ShouldThrowException() {
    // 断言当尝试使用无效的邮箱创建用户时，会抛出运行时异常
    Exception exception = assertThrows(RuntimeException.class, () -> {
        userService.createUser(invalidEmailUserAddReq);
    });

    // 验证抛出的异常消息是否为预期的错误消息
    assertEquals("邮箱格式不正确", exception.getMessage());

    // 验证用户映射器的 insert 方法是否从未被调用
    verify(userMapper, never()).insert(any(UserAddReq.class));
}


/**
     * 测试创建用户时，如果邮箱为null，应该抛出异常。
     * 这个测试方法不接受任何参数，也不会返回任何值。
     * 它主要通过断言验证在尝试使用null邮箱创建用户时，是否会抛出运行时异常，并且异常的消息文本是否正确。
     */
@Test
public void createUser_WithNullEmail_ShouldThrowException() {
    // Act & Assert: 尝试使用null邮箱创建用户，并验证是否抛出了预期的运行时异常
    Exception exception = assertThrows(RuntimeException.class, () -> {
        userService.createUser(nullEmailUserAddReq);
    });

    assertEquals("邮箱不能为空", exception.getMessage()); // 验证异常消息是否正确
    verify(userMapper, never()).insert(any(UserAddReq.class)); // 验证用户映射器的insert方法是否从未被调用
}

五、统计单元测试覆盖率

一、单元测试覆盖率概念

单元测试覆盖率是指程序中被执行的单元测试所覆盖的源代码行数或分支数占总行数或分支数的比例。通常分为行覆盖率、分支覆盖率、语句覆盖率、方法覆盖率等多种度量维度。理想的覆盖率并非追求100%，而是力求覆盖所有关键路径和边界条件，以最大程度地暴露潜在错误。

二、单元测试覆盖率的重要性

保证代码质量：高覆盖率意味着更多的代码逻辑经过了直接或间接的验证，有助于减少因未测试代码引入的缺陷。
推动重构与优化：覆盖率数据可以帮助识别冗余或难以测试的代码段，进而推动代码结构的改进。
持续集成与持续部署：在CI/CD流程中，设定合理的覆盖率阈值，可以作为构建是否通过的门槛，防止低质量代码流入生产环境。

三、主流覆盖率统计工具

JaCoCo：JaCoCo是一款适用于Java字节码的开源覆盖率工具，它支持无缝集成到Maven、Gradle构建工具和Eclipse、IntelliJ IDEA等IDE中。对于Spring Boot应用，可以通过JaCoCo插件轻松获取和报告单元测试覆盖率。

<!-- Maven中JaCoCo配置示例 -->
<build>
    <plugins>
        <plugin>
            <groupId>org.jacoco</groupId>
            <artifactId>jacoco-maven-plugin</artifactId>
            <version>0.8.7</version>
            <executions>
                <execution>
                    <goals>
                        <goal>prepare-agent</goal>
                    </goals>
                </execution>
                <execution>
                    <id>report</id>
                    <phase>test</phase>
                    <goals>
                        <goal>report</goal>
                    </goals>
                </execution>
            </executions>
        </plugin>
    </plugins>
</build>