如何告诉Mockito Mock注释返回一个不为空的模拟对象？

Question

我正在尝试对一个方法进行单元测试，根据在其中创建的对象的值，该方法有不同的分支。下面的代码演示了它。

public class AClass {

public void method2() {
  //Some code goes here
}
public void method1(BClass bObject) {
    C_Class cObject = bObject.someMethodThatReturnsC();
    if(cObject != null) {
        method2();
        method2();
    }
}}

下面是TestClass：

public class AClassTest {
@InjectMocks
AClass AClassSpy;

@Mock
BClass b_objectMock;

@Mock
C_Class c_objectMock;

@BeforeMethod
public void beforeMethod() {
  AClassSpy = spy(new AClass());      
  MockitoAnnotations.initMocks(this);
}

public void method1Test () {
    doReturn(c_objectMock).when(b_objectMock).someMethodThatReturnsC());
    AClassSpy.method1(b_objectMock);
    verify(AClassSpy, times(2).method2();
}
}

但是，它总是失败，因为c_objectMock始终为空。我该怎么做才能告诉Mockito不要返回空对象？

Answer 1

您之所以出现这种行为，是因为您没有模仿对someMethodThatReturnsC的属性调用。

应该是：

doReturn(c_objectMock).when(b_objectMock).someMethodThatReturnsC();

Answer 2

它工作得很好，只需使用jUnit中的@前注解，而不是@Before方法，并将您的测试方法标记为@test并从
do的第二个括号中删除（c_objectMock）。当（b_objectMock）。

这应该行得通：

public class AClassTest {

  @InjectMocks
  private AClass AClassSpy;

  @Mock
  private BClass b_objectMock;

  @Mock
  private C_Class c_objectMock;

  @Before
  public void beforeMethod() {
      AClassSpy = spy(new AClass());
      MockitoAnnotations.initMocks(this);
  }

  @Test
  public void method1Test() {
      doReturn(c_objectMock).when(b_objectMock).someMethodThatReturnsC();
      AClassSpy.method1(b_objectMock);
      verify(AClassSpy, times(2)).method2();
  }

}

您可以使用注释@RunWith代替 before 方法。它看起来很清楚：

@RunWith(MockitoJUnitRunner.class)
public class AClassTest {

  @Spy
  @InjectMocks
  private AClass AClassSpy;

  @Mock
  private BClass b_objectMock;

  @Mock
  private C_Class c_objectMock;

  @Test
  public void method1Test() {
      doReturn(c_objectMock).when(b_objectMock).someMethodThatReturnsC();
      AClassSpy.method1(b_objectMock);
      verify(AClassSpy, times(2)).method2();
  }

}