小编典典

模拟第一次调用失败,第二次调用成功

all

我想使用 Mockito 来测试下面的(简化的)代码。我不知道如何告诉 Mockito 第一次失败,然后第二次成功。

for(int i = 1; i < 3; i++) {
  String ret = myMock.doTheCall();

  if("Success".equals(ret)) {
    log.write("success");
  } else if ( i < 3 ) {
    log.write("failed, but I'll try again. attempt: " + i);
  } else {
    throw new FailedThreeTimesException();
  }
}

我可以设置成功测试:

Mockito.when(myMock).doTheCall().thenReturn("Success");

和失败测试:

Mockito.when(myMock).doTheCall().thenReturn("you failed");

但是我如何测试它是否失败了一次(或两次)然后成功了,没关系?


阅读 70

收藏
2022-08-16

共1个答案

小编典典

文档

有时我们需要为同一个方法调用使用不同的返回值/异常存根。典型的用例可能是模拟迭代器。原始版本的 Mockito
没有这个功能来促进简单的模拟。例如,可以使用 Iterable
或简单的集合来代替迭代器。那些提供了自然的存根方式(例如使用真实的集合)。但是,在极少数情况下,存根连续调用可能会很有用:

when(mock.someMethod("some arg"))
   .thenThrow(new RuntimeException())
  .thenReturn("foo");

//First call: throws runtime exception:
mock.someMethod("some arg");

//Second call: prints "foo"
System.out.println(mock.someMethod("some arg"));

所以在你的情况下,你会想要:

when(myMock.doTheCall())
   .thenReturn("You failed")
   .thenReturn("Success");
2022-08-16