Mockito - 注入模拟列表

Question

我有以下代码：

@Component 
public class Wrapper
{ 
    @Resource 
    private List<Strategy> strategies;

    public String getName(String id)
    {
    // the revelant part of this statement is that I would like to iterate over "strategies"
        return strategies.stream()
            .filter(strategy -> strategy.isApplicable(id))
            .findFirst().get().getAmount(id);
    } 
}

---

@Component 
public class StrategyA implements Strategy{...}

@Component 
public class StrategyB implements Strategy{...}

我想使用 Mockito 为其创建一个测试。 我写的测试如下：

@InjectMocks
private Wrapper testedObject = new Wrapper ();

// I was hoping that this list will contain both strategies: strategyA and strategyB
@Mock
private List<Strategy> strategies;

@Mock
StrategyA strategyA;

@Mock
StrategyB strategyB;

@Test
public void shouldReturnNameForGivenId()
{   // irrevelant code...
    //when
    testedObject.getName(ID);
}

我在线收到 NullPointerException：

filter(strategy -> strategy.isApplicable(id))

，它表示“策略”列表已初始化但为空。 Mohito 有没有办法像 Spring 一样表现得一样？将所有实现接口“策略”的实例自动添加到列表中？

顺便说一句，我在 Wrapper 类中没有任何设置器，如果可能的话，我想以这种方式保留它。

Answer 1

使用@Spy 而不是@Mock 对其进行注释。由于 Mockito 无法监视接口，因此请使用具体实现，例如 ArrayList。 在测试设置期间，将模拟添加到 List spy。 这样您就无需仅仅为了测试目的而更改您的测试对象。

@InjectMocks
private Wrapper testedObject = new Wrapper();

@Spy
private ArrayList<Strategy> mockedStrategies;

@Mock
private StrategyA strategyA;

@Mock
private StrategyB strategyB;

@Before
public void setup() throws Exception {
    mockedStrategies.add(strategyA);
    mockedStrategies.add(strategyB);
}

Answer 2

Mockito 无法知道您想将某些东西放入列表中策略。

你应该重新考虑这个并做这样的事情

@InjectMocks
private Wrapper testedObject = new Wrapper ();

private List<Strategy> mockedStrategies;

@Mock
StrategyA strategyA;

@Mock
StrategyB strategyB;

@Before
public void setup() throws Exception {
    mockedStrategies = Arrays.asList(strategyA, strategyB);
    wrapper.setStrategies(mockedStrategies);
}

Answer 3

你不应该模拟集合。

创建您需要的模拟并将它们放入列表中：

private List<Strategy> strategies; // not mocked!

@Mock
StrategyA strategyA;

@Mock
StrategyB strategyB;

@Before
public void setup(){
  strategies= Arrays.asList(strategyA,strategyB);
  testedObject.strategies= strategies;
}

@Test
public void shouldReturnNameForGivenId()
{   // irrevelant code...
    //when
    testedObject.getName(ID);
}

Answer 4

为什么不直接模拟您对toStream() 的调用？

@InjectMocks
private Wrapper testedObject = new Wrapper();

private List<Strategy> mockedStrategies;

@Mock
StrategyA strategyA;

@Mock
StrategyB strategyB;

@Before
public void setup() {
    when(strategies.stream()).thenReturn(Stream.of(strategyA, strategyB));
}

对我来说，这要优雅得多，因为它不需要您添加仅与测试代码相关的辅助方法。

Answer 5

Erwin Dupont 的解决方案很好，但是当您需要在测试对象的构造函数中注入模拟列表时不起作用。

这就是我解决这个问题的方法。我只展示了列表中一项的解决方案，但您可以通过将 switch(index) 放入 get() 方法将其扩展到 N 项：

class Wrapper {
  private final List<Strategy> strategies;
  Wrapper(List<Strategy> strategies) { this.strategies = new ArrayList<>(strategies); }
  // ...
}

class WrapperTest {
  @InjectMocks
  private Wrapper testedObject;

  @Spy
  private List<Strategy> mockedStrategies new AbstractList<Strategy>() {
      @Override public Trigger get(int index) { return trigger; } // can get away without bounds-checking
      @Override public int size() { return 1; }
  };

  @Mock
  private Strategy strategy;

  @Test
  public void testSomething() {
    assertThat(testedObject).isNotNull();
    assertThat(testedObject.getStrategies()).hasSize(1);
  }
}