Awaitility：让异步变成同步，RocketMQ案例实战

2024-02-13 19:38:05

引言

在编写测试用例的时候遇到有异步或者队列处理的时候经常会用到 Thread.sleep() 等待来进行测试。例如：DLedger 测试选举的过程。当DLedger Leader下线。此时DLedger会选举出一个新的Leader。这个过程是异步的，我们需要等待一段时间才能知道选举结果。

使用 Thread.sleep() 来等待显然不是一个好的做法。首先，Thread.sleep() 会阻塞当前线程，导致测试用例的执行时间变长。其次，Thread.sleep() 的等待时间是固定的，如果选举过程比预期的要长，那么测试用例就会失败。

Awaitility 是一个 Java 库，它可以帮助我们在异步或并发的代码中编写更简洁、更可读的测试代码。通过使用 Awaitility，我们可以很容易地等待某些条件满足，而不用诉诸于 Thread.sleep() 或其他繁琐的等待机制。

Awaitility 的基本用法

Awaitility 的基本用法非常简单。首先，我们需要创建一个 Awaitility 实例。然后，我们可以使用 await() 方法来等待某个条件满足。例如：

Awaitility await = Awaitility.await();
await.until(new Predicate<Object>() {
    @Override
    public boolean apply(Object input) {
        return DLedger.isLeader();
    }
});

这段代码会等待直到 DLedger 成为 Leader。如果等待时间超过了默认的 10 秒，那么 await() 方法就会抛出 TimeoutException 异常。

我们可以使用 atMost() 和 until() 方法来指定等待的时间和条件。例如：

Awaitility await = Awaitility.await();
await.atMost(10, TimeUnit.SECONDS).until(new Predicate<Object>() {
    @Override
    public boolean apply(Object input) {
        return DLedger.isLeader();
    }
});

这段代码会等待最多 10 秒，直到 DLedger 成为 Leader。如果等待时间超过了 10 秒，那么 await() 方法就会抛出 TimeoutException 异常。

RocketMQ 案例

我们来看一个 RocketMQ 的案例。我们想要测试 RocketMQ 的消息发送功能。我们可以使用 Awaitility 来等待消息被发送到指定的消息队列。例如：

Awaitility await = Awaitility.await();
await.atMost(10, TimeUnit.SECONDS).until(() -> {
    return RocketMQClient.getInstance().getDefaultMQProducer().send(message);
});

这段代码会等待最多 10 秒，直到消息被发送到指定的消息队列。如果等待时间超过了 10 秒，那么 await() 方法就会抛出 TimeoutException 异常。