PAT - small fixes, add tests (#2403)

Author: sfc-gh-pczajka

src/snowflake/connector/auth/by_plugin.py

@@ -50,7 +50,7 @@ class AuthType(Enum):
     ID_TOKEN = "ID_TOKEN"
     USR_PWD_MFA = "USERNAME_PASSWORD_MFA"
     OKTA = "OKTA"
-    PAT = "PROGRAMMATIC_ACCESS_TOKEN'"
+    PAT = "PROGRAMMATIC_ACCESS_TOKEN"
     NO_AUTH = "NO_AUTH"
     WORKLOAD_IDENTITY = "WORKLOAD_IDENTITY"
     PAT_WITH_EXTERNAL_SESSION = "PAT_WITH_EXTERNAL_SESSION"

src/snowflake/connector/auth/pat.py

@@ -13,6 +13,7 @@ def __init__(self, pat_token: str, **kwargs) -> None:
         super().__init__(**kwargs)
         self._pat_token: str | None = pat_token
 
+    @property
     def type_(self) -> AuthType:
         return AuthType.PAT
 

test/unit/test_auth_pat.py

@@ -0,0 +1,69 @@
+#!/usr/bin/env python
+#
+# Copyright (c) 2012-2023 Snowflake Computing Inc. All rights reserved.
+#
+
+from __future__ import annotations
+
+from snowflake.connector.auth import AuthByPAT
+from snowflake.connector.auth.by_plugin import AuthType
+from snowflake.connector.network import PROGRAMMATIC_ACCESS_TOKEN
+
+
+def test_auth_pat():
+    """Simple PAT test."""
+    token = "patToken"
+    auth = AuthByPAT(token)
+    assert auth.type_ == AuthType.PAT
+    assert auth.assertion_content == token
+    body = {"data": {}}
+    auth.update_body(body)
+    assert body["data"]["TOKEN"] == token, body
+    assert body["data"]["AUTHENTICATOR"] == PROGRAMMATIC_ACCESS_TOKEN, body
+
+    auth.reset_secrets()
+    assert auth.assertion_content is None
+
+
+def test_auth_pat_reauthenticate():
+    """Test PAT reauthenticate."""
+    token = "patToken"
+    auth = AuthByPAT(token)
+    result = auth.reauthenticate()
+    assert result == {"success": False}
+
+
+def test_pat_authenticator_creates_auth_by_pat(monkeypatch):
+    """Test that using PROGRAMMATIC_ACCESS_TOKEN authenticator creates AuthByPAT instance."""
+    import snowflake.connector
+
+    def mock_post_request(request, url, headers, json_body, **kwargs):
+        return {
+            "success": True,
+            "message": None,
+            "data": {
+                "token": "TOKEN",
+                "masterToken": "MASTER_TOKEN",
+                "idToken": None,
+                "parameters": [{"name": "SERVICE_NAME", "value": "FAKE_SERVICE_NAME"}],
+            },
+        }
+
+    monkeypatch.setattr(
+        snowflake.connector.network.SnowflakeRestful, "_post_request", mock_post_request
+    )
+
+    # Create connection with PAT authenticator
+    conn = snowflake.connector.connect(
+        user="user",
+        account="account",
+        database="TESTDB",
+        warehouse="TESTWH",
+        authenticator=PROGRAMMATIC_ACCESS_TOKEN,
+        token="test_pat_token",
+    )
+
+    # Verify that the auth_class is an instance of AuthByPAT
+    assert isinstance(conn.auth_class, AuthByPAT)
+
+    conn.close()