psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-11-04 09:47:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1105 from rafiss/with-block-connection-test

Browse Source 
Handle failure in setup of ConnectionTwoPhaseTest
This commit is contained in:
Daniele Varrazzo 2020-05-30 04:43:26 +12:00 committed by GitHub
commit 779a1370ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -913,9 +913,11 @@ class ConnectionTwoPhaseTests(ConnectingTestCase):
cur.execute("DROP TABLE test_tpc;") cur.execute("DROP TABLE test_tpc;")
except psycopg2.ProgrammingError: except psycopg2.ProgrammingError:
cnn.rollback() cnn.rollback()
cur.execute("CREATE TABLE test_tpc (data text);") try:
cnn.commit() cur.execute("CREATE TABLE test_tpc (data text);")
cnn.close() cnn.commit()
finally:
cnn.close()
def count_xacts(self): def count_xacts(self):
"""Return the number of prepared xacts currently in the test db.""" """Return the number of prepared xacts currently in the test db."""