diff options
Diffstat (limited to 'lib/sqlalchemy/testing/suite/test_results.py')
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_results.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/lib/sqlalchemy/testing/suite/test_results.py b/lib/sqlalchemy/testing/suite/test_results.py index 5186e189c..7a3e92564 100644 --- a/lib/sqlalchemy/testing/suite/test_results.py +++ b/lib/sqlalchemy/testing/suite/test_results.py @@ -15,6 +15,7 @@ from ... import sql from ... import String from ... import testing from ... import text +from ... import util class RowFetchTest(fixtures.TablesTest): @@ -287,7 +288,10 @@ class ServerSideCursorsTest( ): engine = self._fixture(engine_ss_arg) with engine.begin() as conn: - result = conn.execute(statement) + if isinstance(statement, util.string_types): + result = conn.exec_driver_sql(statement) + else: + result = conn.execute(statement) eq_(self._is_server_side(result.cursor), cursor_ss_status) result.close() @@ -298,7 +302,7 @@ class ServerSideCursorsTest( result = ( engine.connect() .execution_options(stream_results=True) - .execute("select 1") + .exec_driver_sql("select 1") ) assert self._is_server_side(result.cursor) |