Skip to content

Commit f87fdfe

Browse files
chore: update system test
1 parent dc2a15f commit f87fdfe

File tree

1 file changed

+14
-4
lines changed

1 file changed

+14
-4
lines changed

tests/system/test_connector_object.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -79,11 +79,21 @@ def test_multiple_connectors() -> None:
7979
conn.execute(sqlalchemy.text("SELECT 1"))
8080

8181
instance_connection_string = os.environ["MYSQL_CONNECTION_NAME"]
82-
assert instance_connection_string in first_connector._cache
83-
assert instance_connection_string in second_connector._cache
8482
assert (
85-
first_connector._cache[instance_connection_string]
86-
!= second_connector._cache[instance_connection_string]
83+
instance_connection_string,
84+
first_connector._enable_iam_auth,
85+
) in first_connector._cache
86+
assert (
87+
instance_connection_string,
88+
second_connector._enable_iam_auth,
89+
) in second_connector._cache
90+
assert (
91+
first_connector._cache[
92+
(instance_connection_string, first_connector._enable_iam_auth)
93+
]
94+
!= second_connector._cache[
95+
(instance_connection_string, second_connector._enable_iam_auth)
96+
]
8797
)
8898
except Exception:
8999
raise

0 commit comments

Comments
 (0)