First Commit

ext/libpqxx-7.7.3/test/unit/test_subtransaction.cxx

@@ -0,0 +1,78 @@
+#include <pqxx/subtransaction>
+#include <pqxx/transaction>
+
+#include "../test_helpers.hxx"
+
+namespace
+{
+void make_table(pqxx::transaction_base &trans)
+{
+  trans.exec0("CREATE TEMP TABLE foo (x INTEGER)");
+}
+
+
+void insert_row(pqxx::transaction_base &trans)
+{
+  trans.exec0("INSERT INTO foo(x) VALUES (1)");
+}
+
+
+int count_rows(pqxx::transaction_base &trans)
+{
+  return trans.query_value<int>("SELECT count(*) FROM foo");
+}
+
+
+void test_subtransaction_commits_if_commit_called(pqxx::connection &conn)
+{
+  pqxx::work trans(conn);
+  make_table(trans);
+  {
+    pqxx::subtransaction sub(trans);
+    insert_row(sub);
+    sub.commit();
+  }
+  PQXX_CHECK_EQUAL(
+    count_rows(trans), 1, "Work done in committed subtransaction was lost.");
+}
+
+
+void test_subtransaction_aborts_if_abort_called(pqxx::connection &conn)
+{
+  pqxx::work trans(conn);
+  make_table(trans);
+  {
+    pqxx::subtransaction sub(trans);
+    insert_row(sub);
+    sub.abort();
+  }
+  PQXX_CHECK_EQUAL(
+    count_rows(trans), 0, "Aborted subtransaction was not rolled back.");
+}
+
+
+void test_subtransaction_aborts_implicitly(pqxx::connection &conn)
+{
+  pqxx::work trans(conn);
+  make_table(trans);
+  {
+    pqxx::subtransaction sub(trans);
+    insert_row(sub);
+  }
+  PQXX_CHECK_EQUAL(
+    count_rows(trans), 0,
+    "Uncommitted subtransaction was not rolled back uring destruction.");
+}
+
+
+void test_subtransaction()
+{
+  pqxx::connection conn;
+  test_subtransaction_commits_if_commit_called(conn);
+  test_subtransaction_aborts_if_abort_called(conn);
+  test_subtransaction_aborts_implicitly(conn);
+}
+
+
+PQXX_REGISTER_TEST(test_subtransaction);
+} // namespace