Skip to content

Commit 1363cc5

Browse files
Fix PostgresHook bug when getting AWS Redshift Serverless credentials (#43807)
* correct test value * fix production code
1 parent f6e0900 commit 1363cc5

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

providers/src/airflow/providers/postgres/hooks/postgres.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -293,8 +293,8 @@ def get_iam_token(self, conn: Connection) -> tuple[str, str, int]:
293293
dbName=self.database or conn.schema,
294294
workgroupName=workgroup_name,
295295
)
296-
token = cluster_creds["DbPassword"]
297-
login = cluster_creds["DbUser"]
296+
token = cluster_creds["dbPassword"]
297+
login = cluster_creds["dbUser"]
298298
else:
299299
port = conn.port or 5432
300300
rds_client = AwsBaseHook(aws_conn_id=aws_conn_id, client_type="rds").conn

providers/tests/postgres/hooks/test_postgres.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -274,8 +274,8 @@ def test_get_conn_rds_iam_redshift_serverless(
274274
}
275275
if aws_conn_id is not NOTSET:
276276
mock_conn_extra["aws_conn_id"] = aws_conn_id
277-
if conn_workgroup_name is not NOTSET: # change to workgroup
278-
mock_conn_extra["workgroup-name"] = conn_workgroup_name # change to workgroup
277+
if conn_workgroup_name is not NOTSET:
278+
mock_conn_extra["workgroup-name"] = conn_workgroup_name
279279

280280
self.connection.extra = json.dumps(mock_conn_extra)
281281
self.connection.host = host
@@ -287,8 +287,8 @@ def test_get_conn_rds_iam_redshift_serverless(
287287
mock_aws_hook_instance = mock_aws_hook_class.return_value
288288
mock_client = mock.MagicMock()
289289
mock_client.get_credentials.return_value = {
290-
"DbPassword": mock_db_pass,
291-
"DbUser": mock_db_user,
290+
"dbPassword": mock_db_pass,
291+
"dbUser": mock_db_user,
292292
}
293293
type(mock_aws_hook_instance).conn = mock.PropertyMock(return_value=mock_client)
294294

0 commit comments

Comments
 (0)