]> git.ipfire.org Git - thirdparty/sqlalchemy/sqlalchemy.git/commitdiff
- correct server side cursor assertion for driver names
authorMike Bayer <mike_mp@zzzcomputing.com>
Sat, 19 Aug 2017 21:30:06 +0000 (17:30 -0400)
committerMike Bayer <mike_mp@zzzcomputing.com>
Sat, 19 Aug 2017 21:30:06 +0000 (17:30 -0400)
Change-Id: Id8156f6af15a36f7d5653dff56f68e81431692a6

lib/sqlalchemy/testing/suite/test_results.py

index 98ddc7efcc8e6479be5bc52c8ea51cac9aa4f786..c00e0465733aabde7a53c06212d85542b67342d9 100644 (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