refactor(auth, iam): support serverless get_workgroup

Author: Brooke-white

redshift_connector/iam_helper.py

@@ -94,9 +94,9 @@ def set_iam_properties(info: RedshiftProperty) -> RedshiftProperty:
         IamHelper.set_auth_properties(info)
 
         if info._is_serverless and info.iam:
-            if Version(pkg_resources.get_distribution("boto3").version) < Version("1.24.5"):
+            if Version(pkg_resources.get_distribution("boto3").version) < Version("1.24.11"):
                 raise pkg_resources.VersionConflict(
-                    "boto3 >= 1.24.5 required for authentication with Amazon Redshift serverless. "
+                    "boto3 >= 1.24.11 required for authentication with Amazon Redshift serverless. "
                     "Please upgrade the installed version of boto3 to use this functionality."
                 )
 
@@ -274,9 +274,11 @@ def set_cluster_credentials(
                 response: dict
 
                 if info._is_serverless:
-                    response = client.describe_configuration()
-                    info.put("host", response["endpoint"]["address"])
-                    info.put("port", response["endpoint"]["port"])
+                    if not info.serverless_work_group:
+                        raise InterfaceError("Serverless workgroup is not set.")
+                    response = client.get_workgroup(workgroupName=info.serverless_work_group)
+                    info.put("host", response["workgroup"]["endpoint"]["address"])
+                    info.put("port", response["workgroup"]["endpoint"]["port"])
                 else:
                     response = client.describe_clusters(ClusterIdentifier=info.cluster_identifier)
                     info.put("host", response["Clusters"][0]["Endpoint"]["Address"])

test/manual/test_redshift_serverless.py

@@ -40,3 +40,33 @@ def test_idp_auth(okta_idp):
 
     with redshift_connector.connect(**okta_idp):
         pass
+
+
+@pytest.mark.skip()
+def test_connection_without_host(serverless_iam_db_kwargs):
+    serverless_iam_db_kwargs["is_serverless"] = True
+    serverless_iam_db_kwargs["host"] = None
+    serverless_iam_db_kwargs["serverless_work_group"] = "default"
+    with redshift_connector.connect(**serverless_iam_db_kwargs) as conn:
+        with conn.cursor() as cursor:
+            cursor.execute("select 1")
+
+
+@pytest.mark.skip()
+def test_nlb_connection(serverless_iam_db_kwargs):
+    serverless_iam_db_kwargs["is_serverless"] = True
+    serverless_iam_db_kwargs["host"] = "my_nlb_endpoint"
+    serverless_iam_db_kwargs["serverless_work_group"] = "default"
+    with redshift_connector.connect(**serverless_iam_db_kwargs) as conn:
+        with conn.cursor() as cursor:
+            cursor.execute("select 1")
+
+
+@pytest.mark.skip()
+def test_vpc_endpoint_connection(serverless_iam_db_kwargs):
+    serverless_iam_db_kwargs["is_serverless"] = True
+    serverless_iam_db_kwargs["host"] = "my_vpc_endpoint"
+    serverless_iam_db_kwargs["serverless_work_group"] = "default"
+    with redshift_connector.connect(**serverless_iam_db_kwargs) as conn:
+        with conn.cursor() as cursor:
+            cursor.execute("select 1")