【Java】 使用Mockito轻松模拟静态方法

>
> 【痕迹】QQ+微信朋友圈和聊天记录分析工具
>
> （1）纯Python语言实现，使用Flask后端，本地分析，不上传个人数据。
>
> （2）内含QQ、微信聊天记录保存到本地的方法，真正实现自己数据自己管理。
>
> （3）数据可视化分析QQ、微信聊天记录，提取某一天的聊天记录与大模型对话。
> 
>  下载地址：https://www.alipan.com/s/x6fqXe1jVg1
>

基本原理

在软件开发过程中，单元测试是保证代码质量的重要手段。但是，当测试涉及到静态方法时，传统的Mocking工具往往显得无能为力。幸运的是，Mockito为我们提供了一种解决方案，让我们可以轻松模拟静态方法，从而实现对静态方法的单元测试。

Mockito是一个流行的Java测试框架，它允许开发者模拟（mock）对象的行为，以便于在单元测试中隔离测试对象，避免依赖外部系统或服务。然而，Mockito在早期版本中并不支持静态方法的模拟。随着Mockito 3.4.0版本的发布，这一限制被打破，我们可以使用Mockito.mockStatic方法来模拟静态方法了。

代码示例

示例1：模拟单个静态方法

假设我们有一个工具类MathUtils，其中包含一个静态方法add，我们想要测试这个方法的行为。

public class MathUtils {
    public static int add(int a, int b) {
        return a + b;
    }
}

使用Mockito模拟这个静态方法的示例代码如下：

import static org.mockito.Mockito.*;

public class MathUtilsTest {
    @Test
    public void testAdd() {
        // 模拟静态方法
        mockStatic(MathUtils.class);

        // 配置模拟对象的行为
        when(MathUtils.add(10, 5)).thenReturn(20);

        // 调用静态方法并验证结果
        int result = MathUtils.add(10, 5);
        assertEquals(20, result);

        // 验证静态方法是否被调用
        verifyStatic(MathUtils.class);
        MathUtils.add(10, 5);
    }
}

示例2：模拟多个静态方法

如果我们想要同时模拟MathUtils类中的多个静态方法，可以这样做：

public class MathUtils {
    public static int add(int a, int b) {
        return a + b;
    }

    public static int subtract(int a, int b) {
        return a - b;
    }
}

模拟多个静态方法的示例代码：

import static org.mockito.Mockito.*;

public class MathUtilsTest {
    @Test
    public void testStaticMethods() {
        // 模拟静态方法
        mockStatic(MathUtils.class);

        // 配置第一个静态方法的行为
        when(MathUtils.add(10, 5)).thenReturn(20);

        // 配置第二个静态方法的行为
        when(MathUtils.subtract(10, 5)).thenReturn(5);

        // 调用静态方法并验证结果
        assertEquals(20, MathUtils.add(10, 5));
        assertEquals(5, MathUtils.subtract(10, 5));

        // 验证静态方法是否被调用
        verifyStatic(MathUtils.class);
        MathUtils.add(10, 5);
        MathUtils.subtract(10, 5);
    }
}

示例3：模拟静态方法并使用参数匹配

有时候，我们可能需要根据输入参数的不同来返回不同的结果，这时可以使用参数匹配：

import static org.mockito.Mockito.*;

public class MathUtilsTest {
    @Test
    public void testAddWithParameterMatching() {
        // 模拟静态方法
        mockStatic(MathUtils.class);

        // 使用参数匹配来配置静态方法的行为
        when(MathUtils.add(anyInt(), anyInt())).thenAnswer(invocation -> {
            int a = invocation.getArgument(0);
            int b = invocation.getArgument(1);
            return a * b; // 这里我们返回的是乘积，而不是和
        });

        // 调用静态方法并验证结果
        assertEquals(50, MathUtils.add(10, 5));
        assertEquals(18, MathUtils.add(2, 9));

        // 验证静态方法是否被调用
        verifyStatic(MathUtils.class);
        MathUtils.add(10, 5);
        MathUtils.add(2, 9);
    }
}

注意事项

1. 版本要求：请确保你使用的Mockito版本至少是3.4.0，因为这是支持静态方法模拟的最低版本。
2. 线程安全：mockStatic是线程安全的，但是每个静态模拟都是针对当前线程的。如果你的测试在多线程环境中运行，需要确保每个线程都正确地设置了静态模拟。
3. 资源清理：在测试结束后，应该使用mockStatic(YourClass.class).close()来关闭静态模拟，以避免潜在的内存泄漏。
4. Mockito验证：使用verifyStatic来确保静态方法被正确调用，这有助于验证测试覆盖率。

结论

Mockito的静态方法模拟功能极大地简化了对静态方法的单元测试。通过上述示例，我们可以看到如何模拟单个或多个静态方法，以及如何使用参数匹配来控制方法的行为。正确使用Mockito的这一功能，可以帮助我们构建更加健壮和可靠的测试套件，从而提高软件质量。记住，始终确保使用最新版本的Mockito，并在测试结束后清理资源，以避免潜在的问题。

>
> 【痕迹】QQ+微信朋友圈和聊天记录分析工具
>
> （1）纯Python语言实现，使用Flask后端，本地分析，不上传个人数据。
>
> （2）内含QQ、微信聊天记录保存到本地的方法，真正实现自己数据自己管理。
>
> （3）数据可视化分析QQ、微信聊天记录，提取某一天的聊天记录与大模型对话。
>
> 下载地址：https://www.alipan.com/s/x6fqXe1jVg1
>