From ecdd922d9627c60058833fbcc2c8c2bc06d7332b Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Sat, 19 Aug 2017 17:30:06 -0400 Subject: [PATCH] - correct server side cursor assertion for driver names Change-Id: Id8156f6af15a36f7d5653dff56f68e81431692a6 --- lib/sqlalchemy/testing/suite/test_results.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/lib/sqlalchemy/testing/suite/test_results.py b/lib/sqlalchemy/testing/suite/test_results.py index 98ddc7efcc..c00e046573 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 -- 2.47.3