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
|
#include "test_helpers.hxx"
using namespace PGSTD;
using namespace pqxx;
// Test program for libpqxx. Verify abort behaviour of transactor.
//
// The program will attempt to add an entry to a table called "pqxxevents",
// with a key column called "year"--and then abort the change.
//
// Note for the superstitious: the numbering for this test program is pure
// coincidence.
namespace
{
// Let's take a boring year that is not going to be in the "pqxxevents" table
const unsigned int BoringYear = 1977;
// Count events and specifically events occurring in Boring Year, leaving the
// former count in the result pair's first member, and the latter in second.
class CountEvents : public transactor<>
{
string m_Table;
pair<int, int> &m_Results;
public:
CountEvents(string Table, pair<int,int> &Results) :
transactor<>("CountEvents"), m_Table(Table), m_Results(Results) {}
void operator()(argument_type &T)
{
const string CountQuery = "SELECT count(*) FROM " + m_Table;
result R;
R = T.exec(CountQuery);
R.at(0).at(0).to(m_Results.first);
R = T.exec(CountQuery + " WHERE year=" + to_string(BoringYear));
R.at(0).at(0).to(m_Results.second);
}
};
struct deliberate_error : exception
{
};
class FailedInsert : public transactor<>
{
string m_Table;
public:
FailedInsert(string Table) :
transactor<>("FailedInsert"),
m_Table(Table)
{
}
void operator()(argument_type &T)
{
result R( T.exec("INSERT INTO " + m_Table + " VALUES (" +
to_string(BoringYear) + ", "
"'yawn')") );
PQXX_CHECK_EQUAL(R.affected_rows(), 1u, "Bad affected_rows().");
throw deliberate_error();
}
void on_abort(const char Reason[]) throw ()
{
pqxx::test::expected_exception(Name() + " failed: " + Reason);
}
};
void test_013(transaction_base &T)
{
connection_base &C(T.conn());
T.abort();
{
work T2(C);
test::create_pqxxevents(T2);
T2.commit();
}
const string Table = "pqxxevents";
pair<int,int> Before;
C.perform(CountEvents(Table, Before));
PQXX_CHECK_EQUAL(
Before.second,
0,
"Already have event for " + to_string(BoringYear) + "--can't test.");
const FailedInsert DoomedTransaction(Table);
quiet_errorhandler d(C);
PQXX_CHECK_THROWS(
C.perform(DoomedTransaction),
deliberate_error,
"Failing transactor failed to throw correct exception.");
pair<int,int> After;
C.perform(CountEvents(Table, After));
PQXX_CHECK_EQUAL(
After.first,
Before.first,
"abort() didn't reset event count.");
PQXX_CHECK_EQUAL(
After.second,
Before.second,
"abort() didn't reset event count for " + to_string(BoringYear));
}
} // namespace
PQXX_REGISTER_TEST_T(test_013, nontransaction)
|