From: Mike Bayer Date: Sat, 19 Aug 2017 21:30:06 +0000 (-0400) Subject: - correct server side cursor assertion for driver names X-Git-Tag: origin~67 X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=ecdd922d9627c60058833fbcc2c8c2bc06d7332b;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git - correct server side cursor assertion for driver names Change-Id: Id8156f6af15a36f7d5653dff56f68e81431692a6 --- 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