Mockito thenAnswer가 예상대로 작동하지 않는 것 같습니다.

Question

AsyncHttpClient를 사용하는 다음 코드에 대한 단위 테스트를 작성하고 있습니다. FutureCallback의 구현에서 CountDownLatch를 사용하고 CountDownLatch를 감소 시키면 카운트 다운 래치가 감소하기를 기다리는 동안 내 JUnit 테스트 케이스가 정지하게됩니다. JUnit 테스트 내에서 ArgumentCaptor를 사용하여 FutureCallback을 캡처 한 다음 thenAnswer를 사용하여 완성 된 메서드를 호출하여 카운트 다운 래치를 감소시킵니다. 그러나 효과가없는 것 같아서 어떤 아이디어라도 도움이 될 것입니다.

@Test 
public void testHttpError() throws InterruptedException, ExecutionException, TimeoutException { 
    StatusLine status = Mockito.mock(StatusLine.class); 
    when(status.getStatusCode()).thenReturn(400); 

    HttpResponse response = Mockito.mock(HttpResponse.class); 
    when(response.getStatusLine()).thenReturn(status); 

    Future<HttpResponse> future = Mockito.mock(Future.class); 
    when(future.get(anyLong(), any())).thenReturn(response); 

    CloseableHttpAsyncClient httpClient = Mockito.mock(CloseableHttpAsyncClient.class); 
    ArgumentCaptor<HttpUriRequest> requestCaptor = ArgumentCaptor.forClass(HttpUriRequest.class); 
    ArgumentCaptor<FutureCallback<HttpResponse>> futureCallbackCaptor = ArgumentCaptor.forClass((Class)FutureCallback.class); 
    when(httpClient.execute(any(), any())).thenReturn(future).thenAnswer(new Answer() { 

     @Override 
     public Object answer(InvocationOnMock invocation) throws Throwable { 
      verify(httpClient).execute(requestCaptor.capture(), futureCallbackCaptor.capture()); 
      futureCallbackCaptor.getValue().completed(response); 
      return null; 
     } 
    }); 

    StubbedRcaClient rcaClient = new StubbedRcaClient(httpClient); 
    Query query = new Query("abc", "xyz", RcaHttpRequestType.GET, 1000); 
    List<QueryResponse> res = rcaClient.execute(query); 
    assertEquals(0, res.size()); 
    IOUtils.closeQuietly(rcaClient); 
} 

Answer 1

나는 다음과 같이 내 JUnit을 업데이트하여이 작품을 만든 :

when(httpClient.execute(any(), any())).thenAnswer(new Answer<Future<HttpResponse>>() { 

    @Override 
    public Future<HttpResponse> answer(InvocationOnMock invocation) throws Throwable { 
     verify(httpClient).execute(requestCaptor.capture(), futureCallbackCaptor.capture()); 
     futureCallbackCaptor.getValue().completed(response); 
     return future; 
    } 
});