Mockito passt zu jedem Klassenargument

Gibt es eine Möglichkeit, mit einem Klassenargument der folgenden Beispielroutine übereinzustimmen?

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

Wie kann ich immer eine zurückgeben, new B()unabhängig davon, an welche Klasse übergeben wird method? Der folgende Versuch funktioniert nur für den speziellen Fall, in dem Aeine Übereinstimmung vorliegt.

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

EDIT : Eine Lösung ist

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

Zwei weitere Möglichkeiten (siehe meinen Kommentar zur vorherigen Antwort von @Tomasz Nurkiewicz):

Der erste beruht auf der Tatsache, dass der Compiler Sie einfach nicht zulässt, dass etwas vom falschen Typ übergeben wird:

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

Sie verlieren die genaue Eingabe (die Class<? extends A>), aber es funktioniert wahrscheinlich so, wie Sie es brauchen.

Die zweite ist viel komplizierter, aber wahrscheinlich eine bessere Lösung, wenn Sie wirklich sicher sein möchten, dass das Argument method()eine Aoder eine Unterklasse von A:

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

Wo ClassOrSubclassMatcherist ein org.hamcrest.BaseMatcherdefiniert als:

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

Puh! Ich würde mit der ersten Option gehen, bis Sie wirklich eine genauere Kontrolle darüber bekommen müssen, was method()tatsächlich zurückkommt :-)

Es gibt einen anderen Weg, dies ohne Besetzung zu tun:

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

Diese Lösung zwingt die Methode any(), den Class<A>Typ und nicht den Standardwert ( Object) zurückzugeben.

Wenn Sie keine Ahnung haben, welches Paket Sie importieren müssen:

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

ODER

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

Wie wäre es mit:

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

oder:

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

Die Lösung von Millhouse funktioniert mit der neuesten Version von Mockito nicht mehr

Diese Lösung funktioniert mit Java 8 und Mockito 2.2.9

wo ArgumentMatcherist eine Instanz vonorg.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;
    }
}

Und die Verwendung

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