Mockito correspond à n'importe quel argument de classe

Existe-t-il un moyen de faire correspondre n'importe quel argument de classe de l'exemple de routine ci-dessous?

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

Comment puis-je toujours renvoyer un, new B()quelle que soit la classe à laquelle est passé method? La tentative suivante ne fonctionne que pour le cas spécifique où Acorrespond.

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

EDIT : Une solution est

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

Deux autres façons de le faire (voir mon commentaire sur la réponse précédente de @Tomasz Nurkiewicz):

Le premier repose sur le fait que le compilateur ne vous laissera tout simplement pas passer quelque chose du mauvais type:

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

Vous perdez la frappe exacte (le Class<? extends A>) mais cela fonctionne probablement comme vous en avez besoin.

Le second est beaucoup plus complexe mais est sans doute une meilleure solution si vous voulez vraiment être sûr que l'argument de method()est une Aou une sous-classe de A:

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

Où ClassOrSubclassMatcherest org.hamcrest.BaseMatcherdéfini comme:

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

Phew! J'irais avec la première option jusqu'à ce que vous ayez vraiment besoin d'un contrôle plus fin sur ce method()qui revient réellement :-)

Il existe une autre façon de faire cela sans lancer:

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

Cette solution force la méthode any()à renvoyer le Class<A>type et non sa valeur par défaut ( Object).

Si vous ne savez pas quel package vous devez importer:

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

OU

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

Que diriez-vous:

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

ou:

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

la solution de Millhouse ne fonctionne plus avec la version récente de mockito

Cette solution fonctionne avec java 8 et mockito 2.2.9

où ArgumentMatcherest une instance deorg.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;
    }
}

Et l'utilisation

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