Wrap query strings in sqlalchemy.text.

Author: tillahoffmann

mssql/testcontainers/mssql/__init__.py

@@ -17,7 +17,7 @@ class SqlServerContainer(DbContainer):
             >>> with SqlServerContainer() as mssql:
             ...    engine = sqlalchemy.create_engine(mssql.get_connection_url())
             ...    with engine.begin() as connection:
-            ...        result = connection.execute("select @@VERSION")
+            ...        result = connection.execute(sqlalchemy.text("select @@VERSION"))
     """
 
     def __init__(self, image: str = "mcr.microsoft.com/mssql/server:2019-latest", user: str = "SA",

mssql/tests/test_mssql.py

@@ -8,13 +8,13 @@ def test_docker_run_mssql():
     with SqlServerContainer(image, dialect=dialect) as mssql:
         engine = sqlalchemy.create_engine(mssql.get_connection_url())
         with engine.begin() as connection:
-            result = connection.execute('select @@servicename')
+            result = connection.execute(sqlalchemy.text('select @@servicename'))
             for row in result:
                 assert row[0] == 'MSSQLSERVER'
 
     with SqlServerContainer(image, password="1Secure*Password2", dialect=dialect) as mssql:
         engine = sqlalchemy.create_engine(mssql.get_connection_url())
         with engine.begin() as connection:
-            result = connection.execute('select @@servicename')
+            result = connection.execute(sqlalchemy.text('select @@servicename'))
             for row in result:
                 assert row[0] == 'MSSQLSERVER'

mysql/testcontainers/mysql/__init__.py

@@ -34,7 +34,7 @@ class MySqlContainer(DbContainer):
             >>> with MySqlContainer('mysql:5.7.17') as mysql:
             ...     engine = sqlalchemy.create_engine(mysql.get_connection_url())
             ...     with engine.begin() as connection:
-            ...         result = connection.execute("select version()")
+            ...         result = connection.execute(sqlalchemy.text("select version()"))
             ...         version, = result.fetchone()
     """
 

mysql/tests/test_mysql.py

@@ -12,7 +12,7 @@ def test_docker_run_mysql():
     with config as mysql:
         engine = sqlalchemy.create_engine(mysql.get_connection_url())
         with engine.begin() as connection:
-            result = connection.execute("select version()")
+            result = connection.execute(sqlalchemy.text("select version()"))
             for row in result:
                 assert row[0].startswith('5.7.17')
 
@@ -21,7 +21,7 @@ def test_docker_run_mariadb():
     with MySqlContainer("mariadb:10.6.5").maybe_emulate_amd64() as mariadb:
         engine = sqlalchemy.create_engine(mariadb.get_connection_url())
         with engine.begin() as connection:
-            result = connection.execute("select version()")
+            result = connection.execute(sqlalchemy.text("select version()"))
             for row in result:
                 assert row[0].startswith('10.6.5')
 

oracle/testcontainers/oracle/__init__.py

@@ -15,7 +15,7 @@ class OracleDbContainer(DbContainer):
             >>> with OracleDbContainer() as oracle:
             ...     engine = sqlalchemy.create_engine(oracle.get_connection_url())
             ...     with engine.begin() as connection:
-            ...         result = connection.execute("select * from V$VERSION")
+            ...         result = connection.execute(sqlalchemy.text("select * from V$VERSION"))
     """
 
     def __init__(self, image: str = "wnameless/oracle-xe-11g-r2:latest", **kwargs) -> None:

oracle/tests/test_oracle.py

@@ -13,5 +13,5 @@ def test_docker_run_oracle():
     with OracleDbContainer() as oracledb:
         engine = sqlalchemy.create_engine(oracledb.get_connection_url())
         with engine.begin() as connection:
-            result = connection.execute("select * from V$VERSION")
+            result = connection.execute(sqlalchemy.text("select * from V$VERSION"))
             assert {row[0] for row in result} == versions

postgres/testcontainers/postgres/__init__.py

@@ -33,7 +33,7 @@ class PostgresContainer(DbContainer):
             >>> with postgres_container as postgres:
             ...     engine = sqlalchemy.create_engine(postgres.get_connection_url())
             ...     with engine.begin() as connection:
-            ...         result = connection.execute("select version()")
+            ...         result = connection.execute(sqlalchemy.text("select version()"))
             ...         version, = result.fetchone()
             >>> version
             'PostgreSQL 9.5...'

postgres/tests/test_postgres.py

@@ -7,7 +7,7 @@ def test_docker_run_postgres():
     with postgres_container as postgres:
         engine = sqlalchemy.create_engine(postgres.get_connection_url())
         with engine.begin() as connection:
-            result = connection.execute("select version()")
+            result = connection.execute(sqlalchemy.text("select version()"))
             for row in result:
                 assert row[0].lower().startswith("postgresql 9.5")
 
@@ -17,4 +17,4 @@ def test_docker_run_postgres_with_driver_pg8000():
     with postgres_container as postgres:
         engine = sqlalchemy.create_engine(postgres.get_connection_url())
         with engine.begin() as connection:
-            connection.execute("select 1=1")
+            connection.execute(sqlalchemy.text("select 1=1"))