使用mockito的具有CompletableFuture.allOf的测试方法

import static org.mockito.Mockito.*;

import org.junit.Test;

import java.util.concurrent.CompletableFuture;

public class MyAsyncServiceTest {

    @Test
    public void testAsyncMethod() {
        // 创建模拟对象
        MyDependency mockDependency = mock(MyDependency.class);
        
        // 创建CompletableFuture对象
        CompletableFuture<String> future1 = CompletableFuture.completedFuture("Result 1");
        CompletableFuture<String> future2 = CompletableFuture.completedFuture("Result 2");
        
        // 设置模拟对象的行为
        when(mockDependency.doSomethingAsync()).thenReturn(future1);
        when(mockDependency.doSomethingElseAsync()).thenReturn(future2);
        
        // 调用被测试的异步方法
        MyAsyncService asyncService = new MyAsyncService(mockDependency);
        CompletableFuture<String> result = asyncService.asyncMethod();
        
        // 使用CompletableFuture.allOf等待所有异步操作完成
        CompletableFuture.allOf(result).join();
        
        // 断言和验证
        verify(mockDependency).doSomethingAsync();
        verify(mockDependency).doSomethingElseAsync();
        // 其他断言和验证...
    }
}