Mockito如何为静态类的多次调用配置不同返回值？

嘿，这个问题我太熟悉了！你遇到的是静态方法mock时，后续返回值覆盖前序的典型问题。下面我给你一步步讲解如何用Mockito让静态类的每次调用返回不同的对象：

核心前提：使用Mockito 3.4+版本 #

从Mockito 3.4开始，官方原生支持静态方法的mock（之前需要依赖PowerMock），所以先确保你的项目依赖的是这个版本及以上的Mockito。

方法1：链式调用thenReturn()指定顺序返回 #

这是最直接的方式，适合按调用顺序返回不同对象的场景。假设你的静态类是StaticResourceLoader，方法是getInputStream(String filePath)，我们需要第一次调用返回stream1，第二次返回stream2：

首先准备两个独立的InputStream实例：

// 创建两个不同的输入流，确保内容不共享
InputStream stream1 = new ByteArrayInputStream("First call content".getBytes(StandardCharsets.UTF_8));
InputStream stream2 = new ByteArrayInputStream("Second call content".getBytes(StandardCharsets.UTF_8));

然后用MockedStatic包裹静态类，设置顺序返回：

// 使用try-with-resources确保静态mock自动清理，避免影响其他测试
try (MockedStatic<StaticResourceLoader> mockedLoader = Mockito.mockStatic(StaticResourceLoader.class)) {
    // 链式调用thenReturn，依次指定每次调用的返回值
    mockedLoader.when(() -> StaticResourceLoader.getInputStream(Mockito.anyString()))
               .thenReturn(stream1)
               .thenReturn(stream2);

    // 第一次调用
    InputStream result1 = StaticResourceLoader.getInputStream("/test1.txt");
    // 第二次调用
    InputStream result2 = StaticResourceLoader.getInputStream("/test2.txt");

    // 验证结果
    assertEquals("First call content", new String(result1.readAllBytes()));
    assertEquals("Second call content", new String(result2.readAllBytes()));
}

方法2：用thenAnswer()自定义灵活返回逻辑 #

如果你的场景需要根据调用参数、调用次数等动态返回不同对象，thenAnswer()会更适合：

try (MockedStatic<StaticResourceLoader> mockedLoader = Mockito.mockStatic(StaticResourceLoader.class)) {
    mockedLoader.when(() -> StaticResourceLoader.getInputStream(Mockito.anyString()))
               .thenAnswer(invocation -> {
                   // 获取调用时传入的参数
                   String filePath = invocation.getArgument(0);
                   // 根据参数返回不同的流
                   if (filePath.equals("/test1.txt")) {
                       return stream1;
                   } else if (filePath.equals("/test2.txt")) {
                       return stream2;
                   }
                   // 默认返回null或其他默认值
                   return new ByteArrayInputStream(new byte[0]);
               });

    // 调用测试...
}

关键小贴士 #

• 一定要使用独立的InputStream实例，如果两个调用返回同一个流对象，即使mock设置正确，后续的读取操作也会覆盖或影响前一次的内容。
• 必须用try-with-resources管理MockedStatic，它会在代码块结束后自动清理静态mock，防止污染其他测试用例。
• 如果是旧版本Mockito，建议升级到3.4+；实在无法升级的话，可以用PowerMock的mockStatic()配合thenReturn()链式调用实现类似效果，但PowerMock的侵入性更强，不推荐。

内容的提问来源于stack exchange，提问作者anjana