JUnit / Mockito：在新线程中抛出异常

Question

我正在为以下 Java 代码编写单元测试：

try {
    new Thread(() -> myService.myMethod()).start();
    return new MyResponse("Done");
} catch (Exception exception) {
    throw new MyException("Internal Error", exception);
}

我正在尝试模拟 myService.myMethod() 引发的错误，以便在上面代码的 catch 块中拾取它并引发 MyException 的实例。

在我的 JUnit 测试中（使用 org.springframework.test.context.junit4.SpringJUnit4ClassRunner 和 Mockito），我模拟了 MyService 并将其作为 Spring bean 注入到单元测试中：

MyService myService = Mockito.mock(MyService.class);

然后我尝试了这个来模拟服务抛出异常：

IllegalArgumentException myException = new IllegalArgumentException();
Mockito.doThrow(myException).when(myService).myMethod();

...但是我正在测试的方法的 catch 块中永远不会发现异常。

这可能与代码使用另一个线程来调用 myService.myMethod() 的事实有关。

有什么方法可以让 myService.myMethod() 抛出的异常被正确捕获？

提前感谢您的帮助。

Answer 1

“外部”线程可以在另一个线程中捕获异常是没有意义的。 如果您在代码中添加上下文，例如

Thread th = new Thread(() -> { Thread.sleep(10000); throw new RuntimeException(); });
try { th.start(); } catch (Exception e) {
    System.out.println("uh-oh");
}
for (int i = 0; i < 1000; i++){
    System.out.println("running " + i);
    Thread.sleep(1000);
}

线程已启动，10 秒后将抛出异常。然而，开始线程已经开始每秒计数。如果启动线程的异常会被捕获，您是否希望主线程回到它捕获异常的地方？它应该中断计数然后继续吗？不继续？

您能做的最好的事情可能是指定一个错误处理程序（就像 Roger 在他的评论中提到的那样），但是您必须每隔一段时间从主线程查询它的状态，因为它将在并发启动的线程中运行。

考虑到t = new Thread(); t.start(); 不建议再使用，最简单的方法是使用CompletableFuture IMO：

CompletableFuture<Void> future = 
    CompletableFuture.runAsync(() -> myService.myMethod());

现在您可以稍后查询发生了什么：

if (future.isDone()) // task has finished in any way

if (future.isDone() && future.isCompletedExceptionally()) // task was finished because an exception was thrown

Answer 2

你的代码执行顺序如何？首先你需要设置你的模拟，然后运行它。

MyService myService = Mockito.mock(MyService.class);
Mockito.when(myService.myMethod()).doThrow(myexception.class); // try in this order


try {
    new Thread(() -> myService.myMethod()).start();
    return new MyResponse("Done");
} catch (Exception exception) {
    throw new MyException("Internal Error", exception);
}

顺便说一句，如果您使用的是 java 8 或更高版本，您可能需要使用 CompletableFuture 包。即CompletableFuture.supplyAsync(() -> {})