diff options
author | Mike Bayer <mike_mp@zzzcomputing.com> | 2017-08-19 17:30:06 -0400 |
---|---|---|
committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2017-08-19 17:30:06 -0400 |
commit | ecdd922d9627c60058833fbcc2c8c2bc06d7332b (patch) | |
tree | fc0b4dd74eb54a2acead967b352d0e68ca09e88b | |
parent | 9e6393c6b12397525053f4df9146894050b480d2 (diff) | |
download | sqlalchemy-ecdd922d9627c60058833fbcc2c8c2bc06d7332b.tar.gz |
- correct server side cursor assertion for driver names
Change-Id: Id8156f6af15a36f7d5653dff56f68e81431692a6
-rw-r--r-- | lib/sqlalchemy/testing/suite/test_results.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/lib/sqlalchemy/testing/suite/test_results.py b/lib/sqlalchemy/testing/suite/test_results.py index 98ddc7efc..c00e04657 100644 --- a/lib/sqlalchemy/testing/suite/test_results.py +++ b/lib/sqlalchemy/testing/suite/test_results.py @@ -228,14 +228,14 @@ class ServerSideCursorsTest(fixtures.TestBase, testing.AssertsExecutionResults): __backend__ = True def _is_server_side(self, cursor): - if self.engine.url.drivername == 'postgresql': + if self.engine.dialect.driver == "psycopg2": return cursor.name - elif self.engine.url.drivername == 'mysql': - sscursor = __import__('MySQLdb.cursors').cursors.SSCursor - return isinstance(cursor, sscursor) - elif self.engine.url.drivername == 'mysql+pymysql': + elif self.engine.dialect.driver == 'pymysql': sscursor = __import__('pymysql.cursors').cursors.SSCursor return isinstance(cursor, sscursor) + elif self.engine.dialect.driver == "mysqldb": + sscursor = __import__('MySQLdb.cursors').cursors.SSCursor + return isinstance(cursor, sscursor) else: return False |