调用未存根方法时抛出RuntimeException

Question

我正在使用Mockito。我想在RuntimeException调用未存根方法时抛出一个。

有什么办法吗？

Answer 1

您可以为模拟设置默认答案。所有未存根的方法都将使用此默认答案。

public void testUnstubbedException() {
    // Create a mock with all methods throwing a RuntimeException by default
    SomeClass someClass = mock( SomeClass .class, new RuntimeExceptionAnswer() );

    doReturn(1).when(someClass).getId(); // Must use doReturn

    int id = someClass.getId(); // Will return 1

    someClass.unstubbedMethod(); // Will throw RuntimeException
}

public static class RuntimeExceptionAnswer implements Answer<Object> {

    public Object answer( InvocationOnMock invocation ) throws Throwable {
        throw new RuntimeException ( invocation.getMethod().getName() + " is not stubbed" );
    }

}

请注意，您 不能
使用when此功能，因为该方法是在方法调用之前when（嘲笑when（）调用如何工作的？），并且它将RuntimeException在模拟进入存根模式之前抛出a 。

因此，您必须使用doReturn它来工作。