Jak prawidłowo dopasować varargs w Mockito

Question

Próbowałem dostać się do mock metodę z użyciem parametrów vararg 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)); 

ten nie działa, więc jeśli mogę to zrobić w zamian:

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

To działa, pomimo tego, że całkowicie pominąłem argument varargs po upuszczeniu metody.

Jakieś wskazówki?

Answer 1

Mockito 1.8.1 wprowadzono anyVararg() matcher:

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

Zobacz także historię tego: https://code.google.com/archive/p/mockito/issues/62

Answer 2

nieco nieudokumentowanych funkcji: Jeśli chcesz rozwijać własny Matcher pasujący vararg argumentów musisz mieć implementuje on org.mockito.internal.matchers.VarargMatcher, aby działał poprawnie. Jest to pusty interfejs znacznika, bez którego Mockito nie będzie poprawnie porównywał argumentów podczas wywoływania metody z varargs przy użyciu Matchera.

Na przykład:

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

Opierając się na odpowiedź Eli Levine'a o to bardziej ogólne rozwiązanie:

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); 
    } 

} 

Następnie można używać go z tablicy hamcrest za dopasowujące tak:

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

(Oczywiście, że statyczny import uczyni to bardziej czytelnym.)

Answer 4

Opierając się na odpowiedź topchef, w

Dla 2.0.31 beta musiałem użyć Mockito.anyVararg zamiast Matchers.anyVararrg:

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

Answer 5

W moim przypadku podpis metody, które chcę uchwycić swój argument brzmi:

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

W tym przypadku należy przesyłać bajtów tablicy extenso:

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

używam wersji Mockito 1.10.19

Answer 6

I zostały przy użyciu kodu w odpowiedzi Petera Westmacott jest jednak z Mockito 2.2.15 można teraz wykonać następujące czynności:

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

gdzie arg1, arg2, arg3 są varargs.

Answer 7

Można również pętla nad argumentami:

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

na przykład sprawdzić swoje typy i oddać je odpowiednio, dodać do listy lub cokolwiek innego.