fix: remove flakyness from elastic tests (#243)

docarray · Mar 31, 2022 · 0cae644 · 0cae644
1 parent 4722268
commit 0cae644
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/unit/array/conftest.py b/tests/unit/array/conftest.py
@@ -2,6 +2,7 @@
 import time
 
 import pytest
+from elasticsearch import Elasticsearch
 
 cur_dir = os.path.dirname(os.path.abspath(__file__))
 compose_yml = os.path.abspath(os.path.join(cur_dir, 'docker-compose.yml'))
@@ -13,7 +14,10 @@ def start_storage():
         f"docker-compose -f {compose_yml} --project-directory . up  --build -d "
         f"--remove-orphans"
     )
-    time.sleep(20)
+    es = Elasticsearch(hosts='http://localhost:9200/')
+    while not es.ping():
+        time.sleep(0.5)
+
     yield
     os.system(
         f"docker-compose -f {compose_yml} --project-directory . down "