方法后的增量数称为Mockito

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.你需要使用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();