使用 Mockito 是否可以在没有有效匹配器时在方法调用上抛出异常

Question

我有一个模拟的 class，我想在没有有效匹配器时抛出异常而不是返回null 。 mockito 这可能吗？ 这个想法是我已经模拟了使用某些参数的方法。 当这些都不匹配时，抛出异常而不是返回null 。

Answer 1

您可以编写自定义org.mockito.stubbing.Answer<T>并将其与thenAnswer一起使用：

private final static String EXCEPTION_MSG = "no valid matchers";
private final static String A = "A", B = "B";
private final static Map<String, String> argsReturnVal = new HashMap<>();

static {
    argsReturnVal.put(A, B);
}

private static final Answer<String> throwAnswer = a ->
        argsReturnVal.computeIfAbsent(
                a.getArgument(0),
                mf -> { throw new IllegalArgumentException(EXCEPTION_MSG); }
        );

@Mock
private Checker<String> checker;

@Before
public void init() {
    MockitoAnnotations.initMocks(this);
    Mockito.when(checker.check(Mockito.anyString())).thenAnswer(throwAnswer);
}

@Test
public void testThrow() {
    Assert.assertEquals(B, checker.check(A));
    try {
        checker.check("X-X-X");
    } catch (IllegalArgumentException ex) {
        Assert.assertEquals(EXCEPTION_MSG, ex.getMessage());
    }
}

private interface Checker<T> {
    String check(T in);
}