diff --git a/test/__init__.py b/test/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/tests/conftest.py b/tests/conftest.py new file mode 100644 index 0000000..7c80754 --- /dev/null +++ b/tests/conftest.py @@ -0,0 +1,15 @@ +from distutils.command.config import config +import pytest +from pyinfra.config import get_config +import os + +@pytest.fixture(params=) +def storage_config(request): + if request.param == "s3": + os.environ["STORAGE_BACKEND"] = "s3" + os.environ["STORAGE_BUCKET_NAME"] = "pyinfra-test-bucket" + else: + os.environ["STORAGE_BACKEND"] = "aks" + os.environ["STORAGE_AZURECONTAINERNAME"] = "pyinfra-test-bucket" + + return get_config() diff --git a/tests/test_storage.py b/tests/test_storage.py new file mode 100644 index 0000000..830f9df --- /dev/null +++ b/tests/test_storage.py @@ -0,0 +1,5 @@ +from pyinfra.storage import get_storage + +def test_storage(storage_config) -> None: + storage = get_storage(storage_config) + assert storage.has_bucket(storage_config.storage_bucket)