dynamodb 我如何模拟映射器？

Question

我有一段 dynamoDb 相关代码，我想使用mockito 进行测试。

我要测试的方法包含以下行：

List<NotificationFeedRecord> listResult = mapper.query(NotificationFeedRecord.class, queryExpression);

当我手动测试时，它工作正常，我提交查询并从 dynamodb 获取预期结果。

我正在编写单元测试并想要模拟 mapper.query。

我有：

mapper = mock(DynamoDBMapper.class);
List<NotificationFeedRecord> testList = new ArrayList<>();
when(mapper.query(any(), any())).thenReturn(testList);

这里我得到一个错误

Error:(133, 37) java: no suitable method found for thenReturn(java.util.List<notificationfeed.lib.db.NotificationFeedRecord>)
      (argument mismatch; java.util.List<notificationfeed.lib.db.NotificationFeedRecord> cannot be converted to com.amazonaws.services.dynamodbv2.datamodeling.PaginatedQueryList<java.lang.Object>)

我已经尝试了一系列修复（例如，创建PaginatedQueryList 并返回它，更改查询匹配器），但都给出了错误。

.query方法声明如下：

public <T> PaginatedQueryList<T> query(Class<T> clazz, DynamoDBQueryExpression<T> queryExpression) {
        return query(clazz, queryExpression, config);
    }

如何模拟mapper.query？有什么特别的吗？

Answer 1

这真的很简单，我必须模拟 PaginatedQueryList 然后：

when(mapper.query(any(), anyObject())).thenReturn(paginatedQueryList)

;

这对我有用。

这些是我为测试设置的所有条件：

@Mock private PaginatedQueryList paginatedQueryList;
doReturn(mockQuery).when(utilSpy).getQueryExpression();

when(mockQuery.withFilterExpression(anyString())).thenReturn(mockQuery);
when(mockQuery.withLimit(anyInt())).thenReturn(mockQuery);
when(mockQuery.withExpressionAttributeValues(anyMap())).thenReturn(mockQuery);
when(mockQuery.withIndexName(anyString())).thenReturn(mockQuery);
when(mockQuery.withHashKeyValues(anyString())).thenReturn(mockQuery);
when(mockQuery.withConsistentRead(anyBoolean())).thenReturn(mockQuery); 
when(mockQuery.withRangeKeyCondition(anyString(), anyObject())).thenReturn(mockQuery);
when(mapper.query(any(), anyObject())).thenReturn(paginatedQueryList);