diff options
Diffstat (limited to 'lib/sqlalchemy/testing/suite/test_results.py')
| -rw-r--r-- | lib/sqlalchemy/testing/suite/test_results.py | 10 |
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 |