java - 在带有 Mockito 的 JUnit 中,如何等待异步方法完成?
问题描述
我正在针对一些代码编写集成测试,这些代码在数据库中更新不同的值时异步创建记录。我想在创建记录后检查系统的状态,验证它是否按预期创建。因此,测试需要等到创建记录。
我可以使用 Mockito 为创建记录的函数创建一个间谍。Mockito 甚至可以选择等待通过调用该方法Mockito.timeout
,如果经过一定时间而没有调用该方法,则放弃:
// Use or create/wire in spy. In my case, this is set up with @SpyBean from spring-boot-test.
RecordCreationService recordCreationServiceSpy = ...;
testClass.update(someValue);
Mockito.verify(recordCreationServiceSpy, Mockito.timeout(10_000)).createRecord(ArgumentMatchers.any());
但是,这只是等待调用开始,而不是等待调用完成。因此,这进入了一个竞争条件,其中验证可以在所需的调用完成之前完成。
在使用 Mockito 在 JUnit 中进行验证之前,如何干净利落地等待一个过程完成?
解决方案
此功能在 Mockito 中并不直接存在,因为目前有一个未解决的问题可以将此功能添加到 Mockito(Mockito 问题 #1089)。
我目前使用的解决方案是为 spied 方法编写一个自定义答案,等待调用完成后再返回。然后我之后正常验证结果。
@SpyBean
private RecordCreationService recordCreationServiceSpy;
@Test(timeout = 10_000)
public void recordShouldBeCreatedWhenDataIsUpdated() {
// Set up test here
updateValueAndWait(value);
assertEquals(1, recordRepository.findAll().size());
// Perform any additional verifications
}
private void updateValueAndWait(String value) {
CountDownLatch latch = new CountDownLatch(1);
Mockito.doAnswer(invocation -> {
Object result = invocation.callRealMethod();
latch.countDown();
return result;
}).when(recordCreationServiceSpy).insertRecord(any());
testClass.update(value);
try {
latch.await();
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
}
推荐阅读
- r - 在 CentOs Linux 上安装 RStudio 服务器后,无法安装包非零退出状态
- vb.net - 在 VB.net 中使用 NDDE 如何正确捕获错误?
- python-3.x - 选择列表列表的子列表以定义关系
- webpack - Angular CLI 6 自定义 Webpack SCSS 加载器
- angular - 如何遍历 Angular 组件中数组中的每个模型?
- c# - 如何上传网站 (ASP.NET)
- c++ - LOWORD 和 HIWORD 操作
- javascript - 未捕获(承诺中)TypeError:无法将属性“innerHTML”设置为空
- c# - 与接口一起使用的新关键字
- java - 从 BlueJ 创建 Jar 文件时如何处理“发生 Java 异常”问题