为什么 assertThrows 需要 () ->:Lambda 延迟执行与异常捕获

8 阅读2分钟

下面这段代码用来检查:用户名为空时,createUser 是否会抛出 IllegalArgumentException。

assertThrows(
        IllegalArgumentException.class,
        () -> service.createUser("")
);

比较容易让人困惑的是第二个参数。既然真正要测试的是 service.createUser(""),为什么还要在外面套一层 () ->?

原因和 Java 计算方法参数的顺序有关。

直接调用会发生什么

如果去掉 Lambda,代码会变成这样:

assertThrows(
        IllegalArgumentException.class,
        service.createUser("")
);

Java 在进入 assertThrows 之前,必须先算出两个参数的值。

第一个参数 IllegalArgumentException.class 没什么问题。计算第二个参数时,Java 会立刻调用 service.createUser("")。而这个方法收到空用户名后会抛出异常:

throw new IllegalArgumentException("username cannot be blank");

这时 assertThrows 还没有开始执行,自然也没有机会捕获异常。

不过实际写代码时,更可能先遇到编译错误。assertThrows 的第二个参数需要一个 Executable,也就是“可执行的代码”。直接调用 createUser 得到的是返回值,不是 Executable,类型对不上。

() -> 把调用留到了后面

() -> service.createUser("")

这行代码没有调用 createUser。它创建了一个 Lambda,内容是“执行 service.createUser("")”。

可以先给它取个名字:

Executable task = () -> service.createUser("");

此时只是创建了 task。只有调用 task.execute(),里面的 createUser 才会运行。

assertThrows 拿到这个任务后,会在自己的异常捕获逻辑中执行它。内部过程可以粗略理解成:

try {
    task.execute();
} catch (Throwable e) {
    // 检查异常类型是否符合预期
}

这样顺序就对了:assertThrows 先开始运行并准备捕获异常,然后执行 Lambda,最后检查 createUser 抛出的异常是不是 IllegalArgumentException。

因此,Lambda 传给 assertThrows 的不是异常,也不是方法运行后的结果,而是那段可能抛出异常的代码。

方法调用和 Lambda 的区别

service.createUser("") 表示现在调用方法。

() -> service.createUser("") 表示先保存这段调用,等 Lambda 被执行时再调用方法。

这也是 Lambda 很常见的一种用途:把一段行为交给其他代码,由接收方决定执行时机。按钮点击事件、线程任务和 Stream API 都会用到相同的思路。

还有一个容易混淆的地方:throw 不一定会终止整个程序。异常抛出后,Java 会沿着调用栈寻找能够处理它的 catch。assertThrows 正是利用这一点捕获异常,再判断异常类型是否符合测试要求。

这段代码的执行顺序可以压缩成一句话:

先进入 assertThrows,再执行 createUser,这样抛出的异常才在它的捕获范围内。