/usr/lib/python3/dist-packages/psycopg2/tests/dbapi20_tpc.py is in python3-psycopg2 2.5.4+dfsg-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 | """ Python DB API 2.0 driver Two Phase Commit compliance test suite.
"""
import unittest
class TwoPhaseCommitTests(unittest.TestCase):
driver = None
def connect(self):
"""Make a database connection."""
raise NotImplementedError
_last_id = 0
_global_id_prefix = "dbapi20_tpc:"
def make_xid(self, con):
id = TwoPhaseCommitTests._last_id
TwoPhaseCommitTests._last_id += 1
return con.xid(42, "%s%d" % (self._global_id_prefix, id), "qualifier")
def test_xid(self):
con = self.connect()
try:
xid = con.xid(42, "global", "bqual")
except self.driver.NotSupportedError:
self.fail("Driver does not support transaction IDs.")
self.assertEqual(xid[0], 42)
self.assertEqual(xid[1], "global")
self.assertEqual(xid[2], "bqual")
# Try some extremes for the transaction ID:
xid = con.xid(0, "", "")
self.assertEqual(tuple(xid), (0, "", ""))
xid = con.xid(0x7fffffff, "a" * 64, "b" * 64)
self.assertEqual(tuple(xid), (0x7fffffff, "a" * 64, "b" * 64))
def test_tpc_begin(self):
con = self.connect()
try:
xid = self.make_xid(con)
try:
con.tpc_begin(xid)
except self.driver.NotSupportedError:
self.fail("Driver does not support tpc_begin()")
finally:
con.close()
def test_tpc_commit_without_prepare(self):
con = self.connect()
try:
xid = self.make_xid(con)
con.tpc_begin(xid)
cursor = con.cursor()
cursor.execute("SELECT 1")
con.tpc_commit()
finally:
con.close()
def test_tpc_rollback_without_prepare(self):
con = self.connect()
try:
xid = self.make_xid(con)
con.tpc_begin(xid)
cursor = con.cursor()
cursor.execute("SELECT 1")
con.tpc_rollback()
finally:
con.close()
def test_tpc_commit_with_prepare(self):
con = self.connect()
try:
xid = self.make_xid(con)
con.tpc_begin(xid)
cursor = con.cursor()
cursor.execute("SELECT 1")
con.tpc_prepare()
con.tpc_commit()
finally:
con.close()
def test_tpc_rollback_with_prepare(self):
con = self.connect()
try:
xid = self.make_xid(con)
con.tpc_begin(xid)
cursor = con.cursor()
cursor.execute("SELECT 1")
con.tpc_prepare()
con.tpc_rollback()
finally:
con.close()
def test_tpc_begin_in_transaction_fails(self):
con = self.connect()
try:
xid = self.make_xid(con)
cursor = con.cursor()
cursor.execute("SELECT 1")
self.assertRaises(self.driver.ProgrammingError,
con.tpc_begin, xid)
finally:
con.close()
def test_tpc_begin_in_tpc_transaction_fails(self):
con = self.connect()
try:
xid = self.make_xid(con)
cursor = con.cursor()
cursor.execute("SELECT 1")
self.assertRaises(self.driver.ProgrammingError,
con.tpc_begin, xid)
finally:
con.close()
def test_commit_in_tpc_fails(self):
# calling commit() within a TPC transaction fails with
# ProgrammingError.
con = self.connect()
try:
xid = self.make_xid(con)
con.tpc_begin(xid)
self.assertRaises(self.driver.ProgrammingError, con.commit)
finally:
con.close()
def test_rollback_in_tpc_fails(self):
# calling rollback() within a TPC transaction fails with
# ProgrammingError.
con = self.connect()
try:
xid = self.make_xid(con)
con.tpc_begin(xid)
self.assertRaises(self.driver.ProgrammingError, con.rollback)
finally:
con.close()
|