Mockito - 在单元测试中期望模拟时返回 null

Question

我在一个名为 FruitBasket 的类中有一个方法，我想这样测试它：

// Method to test
public Fruit getFruit(String fruitName) {
    Fruit fruit = new Fruit();

    if(fruitExists(fruitName)) {
        fruit = getFruitByName(fruitName);
    }
    else {
        fruit.setFruitName(fruitName);
        saveFruit(fruit);
        fruit = getFruitByName(fruitName);
    }

    return fruit;
}



private Fruit getFruitByName(String fruitName) {
    return fruitDao.getFruitByName(fruitName);
}

public boolean fruitExists(String fruitName) {
    return fruitDao.fruitExists(fruitName);
}

我为这个方法写了一个单元测试如下：

@Mock
FruitDao fruitDao;
@Mock
Fruit mockFruit;

@Before
public void setUp() {
    MockitoAnnotations.initMocks(this);
}

@Test
public void testgetFruitMethod() {
    FruitBasket fruitBasket = new FruitBasket (fruitDao);
    Fruit apple = fruitBasket.getFruit("Apple");

    when(fruitDao.fruitExists(anyString())).thenReturn(true);
    when(fruitDao.getFruitByName(anyString())).thenReturn(mockFruit);

    assertThat(apple, instanceOf(Fruit.class));
}

但是测试失败并出现断言错误。应该是 Fruit 的实例，但返回的是 null。

谁能发现我为什么会变成空的问题？

Answer 1

您对fruitBasket.getFruit("Apple"); 的调用会导致对FruitDao 的调用，该调用尚未配置，以便了解预期的调用以及如何响应它们。我想当 mockito 不知道该怎么做时，它的默认行为是只返回 null 而不是抛出异常，所以你的模拟 FruitDao 返回一个 null Fruit，你的测试就会失败。

因此，您需要先执行when()s，然后调用Fruit apple = fruitBasket.getFruit("Apple");

Answer 2

在使用前尝试准备模拟：

@Test
public void testgetFruitMethod() {
    // given
    when(fruitDao.fruitExists(anyString())).thenReturn(true);
    when(fruitDao.getFruitByName(anyString())).thenReturn(mockFruit);
    FruitBasket fruitBasket = new FruitBasket (fruitDao);

    // when
    Fruit apple = fruitBasket.getFruit("Apple"); 

    // then 
    assertThat(apple, instanceOf(Fruit.class));
}