java - Spring Webflux如何验证所有步骤都执行成功
问题描述
我是 Spring webflux 的新手,我刚刚开始 webflux 项目,但我被困在一个地方。
我正在使用Id和FileName创建一个学生对象。
当我在 studentService 上调用 deleteById(Long) 方法时,我想先从存储中删除文件,然后我想从Repository中删除记录,
所有这些方法都返回 Mono
以下是我为学生提供的服务和存储库代码
public class Student {
private Long id;
private String fileName;
//getter & setter
}
public class StudentRepository {
public Mono<Student> findById(long l){
Student sample = new Student();
sample.setFileName("file-name");
sample.setId(1L);
return Mono.just(sample);
}
public Mono<Void> deleteFile(String fileName) {
return Mono.empty();
}
public Mono<Void> deleteById(Long id) {
return Mono.empty();
}
}
public class StudentService {
private StudentRepository repository;
public void setRepository(StudentRepository repository) {
this.repository = repository;
}
public Mono<Student> findById(long l){
Student sample = new Student();
sample.setFileName("file-name");
sample.setId(1L);
return Mono.just(sample);
}
public Mono<Void> deleteById(Long id){
return repository.findById(id)
.flatMap(student ->
repository.deleteFile(student.getFileName()).thenReturn(student)
)
.flatMap(studentMono ->
repository.deleteById(studentMono.getId())
);
}
}
现在我想验证第一个文件是否从文件存储中删除,然后记录应该从数据库中删除。
我写了如下测试
public void test(){
StudentService studentService = new StudentService();
StudentRepository studentRepository = mock(StudentRepository.class);
studentService.setRepository(studentRepository);
Student student = new Student();
student.setFileName("file-name");
student.setId(1L);
Mono<Student> studentMono = Mono.just(student);
Mockito.when(studentRepository.findById(1L)).thenReturn(studentMono);
studentService.deleteById(1L);
StepVerifier.create(studentMono)
.expectNextMatches(leadImport->leadImport.getFileName().equals("file-name"))
.expectNextMatches(leadImport -> leadImport.getId() == 1L)
.verifyComplete();
}
但是有些我的测试失败了。
有人可以帮助我如何验证我所有的预期步骤,比如
- 首先删除文件
- 第二次从数据库中删除
解决方案
you create a StepVerifier from a Mono.just(student), it means you are watching a single element
chaining the expectNextMatches will work if you have more than one element : Flux.just(student1, student2,..) as it will verify each student for each calls
change your code by calling expectNext only once :
StepVerifier.create(studentMono)
.expectNextMatches(leadImport->leadImport.getFileName().equals("file-name") && leadImport.getId() == 1L)
.verifyComplete();