在返回新对象时使用mockito

Question

我的实现中有以下代码行：

myService.getTypes(new TypeRequest(myList.stream().map(Object::toString)
                        .collect(Collectors.toList())))

我试着在我的测试课上模仿上述内容，如下所示：

when(myService.getTypes(new TypeRequest(Lists.newArrayList("15", "16"))))
                .thenReturn(Lists.newArrayList(type1, type2));
when(myService.getTypes(new TypeRequest(Lists.newArrayList("25", "26"))))
                .thenReturn(Lists.newArrayList(type3, type4));

但是当我调试测试代码时，我可以看到myService。getTypes正在返回一个空列表。有什么问题？

还有，有没有办法使用args来简化上述模拟？我发现的只是使用传递到方法中的参数，而不是传递到方法中的参数。

Answer 1

找到解决方案：

when(myService.getTypes(any(TypeRequest.class)))
    .thenAnswer(invocationOnMock -> {
        TypeRequest r = invocationOnMock.getArgument(0);
        //if r contains "15"
            return Lists.newArrayList(type1, type2);
        } else {
            return Lists.newArrayList(type3, type2);
        }
});