Retrofit for Callback中的单元测试

import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import retrofit2.Call;
import retrofit2.Callback;
import retrofit2.Response;

import static org.mockito.Mockito.*;

public class RetrofitUnitTest {

    @Mock
    private ApiService mockApiService;

    @Mock
    private Call<ApiResponse> mockCall;

    @Before
    public void setup() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testApiRequest_Success() {
        // 模拟请求成功的响应结果
        ApiResponse successResponse = new ApiResponse(true, "Success");
        when(mockCall.execute()).thenReturn(Response.success(successResponse));
        when(mockApiService.apiRequest()).thenReturn(mockCall);

        // 创建一个Callback对象，并调用请求方法
        Callback<ApiResponse> callback = new Callback<ApiResponse>() {
            @Override
            public void onResponse(Call<ApiResponse> call, Response<ApiResponse> response) {
                // 在这里可以对响应结果进行断言
                assert response.body() != null;
                assert response.body().isSuccess();
            }

            @Override
            public void onFailure(Call<ApiResponse> call, Throwable t) {
                // 请求失败的处理
            }
        };

        // 执行网络请求
        mockApiService.apiRequest().enqueue(callback);

        // 验证网络请求方法是否被调用
        verify(mockApiService).apiRequest();
    }

    @Test
    public void testApiRequest_Failure() {
        // 模拟请求失败的情况
        when(mockCall.execute()).thenThrow(new IOException());
        when(mockApiService.apiRequest()).thenReturn(mockCall);

        // 创建一个Callback对象，并调用请求方法
        Callback<ApiResponse> callback = new Callback<ApiResponse>() {
            @Override
            public void onResponse(Call<ApiResponse> call, Response<ApiResponse> response) {
                // 请求成功的处理
            }

            @Override
            public void onFailure(Call<ApiResponse> call, Throwable t) {
                // 在这里可以对错误信息进行断言
                assert t instanceof IOException;
            }
        };

        // 执行网络请求
        mockApiService.apiRequest().enqueue(callback);

        // 验证网络请求方法是否被调用
        verify(mockApiService).apiRequest();
    }
}