Allow for ert4 or ert5 storage service

equinor · May 31, 2023 · 91a53de · 91a53de
1 parent 4660382
commit 91a53de
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/data/spe1_st/tests/conftest.py b/tests/data/spe1_st/tests/conftest.py
@@ -3,8 +3,11 @@
 
 @pytest.fixture(scope="session", autouse=True)
 def start_storage_server():
-    from ert.services import StorageService
+    try:
+        from ert.services import StorageService as Storage
+    except ImportError:
+        from ert.services import Storage
 
-    with StorageService.start_server() as service:
+    with Storage.start_server() as service:
         service.wait_until_ready(timeout=30)
         yield service