Skip to content

Commit 2f8c0f2

Browse files
chore: update tests
1 parent bb31e2c commit 2f8c0f2

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

tests/system/test_pymysql_connection.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,7 @@ def test_lazy_pymysql_connection() -> None:
127127

128128

129129
def test_MCP_pymysql_connection() -> None:
130-
"""Basic test to get time from database."""
130+
"""Basic test to get time from database using MCP enabled instance."""
131131
inst_conn_name = os.environ["MYSQL_MCP_CONNECTION_NAME"]
132132
user = os.environ["MYSQL_USER"]
133133
password = os.environ["MYSQL_MCP_PASS"]

tests/system/test_pymysql_iam_auth.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -116,15 +116,13 @@ def test_lazy_pymysql_iam_authn_connection() -> None:
116116

117117

118118
def test_MCP_pymysql_iam_authn_connection() -> None:
119-
"""Basic test to get time from database."""
119+
"""Basic test to get time from database using MCP enabled instance."""
120120
inst_conn_name = os.environ["MYSQL_MCP_CONNECTION_NAME"]
121121
user = os.environ["MYSQL_IAM_USER"]
122122
db = os.environ["MYSQL_DB"]
123123
ip_type = os.environ.get("IP_TYPE", "public")
124124

125-
engine, connector = create_sqlalchemy_engine(
126-
inst_conn_name, user, db, ip_type, "lazy"
127-
)
125+
engine, connector = create_sqlalchemy_engine(inst_conn_name, user, db, ip_type)
128126
with engine.connect() as conn:
129127
time = conn.execute(sqlalchemy.text("SELECT NOW()")).fetchone()
130128
conn.commit()

0 commit comments

Comments
 (0)