Mockito 验证在 Callable 对象上没有失败

Question

我有以下代码用于对 Callable 参数执行操作的方法：

public static <T> T queryWithRetry(Callable<T> query, int maxTries, int retryIntervalInMilliseconds) throws MongoServiceException, InterruptedException {
    int tries = MAX_TRIES;
    while (tries-- > 0) {
      try {
        return query.call();
      } catch (TimeoutException e) {
        LOGGER.debug(String.format("Query timed out. Retrying attempt %d/%d", MAX_TRIES - tries, MAX_TRIES));
        Thread.sleep(RETRY_INTERVAL_IN_MILLISECONDS);
        continue;
    }
    throw new RandomException();
  }

我正在使用 Mockito 尝试验证 query.call() 行在抛出 RandomException 之前被尝试了 MAX_TRIES 的次数。我尝试使用以下测试代码来做到这一点：

public class CallableQueryTest {
  private static final int MAX_TRIES = 3;
  private static final int RETRY_INTERVAL_IN_MILLISECONDS = 100;

  @Mock
  private Callable<Document> mockCallable;

  @Rule
  public ExpectedException thrown = ExpectedException.none();

  @Before
  public void setUp() throws Exception {
    mockCallable = Mockito.mock(Callable.class);
  }

  @Test
  public void testQueryConfigThrowsRandomExceptionOnTimeout() throws Exception {
    Mockito.when(mockCallable.call()).thenThrow(new TimeoutException("timeout"));

    thrown.expect(RandomException.class);    
    Mockito.verify(mockCallable, Mockito.atMost(1)).call();
    MongoQueryUtils.queryWithRetry(mockCallable, MAX_TRIES, RETRY_INTERVAL_IN_MILLISECONDS);
  }
}

Mockito 代码成功测试了该方法抛出了 RandomException，但错误地表示测试通过了。

这个测试应该失败，因为我写道 Mockito 应该验证 mockCallable.call() 最多执行一次，但据我了解，它被称为 MAX_TRIES 次（设置为 3）。

有人可以解释这种行为并就如何正确测试mockCallable.call() 被调用的次数提供建议吗？

Answer 1

您的测试有很多问题。这是你想做的事情：

private static final int MAX_TRIES = 3;
private static final int RETRY_INTERVAL_IN_MILLISECONDS = 100;

@Mock
private Callable<String> mockCallable;

@Test
public void testQueryConfigThrowsRandomExceptionOnTimeout() throws Exception {
    when(mockCallable.call()).thenThrow(new IllegalArgumentException("timeout"));
    try {
        queryWithRetry(mockCallable, MAX_TRIES, RETRY_INTERVAL_IN_MILLISECONDS);
        fail("should have thrown");
    } catch (RuntimeException re) {
        // as expected
    }
    verify(mockCallable, Mockito.times(3)).call();
}

（请注意：我更改了测试的异常类型；但这应该很明显）。

所以，你错了：

• 我建议使用@RunWith(MockitoJUnitRunner.class)，然后使用@Mock 注释。您有@Mock 注释加上 一个设置方法，该方法为同一字段调用 Mockito.mock()。那是多余的。将注解与@RunWith 一起使用，或在您的设置方法中调用 MockitoAnnotations.initMocks()
• verify() 必须在您与模拟对象交互之后调用。想想那个有计数器的模拟。在您进行交互之前，计数器都是 0。因此，在您触发对模拟的调用之前询问模拟的计数器是没有意义的。
• 我不熟悉 JUnit 规则；因此我重写了测试以在没有它的情况下工作。如果要使用 JUnit 规则进行异常处理；留给读者作为练习。