SNOW-1917265: Add OAUTH_TYPE for authorization flow (#2174)

Co-authored-by: Mark Keller <[email protected]>
Co-authored-by: Michał Hofman <[email protected]>

Author: 3 people

src/snowflake/connector/auth/oauth_code.py

@@ -17,6 +17,7 @@
 import urllib3
 
 from ..compat import parse_qs, urlparse, urlsplit
+from ..constants import OAUTH_TYPE_AUTHORIZATION_CODE
 from ..errorcode import (
     ER_IDP_CONNECTION_ERROR,
     ER_OAUTH_STATE_CHANGED,
@@ -92,6 +93,7 @@ def update_body(self, body: dict[Any, Any]) -> None:
         """
         body["data"]["AUTHENTICATOR"] = OAUTH_AUTHENTICATOR
         body["data"]["TOKEN"] = self._oauth_token
+        body["data"]["OAUTH_TYPE"] = OAUTH_TYPE_AUTHORIZATION_CODE
 
     def construct_url(self) -> str:
         params = {

src/snowflake/connector/constants.py

@@ -436,3 +436,4 @@ class IterUnit(Enum):
 )
 
 _OAUTH_DEFAULT_SCOPE = "session:role:{role}"
+OAUTH_TYPE_AUTHORIZATION_CODE = "authorization_code"

test/unit/test_auth_oauth_auth_code.py

@@ -0,0 +1,22 @@
+#!/usr/bin/env python
+#
+# Copyright (c) 2012-2023 Snowflake Computing Inc. All rights reserved.
+#
+
+from snowflake.connector.auth import AuthByOauthCode
+
+
+def test_auth_oauth_auth_code_oauth_type():
+    """Simple OAuth Auth Code oauth type test."""
+    auth = AuthByOauthCode(
+        "app",
+        "clientId",
+        "clientSecret",
+        "auth_url",
+        "tokenRequestUrl",
+        "redirectUri:{port}",
+        "scope",
+    )
+    body = {"data": {}}
+    auth.update_body(body)
+    assert body["data"]["OAUTH_TYPE"] == "authorization_code"