Question

我想对一些代码加桩，以便当参数之一与特定值匹配时，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数组的一个特定元素。

Answer 1

您必须将方法存根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")
}

Answer 2

您可以创建自己的matcher：

public class MyVarargMatcher extends ArgumentMatcher<String[]> implements VarargMatcher
{
    private String expectedFirstValue;

    public MyVarargMatcher(String expectedFirstValue)
    {
        this.expectedFirstValue = expectedFirstValue;
    }

    @Override
    public boolean matches(Object varargArgument) {
        if (varargArgument instanceof String[])
        {
            String[] args = (String[]) varargArgument;
            return Stream.of(args).anyMatch(expectedFirstValue::equals);
        }
        return false;
    }
}

然后像这样使用它（Java代码）：

Foo foo = Mockito.mock(Foo.class);
Mockito.doReturn(true).when(foo).bar(Mockito.argThat(new MyVarargMatcher("AAA")));

根据操作员的评论进行编辑：只要“ AAA”是args之一，它就应该返回true

如何在Mockito中对varargs存根以仅匹配一个参数

2 个答案: