java - 如何模拟DynamoDB批量保存？

Question

我编写了存储项目列表的 dynamoDB 代码。

mapper.batchSave(trafficSensorReadings)

这将返回。

List<FailedBatch>

我想模拟mapper.batchSave，然后返回一个失败的作业。我怎样才能实现它？我正在使用mockito和Junit。

我写过这样的东西。但没啥用。

        when(dynamoDBMapper.batchSave(eq(List.class))).thenReturn(mock(List.class));

Answer 1

完整示例如下

@Test
public void test() {
    FailedBatch failedBatch = mock(FailedBatch.class);
    List<FailedBatch> failedBatchList = new ArrayList<>();
    failedBatchList.add(failedBatch);
    DynamoDBMapper dynamoDBMapperMock = mock(DynamoDBMapper.class);
    when(dynamoDBMapperMock.batchSave(any(List.class))).thenReturn(failedBatchList);

    tested.testedMethodCall();

    verify(dynamoDBMapperMock).batchSave(any(List.class));
}