java - 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);