Fixing test

apache · May 8, 2021 · ef6e45d · ef6e45d
1 parent 1aba800
commit ef6e45d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/core/test_sqlalchemy_config.py b/tests/core/test_sqlalchemy_config.py
@@ -76,12 +76,15 @@ def test_sql_alchemy_connect_args(
         }
         with conf_vars(config):
             settings.configure_orm()
+            engine_args = {}
+            if settings.SQL_ALCHEMY_CONN.startswith('mysql'):
+                engine_args['isolation_level'] = 'READ COMMITTED'
             mock_create_engine.assert_called_once_with(
                 settings.SQL_ALCHEMY_CONN,
                 connect_args=SQL_ALCHEMY_CONNECT_ARGS,
                 poolclass=NullPool,
                 encoding='utf-8',
-                isolation_level='READ COMMITTED',
+                **engine_args,
             )
 
     @patch('airflow.settings.setup_event_handlers')