在同一个模拟上使用多个 ArgumentMatchers

Question

我正在尝试在 Mock 上使用 Mockito 来做到这一点：
当使用参数 1 调用 Mock.someMethod(..) 时 --> 返回结果 1
当使用 argument2 调用 Mock.someMethod(..) 时 --> return result2
当使用 argument3 调用 Mock.someMethod(..) 时 --> return result3

    when(mock.method(Matchers.argThat(new MyMatcher1() {

        @Override
        public boolean matches(Object arg0) {
                   // comparision logic
        }
    }))).thenReturn(result1);

    when(mock.method(Matchers.argThat(new MyMatcher2() {

        @Override
        public boolean matches(Object arg0) {
                   // comparision logic
        }
    }))).thenReturn(result2);

    when(mock.method(Matchers.argThat(new MyMatcher3() {

        @Override
        public boolean matches(Object arg0) {
                   // comparision logic
        }
    }))).thenReturn(result3);

但是 Mockito 正确地存根第一个，但是在第二个上它会抛出 NullPointer 异常，因为它出于某种原因尝试使用 null agrument 运行 Matcher。我不确定它是否受支持。

如果这不是正确的方法，如何使用 Mockito 来实现？谢谢。

Answer 1

而不是写

when(mock.method(Matchers.argThat(new MyMatcher1() {
    @Override
    public boolean matches(Object arg0) {
           // comparison logic
    }
}))).thenReturn(result1);

试试这个。

doReturn(result1).when(mock).method(Matchers.argThat(new MyMatcher1() {

    @Override
    public boolean matches(Object arg0) {
               // comparison logic
    }
}));

result2 和 result3 也是如此。

http://docs.mockito.googlecode.com/hg/latest/org/mockito/Mockito.html#12 对此进行了描述，但在我看来，文档并不清楚在这种情况下实际上需要这种构造。我将在这里与 Mockito 团队的其他成员讨论如何改进文档。

祝你好运。

Answer 2

根据经验，如果测试失败，您应该能够查明被测单元的问题所在。避免专门为一项测试编写自定义匹配器。 如果一个方法需要返回多个值，通常只需通过按测试预测的顺序返回值来简单地对方法进行存根即可。

例如

when(mock.method(any(Object.class))).thenReturn(result1, result2, result3);

这将在第一次调用时返回结果 1，在第二次调用时返回结果 2，依此类推。

这显然是不够的，但通常情况下，测试越简单越好。

Answer 3

我现在可以通过在自定义 ArgumentMatcher 中进行空检查来解决这个问题。它起作用了，因为 NPE 仅在 Mockito 调用 when() 语句时才在启动期间。此时它甚至不应该调用 ArgumentMatcher.matches() ！感觉就像 Mockito 中的一个错误。