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。然后在测试中使用mocks。如果你想使用@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.