如何在JUnit中刷新App Engine SearchServiceFactory异步操作？

Question

I'm using App engine Standard env and my services use the Search API .

So basically in my service I do something like:

IndexSpec indexSpec = IndexSpec.newBuilder().setName(indexName).build();
Index index = SearchServiceFactory.getSearchService().getIndex(indexSpec);
index.putAsync(document);

I'd like to implement a JUnit test to validate my document has been saved in the index.

But if I do in the same JUnit Thread:

...
index.putAsync(document);
index.search("");

it won't return my document (because I used Index#putAsync).

A simple hack is to use Thread#sleep:

...
index.putAsync(document);
Thread.sleep(10);
index.search("");

But that's really ugly.

How can I flush/execute all waiting async operations before running my search?

(I know there is a synchronous Index#put method but I want to use Index#putAsync in my service so the goal here is really to flush all waiting async operations)

Thanks

Answer 1

I ended up doing something like that:

...
Future future = index.putAsync(document);
while (!future.isDone()) {
  Thread.yield();
}
index.search("");