如何验证调用了非模拟对象的方法?

2022-01-14 00:00:00 unit-testing mockito java

mockito 似乎只验证是否调用了模拟对象的方法，并且模拟对象总是有类似 doReturn().when(mock object)

It seems mockito only verifies whether a method of a mock object is called and the mock object always have something like doReturn().when(mock object)

但是我可以创建一个模拟对象并定义 doReturn().when(mock object)然后验证另一个对象的方法是否被调用?

But can I create a mock object and define doReturn().when(mock object) and then verify a method of another object is called?

这就是我想要做的:我定义了一个 mockEnvironment 并且无论发生什么都返回一个响应.但是接下来我想验证一下 anotherObj 的不同方法是在不同的情况下被调用的.

Here is what I want to do: I define a mockEnvironment and return a response no matter what happens. But then I want to verify different methods of anotherObj is called in different cases.

怎么做?

public class BaseClass {
    private Environment mockEnvironment;
    @Test
    public void testcase () {
     setMockitoEnvironment(); 
     response = foo(mockEnvironment, argument1);
     verify(anotherObj).codePath1(...);
     response = foo(mockEnvironment, argument2);
     verify(anotherObj).codePath2(...);
   }
}

//this method successfully return a response with any input 
//because I do not care how response is eventually generated, 
//I only care whether code path reaches createResponse() via 
//code path 1 or code path 2.
private void setMockitoEnvironment() {
    mockEnvironment = mock(Environment.class);
    doReturn (response).when(mockEnvironment).createResponse(for any input);
}
private Response foo(...) {
    ... 
    return createResponse(...);
}

推荐答案

你可以使用 Mockito 间谍 .如果您将 anotherObj 设置为间谍，您可以验证对该对象的方法调用.在您的示例中，您需要确保对 foo 的调用使用 spy 而不是 anotherObj 的普通实现.间谍是这样设置的:

You can use a Mockito Spy for this. If you setup anotherObj as a spy you can verify method calls on that object. In your example you need to make sure that the call to foo uses the spy instead of an ordinary implementation of anotherObj. The spy is setup something like this:

AnotherClass anotherObjSpy = Mockito.spy(new AnotherClass());
// do stuff -- e.g. anotherObjSpy.foo(...);
verify(anotherObjSpy).codePath1(...);

相关文章