Adding upstream version 3.1.7.upstream/3.1.7upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 34 insertions, 0 deletions

diff --git a/tests/crdb/test_no_crdb.py b/tests/crdb/test_no_crdb.py
new file mode 100644
index 0000000..df43f3b
--- /dev/null
+++ b/tests/crdb/test_no_crdb.py
@@ -0,0 +1,34 @@
+from psycopg.pq import TransactionStatus
+from psycopg.crdb import CrdbConnection
+
+import pytest
+
+pytestmark = pytest.mark.crdb("skip")
+
+
+def test_is_crdb(conn):
+    assert not CrdbConnection.is_crdb(conn)
+    assert not CrdbConnection.is_crdb(conn.pgconn)
+
+
+def test_tpc_on_pg_connection(conn, tpc):
+    xid = conn.xid(1, "gtrid", "bqual")
+    assert conn.info.transaction_status == TransactionStatus.IDLE
+
+    conn.tpc_begin(xid)
+    assert conn.info.transaction_status == TransactionStatus.INTRANS
+
+    cur = conn.cursor()
+    cur.execute("insert into test_tpc values ('test_tpc_commit')")
+    assert tpc.count_xacts() == 0
+    assert tpc.count_test_records() == 0
+
+    conn.tpc_prepare()
+    assert conn.info.transaction_status == TransactionStatus.IDLE
+    assert tpc.count_xacts() == 1
+    assert tpc.count_test_records() == 0
+
+    conn.tpc_commit()
+    assert conn.info.transaction_status == TransactionStatus.IDLE
+    assert tpc.count_xacts() == 0
+    assert tpc.count_test_records() == 1