mirror of
https://github.com/zerotier/ZeroTierOne.git
synced 2024-10-18 05:00:48 -07:00
131 lines
3.5 KiB
C++
131 lines
3.5 KiB
C++
#include <pqxx/transaction>
|
|
#include <pqxx/transactor>
|
|
|
|
#include "../test_helpers.hxx"
|
|
|
|
namespace
|
|
{
|
|
void test_transactor_newstyle_executes_simple_query()
|
|
{
|
|
pqxx::connection conn;
|
|
auto const r{pqxx::perform([&conn] {
|
|
return pqxx::work{conn}.exec("SELECT generate_series(1, 4)");
|
|
})};
|
|
|
|
PQXX_CHECK_EQUAL(std::size(r), 4, "Unexpected result size.");
|
|
PQXX_CHECK_EQUAL(r.columns(), 1, "Unexpected number of columns.");
|
|
PQXX_CHECK_EQUAL(r[0][0].as<int>(), 1, "Unexpected first row.");
|
|
PQXX_CHECK_EQUAL(r[3][0].as<int>(), 4, "Unexpected last row.");
|
|
}
|
|
|
|
|
|
void test_transactor_newstyle_can_return_void()
|
|
{
|
|
bool done{false};
|
|
pqxx::perform([&done]() noexcept { done = true; });
|
|
PQXX_CHECK(done, "Callback was not executed.");
|
|
}
|
|
|
|
|
|
void test_transactor_newstyle_completes_upon_success()
|
|
{
|
|
int attempts{0};
|
|
pqxx::perform([&attempts]() noexcept { attempts++; });
|
|
PQXX_CHECK_EQUAL(attempts, 1, "Successful transactor didn't run 1 time.");
|
|
}
|
|
|
|
|
|
void test_transactor_newstyle_retries_broken_connection()
|
|
{
|
|
int counter{0};
|
|
auto const &callback{[&counter] {
|
|
++counter;
|
|
if (counter == 1)
|
|
throw pqxx::broken_connection();
|
|
return counter;
|
|
}};
|
|
|
|
int const result{pqxx::perform(callback)};
|
|
PQXX_CHECK_EQUAL(result, 2, "Transactor run returned wrong result.");
|
|
PQXX_CHECK_EQUAL(counter, result, "Number of retries does not match.");
|
|
}
|
|
|
|
|
|
void test_transactor_newstyle_retries_rollback()
|
|
{
|
|
int counter{0};
|
|
auto const &callback{[&counter] {
|
|
++counter;
|
|
if (counter == 1)
|
|
throw pqxx::transaction_rollback("Simulated error");
|
|
return counter;
|
|
}};
|
|
|
|
int const result{pqxx::perform(callback)};
|
|
PQXX_CHECK_EQUAL(result, 2, "Transactor run returned wrong result.");
|
|
PQXX_CHECK_EQUAL(counter, result, "Number of retries does not match.");
|
|
}
|
|
|
|
|
|
void test_transactor_newstyle_does_not_retry_in_doubt_error()
|
|
{
|
|
int counter{0};
|
|
auto const &callback{[&counter] {
|
|
++counter;
|
|
throw pqxx::in_doubt_error("Simulated error");
|
|
}};
|
|
|
|
PQXX_CHECK_THROWS(
|
|
pqxx::perform(callback), pqxx::in_doubt_error,
|
|
"Transactor did not propagate in_doubt_error.");
|
|
PQXX_CHECK_EQUAL(counter, 1, "Transactor retried after in_doubt_error.");
|
|
}
|
|
|
|
|
|
void test_transactor_newstyle_does_not_retry_other_error()
|
|
{
|
|
int counter{0};
|
|
auto const &callback{[&counter] {
|
|
++counter;
|
|
throw std::runtime_error("Simulated error");
|
|
}};
|
|
|
|
PQXX_CHECK_THROWS(
|
|
pqxx::perform(callback), std::runtime_error,
|
|
"Transactor did not propagate std exception.");
|
|
PQXX_CHECK_EQUAL(counter, 1, "Transactor retried after std exception.");
|
|
}
|
|
|
|
|
|
void test_transactor_newstyle_repeats_up_to_given_number_of_attempts()
|
|
{
|
|
int const attempts{5};
|
|
int counter{0};
|
|
auto const &callback{[&counter] {
|
|
++counter;
|
|
throw pqxx::transaction_rollback("Simulated error");
|
|
}};
|
|
|
|
PQXX_CHECK_THROWS(
|
|
pqxx::perform(callback, attempts), pqxx::transaction_rollback,
|
|
"Not propagating original exception.");
|
|
PQXX_CHECK_EQUAL(counter, attempts, "Number of retries does not match.");
|
|
}
|
|
|
|
|
|
void test_transactor()
|
|
{
|
|
test_transactor_newstyle_executes_simple_query();
|
|
test_transactor_newstyle_can_return_void();
|
|
test_transactor_newstyle_completes_upon_success();
|
|
test_transactor_newstyle_retries_broken_connection();
|
|
test_transactor_newstyle_retries_rollback();
|
|
test_transactor_newstyle_does_not_retry_in_doubt_error();
|
|
test_transactor_newstyle_does_not_retry_other_error();
|
|
test_transactor_newstyle_repeats_up_to_given_number_of_attempts();
|
|
}
|
|
|
|
|
|
PQXX_REGISTER_TEST(test_transactor);
|
|
} // namespace
|