单元测试DynamoDB PaginatedQueryList时如何返回列表

Question

我有一个查询语句，它返回PaginatedQueryList，并将其分配给一个列表。然后我在单子上操作。

List<MyClass> entries= mapper.query(MyClass.class, queryExpression);
//Logic to test
Map<String, Map> map = entries.stream()
                        .collect(Collectors.toMap(MyClass::getA, MyClass::getB));
....
....

when(mapper.query(eq(MyClass.class), any(DynamoDBQueryExpression.class)))
                              .thenReturn((PaginatedQueryList<MyClass>) list);

如何在列表执行mappper.query(..)时发送该列表。

其动机是在执行mapper.query(..)时将构造的列表发送到Entries。

注：

when(paginatedQueryList.stream()).thenReturn(Stream.of(list.get(0), list.get(1)));

Answer 1

Mockito对此有一些强大的工具。

List<MyClass> list = new ArrayList<>();
when(mapper.query(eq(MyClass.class), any(DynamoDBQueryExpression.class)))
    .thenReturn(mock(PaginatedQueryList.class, withSettings().defaultAnswer(new ForwardsInvocations(list))));

这将返回一个模拟的PaginatedQueryList，它将所有方法调用转发到List。