psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-22 08:56:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixed exception handling in disconnection test

Browse Source 
OperationalError is DatabaseError's subclass.
This commit is contained in:
Daniele Varrazzo 2013-03-16 21:42:41 +00:00
parent c5de7b5ac6
commit af18c29f0b
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_connection.py
View File

@ -75,12 +75,14 @@ class ConnectionTests(unittest.TestCase):
cur = conn.cursor()
try:
cur.execute("select pg_terminate_backend(pg_backend_pid())")
except psycopg2.DatabaseError, e:
# curiously in green mode we get a DatabaseError without pgcode
pass
except psycopg2.OperationalError, e:
if e.pgcode != psycopg2.errorcodes.ADMIN_SHUTDOWN:
raise
except psycopg2.DatabaseError, e:
# curiously when disconnected in green mode we get a DatabaseError
# without pgcode.
if e.pgcode is not None:
raise
self.assertEqual(conn.closed, 2)
conn.close()