From eefa4b68f5dce3770d9b95455e9f14439f6bbd72 Mon Sep 17 00:00:00 2001 From: John Anderson Date: Mon, 18 Mar 2013 15:04:30 -0700 Subject: [PATCH] cleaned up errors, only check for connection timeout for now --- lib/sqlalchemy/dialects/mssql/pymssql.py | 5 ++--- test/dialect/test_mssql.py | 4 ++-- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/lib/sqlalchemy/dialects/mssql/pymssql.py b/lib/sqlalchemy/dialects/mssql/pymssql.py index 0754f7f764..08e8fca6a2 100644 --- a/lib/sqlalchemy/dialects/mssql/pymssql.py +++ b/lib/sqlalchemy/dialects/mssql/pymssql.py @@ -85,9 +85,8 @@ class MSDialect_pymssql(MSDialect): def is_disconnect(self, e, connection, cursor): for msg in ( - "Error 20003", # Connection Timeout - "Error 20004", # Read from SQL server failed - "Error 20006", # Write to SQL Server failed + "Adaptive Server connection timed out", + "message 20003", # connection timeout "Error 10054", "Not connected to any MS SQL server", "Connection is closed" diff --git a/test/dialect/test_mssql.py b/test/dialect/test_mssql.py index b38d763723..bddc887779 100644 --- a/test/dialect/test_mssql.py +++ b/test/dialect/test_mssql.py @@ -1323,8 +1323,8 @@ class ParseConnectTest(fixtures.TestBase, AssertsCompiledSQL): def test_pymssql_disconnect(self): dialect = pymssql.dialect() - for error in ['20003', '20003', '20006']: - eq_(dialect.is_disconnect('Error %s' % error, None, None), True) + for error in ['Adaptive Server connection timed out', 'message 20003']: + eq_(dialect.is_disconnect(error, None, None), True) @testing.only_on(['mssql+pyodbc', 'mssql+pymssql'], "FreeTDS specific test") def test_bad_freetds_warning(self): -- 2.47.2