java - 如何测试异步方法

Question

我需要从数据库异步获取一些数据，然后用获取的数据同步更新当前缓存的数据。

目前我的代码是这样的:

@Asynchronous
public void fetchData() {
    CompletableFuture.supplyAsync(() -> {
        //Do some logic to fetch data
        return fetchedData;
    }).thenAccept(fetchedData -> updateCache(fetchedData));
}

我的问题是，当我在测试中调用此方法时，它会立即转到 thenAccept但是fetchedData是空的。我试图改变我的方法，并返回 CompletableFuture<List<Data>> , 并调用 fetchData().join()在测试中。它工作正常，但首先 - 我的应用程序无法构建，因为 @Asynchronous需要返回Future或 void ，而且我认为改变方法只是为了测试它不是一个好主意。我听说过 Await 库，但不能在当前项目中使用它。

我也不确定，如果 @Asynchronous在我的情况下需要吗？如果我没记错的话，CompletableFuture默认情况下应该在自己的线程中运行吗？

Answer 1

Jacek，我认为不需要@Asynchronous 注释。您可以使用可完成 future 的纯粹形式。我在下面提供了基于您提供的代码的示例代码。

public String getInfoFromDB() {
    try {
      TimeUnit.SECONDS.sleep(10);
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
    return "Data from DB";
  }

  public void updateCache(String dataFromDB) {
    System.out.println("Data from D : " + dataFromDB);
    System.out.println("Doing some operations with data from DB");
  }

  public void fetchData() {
    CompletableFuture cf =
        CompletableFuture.supplyAsync(() -> getInfoFromDB())
            .thenAccept(dataFromDB -> updateCache(dataFromDB));
    cf.join();
  }