验证是否使用mockito调用了三个方法之一

Question

我有这样的三个方法：

public void method1(String str){
    ...
}

public void method1(String str, String str2, String str3){
    ...
}

public void method1(String str, String str2, Object[] objs, String str3){
    ...
}

我想在 Mockito 中检查是否调用了这些方法中的任何一个，因此我尝试使用 anyVararg Matcher：

verify(foo).method1(anyVararg());

但这不会编译“类型中的方法 method1(String, String)”错误不适用于参数（对象）”

我有两个问题：

1. 我该如何解决这个问题？
2. 有没有办法检查是否调用了两个方法中的任何一个？想象一下我有另一种方法，称为方法 2 和方法 3。我想检查是否调用了其中任何一个（但至少有一个）。

谢谢。

Answer 1

您可以通过使用 Answer 在调用任何方法时增加计数器来实现此目的。

private Answer incrementCounter = new Answer() {
    public Object answer(InvocationOnMock invocation) throws Throwable {
        counter++;
        return null;
    }        
};

请注意，您需要存根所有方法。方法的唯一性基于其签名，而不仅仅是方法名称。两个同名的方法仍然是两个不同的方法。

doAnswer(incrementCounter).when(mockObj.method1(anyString()));
doAnswer(incrementCounter).when(mockObj.method1(anyString(), anyString()));
doAnswer(incrementCounter).when(mockObj.method2(anyString()));

请参阅 doAnswer 文档此处< /a>.

Answer 2

可变参数方法具有如下签名：

public void myMethod(String ... arguments){}

没有一个方法是可变参数方法。

我不了解 Mockito，所以我无法为您解决您的问题，但是除非您使用反射，否则不可能对上述所有三种方法进行抽象，所以我想您将不得不对上述每种方法使用单独的情况方法。

Answer 3

您可以拦截对对象的所有调用，如下答案所示：
Mockito - 拦截模拟上的任何方法调用
然后将其与@Bozho建议的方法结合起来：

private Answer incrementCounter = new Answer() {
    public Object answer(InvocationOnMock invocation) throws Throwable {
        if (invocation.getMethod().getName().eqauls("someMethid")) {
            counter++;
        }
        return null;
    }        
};

VendorObject mockObj = mock(SomeClass.class, incrementCounter);

注意：如果您为任何此方法提供其他拦截器 - 默认将不会被调用。