每当我运行以下代码时,我都会收到错误:
Wanted but not invoked:
dynamoDBWriter.addItemsToDynamoTable(
<any>,
<any>
);
-> at DynamoDBWriterTest.testAllItemsAdded(DynamoDBWriterTest.java:32)
However, there were other interactions with this mock:
dynamoDBWriter.write(
[{ Item: {} }, { Item: {} }, { Item: {} }, { Item: {} }, { Item: {} }]
);
-> at DynamoDBWriterTest.testAllItemsAdded(DynamoDBWriterTest.java:31)
以下是我的代码:
public class DynamoDBWriterTest {
DynamoDBWriter dynamoDbWriter;
@Before
public void setup() {
dynamoDbWriter = mock(DynamoDBWriter.class);
}
@Test
public void testAllItemsAdded() {
List<Item> items = new ArrayList<>();
for (int index = 0; index < 5; index++) {
items.add(new Item());
}
dynamoDbWriter.write(items);
verify(dynamoDbWriter, times(5)).addItemsToDynamoTable(any(), any());
}
}
DynamoDBWriter的代码:
public void write(List<Item> items) {
// Initialize the rate limiter to allow capacity units / sec
// Since we know that the Item we are putting consumes 1 unit throughput.
RateLimiter rateLimiter = RateLimiter.create(1);
// Track how much throughput we consume on each put operation
for (Item item: items) {
// Let the rate limiter wait until our desired throughput "recharges"
rateLimiter.acquire();
addItemsToDynamoTable(table, item);
}
}
protected void addItemsToDynamoTable(Table table, Item item) {
try {
table.putItem(item);
} catch (RuntimeException e) {
logger.fatal("dynamoDB table.putItem threw exception for:" + tableName, e);
throw e;
}
}
谢谢你的帮助。我还添加了我使用的实际"纠正/工作"代码:
@Before
public void setup() {
dynamoDbWriter = spy(DynamoDBWriter....);
doNothing().when(dynamoDbWriter).addItemsToDynamoTable(any(), any());
}
// Method makes sure that irrespective of the throughput, all the items are added to dynamoDB
@Test
public void testAllItemsAdded() {
List<Item> items = new ArrayList<>();
for (int index = 0; index < 5; index++) {
items.add(new Item());
}
dynamoDbWriter.write(items);
verify(dynamoDbWriter, times(5)).addItemsToDynamoTable(any(), any());
}
当您创建mockDynamoDBWriter时,默认情况下,它会用不执行任何操作的存根覆盖所有方法。当您调用write
时,Mockito的替换实现允许验证write
是否被调用,但不允许write
实现进行的任何调用。从未调用过该实现。
当然,真正的问题是你在嘲笑你正在测试的类。即使有了技术上的变通方法,测试mocking框架也非常容易,而不是在测试中模拟您的类。一般来说,保留对被测试系统的合作者的嘲讽,而不是对被测试的系统本身的嘲讽。(另请参阅JB Nizet在这里的回答中的类比。)
也就是说,如果绝对必要,您可以使用真实实例的spy
,并根据需要选择性地重写方法,或者更危险的是,您可以在mock:上使用thenCallRealMethod
when(dynamoDbWriter.write(any())).thenCallRealMethod();
关于两者之间的区别,以及为什么thenCallRealMethod
如此危险,请参阅这个so问题。
您正在模拟DynamoDBWriter
,这意味着对它的调用并没有真正被调用(它们被调用,但其中的内部代码没有执行,它只是一个模拟)。所以你在这里得到的错误是因为只有write
被执行,而不是addItemsToDynamoTable
你在这里模拟了错误的对象——如果你想测试DynamoDBWriter
,你不应该模拟它,你应该模拟它使用的外围对象
在这种情况下,我会模拟表,验证table.put
是否执行了5次(从代码中不确定如何构造DynamoDBWriter
并将表传递给它,但我想你可以将模拟表传递给他)