单元测试：如何测试 Spring 中的 @Async 注解方法

在单元测试中测试带有@Async注解的方法时，我们可以使用一些特定的方式来验证异步方法的行为和结果。

使用@Async注解的方法返回值为void时，可以使用CountDownLatch来等待异步方法执行完成。在测试方法中，创建一个CountDownLatch，并将其初始化为1。然后调用异步方法，并在异步方法执行完成之前调用CountDownLatch的await()方法进行等待。在异步方法执行完成后，调用CountDownLatch的countDown()方法，使其计数减一。最后，使用assert方法验证异步方法的行为和结果。

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;

@SpringJUnitConfig
@SpringBootTest
public class AsyncMethodTest {

    @Autowired
    private AsyncMethod asyncMethod;

    @Test
    public void testAsyncMethod() throws InterruptedException {
        CountDownLatch latch = new CountDownLatch(1);

        asyncMethod.asyncMethodWithVoidReturnType(latch);

        assertTrue(latch.await(5, TimeUnit.SECONDS));
        // 这里可以添加自定义的断言，验证异步方法的行为和结果
    }
}

使用@Async注解的方法返回值为CompletableFuture时，可以使用CompletableFuture的一些方法来验证异步方法的行为和结果。在测试方法中，调用异步方法并获取返回的CompletableFuture对象。然后使用CompletableFuture的各种方法进行验证，如使用get()方法获取异步方法的返回值，并使用assert方法进行断言。

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;

import java.util.concurrent.CompletableFuture;

import static org.junit.jupiter.api.Assertions.assertEquals;

@SpringJUnitConfig
@SpringBootTest
public class AsyncMethodTest {

    @Autowired
    private AsyncMethod asyncMethod;

    @Test
    public void testAsyncMethod() throws Exception {
        CompletableFuture<String> future = asyncMethod.asyncMethodWithCompletableFuture();

        assertEquals('Async method result', future.get());
        // 这里可以添加自定义的断言，验证异步方法的行为和结果
    }
}

这些是两种常见的测试异步方法的方式，具体使用哪种方式可以根据实际情况进行选择。