added DSQL unit tests

Author: leszekamz

aws_advanced_python_wrapper/utils/dsql_token_utils.py

@@ -44,7 +44,20 @@ def generate_authentication_token(
         context = telemetry_factory.open_telemetry_context("fetch authentication token", TelemetryTraceLevel.NESTED)
 
         try:
-            client = boto3.client("dsql", region_name=region)
+            session = client_session if client_session else boto3.Session()
+            if credentials is not None:
+                client = session.client(
+                    'dsql',
+                    region_name=region,
+                    aws_access_key_id=credentials.get('AccessKeyId'),
+                    aws_secret_access_key=credentials.get('SecretAccessKey'),
+                    aws_session_token=credentials.get('SessionToken')
+                )
+            else:
+                client = session.client(
+                    'dsql',
+                    region_name=region
+                )
 
             if user == "admin":
                 token = client.generate_db_connect_admin_auth_token(host_name, region)

docs/examples/DSQLIamAuthentication.py

@@ -24,7 +24,7 @@
             user="admin",
             plugins="iam_dsql",
             iam_region="us-east-1",
-            wrapper_dialect="aurora-pg",
+            wrapper_dialect="pg",
             autocommit=True
     ) as awsconn, awsconn.cursor() as awscursor:
         awscursor.execute("CREATE TABLE IF NOT EXISTS bank_test (id int primary key, name varchar(40), account_balance int)")