Làm thế nào để khớp đúng varargs trong Mockito

Tôi đã cố gắng để giả định một phương thức với các tham số vararg bằng 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));

Điều này không hoạt động, tuy nhiên nếu tôi làm điều này thay vào đó:

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

Điều này hoạt động, mặc dù tôi đã hoàn toàn bỏ qua đối số varargs khi khai thác phương thức.

Bất kì manh mối nào?

Mockito 1.8.1 đã giới thiệu trình so khớp anyVararg () :

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

Đồng thời xem lịch sử cho việc này: https://code.google.com.vn/archive/p/mockito/issues/62

Chỉnh sửa cú pháp mới sau khi phản đối:

when(a.b(anyInt(), anyInt(), ArgumentMatchers.<String>any())).thenReturn(b);

Một tính năng hơi không có giấy tờ: Nếu bạn muốn phát triển Trình so khớp tùy chỉnh phù hợp với các đối số khác nhau, bạn cần phải thực hiện org.mockito.internal.matchers.VarargMatchernó để nó hoạt động chính xác. Đó là một giao diện đánh dấu trống, không có Mockito sẽ không so sánh chính xác các đối số khi gọi một phương thức với các varargs bằng Matcher của bạn.

Ví dụ:

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

Dựa trên câu trả lời của Eli Levine ở đây là một giải pháp chung chung hơn:

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

}

Sau đó, bạn có thể sử dụng nó với bộ so khớp mảng của hamcrest, do đó:

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

(Rõ ràng nhập tĩnh sẽ khiến điều này dễ đọc hơn.)

Tôi đã sử dụng mã trong câu trả lời của Peter Westmacott, tuy nhiên với Mockito 2.2.15 bây giờ bạn có thể làm như sau:

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

nơi arg1, arg2, arg3là varargs.

Dựa trên câu trả lời của topchef,

Đối với 2.0.31-beta, tôi đã phải sử dụng Mockito.anyVararg thay vì Matchers.anyVararrg:

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

Trong trường hợp của tôi, chữ ký của phương thức mà tôi muốn nắm bắt đối số của nó là:

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

Trong trường hợp này, bạn nên chuyển sang mảng byte một cách rõ ràng:

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

Tôi đang sử dụng phiên bản mockito 1.10.19

Bạn cũng có thể lặp qua các đối số:

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

ví dụ kiểm tra các loại của chúng và bỏ chúng một cách thích hợp, thêm vào danh sách hoặc bất cứ thứ gì.

Điều chỉnh câu trả lời từ @topchef,

Mockito.when(a.b(Mockito.anyInt(), Mockito.anyInt(), Mockito.any())).thenReturn(b);

Theo các tài liệu java cho Mockito 2.23.4, Mockito.any () "Phù hợp với mọi thứ, bao gồm null và varargs."

Bạn có thể thực hiện điều này bằng cách vượt qua một bản chụp ArgumentCaptor và sau đó truy xuất các varargs dưới dạng danh sách bằng cách sử dụng "getAllValues", xem: https://stackoverflow.com/a/55621731/11342928