Update tests/deployments/test_steps.py

Co-authored-by: Chris Guidry <[email protected]>
PrefectHQ · Jan 19, 2024 · e7d0d3c · e7d0d3c
1 parent 5315709
commit e7d0d3c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/deployments/test_steps.py b/tests/deployments/test_steps.py
@@ -10,7 +10,15 @@
 from prefect_aws import AwsCredentials
 from prefect_aws.deployments.steps import get_s3_client, pull_from_s3, push_to_s3
 
-os.environ["MOTO_S3_CUSTOM_ENDPOINTS"] = "http://custom.minio.endpoint:9000"
+
+@pytest.fixture(scope="module", autouse=True)
+def set_custom_endpoint():
+    original = os.environ.get("MOTO_S3_CUSTOM_ENDPOINTS")
+    os.environ["MOTO_S3_CUSTOM_ENDPOINTS"] = "http://custom.minio.endpoint:9000"
+    yield
+    os.environ.pop("MOTO_S3_CUSTOM_ENDPOINTS")
+    if original is not None:
+        os.environ["MOTO_S3_CUSTOM_ENDPOINTS"] = original
 
 
 @pytest.fixture