使用 Mockito 和 jUnit 为具有两个 HTTP 调用的方法编写单元测试

Question

我有三种方法，

• 无效保存（字符串 a，字符串 b）
• String getId(String a, Pojo p)
• 无效更新（字符串 a，字符串 b，字符串实体 c）

这里是实现（不是完整的实现），

String getId(String accessToken, Pojo p) {

    //Note that the pojo is for formatting the payload for the request, nothing more

    HttpResponse response = httpRequestService.makeGetRequest(url, TOKEN_PREFIX, accessToken,
            CONTENT_TYPE);
    if (response.getStatusLine().getStatusCode() == 200) {
        log.debug("Success");
    }

    //Code for getting the id from the response

    return id;
}

void update(String accassToken, String id, StringEntity payload) {

    HttpResponse response = httpRequestService.makePutRequest(url + id,
            TOKEN_PREFIX, accessToken, CONTENT_TYPE, payload);

    if (response.getStatusLine().getStatusCode() == 200) {
        log.debug("Success");
    }
}

void save(String accessToken, String payload) {

    //The getId() is called here
    String id = getId(/*arguments*/);

    if(id == null) {
        log.error("Error");
    } else {
        //The update() is called here
        update(/*arguments*/);
    }
}

如前所述，getId 和 update 方法在 save 方法中调用，getId 和 update 方法都有 HTTP 调用。

我必须为 save() 方法编写一个单元测试。这是我尝试过的。

    //This is an interface with methods to call HTTP requests.
    HttpRequestService httpRequestService = Mockito.mock(HttpRequestService.class);

    //Constructor injection
    ClassWithMethods a = new ClasswithMethods(httpRequestService);

    HttpResponse responseGet = Mockito.mock(HttpResponse.class);
    StatusLine statusLineGet = Mockito.mock(StatusLine.class);
    HttpEntity httpEntity = Mockito.mock(HttpEntity.class);
    Mockito.when(responseGet.getStatusLine()).thenReturn(statusLineGet);
    Mockito.when(statusLineGet.getStatusCode()).thenReturn(200);
    Mockito.when(responseGet.getEntity()).thenReturn(httpEntity);
    Mockito.when(httpEntity.getContent()).thenReturn(IOUtils.toInputStream(stream));
    Mockito.when(httpRequestService.makeGetRequest(url, TOKEN_PREFIX, accessToken,
            CONTENT_TYPE).thenReturn(responseGet);

    HttpResponse responsePut = Mockito.mock(HttpResponse.class);
    StatusLine statusLinePut = Mockito.mock(StatusLine.class);
    Mockito.when(responsePut.getStatusLine()).thenReturn(statusLinePut);
    Mockito.when(statusLinePut.getStatusCode()).thenReturn(200);
    Mockito.when(httpRequestService.makePutRequest(url + id, TOKEN_PREFIX,
            accessToken, CONTENT_TYPE, payloadEntity).thenReturn(responsePut);

    a.save(accessToken, payload);

但是在测试时，responsePut 返回一个空值。参数也匹配。

问题：如何测试这个调用两个HTTP方法的save方法？

这可能不是最好的方法。如果有更好的方法来测试这样的方法，请提出建议。

谢谢

请注意，httpRequestService 是一个带有 HTTP 调用方法的接口，也使用了构造函数注入。

Answer 1

您应该将以下内容提取到一个单独的方法中（PUT 也是如此）：

HttpResponse response = httpRequestService.makeGetRequest(url, TOKEN_PREFIX, accessToken,
            CONTENT_TYPE);

然后你可以使用 mockito 来存根该方法：

when(obj.methodGET(...)).thenReturns(...)

记住：您的单元测试应该不通过网络进行调用！

编辑

private HttpResponse get(String url, ...) {
    return httpRequestService.makeGetRequest(url, TOKEN_PREFIX, accessToken,
            CONTENT_TYPE);
}


private HttpResponse put(String url, ...) {
    return httpRequestService.makePutRequest(url + id,
        TOKEN_PREFIX, accessToken, CONTENT_TYPE, payload);
}

然后：

...
when(mockedStatusLineObj.getStatusCode()).thenReturns(200);
HttpResponse mockedGet = mock(HttpResponse.class);
when(mockedGet.getStatusLine()).thenReturns(mockedStatusLineObj);
when(object.get(...)).thenReturns(mockedGet);
when(object.put(...)).thenReturns(...);

Answer 2

关于您的测试方式

实际上，您的测试设置很好，但阅读起来很复杂。
测试必须保持可读性和简单性。
否则理解和维护它会变得很困难。 你想模拟HttpRequestService。
如果您想 unit-test 您的班级并且不想考虑集成问题，这是一个好方法。
但是你模拟被模拟的HttpRequestService 的依赖项的方式使得代码不可读：

HttpResponse responsePut = Mockito.mock(HttpResponse.class);
StatusLine statusLinePut = Mockito.mock(StatusLine.class);
Mockito.when(responsePut.getStatusLine()).thenReturn(statusLinePut);
Mockito.when(statusLinePut.getStatusCode()).thenReturn(200);

您在每次调用时重复此操作。
在方法中提取它们：

public HttpResponse createMockedHttpResponse(String url){
    HttpResponse responsePut = Mockito.mock(HttpResponse.class);
    StatusLine statusLinePut = Mockito.mock(StatusLine.class);
    Mockito.when(responsePut.getStatusLine()).thenReturn(statusLinePut);
    Mockito.when(statusLinePut.getStatusCode()).thenReturn(200);
    return responsePut;
}

关于测试失败

它只是在线路上抛出一个NullPointerException，

if (response.getStatusLine().getStatusCode() == 200) 在update() 方法。

当我调试它时，它显示 responsePut 为空。

如果我查看您的实际代码，这些在被测类中的调用：

getStatusLine().getStatusCode()

在这里被正确地模拟了：

HttpResponse responsePut = Mockito.mock(HttpResponse.class);
StatusLine statusLinePut = Mockito.mock(StatusLine.class);
Mockito.when(responsePut.getStatusLine()).thenReturn(statusLinePut);
Mockito.when(statusLinePut.getStatusCode()).thenReturn(200);

通过排除，表示没有遇到下一条语句的记录行为：

Mockito.when(httpRequestService.makePutRequest(url + id, TOKEN_PREFIX,
        accessToken, CONTENT_TYPE, payloadEntity).thenReturn(responsePut);

因此请确保makePutRequest() 的记录行为和makePutRequest() 的实际调用具有完全相同/equals() 传递的参数。
如果某些参数对于测试逻辑来说并不是真正需要的，请不要犹豫，通过Mockito.any()