fix: remove flakyness from elastic tests

alaeddine-13 · alaeddine-13 · commit cad959472735 · 2022-03-31T13:29:46.000+02:00
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 "
