add logic to skip test in test_file itself

Author: suni72

gcsfs/tests/test_extended_gcsfs.py

@@ -154,9 +154,7 @@ def test_read_small_zb(extended_gcsfs, zonal_mocks):
             # cache drop
             assert len(f.cache.cache) < len(out)
             if mocks:
-                mocks["get_bucket_type"].assert_called_once_with(
-                    TEST_ZONAL_BUCKET
-                )
+                mocks["get_bucket_type"].assert_called_once_with(TEST_ZONAL_BUCKET)
 
 
 def test_readline_zb(extended_gcsfs, zonal_mocks):

gcsfs/tests/test_zonal_file.py

@@ -1,5 +1,6 @@
 """Tests for ZonalFile write operations."""
 
+import os
 from unittest import mock
 
 import pytest
@@ -10,6 +11,17 @@
 file_path = f"{TEST_BUCKET}/zonal-file-test"
 test_data = b"hello world"
 
+REQUIRED_ENV_VAR = "GCSFS_EXPERIMENTAL_ZB_HNS_SUPPORT"
+
+# If the condition is True, only then tests in this file are run.
+should_run = os.getenv(REQUIRED_ENV_VAR, "false").lower() in (
+    "true",
+    "1",
+)
+pytestmark = pytest.mark.skipif(
+    not should_run, reason=f"Skipping tests: {REQUIRED_ENV_VAR} env variable is not set"
+)
+
 
 @pytest.fixture
 def zonal_write_mocks():