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

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，而不是所有的映射。

你知道我是如何模仿提供者的吗。get（）。获取（Enum.ONE）以便它返回ObjectOne？

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

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

Answer 1

我通过使用mockito搜索模拟嵌套方法调用找到了答案。

@Mock
private ObjectOne mockObjectOne;

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

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

Answer 2

感觉你处理问题的方式不对。ObjectOne是您似乎想要存根的对象，而不是映射本身

定义一个构造函数

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