Mockito：当任何实例调用方法时，都会返回一些内容

Question

我对Mockito很陌生，有一门课

public class A {
   private B b;
   public B getB() { return b; }
   // ...
}

和

public class B {
   private C c;
   public C getC() { return c; }
   // ...
}

我正在使用Mockito为A类编写单元测试，并试图实现以下行为，其中B是一个模拟：

when(a.getB().getC()).thenReturn(mock(C.class))
when(any(mock(C.class)).doSomething()).thenReturn("somethingElse") // This doesn't work.

当任何模拟的C实例调用doSomething（）时，如何返回“somethingElse”？我不想将此行为绑定到任何一个特定的mock（C.class）实例。

Answer 1

不要添加用于创建新模拟的mock（）。

就像这样

when(any(C.class).doSomething()).thenReturn("somethingElse");

Answer 2

首先，你必须把你的测试对象作为一个单元来对待，并模仿其他一切。因此，既然您正在测试类A，就不能模拟As方法。

下面是如何使用Mockito使用注释来模拟B的实例：

@MockitoSettings
public class ATest {
  @Mock
  private B b;
  @InjectMocks
  private A a;

  @Test
  void testExample() {
    C c = mock(C.class):
    when(b.getC()).thenReturn(c);
    when(c.doSomething()).thenReturn("somethingElse");
    
    // a.getB() == b
  }
}

那么这里发生了什么？

注释@Mock初始化类型为B的模拟对象，并将字段变量B设置为该对象。

@InjectMocks正在收集所有@Mock字段并设置对象中的字段。这是你的测试对象。

现在，您的A实例有一个模拟的B对象。在B中，你有C。因为B是一个模拟对象，所以方法getC（）根本不做任何事情。你必须用一些功能来存根。在测试中，我初始化了一个本地模拟的C实例，从getC（）返回。

所以现在当A对象访问模拟的B并获取C时，它将获取这个局部变量C。所以你可以用模拟函数来存根这个对象。