嘲笑 okhttp 响应

exception java mockito okhttp

2022-09-03 04:07:51

我有下面的方法，它执行GET请求3次，直到成功。

嘲笑这种方法的更好方法是什么？我想模拟和测试status_code 401，状态代码500，并希望测试该方法是否执行三次。

在python中，我们有 https://github.com/getsentry/responses 它直接模拟请求，因此很容易测试这些方法。

Java中有什么等效的东西吗？

@Override
    public <T> UResponse<T> get(Request request, JSONUnmarshaler<T> unmarshaller, Gson gson) throws UException {
        int status_code = 0;
        String next = null;
        String rawJSON = null;
        JsonElement jsonelement = null;
        Boolean retry = true;
        try {
            int attempts = 3;
            while ((attempts >= 0)  && (retry) && status_code != 200){
                Response response = this.client.newCall(request).execute();

                rawJSON = response.body().string();

                jsonelement = gson.fromJson(rawJSON, JsonElement.class);

                next = gson.fromJson(jsonelement.getAsJsonObject().get("next"), String.class);

                status_code = response.code();

                if (status_code == 401) {
                    try {
                        logger.warn("token expired");
                        TimeUnit.SECONDS.sleep(5);
                        retry = true;
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    }
                }

                if ((status_code / 100 ) == 5){
                    logger.warn("gateway error");
                    retry = true;                   
                }
                attempts -= 1;
                retry = false;

            }
            if (status_code != 200){
                throw new UException();
            }
            return new UResponse<T>(status_code, next, rawJSON,
                    unmarshaller.fromJSON(gson, jsonelement.getAsJsonObject()));

        } catch (IOException e) {
            e.printStackTrace();
            throw new UException();

        }

答案 1

这将很难测试，因为它会循环。一种解决方案是分解成一个单独的函数。然后，您可以使用间谍来存根该方法，以根据剩余的尝试次数返回不同的响应。
我意识到okhttp的好处是，你可以自己建立一个虚假的响应。例如this.client.newCall(request).execute()sendRequest(Request request, int attemptsRemaining)

Request mockRequest = new Request.Builder()
                .url("https://some-url.com")
                .build();
return new Response.Builder()
                .request(mockRequest)
                .protocol(Protocol.HTTP_2)
                .code(401) // status code
                .message("")
                .body(ResponseBody.create(
                        MediaType.get("application/json; charset=utf-8"),
                        "{}"
                ))
                .build();

因此，您可以创建并存根一个返回响应的简单方法，并且基本上可以进行测试。

答案 2

有一些模拟框架可以添加到你的依赖项列表中，这一切都很好，但是，我是老式的，更喜欢模拟和方法：remoteCallexecute

将此方法复制到测试类中：

private static OkHttpClient mockHttpClient(final String serializedBody) throws IOException {
        final OkHttpClient okHttpClient = mock(OkHttpClient.class);

        final Call remoteCall = mock(Call.class);

        final Response response = new Response.Builder()
                .request(new Request.Builder().url("http://url.com").build())
                .protocol(Protocol.HTTP_1_1)
                .code(200).message("").body(
                   ResponseBody.create(
                        MediaType.parse("application/json"),
                        serializedBody
                ))
                .build();

        when(remoteCall.execute()).thenReturn(response);
        when(okHttpClient.newCall(any())).thenReturn(remoteCall);

        return okHttpClient;
    }

用法：

final OkHttpClient mockedClient = mockHttpClient("{\"key\": \"val\"}");