JUnit-测试Spring @Async void服务方法

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

import java.util.concurrent.CountDownLatch;

@SpringBootTest
@EnableAsync
public class AsyncServiceTest {

    @Autowired
    private AsyncService asyncService;

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

        asyncService.asyncMethod(() -> latch.countDown());

        latch.await(); // 等待异步方法执行完成
    }
}

import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

@Service
public class AsyncService {

    @Async
    public void asyncMethod(Runnable callback) {
        // 异步执行的代码
        callback.run(); // 执行回调
    }
}