Мокіто відповідає будь-якому аргументу класу

Чи є спосіб узгодити будь-який аргумент класу з наведеної нижче схеми зразка?

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

Як я завжди можу повернути new B()незалежно від того, у який клас передано method? Наступна спроба працює лише для конкретного випадку, де Aвона відповідає.

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

EDIT : Одне рішення

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

Ще два способи зробити це (дивіться мій коментар до попередньої відповіді від @Tomasz Nurkiewicz):

Перший покладається на те, що компілятор просто не дозволить вам передати щось неправильне:

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

Ви втрачаєте точне введення тексту ( Class<? extends A> ), але воно, ймовірно, працює так, як вам потрібно.

Друга - це набагато більше, але це, мабуть, краще рішення, якщо ви дійсно хочете бути впевненими, що аргумент method()- це Aабо підклас A:

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

Там , де ClassOrSubclassMatcherнаведено org.hamcrest.BaseMatcherвизначається як:

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

Фу! Я б пішов з першим варіантом, поки вам справді не потрібно отримати більш тонкий контроль над тим, що method()насправді повертається :-)

Є ще один спосіб зробити це без кастингу:

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

Це рішення змушує метод any()повернути Class<A>тип, а не його значення за замовчуванням ( Object).

Якщо ви не знаєте, який пакунок потрібно імпортувати:

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

АБО

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

Як щодо:

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

або:

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

рішення з millhouse більше не працює з останньою версією mockito

Це рішення працює з java 8 і mockito 2.2.9

де ArgumentMatcherє 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;
    }
}

І використання

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