From c568fde70f26838c8673a0a14ece8b5f10254f6c Mon Sep 17 00:00:00 2001 From: Austin Raney Date: Tue, 30 Jan 2024 11:47:32 -0500 Subject: [PATCH] test: can specify minio keys using env vars --- .../dmod/test/it_object_store_dataset_manager.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 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 index 3fec85778..7b1edf5f3 100644 --- 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()