在使用Mockito的JUnit中，如何等待异步方法完成？

import org.junit.Test;
import org.mockito.Mockito;

import java.util.concurrent.CountDownLatch;

public class AsyncTest {

    @Test
    public void testAsyncMethod() throws InterruptedException {
        // 创建CountDownLatch对象并初始化为1
        CountDownLatch latch = new CountDownLatch(1);

        // 创建被测试的对象
        MyClass myClass = new MyClass();

        // 创建异步方法的回调
        MyCallback callback = new MyCallback() {
            @Override
            public void onSuccess() {
                // 异步方法完成时调用countDown()方法
                latch.countDown();
            }

            @Override
            public void onFailure() {
                // 处理异步方法失败的情况
            }
        };

        // 调用被测试的异步方法
        myClass.asyncMethod(callback);

        // 等待异步方法完成
        latch.await();
        
        // 进行断言或其他验证操作
        Mockito.verify(callback).onSuccess();
    }
}

interface MyCallback {
    void onSuccess();
    void onFailure();
}

class MyClass {
    void asyncMethod(MyCallback callback) {
        // 异步方法的实现
    }
}