Skip to content

Commit e340e50

Browse files
committed
Enforce using pymysql as mysql dialect when creating mysql containers.
1 parent 847cbfa commit e340e50

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

sdks/python/apache_beam/io/external/xlang_jdbcio_it_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ class CrossLanguageJdbcIOTest(unittest.TestCase):
121121
'postgresql',
122122
'org.postgresql.Driver'),
123123
'mysql': DbData(
124-
lambda: MySqlContainer(), ['mysql:mysql-connector-java:8.0.28'],
124+
lambda: MySqlContainer(dialect='pymysql'), ['mysql:mysql-connector-java:8.0.28'],
125125
'mysql',
126126
'com.mysql.cj.jdbc.Driver')
127127
}

sdks/python/apache_beam/yaml/integration_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -228,7 +228,7 @@ def temp_mysql_database():
228228
with the MySQL database during setup.
229229
Exception: Any other exception encountered during the setup process.
230230
"""
231-
with MySqlContainer(init=True) as mysql_container:
231+
with MySqlContainer(init=True, dialect='pymysql') as mysql_container:
232232
try:
233233
# Make connection to temp database and create tmp table
234234
engine = sqlalchemy.create_engine(mysql_container.get_connection_url())

0 commit comments

Comments
 (0)