Skip to content

Commit e9fe082

Browse files
skip if cloud provider not set
1 parent ca0bddc commit e9fe082

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

test/integ/aio_it/test_put_get_with_aws_token_async.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,14 @@
3838
pytestmark = [pytest.mark.asyncio, pytest.mark.aws]
3939

4040

41+
# Skip entire module if not running on AWS
42+
if (CLOUD_PROVIDER := os.getenv("cloud_provider", "not-set").upper()) != "AWS":
43+
pytest.skip(
44+
f"AWS-specific test module, skipping on {CLOUD_PROVIDER}",
45+
allow_module_level=True,
46+
)
47+
48+
4149
@pytest.mark.parametrize(
4250
"from_path", [True, pytest.param(False, marks=pytest.mark.skipolddriver)]
4351
)
@@ -146,7 +154,7 @@ async def test_put_with_invalid_token(tmpdir, aio_connection):
146154
)
147155

148156
client = SnowflakeS3RestClient(meta, creds, stage_info, 8388608)
149-
await client.transfer_accelerate_config(None)
157+
await client.transfer_accelerate_config(False) # Initialize endpoint
150158
await client.get_file_header(meta.name) # positive case
151159

152160
# negative case, no aws token

0 commit comments

Comments
 (0)