Mockito.when表达式求值问题：JUnit测试S3客户端相关逻辑

我太懂你这种困扰了——测试涉及S3客户端的代码时，Mockito的when语法总是容易踩坑，要么触发了真实的网络调用，要么Mock的返回根本不生效。结合你说的场景（类M依赖类P，P持有S3客户端，测试M的processInputFile方法），我给你梳理几个最常见的问题和解决办法：

1. 先搞清楚：你是不是触发了真实方法调用？ #

很多时候Mockito.when()里的表达式会先执行真实方法（哪怕是Mock对象的默认实现），比如你写了：

when(s3Client.getObject("my-bucket", "input.txt")).thenReturn(mockS3Object);

这里的getObject方法会先被调用一次，如果你的Mock对象没正确设置，可能会抛出NPE或者触发真实网络请求。

解决办法：改用doReturn().when()语法
这种方式不会先执行目标方法，直接指定返回值，完美避开求值问题：

// 先创建好Mock的S3Object和输入流
S3Object mockS3Object = Mockito.mock(S3Object.class);
S3ObjectInputStream mockInputStream = new S3ObjectInputStream(
    new ByteArrayInputStream("test data".getBytes()), 
    null
);
when(mockS3Object.getObjectContent()).thenReturn(mockInputStream);

// 用doReturn替代when(...)
doReturn(mockS3Object).when(s3Client).getObject(anyString(), anyString());

2. 确保S3客户端是Mock的，且正确注入到M类里 #

如果你的类P是M的成员变量，那要保证P里的S3客户端是Mock实例，而不是真实的客户端：

场景A：类M通过构造/Setter注入类P #

在@Before的setup方法里，Mock类P和S3客户端，然后把Mock的P传给M：

private M mInstance;
private P mockP;
private AmazonS3 mockS3Client;

@Before
public void setup() {
    // 1. Mock S3客户端
    mockS3Client = Mockito.mock(AmazonS3.class);
    // 2. Mock类P，并让它返回Mock的S3客户端
    mockP = Mockito.mock(P.class);
    when(mockP.getS3Client()).thenReturn(mockS3Client);
    // 3. 初始化M类，注入Mock的P
    mInstance = new M(mockP);
    
    // 4. Mock S3的getObject方法返回
    S3Object mockS3Obj = Mockito.mock(S3Object.class);
    doReturn(mockS3Obj).when(mockS3Client).getObject(anyString(), anyString());
    doReturn(new S3ObjectInputStream(new ByteArrayInputStream("test content".getBytes()), null))
        .when(mockS3Obj).getObjectContent();
}

场景B：类P里是静态方法创建S3客户端（比如AmazonS3ClientBuilder） #

如果类P里是用AmazonS3ClientBuilder.defaultClient()这种静态方法创建的S3客户端，那普通的Mock没法处理，需要用Mockito.mockStatic来Mock静态类：

@Before
public void setup() {
    // Mock AmazonS3ClientBuilder的静态方法
    try (MockedStatic<AmazonS3ClientBuilder> mockedBuilder = Mockito.mockStatic(AmazonS3ClientBuilder.class)) {
        mockS3Client = Mockito.mock(AmazonS3.class);
        // 指定静态方法返回Mock的S3客户端
        mockedBuilder.when(AmazonS3ClientBuilder::defaultClient).thenReturn(mockS3Client);
        
        // 初始化P和M类（此时P会使用Mock的S3客户端）
        mockP = new P();
        mInstance = new M(mockP);
        
        // 后续Mock getObject的逻辑和之前一样
        S3Object mockS3Obj = Mockito.mock(S3Object.class);
        doReturn(mockS3Obj).when(mockS3Client).getObject(anyString(), anyString());
        when(mockS3Obj.getObjectContent()).thenReturn(new S3ObjectInputStream(new ByteArrayInputStream("test data".getBytes()), null));
    }
}

3. 检查参数匹配器是否正确 #

如果你的when里用了硬编码的参数（比如"my-bucket"），但实际代码中传入的参数和这个不匹配，Mock也会失效。这时候建议用参数匹配器：

• 匹配任意字符串：anyString()
• 匹配特定字符串：eq("my-bucket")
• 匹配任意对象：any()

比如：

doReturn(mockS3Object).when(mockS3Client).getObject(eq("my-target-bucket"), anyString());

最后再验证一下 #

写完测试后，你可以加一行验证代码，确认S3客户端的方法确实被调用了：

verify(mockS3Client, times(1)).getObject(anyString(), anyString());

这样能帮你快速排查Mock是否生效。

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