Mockito pasuje do dowolnego argumentu klasy

Czy istnieje sposób dopasowania dowolnego argumentu klasy w poniższej przykładowej procedurze?

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

Jak zawsze mogę zwrócić, new B()niezależnie od tego, do której klasy jest przekazywana method? Poniższa próba działa tylko w konkretnym przypadku, w którym Ajest dopasowany.

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

EDYCJA : jednym rozwiązaniem jest

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

Jeszcze dwa sposoby (zobacz mój komentarz do poprzedniej odpowiedzi autorstwa @Tomasz Nurkiewicz):

Pierwsza polega na tym, że kompilator po prostu nie pozwoli Ci przekazać czegoś niewłaściwego typu:

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

Tracisz dokładne wpisywanie (the Class<? extends A>), ale prawdopodobnie działa tak, jak tego potrzebujesz.

Drugi jest o wiele bardziej skomplikowany, ale jest prawdopodobnie lepszym rozwiązaniem, jeśli naprawdę chcesz mieć pewność, że argument do method()jest Apodklasą lub podklasą A:

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

Gdzie ClassOrSubclassMatcherjest org.hamcrest.BaseMatcherzdefiniowany jako:

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

Uff! Wybrałbym pierwszą opcję, dopóki naprawdę nie będziesz potrzebować dokładniejszej kontroli nad tym, co method()faktycznie zwraca :-)

Jest na to inny sposób bez obsady:

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

To rozwiązanie wymusza na metodzie any()zwracanie Class<A>typu, a nie wartości domyślnej ( Object).

Jeśli nie masz pojęcia, który pakiet musisz zaimportować:

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

LUB

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

Co powiesz na:

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

lub:

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

rozwiązanie od millhouse nie działa już z najnowszą wersją mockito

To rozwiązanie działa z Javą 8 i Mockito 2.2.9

gdzie ArgumentMatcherjest instancjaorg.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;
    }
}

I użycie

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