test(auth): add manual authentication tests

Author: Brooke-white

test/__init__.py

@@ -8,4 +8,5 @@
     jwt_google_idp,
     okta_browser_idp,
     okta_idp,
+    ping_browser_idp,
 )

test/integration/__init__.py

@@ -8,4 +8,5 @@
     jwt_google_idp,
     okta_browser_idp,
     okta_idp,
+    ping_browser_idp,
 )

test/integration/conftest.py

@@ -105,6 +105,18 @@ def jumpcloud_browser_idp():
     return {**_get_default_connection_args(), **db_connect}
 
 
+@pytest.fixture(scope="class")
+def ping_browser_idp():
+    db_connect = {
+        "iam": conf.getboolean("ping-one-idp", "iam"),
+        "credentials_provider": conf.get("ping-one-idp", "credentials_provider"),
+        "login_url": conf.get("ping-one-idp", "login_url"),
+        "listen_port": conf.getint("ping-one-idp", "listen_port"),
+        "idp_response_timeout": conf.getint("ping-one-idp", "idp_response_timeout"),
+    }
+    return {**_get_default_connection_args(), **db_connect}
+
+
 @pytest.fixture(scope="class")
 def azure_idp():
     db_connect = {

test/manual/__init__.py

@@ -0,0 +1,33 @@
+import pytest
+
+import redshift_connector
+
+aws_secret_access_key: str = ""
+aws_access_key: str = ""
+aws_session_token: str = ""
+
+"""
+How to use:
+0) If necessary, create a Redshift cluster
+1) In the connect method below,  specify the connection parameters
+3) Specify the AWS IAM credentials in the variables above
+4) Manually execute this test
+"""
+
+
+@pytest.mark.skip(reason="manual")
+def test_use_aws_credentials_default_profile():
+    with redshift_connector.connect(
+        iam=True,
+        database="my_database",
+        db_user="my_db_user",
+        password="",
+        user="",
+        cluster_identifier="my_cluster_identifier",
+        region="my_region",
+        access_key_id=aws_access_key,
+        secret_access_key=aws_secret_access_key,
+        session_token=aws_session_token,
+    ) as con:
+        with con.cursor() as cursor:
+            cursor.execute("select 1")

test/manual/auth/test_aws_credentials.py

@@ -0,0 +1,44 @@
+import configparser
+import os
+import typing
+from test import (
+    azure_browser_idp,
+    idp_arg,
+    jumpcloud_browser_idp,
+    jwt_azure_v2_idp,
+    jwt_google_idp,
+    okta_browser_idp,
+    ping_browser_idp,
+)
+
+import pytest  # type: ignore
+
+import redshift_connector
+
+conf = configparser.ConfigParser()
+root_path = os.path.dirname(os.path.dirname(os.path.abspath(os.path.join(__file__, os.pardir))))
+conf.read(root_path + "/config.ini")
+
+BROWSER_CREDENTIAL_PROVIDERS: typing.List[str] = [
+    "jumpcloud_browser_idp",
+    "okta_browser_idp",
+    "azure_browser_idp",
+    "jwt_azure_v2_idp",
+    "jwt_google_idp",
+    "ping_browser_idp",
+]
+
+"""
+How to use:
+0) If necessary, create a Redshift cluster and configure it for use with the desired IdP
+1) In config.ini specify the connection parameters required by the desired IdP fixture in test/integration/conftest.py
+2) Ensure browser cookies have been cleared
+3) Manually execute the tests in this file, providing the necessary login information in the web browser
+"""
+
+
+@pytest.mark.skip(reason="manual")
+@pytest.mark.parametrize("idp_arg", BROWSER_CREDENTIAL_PROVIDERS, indirect=True)
+def test_browser_credentials_provider_can_auth(idp_arg):
+    with redshift_connector.connect(**idp_arg):
+        pass

test/manual/plugin/__init__.py

@@ -6,10 +6,12 @@
     azure_browser_idp,
     azure_idp,
     idp_arg,
+    jumpcloud_browser_idp,
     jwt_azure_v2_idp,
     jwt_google_idp,
     okta_browser_idp,
     okta_idp,
+    ping_browser_idp,
 )
 
 import pytest  # type: ignore
@@ -26,6 +28,8 @@
 ALL_IDP: typing.List[str] = [
     "okta_browser_idp",
     "azure_browser_idp",
+    "jumpcloud_browser_idp",
+    "ping_browser_idp",
     "jwt_google_idp",
     "jwt_azure_v2_idp",
 ] + NON_BROWSER_IDP