使用Robolectric回调测试okHttp请求

Question

我有一个函数要测试，它在okHttp回调中运行。我正在尝试使用Robolectrics测试它，但回调从未执行过。我想这是因为测试在请求之后继续进行，而不需要等待okHttp返回。到目前为止，我已经尝试过：

    ShadowLooper.pauseMainLooper();
    Robolectric.flushBackgroundScheduler();
    ShadowLooper.unPauseMainLooper();

但那没用。有什么建议吗？

编辑：

下面是我的代码示例：

ApiClient.sendSomeDataToServer(data, callback);

其中ApiClient是一个包含okHttp客户端的助手类。sendSomeDataToServer API调用看起来像这样：

public static void sendSomeDataToServer(MyObject data, Callback callback){
    final Request request = new Request.Builder()
            .url(API_SOME_URL)
            .post(RequestBody.create(JSON, myObject.getAsJson().toString()))
            .build();
    sHttpClient.newCall(request).enqueue(callback);
}

其中sHttpClient是初始化的OkHttpClient。

我可以通过在测试代码中强制使用Thread.sleep(5000)并提供自定义回调来测试上面的执行。我试图测试的代码在回调中。有什么建议我可以测试一下吗？我真的不想为了适应测试框架而更改主代码——应该是相反的。

Answer 1

让我们假设您有下一个代码。接口：

@GET("/user/{id}/photo")  
void listUsers(@Path("id") int id, Callback<Photo> cb);

实施：

public void fetchData() {
    RestAdapter restAdapter = new RestAdapter.Builder()
                .setServer("baseURL")     
                .build();
    ClientInterface service = restAdapter.create(ClientInterface.class);
    Callback<Photo> callback = new Callback<Photo>() {
        @Override
        public void success(Photo o, Response response) {
        }
        @Override
        public void failure(RetrofitError retrofitError) {
        }
    };
    service.listUsers(435, callback);
}

首先，您需要将service实例化更改为service注入（作为参数或字段）。我会把它作为参数：

public void fetchData(ClientInterface clients) {
}

之后的文本是相当琐碎的：

@Test
public void checkThatServiceSuccessIsProcessed() {
    ClientInterface mockedClients = mock(ClientInterface.class);
    activity.fetchData(mockedClients);
    // get callback
    ArgumentCaptor<Callback<Photo>> captor = (ArgumentCaptor<Callback<Photo>>)ArgumentCaptor.forClass(Callback.class);
    verify(mockedInterface).listUsers(anything(), captor.capture());
    Callback<Photo> passedCallback = captor.value();
    // run callback
    callback.success(...);
    // check your conditions
}

用于模拟和验证的库是Mockito。

由于泛型的原因，captor实例化会有一个警告，但如果您使用@Captor注释而不是手工创建captor，则可以修复。

参数注入并不完美，尤其是在活动的情况下。这是为了简化示例。考虑是否使用库进行正确的注入。我鼓励您尝试Dagger注射

Answer 2

您可以使用ArgumentCaptor（Mockito的类）。参考：http://www.mdswanson.com/blog/2013/12/16/reliable-android-http-testing-with-retrofit-and-mockito.html