diff --git a/taskflow/tests/unit/persistence/test_sql_persistence.py b/taskflow/tests/unit/persistence/test_sql_persistence.py index 7a80d5fea..28c03ca72 100644 --- a/taskflow/tests/unit/persistence/test_sql_persistence.py +++ b/taskflow/tests/unit/persistence/test_sql_persistence.py @@ -173,7 +173,7 @@ class MysqlPersistenceTest(BackendPersistenceTestMixin, test.TestCase): db_uri = _get_connect_string('mysql', USER, PASSWD) engine = sa.create_engine(db_uri) with contextlib.closing(engine.connect()) as conn: - conn.execute("CREATE DATABASE %s" % DATABASE) + conn.execute(sa.text("CREATE DATABASE %s" % DATABASE)) except Exception as e: raise Exception('Failed to initialize MySQL db: %s' % (e)) finally: @@ -190,7 +190,7 @@ class MysqlPersistenceTest(BackendPersistenceTestMixin, test.TestCase): try: engine = sa.create_engine(self.db_uri) with contextlib.closing(engine.connect()) as conn: - conn.execute("DROP DATABASE IF EXISTS %s" % DATABASE) + conn.execute(sa.text("DROP DATABASE IF EXISTS %s" % DATABASE)) except Exception as e: raise Exception('Failed to remove temporary database: %s' % (e)) finally: @@ -215,7 +215,7 @@ class PostgresPersistenceTest(BackendPersistenceTestMixin, test.TestCase): engine = sa.create_engine(db_uri) with contextlib.closing(engine.connect()) as conn: conn.connection.set_isolation_level(0) - conn.execute("CREATE DATABASE %s" % DATABASE) + conn.execute(sa.text("CREATE DATABASE %s" % DATABASE)) conn.connection.set_isolation_level(1) except Exception as e: raise Exception('Failed to initialize PostgreSQL db: %s' % (e)) @@ -239,7 +239,7 @@ class PostgresPersistenceTest(BackendPersistenceTestMixin, test.TestCase): engine = sa.create_engine(db_uri) with contextlib.closing(engine.connect()) as conn: conn.connection.set_isolation_level(0) - conn.execute("DROP DATABASE IF EXISTS %s" % DATABASE) + conn.execute(sa.text("DROP DATABASE IF EXISTS %s" % DATABASE)) conn.connection.set_isolation_level(1) except Exception as e: raise Exception('Failed to remove temporary database: %s' % (e))