Wie man richtig varargs in Mockito

Question

übereinstimmen ich versucht habe zu bekommen, ein Verfahren mit Vararg Parametern zu verspotten mit Mockito:

interface A { 
    B b(int x, int y, C... c); 
} 

A a = mock(A.class); 
B b = mock(B.class); 

when(a.b(anyInt(), anyInt(), any(C[].class))).thenReturn(b); 
assertEquals(b, a.b(1, 2)); 

nicht funktioniert Dies gilt jedoch, wenn ich dies tun, statt:

when(a.b(anyInt(), anyInt())).thenReturn(b); 
assertEquals(b, a.b(1, 2)); 

Dies funktioniert, obwohl ich das varargs Argument beim Stubben der Methode vollständig weggelassen haben.

Irgendwelche Hinweise?

Answer 1

Mockito 1.8.1 anyVararg() matcher eingeführt:

when(a.b(anyInt(), anyInt(), Matchers.<String>anyVararg())).thenReturn(b); 

auch Geschichte für diese sehen: https://code.google.com/archive/p/mockito/issues/62

Answer 2

Eine etwas nicht dokumentierte Funktion: Wenn Sie einen benutzerdefinierten Matcher entwickeln wollen, die Vararg Argumente, die Sie haben müssen passt zu es implementieren org.mockito.internal.matchers.VarargMatcher, damit es richtig funktioniert. Es ist ein leeres Marker-Interface, ohne das Mockito Argumente beim Aufruf einer Methode mit varargs nicht korrekt vergleicht.

Zum Beispiel:

class MyVarargMatcher extends ArgumentMatcher<C[]> implements VarargMatcher { 
    @Override public boolean matches(Object varargArgument) { 
     return /* does it match? */ true; 
    } 
} 

when(a.b(anyInt(), anyInt(), argThat(new MyVarargMatcher()))).thenReturn(b); 

Answer 3

Aufbauend auf Eli Levine Antwort hier ist eine generische Lösung:

import org.hamcrest.Description; 
import org.hamcrest.Matcher; 
import org.mockito.ArgumentMatcher; 
import org.mockito.internal.matchers.VarargMatcher; 

import static org.mockito.Matchers.argThat; 

public class VarArgMatcher<T> extends ArgumentMatcher<T[]> implements VarargMatcher { 

    public static <T> T[] varArgThat(Matcher<T[]> hamcrestMatcher) { 
     argThat(new VarArgMatcher(hamcrestMatcher)); 
     return null; 
    } 

    private final Matcher<T[]> hamcrestMatcher; 

    private VarArgMatcher(Matcher<T[]> hamcrestMatcher) { 
     this.hamcrestMatcher = hamcrestMatcher; 
    } 

    @Override 
    public boolean matches(Object o) { 
     return hamcrestMatcher.matches(o); 
    } 

    @Override 
    public void describeTo(Description description) { 
     description.appendText("has varargs: ").appendDescriptionOf(hamcrestMatcher); 
    } 

} 

Dann können Sie es mit hamcrest des Arrays verwenden Matcher so:

verify(a).b(VarArgMatcher.varArgThat(
      org.hamcrest.collection.IsArrayContaining.hasItemInArray("Test"))); 

(Offensichtlich werden statische Importe dies lesbarer machen.)

Answer 4

Aufbauend auf TopChef Antwort,

Für 2.0.31-beta I Mockito.anyVararg statt Matchers.anyVararrg verwenden musste:

when(a.b(anyInt(), anyInt(), Mockito.<String>anyVararg())).thenReturn(b); 

Answer 5

die Signatur der Methode In meinem Fall, den ich will ihr Argument erfassen ist:

public byte[] write(byte ... data) throws IOException; 

In diesem Fall Sie explizit Byte-Array werfen sollte:

 when(spi.write((byte[])anyVararg())).thenReturn(someValue); 

Ich verwende Mockito Version 1.10.19

Answer 6

Ich habe in Peter Westmacott Antwort jedoch mit Mockito 2.2.15 den Code wurde verwenden, können Sie jetzt wie folgt vorgehen:

verify(a).method(100L, arg1, arg2, arg3)

wo arg1, arg2, arg3 sind Varargs.

Answer 7

Sie können auch eine Schleife über die Argumente:

Object[] args = invocation.getArguments(); 
for(int argNo = 0; argNo < args.length; ++argNo) { 
    // ... do something with args[argNo] 
} 

zum Beispiel ihre Typen überprüfen und sie in geeigneter Weise werfen, fügen Sie zu einer Liste oder was auch immer.