使用PowerMock无法Mock java.net.URL？S3预签名URL上传JUnit求助

我来帮你解决这个PowerMock mock URL失败的问题！你的测试代码核心问题是@PrepareForTest注解的目标类不正确，再加上一些Mockito API的过时用法，导致mock没有生效。下面是修正后的完整测试代码，以及关键修改点的详细解释：

修正后的JUnit测试代码 #

import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.net.HttpURLConnection;
import java.net.URL;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.ArgumentMatchers;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

@RunWith(PowerMockRunner.class)
// 关键：必须指定**调用new操作的类**（也就是App.class），而不是被实例化的类
@PrepareForTest(App.class)
public class AppTest {
    @InjectMocks
    private App app;

    @Test
    public void uploadToS3_should_return_true() throws Exception {
        // 1. Mock所有依赖对象
        URL mockUrl = PowerMockito.mock(URL.class);
        HttpURLConnection mockConnection = PowerMockito.mock(HttpURLConnection.class);
        OutputStream mockOutputStream = PowerMockito.mock(OutputStream.class);
        OutputStreamWriter mockWriter = PowerMockito.mock(OutputStreamWriter.class);

        // 2. Mock URL构造方法：拦截App类中new URL(任意字符串)的调用
        PowerMockito.whenNew(URL.class).withArguments(ArgumentMatchers.anyString()).thenReturn(mockUrl);

        // 3. Mock URL.openConnection()返回我们的mock连接
        PowerMockito.when(mockUrl.openConnection()).thenReturn(mockConnection);

        // 4. Mock连接的void方法（setDoOutput、setRequestMethod）
        PowerMockito.doNothing().when(mockConnection).setDoOutput(true);
        PowerMockito.doNothing().when(mockConnection).setRequestMethod("PUT");

        // 5. Mock连接的输出流
        PowerMockito.when(mockConnection.getOutputStream()).thenReturn(mockOutputStream);

        // 6. Mock OutputStreamWriter构造方法：拦截App类中new OutputStreamWriter的调用
        PowerMockito.whenNew(OutputStreamWriter.class)
            .withArguments(ArgumentMatchers.any(OutputStream.class))
            .thenReturn(mockWriter);

        // 7. Mock响应码为201（S3上传成功的典型响应码）
        PowerMockito.when(mockConnection.getResponseCode()).thenReturn(201);

        // 执行测试方法并断言结果
        boolean uploadResult = app.uploadToS3("http://something.com", "{}");
        org.junit.Assert.assertTrue("Upload should succeed", uploadResult);

        // 验证关键业务逻辑是否执行（可选但推荐，确保代码走了正确的分支）
        PowerMockito.verify(mockWriter).write("{}");
        PowerMockito.verify(mockWriter).close();
        PowerMockito.verify(mockConnection).getResponseCode();
    }
}

关键修改点说明 #

• 正确设置@PrepareForTest
  这是你测试失败的核心原因：PowerMock的whenNew需要拦截在某个类内部执行的new操作，所以必须把执行new URL()和new OutputStreamWriter()的App.class加入@PrepareForTest，而不是URL或OutputStreamWriter这些被实例化的类。原来的代码把被实例化的类加入注解，导致PowerMock无法拦截App类里的构造调用。

• 替换过时的Mockito API
  原来的Matchers类已经被Mockito标记为过时，改用ArgumentMatchers类的静态方法（比如anyString()、any(OutputStream.class)），这是更规范的用法。

• 避免重复实例化被测类
  你原来的代码同时用了@InjectMocks和手动new App()，这会导致测试用的App实例不统一。保留@InjectMocks即可，PowerMock会自动创建App实例并处理依赖注入（即使你的App没有依赖，保持写法一致也更清晰）。

• 添加断言和验证逻辑
  新增了Assert.assertTrue来验证方法返回值是否符合预期，同时用PowerMockito.verify来确认业务代码中的关键步骤（写入JSON内容、关闭流、获取响应码）都被执行，让测试更健壮，能及时发现逻辑问题。

• Mock void方法
  业务代码中调用了connection.setDoOutput(true)和setRequestMethod("PUT")这些void方法，用PowerMockito.doNothing()来mock这些调用，避免不必要的异常抛出。

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