使用 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);
}