java - 如何模拟 Map 的返回值？

Question

我的代码看起来像这样：

for (Map.Entry<Integer, Action> entry : availableActions.entrySet()) {
  ...
}

我试图像这样模拟它：

Map mockAvailableActions = mock(Map.class, Mockito.RETURNS_DEEP_STUBS);
mockAvailableActions.put(new Integer(1), mockAction);

我认为这就足够了。但是 entrySet 是空的。所以我添加了这个：

when(mockAvailableActions.entrySet().iterator()).thenReturn(mockIterator);
when(mockIterator.next()).thenReturn(mockAction);

仍然 entrySet 是空的。我究竟做错了什么？感谢您的任何意见！

Answer 1

也许我错过了一些东西，但为什么不这样做：

Map.Entry<Integer, Action> entrySet = <whatever you want it to return>
Map mockAvailableActions = mock(Map.class);
when(mockAvailableActions.entrySet()).thenReturn(entrySet);

还要考虑你是否真的需要一个模拟地图，不是一个真正的地图吗？模拟通常用于替换您不想参与单元测试的其他代码片段，Map 是核心 Java 语言的一部分，通常不会被模拟出来。