How to make Elasticsearch indexing "blocking" when testing with Python?

I have few new microservices that use Elasticsearch to pass and retrieve data and I want to have few integration tests that use ES. The problem I have is getting data after some_document.save(). I have to have like sleep(1) for the get to retrieve the data via tested code.

Is there a way to make it blocking/synchronous to not use sleep in tests?