- correct server side cursor assertion for driver names

Change-Id: Id8156f6af15a36f7d5653dff56f68e81431692a6
This commit is contained in:
Mike Bayer
2017-08-19 17:30:06 -04:00
parent 9e6393c6b1
commit ecdd922d96
+5 -5
View File
@@ -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