Mockito correspond à n'importe quel argument de classe

Question

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 retourner un new B() quelle que soit la classe passée dans method ? La tentative suivante ne fonctionne que pour le cas spécifique où A est apparié.

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

---

EDIT : Une solution est

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

Answer 1

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

La première repose sur le fait que le compilateur ne vous laissera tout simplement pas passer quelque chose de mauvais type :

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

Vous perdez le typage exact (le Class<? extends A> ) mais cela fonctionne probablement comme vous le souhaitez.

La seconde est beaucoup plus complexe, mais c'est sans doute la meilleure solution si vous souhaitez vraiment veulent être sûrs que l'argument de method() est un A ou une sous-classe de A :

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

Où ClassOrSubclassMatcher est un org.hamcrest.BaseMatcher défini comme suit :

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

Ouf ! J'opterais pour la première option jusqu'à ce que vous vraiment besoin d'avoir un contrôle plus fin sur ce que method() retourne effectivement :-)

Answer 2

Il y a un autre moyen de le faire sans cast :

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

Cette solution force la méthode any() pour revenir Class<A> et non sa valeur par défaut ( Object ).

Answer 3

Si vous n'avez aucune idée du paquet que vous devez importer :

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

OU

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

Answer 4

Pourquoi pas :

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

ou :

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

Answer 5

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.

donde ArgumentMatcher est une instance de org.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);