Question

我有一个带有void方法的简单类。

public class Simple {
     public Void doSomething() {}
}

我想在调用doSomething()之后在我的测试中增加一个数字。

@Test
public void test1() {
     int number = 0;
     Simple simple = Mockito.mock(Simple.class);
     Mockito.when(simple.doSomething()).then(number++);
}

当然它会导致编译错误。如何在致电number后增加doSomething()？

替代解决方案：

这是一种非常糟糕的做法，但它是下面最佳答案的替代解决方案。

private int number = 0;

@Test
public void test1() {
     Simple simple = Mockito.mock(Simple.class);
     Mockito.when(simple.doSomething()).thenReturn(increment());
}

private Void increment() {
     number++;
     return null;
}

Answer 1

从您的问题中不清楚为什么您会想要这样的事情。这个答案中的代码不是很好的做法，我不会推荐它。

第一个问题是您的方法返回void，因此您无法使用Mockito.when将其存根。 You need to use Mockito.doWhen

在此代码中，我使用Mockito.doAnswer在答案部分内编写自定义代码：

@Test
public void test1() {
     int[] number = { 0 };
     Simple simple = Mockito.mock(Simple.class);
     Mockito.doAnswer(new Answer<Void>() {
        @Override
        public Void answer(InvocationOnMock invocation) throws Throwable {
            number[0]++;
            return null;
        }
    }).when(simple).doSomething();
}

诀窍是将数字存储在最终变量（在本例中为数组）中，并在答案部分中增加其唯一值。每次调用doSomething()时，它都会递增，您将在number[0]内进行计数。

使用Java 8可以使相同的代码更紧凑：

Mockito.doAnswer(invocation -> number[0]++).when(simple).doSomething();

方法后的增量数称为Mockito

1 个答案: