psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-22 17:06:33 +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() cur = conn.cursor()
try: try:
cur.execute("select pg_terminate_backend(pg_backend_pid())") 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: except psycopg2.OperationalError, e:
if e.pgcode != psycopg2.errorcodes.ADMIN_SHUTDOWN: if e.pgcode != psycopg2.errorcodes.ADMIN_SHUTDOWN:
raise 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) self.assertEqual(conn.closed, 2)
conn.close() conn.close()