如何在 ExecutorCompletionService 上使用 mock

Question

我正在使用ExecutorCompletionService和下面的方法调用

Future<List<Student>> studentDetails = taskCompletionService.take();
Lis<Student> details =studentDetails.get()

现在正在写Junit和Mockito，我想嘲笑上面的两个电话。我怎样才能做到这一点？

Answer 1

如果你只是想为模拟taskCompletionService的take方法存根，那么你可以做这样的事情：

List<Student> studentDetails = Arrays.asList(fakeStudentDetail); 
when(taskCompletionService.take())
    .thenReturn(CompletableFuture.completedFuture(studentDetails));