SNOW-2183023: fixed tests

Author: sfc-gh-fpawlowski

src/snowflake/connector/wif_util.py

@@ -131,9 +131,6 @@ def extract_iss_and_sub_without_signature_verification(
     return claims["iss"], claims["sub"]
 
 
-# --------------------------------------------------------------------------- #
-# AWS helper utilities (token, credentials, region)                           #
-# --------------------------------------------------------------------------- #
 def _imds_v2_token(session_manager: SessionManager | None = None) -> str | None:
     res = try_metadata_service_call(
         method="PUT",

test/csp_helpers.py

@@ -291,8 +291,7 @@ def sign_request(self, request: AWSRequest) -> None:
         date_stamp = utc_now.strftime("%Y%m%d")
 
         request.headers["X-Amz-Date"] = amz_date
-        if self.util_creds.token:
-            request.headers["X-Amz-Security-Token"] = self.util_creds.token
+        request.headers["X-Amz-Security-Token"] = self.util_creds.token
 
         # Host header is already set by the test; add it if a future test forgets
         if "Host" not in request.headers:

test/unit/test_auth_workload_identity.py

@@ -473,7 +473,7 @@ def test_internal_signer_vs_botocore_prefix(fake_aws_environment: FakeAwsEnviron
     new_prefix = new_hdr.split("Signature=")[0]
 
     # --- headers from real botocore SigV4Auth ----------------------------
-    creds = fake_aws_environment.util_creds  # boto Credentials
+    creds = fake_aws_environment.boto_creds  # boto Credentials
     region = fake_aws_environment.region
     url = (
         f"https://sts.{region}.amazonaws.com/"