使用Mockito多次调用具有相同参数的相同方法

是否有一种方法可以让存根方法在后续调用时返回不同的对象?我希望这样做是为了测试来自ExecutorCompletionService的不确定响应。也就是说，测试不管方法的返回顺序如何，结果都保持不变。

我要测试的代码看起来像这样。

// 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 (...) { }        
}

当前回答

这可能是基本的/明显的，但如果像我一样，你试图模拟一个方法的多次调用，每次调用要测试的方法被调用未知次数，例如:

public String method(String testArg) {
    //...
    while(condition) {
        someValue = someBean.nestedMethod(); // This is called unknown number of times
        //...
    }
    //...
}

你可以这样做:

@Test
public void testMethod() {
    mockNestedMethodForValue("value1");
    assertEquals(method("arg"), "expected1");
    mockNestedMethodForValue("value2");
    assertEquals(method("arg"), "expected2");
    mockNestedMethodForValue("value3");
    assertEquals(method("arg"), "expected3");
}

private void mockNestedMethodForValue(String value) {
    doReturn(value).when(someBeanMock).nestedMethod();
}

2021-12-07 08:24:20

其他回答

Following可以用作一个通用方法，在不同的方法调用上返回不同的参数。我们需要做的唯一一件事是，我们需要传递一个数组，其中对象应该在每次调用中检索。

@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;    
       } 
    }
}

示例:getAnswerForSubsequentCalls(mock1, mock3, mock2);在第一次调用时返回mock1对象，第二次调用时返回mock3对象，第三次调用时返回mock2对象。应该像(something())那样使用。doAnswer(getAnswerForSubsequentCalls(mock1, mock3, mock2)); 这几乎类似于when(something())。然后返回(mock1, mock3, mock2);

2016-06-23 08:06:21

你可以使用LinkedList和Answer。如

MyService mock = mock(MyService.class);
LinkedList<String> results = new LinkedList<>(List.of("A", "B", "C"));
when(mock.doSomething(any())).thenAnswer(invocation -> results.removeFirst());

2021-02-12 16:47:50

如何

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

你可以在thenReturn的括号中放入任意数量的参数，前提是它们都是正确的类型。第一个值将在第一次调用该方法时返回，然后是第二个答案，依此类推。当所有其他值用完后，将重复返回最后一个值。

2011-12-06 05:21:54

doReturn(value1, value2, value3)。When(方法调用)

2020-02-09 17:35:22

几乎所有的调用都是可链的:

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

2015-05-13 15:13:17

使用Mockito多次调用具有相同参数的相同方法

推荐文章

最新文章

标签