统计 Mockito 的间接方法调用

Question

我在使用 Mockito 计算方法调用时遇到问题。问题是，我想要计数的调用方法在测试类中被其他方法间接调用。这是代码：

public class ClassForTest {
    private Integer value;

    public void doSmth() {
        prepareValue("First call");
        prepareValue("Second call");
        prepareValue("Third call");
        System.out.println(value);
    }

    protected void prepareValue(String msg) {
        System.out.println("This is message: " + msg);
        value++;
    }
}

和测试类：

public class ClassForTestTest extends TestCase {
    @Test
    public void testDoSmth() {
        ClassForTest testMock = mock(ClassForTest.class);
        doNothing().when(testMock).prepareValue(anyString());
        testMock.doSmth();
        verify(testMock, times(3)).prepareValue(anyString());
    }
}

有这样的例外：

Wanted but not invoked:
classForTest.prepareValue(<any>);
-> at org.testing.ClassForTestTest.testDoSmth(ClassForTestTest.java:24)

However, there were other interactions with this mock:
-> at org.testing.ClassForTestTest.testDoSmth(ClassForTestTest.java:21)

请有任何想法。提前致谢！

Answer 1

这会起作用。使用spy调用底层方法。确保首先初始化value。

    @Test
    public void testDoSmth() {
        ClassForTest testMock = spy(new ClassForTest());
        testMock.doSmth();
        verify(testMock, times(3)).prepareValue(anyString());
    }

    public class ClassForTest {
        private Integer value = 0;

        public void doSmth() {
            prepareValue("First call");
            prepareValue("Second call");
            prepareValue("Third call");
            System.out.println(value);
        }

        protected void prepareValue(String msg) {
            System.out.println("This is message: " + msg);
            value++;
        }
    }

Answer 2

这表明您需要进行一些重构来改进您的设计。单个类应该是完全可测试的，而不需要模拟它的一部分。无论您认为需要模拟什么部分，都应该将其提取到一个或多个协作对象中。 不要陷入部分模拟的陷阱。听听测试告诉你什么。未来的你会感谢你。

Answer 3

你在嘲笑被测试的班级。模拟旨在针对已测试类的依赖项，而不是类本身。

我怀疑你想要的是Mockito.spy()。然而，这是 Mockito Javadoc 建议反对的部分模拟。

Answer 4

或者，如果您想重构可测试性，您可以执行以下操作：

@Test
public void testDoSmth() {
    Preparer preparer = mock(Preparer.class);
    ClassForTest cft = new ClassForTest(preparer);
    cft.doSmth();
    verify(preparer, times(3)).prepareValue(anyString());
}

public class ClassForTest {
    private final Preparer preparer;

    public ClassForTest(Preparer preparer) {
        this.preparer = preparer;
    }

    public void doSmth() {
        preparer.prepareValue("First call");
        preparer.prepareValue("Second call");
        preparer.prepareValue("Third call");
        System.out.println(preparer.getValue());
    }
}

public class Preparer {
    private Integer value = 0;

    public void prepareValue(String msg) {
        System.out.println("This is message: " + msg);
        value++;
    }

    public Integer getValue() {
        return value;
    }
}