Add stateful dark storage test

src/ert/dark_storage/endpoints/updates.py

@@ -1,4 +1,8 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, Depends
+
+from ert.dark_storage.enkf import get_storage
+
+DEFAULT_STORAGE = Depends(get_storage)
 
 router = APIRouter(tags=["ensemble"])
 

src/ert/dark_storage/enkf.py

@@ -18,11 +18,16 @@
 
 
 def get_storage() -> Storage:
-    global _storage
+    global _storage  # noqa
     if _storage is None:
-        try:
-            return (_storage := open_storage(os.environ["ERT_STORAGE_ENS_PATH"]))
-        except RuntimeError as err:
-            raise InternalServerError(f"{err!s}") from err
+        _storage = update_storage()
     _storage.refresh()
     return _storage
+
+
+def update_storage() -> Storage:
+    global _storage
+    try:
+        return (_storage := open_storage(os.environ["ERT_STORAGE_ENS_PATH"]))
+    except RuntimeError as err:
+        raise InternalServerError(f"{err!s}") from err

tests/ert/unit_tests/dark_storage/test_dark_storage_state.py

@@ -0,0 +1,50 @@
+import os
+from uuid import UUID
+
+import pytest
+from hypothesis import note, settings
+from hypothesis.stateful import rule
+from starlette.testclient import TestClient
+
+from ert.dark_storage.app import app
+from ert.dark_storage.enkf import update_storage
+from tests.ert.unit_tests.storage.test_local_storage import StatefulStorageTest
+
+
+@settings(max_examples=100)
+class DarkStorageStateTest(StatefulStorageTest):
+    def __init__(self):
+        super().__init__()
+        self.prev_no_token = os.environ.get("ERT_STORAGE_NO_TOKEN")
+        self.prev_ens_path = os.environ.get("ERT_STORAGE_ENS_PATH")
+        os.environ["ERT_STORAGE_NO_TOKEN"] = "yup"
+        os.environ["ERT_STORAGE_ENS_PATH"] = str(self.storage.path)
+        update_storage()
+        self.client = TestClient(app)
+
+    @rule()
+    def get_experiment_through_client(self):
+        self.client.get("/updates/storage")
+        response = self.client.get("/experiments")
+        experiment_records = response.json()
+        note(str((experiment_records, list(self.storage.experiments))))
+        assert len(experiment_records) == len(list(self.storage.experiments))
+        for record in experiment_records:
+            storage_experiment = self.storage.get_experiment(UUID(record["id"]))
+            assert {UUID(i) for i in record["ensemble_ids"]} == {
+                ens.id for ens in storage_experiment.ensembles
+            }
+
+    def teardown(self):
+        super().teardown()
+        if self.prev_no_token is not None:
+            os.environ["ERT_STORAGE_NO_TOKEN"] = self.prev_no_token
+        else:
+            del os.environ["ERT_STORAGE_NO_TOKEN"]
+        if self.prev_ens_path is not None:
+            os.environ["ERT_STORAGE_ENS_PATH"] = self.prev_ens_path
+        else:
+            del os.environ["ERT_STORAGE_ENS_PATH"]
+
+
+TestDarkStorage = pytest.mark.integration_test(DarkStorageStateTest.TestCase)

tests/ert/unit_tests/storage/test_local_storage.py

@@ -13,7 +13,7 @@
 import numpy as np
 import pytest
 import xarray as xr
-from hypothesis import assume, given
+from hypothesis import assume, given, note
 from hypothesis.extra.numpy import arrays
 from hypothesis.stateful import Bundle, RuleBasedStateMachine, initialize, rule
 
@@ -585,6 +585,7 @@ def __init__(self):
         super().__init__()
         self.tmpdir = tempfile.mkdtemp(prefix="StatefulStorageTest")
         self.storage = open_storage(self.tmpdir + "/storage/", "w")
+        note(f"storage path is: {self.storage.path}")
         self.model: Dict[UUID, Experiment] = {}
         assert list(self.storage.ensembles) == []