Mockito соответствует любому аргументу класса

Есть ли способ сопоставить любой аргумент класса в приведенном ниже примере подпрограммы?

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

Как я могу всегда возвращать new B () независимо от того, какой класс передан в метод ? Следующая попытка работает только для конкретного случая, когда совпадает A .

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

РЕДАКТИРОВАТЬ : Одно из решений -

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