Update test_pg_user_pass.py

astronomer · Nov 17, 2023 · 91121c6 · 91121c6
1 parent df6d609
commit 91121c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/profiles/postgres/test_pg_user_pass.py b/tests/profiles/postgres/test_pg_user_pass.py
@@ -83,9 +83,9 @@ def test_connection_claiming() -> None:
             assert not profile_mapping.can_claim_connection()
 
     # also test when there's no schema
-    conn = Connection(**potential_values)  # type: ignore
+    conn = Connection(**{k: v for k, v in potential_values.items() if k != 'schema'})
     with patch("airflow.hooks.base.BaseHook.get_connection", return_value=conn):
-        profile_mapping = PostgresUserPasswordProfileMapping(conn, {})
+        profile_mapping = PostgresUserPasswordProfileMapping(conn, {"schema": None})
         assert not profile_mapping.can_claim_connection()
 
     # if we have them all, it should claim