Sử dụng Mockito với nhiều cuộc gọi đến cùng một phương thức với cùng một đối số

Question

Có cách nào để có phương thức gốc được trả về các đối tượng khác nhau trên lời gọi tiếp theo không? Tôi muốn làm điều này để kiểm tra các câu trả lời không xác định từ một số ExecutorCompletionService. tức là để kiểm tra không phân biệt thứ tự trả về của các phương thức, kết quả vẫn không thay đổi.

Mã tôi đang tìm kiếm để kiểm tra trông giống như thế này.

// Create an completion service so we can group these tasks together 
ExecutorCompletionService<T> completionService = 
     new ExecutorCompletionService<T>(service); 

// Add all these tasks to the completion service 
for (Callable<T> t : ts) 
    completionService.submit(request); 

// As an when each call finished, add it to the response set. 
for (int i = 0; i < calls.size(); i ++) { 
    try { 
     T t = completionService.take().get(); 
     // do some stuff that I want to test 
    } catch (...) { }   
} 

Answer 1

Bạn có thể làm điều đó bằng cách sử dụng thenAnswer phương pháp (khi chaining với when):

when(someMock.someMethod()).thenAnswer(new Answer() { 
    private int count = 0; 

    public Object answer(InvocationOnMock invocation) { 
     if (count++ == 1) 
      return 1; 

     return 2; 
    } 
}); 

Hoặc sử dụng tương đương, tĩnh doAnswer phương pháp:

doAnswer(new Answer() { 
    private int count = 0; 

    public Object answer(InvocationOnMock invocation) { 
     if (count++ == 1) 
      return 1; 

     return 2; 
    } 
}).when(someMock).someMethod(); 

Answer 2

Làm thế nào về

when(method-call).thenReturn(value1, value2, value3); 

Bạn có thể đặt bao nhiêu đối số theo ý muốn trong các dấu ngoặc vuông rồiQuay lại, miễn là chúng đều là loại chính xác. Giá trị đầu tiên sẽ được trả về lần đầu tiên phương thức được gọi, sau đó là câu trả lời thứ hai, v.v. Giá trị cuối cùng sẽ được trả về nhiều lần khi tất cả các giá trị khác được sử dụng hết.

Answer 3

Bạn thậm chí có chuỗi doReturn() phương pháp invocations như thế này

doReturn(null).doReturn(anotherInstance).when(mock).method(); 

dễ thương phải không :)

Answer 4

Như previously pointed out gần như tất cả các cuộc gọi được thể kết nối.

Vì vậy, bạn có thể gọi

when(mock.method()).thenReturn(foo).thenReturn(bar).thenThrow(new Exception("test")); 

//OR if you're mocking a void method and/or using spy instead of mock 

doReturn(foo).doReturn(bar).doThrow(new Exception("Test").when(mock).method(); 

Thông tin thêm trong Mockito's Documenation.

Answer 5

Sau đây có thể được sử dụng như một phương pháp phổ biến để trả về các đối số khác nhau về các cuộc gọi phương thức khác nhau. Điều duy nhất chúng ta cần làm là chúng ta cần truyền một mảng với thứ tự các đối tượng cần được lấy ra trong mỗi cuộc gọi.

@SafeVarargs 
public static <Mock> Answer<Mock> getAnswerForSubsequentCalls(final Mock... mockArr) { 
    return new Answer<Mock>() { 
     private int count=0, size=mockArr.length; 
     public Mock answer(InvocationOnMock invocation) throws throwable { 
      Mock mock = null; 
      for(; count<size && mock==null; count++){ 
       mock = mockArr[count]; 
      } 

      return mock;  
     } 
    } 
} 

Ví dụ: getAnswerForSubsequentCalls(mock1, mock3, mock2); sẽ trả về đối tượng mock1 trên cuộc gọi đầu tiên, đối tượng mock3 trên cuộc gọi thứ hai và đối tượng mock2 trên cuộc gọi thứ ba. nên được sử dụng như when(something()).doAnswer(getAnswerForSubsequentCalls(mock1, mock3, mock2)); này là gần như tương tự như when(something()).thenReturn(mock1, mock3, mock2);

Answer 6

Tôi đã thực hiện một lớp MultipleAnswer giúp tôi để còn sơ khai câu trả lời khác nhau trong mỗi cuộc gọi. Đây là đoạn mã:

private final class MultipleAnswer<T> implements Answer<T> { 

    private final ArrayList<Answer<T>> mAnswers; 

    MultipleAnswer(Answer<T>... answer) { 
     mAnswers = new ArrayList<>(); 
     mAnswers.addAll(Arrays.asList(answer)); 
    } 

    @Override 
    public T answer(InvocationOnMock invocation) throws Throwable { 
     return mAnswers.remove(0).answer(invocation); 
    } 
}