test: can specify minio keys using env vars

aaraney · Jan 30, 2024 · c568fde · c568fde
1 parent abc684f
commit c568fde
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/python/lib/modeldata/dmod/test/it_object_store_dataset_manager.py b/python/lib/modeldata/dmod/test/it_object_store_dataset_manager.py
@@ -67,8 +67,14 @@ def setUp(self) -> None:
         self._hostname = 'localhost'
 
         self._secrets_dir: Path = Path(self.find_git_root_dir()).joinpath("docker/secrets/")
-        self._access_key = self._secrets_dir.joinpath("object_store/model_exec_access_key").read_text()
-        self._secret_key = self._secrets_dir.joinpath("object_store/model_exec_secret_key").read_text()
+        if self._secrets_dir.exists():
+            self._access_key = self._secrets_dir.joinpath("object_store/model_exec_access_key").read_text()
+            self._secret_key = self._secrets_dir.joinpath("object_store/model_exec_secret_key").read_text()
+        else:
+            self._access_key = os.environ.get("MODEL_EXEC_ACCESS_KEY")
+            assert self._access_key is not None, "'MODEL_EXEC_ACCESS_KEY' environment variable or 'docker/secrets/object_store/model_exec_access_key' file is required"
+            self._secret_key =  os.environ.get("MODEL_EXEC_SECRET_KEY")
+            assert self._secret_key is not None, "'MODEL_EXEC_SECRET_KEY' environment variable or 'docker/secrets/object_store/model_exec_secret_key' file is required"
 
         # Initialize the manager and its backing minio client
         self._initialize_manager()