协慌网

登录 贡献 社区

如何使用 mockito 模拟 void 方法

如何用 void 返回类型模拟方法?

我实现了一个观察者模式,但我无法用 Mockito 模拟它,因为我不知道如何。

我试图在互联网上找到一个例子,但没有成功。

我的班级看起来像

public class World {

    List<Listener> listeners;

    void addListener(Listener item) {
        listeners.add(item);
    }

    void doAction(Action goal,Object obj) {
        setState("i received");
        goal.doAction(obj);
        setState("i finished");
    }

    private string state;
    //setter getter state
} 

public class WorldTest implements Listener {

    @Test public void word{
    World  w= mock(World.class);
    w.addListener(this);
    ...
    ...

    }
}

interface Listener {
    void doAction();
}

系统不会通过模拟触发。 =(我想显示上面提到的系统状态。并根据它们做出断言。

答案

看看 Mockito API 文档 。正如链接文档所提到的(第 12 点),您可以使用doThrow()任何doThrow()doAnswer()doNothing()doReturn()方法系列来模拟 void 方法。

例如,

Mockito.doThrow(new Exception()).when(instance).methodName();

或者如果你想将它与后续行为相结合,

Mockito.doThrow(new Exception()).doNothing().when(instance).methodName();

假设您正在考虑在下面的类 World 中模拟 setter setState(String s) ,代码使用doAnswer方法来模拟setState

World  mockWorld = mock(World.class); 
doAnswer(new Answer<Void>() {
    public Void answer(InvocationOnMock invocation) {
      Object[] args = invocation.getArguments();
      System.out.println("called with arguments: " + Arrays.toString(args));
      return null;
    }
}).when(mockWorld).setState(anyString());

我想我已经找到了一个更简单的答案,只为一个方法调用真正的方法(即使它有一个 void 返回)你可以这样做:

Mockito.doCallRealMethod().when(<objectInstance>).<method>();
<objectInstance>.<method>();

或者,您可以为该类的所有方法调用实际方法,执行以下操作:

<Object> <objectInstance> = mock(<Object>.class, Mockito.CALLS_REAL_METHODS);

添加 @sateesh 所说的,当你只想模拟一个 void 方法以防止测试调用它时,你可以这样使用Spy

World world = new World();
World spy = Mockito.spy(world);
Mockito.doNothing().when(spy).methodToMock();

如果要运行测试,请确保在spy对象上调用测试方法而不是world对象。例如:

assertEquals(0,spy.methodToTestThatShouldReturnZero());