Mockito：在整个控制流中注入模拟

Question

我仍在学习mockito，现在我正在学习如何注入mock。

我有一个正在测试的对象，其特定方法依赖于其他对象。这些对象又依赖于其他对象。我想模拟某些东西，并让这些模拟在执行过程中随处使用——整个方法的控制流。

例如，假设有这样的类：

public class GroceryStore {
    public double inventoryValue = 0.0;
    private shelf = new Shelf(5);
    public void takeInventory() {
        for(Item item : shelf) {
            inventoryValue += item.price();
        }
    }
}

public class Shelf extends ArrayList<Item> {
    private ProductManager manager = new ProductManager();
    public Shelf(int aisleNumber){
        super(manager.getShelfContents(aisleNumber);
    }
}

public class ProductManager {
    private Apple apple;
    public void setApple(Apple newApple) {
        apple = newApple;
    }
    public Collection<Item> getShelfContents(int aisleNumber) {
        return Arrays.asList(apple, apple, apple, apple, apple);
    }
}

我需要编写测试代码，其部分内容如下：

....
@Mock
private Apple apple;
... 
when(apple.price()).thenReturn(10.0);
... 

...
@InjectMocks
private GroceryStore store = new GroceryStore();
...
@Test
public void testTakeInventory() {
   store.takeInventory();
   assertEquals(50.0, store.inventoryValue);
}

每当调用 apple.price() 时，我希望使用我的模拟苹果。这可能吗？

编辑：
重要说明...
包含我想要模拟的对象的类确实有该对象的设置器。但是，在我正在测试的级别上，我确实无法处理该类。因此，按照这个示例，虽然 ProductManager 有一个适用于 Apple 的 setter，但我没有办法从 GroceryStore 对象获取 ProductManager。

Answer 1

问题是您通过调用 new 而不是注入来创建您所依赖的对象。将ProductManager注入Shelf（例如在构造函数中），并将Shelf注入GroceryStore。然后在测试中使用模拟。如果你想使用@InjectMocks，你必须通过setter方法注入。

通过构造函数，它可能看起来像这样：

public class GroceryStore {
  public double inventoryValue = 0.0;
  private shelf;

  public GroceryStore(Shelf shelf) {
    this.shelf = shelf;
  }

  public void takeInventory() {
    for(Item item : shelf) {
      inventoryValue += item.price();
    }
  }
}

public class Shelf extends ArrayList<Item> {
  private ProductManager manager;

  public Shelf(int aisleNumber, ProductManager manager) {
    super(manager.getShelfContents(aisleNumber);
    this.manager = manager;
  }
}

public class ProductManager {
  private Apple apple;
  public void setApple(Apple newApple) {
    apple = newApple;
  }
  public Collection<Item> getShelfContents(int aisleNumber) {
    return Arrays.asList(apple, apple, apple, apple, apple);
  }
}

然后你可以测试它模拟所有你依赖的对象：

@Mock
private Apple apple;
... 
when(apple.price()).thenReturn(10.0);

@InjectMocks
private ProductManager manager = new ProductManager();

private Shelf shelf = new Shelf(5, manager);
private GroceryStore store = new GroceryStore(shelf);

//Then you can test your store.