如何使用 Mockito 和 Spring Boot 使用 okhttp 编写单元测试

Question

我正在尝试在我的 Spring Boot 应用程序中将 okhttp 与 Mockito 一起使用，但我总是在行时收到 NullPointerException

httpClient.newCall(request).execute()

我们如何将 Mockito 与 okhttp 一起使用？

@Component
public class HStatus {

    @Autowired
    private OkHttpClient httpClient;

    public int HCheck() throws URISyntaxException, IOException {

        String url = new URIBuilder("Some URL").build().toString();

        Request request = new Request.Builder().url(url).get().build();

        try (Response resp = httpClient.newCall(request).execute()) {

            return resp.code();

        }

    }

在junit中，我正在尝试做-

@RunWith(MockitoJUnitRunner.class)
public class HStatusTest {
    
    @InjectMocks
    private HStatus hStatus;

    @Mock
    private OkHttpClient httpClient;


    @Test
    public void TestHCheck_Should_Return_200_StatusCode() throws URISyntaxException, IOException {

        Response resp = new Response.Builder()
                .request(new Request.Builder().url("SOME URL").build())
                .protocol(Protocol.HTTP_1_1)
                .code(200).message("")
                .build();

        Mockito.when(httpClient.newCall(ArgumentMatchers.any()).execute).thenReturn(resp);

// Getting NullPointer at this call -> httpClient.newCall(ArgumentMatchers.any()).execute

        int statusCode = hStatus.HCheck();
        
        assertEquals(200, statusCode);

    }
}

Answer 1

我猜你正在创建 OkHttpClient 的新实例。

对于创建新实例并不容易用 mockito 模拟它们。你可以使用 mockito @Spy 但我不喜欢那样。

否则你可以使用powermock

 OkHttpClient mockClient = mock(OkHttpClient.class);
 PowerMockito.whenNew(OkHttpClient.class).withNoArguments().thenReturn(mockClient);

另一种方式： 为 new OkHttpClient() 创建一个 getter 类；并模拟它。

getClient(){
    return new OkHttpClient();
}

模拟

when(class.getClient()).thenReturn(mockClient);