如何模拟从特定方法返回的地图？

Question

基本上，我有以下几点：

public class Provider {
  public Map<Enum, Object> get() {
    return Map.of(Enum.ONE, ObjectOne,
                Enum.Two, ObjectTwo);
  }
}

Object value = provider.get().get(Enum.ONE);

我试着嘲笑这个

when(provider.get().get(any())).thenReturn(ObjectOne);

但我收到了这个错误

org.mockito.exceptions.misusing.WrongTypeOfReturnValue: 
Object cannot be returned by get()
get() should return Map

我不想模拟地图的返回，因为我只希望这个特定方法的地图返回ObjectOne，而不是一般的所有地图。

任何想法我如何模拟provider.get().get(Enum.ONE) 使其返回ObjectOne？

Object value = provider.get().get(Enum.ONE);

when(provider.get().get(any())).thenReturn(ObjectOne);

Answer 1

感觉你解决问题的方法有误。 ObjectOne 是您似乎想要存根的东西，而不是 Map 本身

所以，定义一个构造函数

public class Provider {

  private ObjectOne one;

  Provider(ObjectOne one) { 
    this.one = one;
  }

  public Map<Enum, Object> get() {
    return Map.of(Enum.ONE, one,
                Enum.Two, ObjectTwo);
  }
}

并使用一些类似的测试

ObjectOne oo = new ObjectOne();
Provider p = new Provider(oo);
assertSame(oo, p.get().get(Enum.ONE));

否则，您需要存根整个方法

Map<Enum, Object> map = new HashMap<>();
map.put(Enum.ONE, "foobar");
Provider mock = mock(Provider.class);
when(mock.get()).thenReturn(map);

// obviously the remaining test is pointless because calling mock.get() returns exactly what you already have

Answer 2

我通过搜索mock nested method calls using mockito找到了答案。

@Mock
private ObjectOne mockObjectOne;

@Mock(answer = Answers.RETURNS_DEEP_STUBS)
private Provider mockProvider;

when(mockProvider.get().get(any())).thenReturn(mockObjectOne);