Lint

Author: sfc-gh-pmansour

src/snowflake/connector/connection.py

@@ -359,7 +359,7 @@ def _get_private_bytes_from_file(
     "enable_single_use_refresh_tokens": (
         False,
         bool,
-    )
+    ),
 }
 
 APPLICATION_RE = re.compile(r"[\w\d_]+")

test/unit/test_auth_oauth_auth_code.py

@@ -3,10 +3,12 @@
 # Copyright (c) 2012-2023 Snowflake Computing Inc. All rights reserved.
 #
 
+from unittest.mock import patch
+
 import pytest
+
 from snowflake.connector.auth import AuthByOauthCode
 from snowflake.connector.network import OAUTH_AUTHORIZATION_CODE
-from unittest.mock import patch
 
 
 def test_auth_oauth_auth_code_oauth_type():
@@ -24,6 +26,7 @@ def test_auth_oauth_auth_code_oauth_type():
     auth.update_body(body)
     assert body["data"]["OAUTH_TYPE"] == "authorization_code"
 
+
 @pytest.mark.parametrize("rtr_enabled", [True, False])
 def test_auth_oauth_auth_code_single_use_refresh_tokens(rtr_enabled: bool):
     """Verifies that the enable_single_use_refresh_tokens option is plumbed into the authz code request."""
@@ -38,12 +41,26 @@ def test_auth_oauth_auth_code_single_use_refresh_tokens(rtr_enabled: bool):
         pkce_enabled=False,
         enable_single_use_refresh_tokens=rtr_enabled,
     )
+
     def fake_get_request_token_response(_, fields: dict[str, str]):
         if rtr_enabled:
             assert fields.get("enable_single_use_refresh_tokens") == "true"
         else:
             assert "enable_single_use_refresh_tokens" not in fields
         return ("access_token", "refresh_token")
-    with patch("snowflake.connector.auth.AuthByOauthCode._do_authorization_request", return_value="abc"):
-        with patch("snowflake.connector.auth.AuthByOauthCode._get_request_token_response", side_effect=fake_get_request_token_response):
-            auth.prepare(conn=None, authenticator=OAUTH_AUTHORIZATION_CODE, service_name=None, account="acc", user="user")
+
+    with patch(
+        "snowflake.connector.auth.AuthByOauthCode._do_authorization_request",
+        return_value="abc",
+    ):
+        with patch(
+            "snowflake.connector.auth.AuthByOauthCode._get_request_token_response",
+            side_effect=fake_get_request_token_response,
+        ):
+            auth.prepare(
+                conn=None,
+                authenticator=OAUTH_AUTHORIZATION_CODE,
+                service_name=None,
+                account="acc",
+                user="user",
+            )

test/unit/test_connection.py

@@ -690,7 +690,9 @@ def test_toml_connection_params_are_plumbed_into_authbyworkloadidentity(
 
 
 @pytest.mark.parametrize("rtr_enabled", [True, False])
-def test_single_use_refresh_tokens_option_is_plumbed_into_authbyauthcode(monkeypatch, rtr_enabled: bool):
+def test_single_use_refresh_tokens_option_is_plumbed_into_authbyauthcode(
+    monkeypatch, rtr_enabled: bool
+):
     with monkeypatch.context() as m:
         m.setattr(
             "snowflake.connector.SnowflakeConnection._authenticate", lambda *_: None