java – 如何在Mockito中存储varargs以仅匹配一个参数

我想存储一些代码,以便当其中一个参数与特定值匹配时,vararg方法返回true.例如,鉴于我无法更改的现有代码：

(在这里使用Kotlin,但我认为这适用于任何Java情况.)

class Foo {
    fun bar(vararg strings : String) : Boolean {
        // Iterates `strings` and returns true when one satisfies some criteria
    }
}

…我想编写与此类似的存根代码：

val foo = Foo()
whenever(foo.bar(eq("AAA"))).thenReturn(true)

当调用正好是foo.bar(“AAA”)时,这可以正常工作.

但是,有时候测试中的代码会调用foo.bar(“AAA”,“BBB”),在这种情况下,它会失败.

如何修改我的存根代码,以便在调用中传递任意数量的varargs时它可以工作？

编辑标记为可能的duplicate;在这种情况下,该场景考虑完全省略了呼叫中的varargs.在这里,我试图匹配varargs数组的一个特定元素.

解决方法

你必须将你的方法存根2次.首先是最不具体的存根：

val foo = Foo()
whenever(foo.bar(any())).thenReturn(false) // or whatever you want to return/throw here

然后是更具体的单一参数方法：

whenever(foo.bar(eq("AAA"))).thenReturn(true)

你的评论之后你也可以使用这样的东西(这次使用Java)：

when(foo.bar(any())).thenAnswer(invocation -> {
    for (Object argument : invocation.getArguments()) {
        if ("AAA".equals(argument)) return true;
    }
    return false;
});

在科特林也一样

whenever(foo.bar(any()).thenAnswer {
    it.arguments.contains("AAA")
}