Mockito phù hợp với bất kỳ đối số lớp

Có cách nào để khớp với bất kỳ đối số lớp nào của thói quen mẫu bên dưới không?

class A {
     public B method(Class<? extends A> a) {}
}

Làm thế nào tôi có thể luôn trả về new B()bất kể lớp nào được truyền vào method? Các nỗ lực sau đây chỉ hoạt động cho trường hợp cụ thể Ađược khớp.

A a = new A();
B b = new B();
when(a.method(eq(A.class))).thenReturn(b);

EDIT : Một giải pháp là

(Class<?>) any(Class.class)

Hai cách khác để làm điều đó (xem nhận xét của tôi về câu trả lời trước của @Tomasz Nurkiewicz):

Cái đầu tiên dựa trên thực tế là trình biên dịch đơn giản sẽ không cho phép bạn vượt qua thứ gì đó không đúng loại:

when(a.method(any(Class.class))).thenReturn(b);

Bạn mất gõ chính xác ( Class<? extends A>nhưng) nhưng nó có thể hoạt động như bạn cần.

Thứ hai là tham gia nhiều hơn nhưng được cho là một giải pháp tốt hơn nếu bạn thực sự muốn chắc chắn rằng đối số method()là một Ahoặc một lớp con của A:

when(a.method(Matchers.argThat(new ClassOrSubclassMatcher<A>(A.class)))).thenReturn(b);

Trường hợp ClassOrSubclassMatcherđược org.hamcrest.BaseMatcherđịnh nghĩa là:

public class ClassOrSubclassMatcher<T> extends BaseMatcher<Class<T>> {

    private final Class<T> targetClass;

    public ClassOrSubclassMatcher(Class<T> targetClass) {
        this.targetClass = targetClass;
    }

    @SuppressWarnings("unchecked")
    public boolean matches(Object obj) {
        if (obj != null) {
            if (obj instanceof Class) {
                return targetClass.isAssignableFrom((Class<T>) obj);
            }
        }
        return false;
    }

    public void describeTo(Description desc) {
        desc.appendText("Matches a class or subclass");
    }       
}

Phù! Tôi sẽ đi với tùy chọn đầu tiên cho đến khi bạn thực sự cần kiểm soát tốt hơn những gì method()thực sự trả về :-)

Có một cách khác để làm điều đó mà không cần diễn viên:

when(a.method(Matchers.<Class<A>>any())).thenReturn(b);

Giải pháp này buộc phương thức any()trả về Class<A>kiểu và không phải giá trị mặc định của nó ( Object).

Nếu bạn không biết gói nào bạn cần nhập:

import static org.mockito.ArgumentMatchers.any;
any(SomeClass.class)

HOẶC LÀ

import org.mockito.ArgumentMatchers;
ArgumentMatchers.any(SomeClass.class)

Làm thế nào về:

when(a.method(isA(A.class))).thenReturn(b);

hoặc là:

when(a.method((A)notNull())).thenReturn(b);

giải pháp từ millhouse không hoạt động nữa với phiên bản gần đây của mockito

Giải pháp này hoạt động với java 8 và mockito 2.2.9

nơi ArgumentMatcherlà một instanceoforg.mockito.ArgumentMatcher

public class ClassOrSubclassMatcher<T> implements ArgumentMatcher<Class<T>> {

   private final Class<T> targetClass;

    public ClassOrSubclassMatcher(Class<T> targetClass) {
        this.targetClass = targetClass;
    }

    @Override
    public boolean matches(Class<T> obj) {
        if (obj != null) {
            if (obj instanceof Class) {
                return targetClass.isAssignableFrom( obj);
            }
        }
        return false;
    }
}

Và việc sử dụng

when(a.method(ArgumentMatchers.argThat(new ClassOrSubclassMatcher<>(A.class)))).thenReturn(b);