summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2017-08-19 17:30:06 -0400
committerMike Bayer <mike_mp@zzzcomputing.com>2017-08-19 17:30:06 -0400
commitecdd922d9627c60058833fbcc2c8c2bc06d7332b (patch)
treefc0b4dd74eb54a2acead967b352d0e68ca09e88b
parent9e6393c6b12397525053f4df9146894050b480d2 (diff)
downloadsqlalchemy-ecdd922d9627c60058833fbcc2c8c2bc06d7332b.tar.gz
- correct server side cursor assertion for driver names
Change-Id: Id8156f6af15a36f7d5653dff56f68e81431692a6
-rw-r--r--lib/sqlalchemy/testing/suite/test_results.py10
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