mirror of
https://github.com/codership/wsrep-lib.git
synced 2025-04-19 21:02:17 +03:00
219 lines
8.3 KiB
C++
219 lines
8.3 KiB
C++
//
|
|
// Copyright (C) 2018 Codership Oy <info@codership.com>
|
|
//
|
|
|
|
#include "mock_server_state.hpp"
|
|
|
|
#include <boost/test/unit_test.hpp>
|
|
|
|
namespace
|
|
{
|
|
struct applying_server_fixture
|
|
{
|
|
applying_server_fixture()
|
|
: ss("s1", "s1",
|
|
wsrep::server_state::rm_sync)
|
|
, cc(ss,
|
|
wsrep::client_id(1),
|
|
wsrep::client_state::m_high_priority)
|
|
, hps(ss, &cc, false)
|
|
, ws_handle(1, (void*)1)
|
|
, ws_meta(wsrep::gtid(wsrep::id("1"), wsrep::seqno(1)),
|
|
wsrep::stid(wsrep::id("1"), 1, 1),
|
|
wsrep::seqno(0),
|
|
wsrep::provider::flag::start_transaction |
|
|
wsrep::provider::flag::commit)
|
|
{
|
|
cc.open(cc.id());
|
|
cc.before_command();
|
|
}
|
|
wsrep::mock_server_state ss;
|
|
wsrep::mock_client cc;
|
|
wsrep::mock_high_priority_service hps;
|
|
wsrep::ws_handle ws_handle;
|
|
wsrep::ws_meta ws_meta;
|
|
};
|
|
|
|
struct sst_first_server_fixture : applying_server_fixture
|
|
{
|
|
sst_first_server_fixture()
|
|
: applying_server_fixture()
|
|
{
|
|
ss.sst_before_init_ = true;
|
|
}
|
|
};
|
|
|
|
struct init_first_server_fixture : applying_server_fixture
|
|
{
|
|
init_first_server_fixture()
|
|
: applying_server_fixture()
|
|
{
|
|
ss.sst_before_init_ = false;
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
// Test on_apply() method for 1pc
|
|
BOOST_FIXTURE_TEST_CASE(server_state_applying_1pc,
|
|
applying_server_fixture)
|
|
{
|
|
char buf[1] = { 1 };
|
|
BOOST_REQUIRE(ss.on_apply(hps, ws_handle, ws_meta,
|
|
wsrep::const_buffer(buf, 1)) == 0);
|
|
const wsrep::transaction& txc(cc.transaction());
|
|
// ::abort();
|
|
BOOST_REQUIRE_MESSAGE(
|
|
txc.state() == wsrep::transaction::s_committed,
|
|
"Transaction state " << txc.state() << " not committed");
|
|
}
|
|
|
|
// Test on_apply() method for 2pc
|
|
BOOST_FIXTURE_TEST_CASE(server_state_applying_2pc,
|
|
applying_server_fixture)
|
|
{
|
|
char buf[1] = { 1 };
|
|
BOOST_REQUIRE(ss.on_apply(hps, ws_handle, ws_meta,
|
|
wsrep::const_buffer(buf, 1)) == 0);
|
|
const wsrep::transaction& txc(cc.transaction());
|
|
BOOST_REQUIRE(txc.state() == wsrep::transaction::s_committed);
|
|
}
|
|
|
|
// Test on_apply() method for 1pc transaction which
|
|
// fails applying and rolls back
|
|
BOOST_FIXTURE_TEST_CASE(server_state_applying_1pc_rollback,
|
|
applying_server_fixture)
|
|
{
|
|
hps.fail_next_applying_ = true;
|
|
char buf[1] = { 1 };
|
|
BOOST_REQUIRE(ss.on_apply(hps, ws_handle, ws_meta,
|
|
wsrep::const_buffer(buf, 1)) == 1);
|
|
const wsrep::transaction& txc(cc.transaction());
|
|
BOOST_REQUIRE(txc.state() == wsrep::transaction::s_aborted);
|
|
}
|
|
|
|
// Test on_apply() method for 2pc transaction which
|
|
// fails applying and rolls back
|
|
BOOST_FIXTURE_TEST_CASE(server_state_applying_2pc_rollback,
|
|
applying_server_fixture)
|
|
{
|
|
hps.fail_next_applying_ = true;
|
|
char buf[1] = { 1 };
|
|
BOOST_REQUIRE(ss.on_apply(hps, ws_handle, ws_meta,
|
|
wsrep::const_buffer(buf, 1)) == 1);
|
|
const wsrep::transaction& txc(cc.transaction());
|
|
BOOST_REQUIRE(txc.state() == wsrep::transaction::s_aborted);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(server_state_streaming)
|
|
{
|
|
wsrep::mock_server_state ss("s1", "s1",
|
|
wsrep::server_state::rm_sync);
|
|
wsrep::mock_client cc(ss,
|
|
wsrep::client_id(1),
|
|
wsrep::client_state::m_high_priority);
|
|
cc.debug_log_level(1);
|
|
wsrep::mock_high_priority_service hps(ss, &cc, false);
|
|
wsrep::ws_handle ws_handle(1, (void*)1);
|
|
wsrep::ws_meta ws_meta(wsrep::gtid(wsrep::id("1"), wsrep::seqno(1)),
|
|
wsrep::stid(wsrep::id("1"), 1, 1),
|
|
wsrep::seqno(0),
|
|
wsrep::provider::flag::start_transaction);
|
|
cc.open(cc.id());
|
|
BOOST_REQUIRE(cc.before_command() == 0);
|
|
BOOST_REQUIRE(ss.on_apply(hps, ws_handle, ws_meta,
|
|
wsrep::const_buffer("1", 1)) == 0);
|
|
BOOST_REQUIRE(ss.find_streaming_applier(
|
|
ws_meta.server_id(), ws_meta.transaction_id()));
|
|
ws_meta = wsrep::ws_meta(wsrep::gtid(wsrep::id("1"), wsrep::seqno(2)),
|
|
wsrep::stid(wsrep::id("1"), 1, 1),
|
|
wsrep::seqno(1),
|
|
0);
|
|
BOOST_REQUIRE(ss.on_apply(hps, ws_handle, ws_meta,
|
|
wsrep::const_buffer("1", 1)) == 0);
|
|
ws_meta = wsrep::ws_meta(wsrep::gtid(wsrep::id("1"), wsrep::seqno(2)),
|
|
wsrep::stid(wsrep::id("1"), 1, 1),
|
|
wsrep::seqno(1),
|
|
wsrep::provider::flag::commit);
|
|
BOOST_REQUIRE(ss.on_apply(hps, ws_handle, ws_meta,
|
|
wsrep::const_buffer("1", 1)) == 0);
|
|
BOOST_REQUIRE(ss.find_streaming_applier(
|
|
ws_meta.server_id(), ws_meta.transaction_id()) == 0);
|
|
}
|
|
|
|
|
|
BOOST_AUTO_TEST_CASE(server_state_state_strings)
|
|
{
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
wsrep::server_state::s_disconnected) == "disconnected");
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
wsrep::server_state::s_initializing) == "initializing");
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
wsrep::server_state::s_initialized) == "initialized");
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
wsrep::server_state::s_connected) == "connected");
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
wsrep::server_state::s_joiner) == "joiner");
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
wsrep::server_state::s_joined) == "joined");
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
wsrep::server_state::s_donor) == "donor");
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
wsrep::server_state::s_synced) == "synced");
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
wsrep::server_state::s_disconnecting) == "disconnecting");
|
|
BOOST_REQUIRE(wsrep::to_string(
|
|
static_cast<enum wsrep::server_state::state>(0xff)) == "unknown");
|
|
}
|
|
|
|
BOOST_FIXTURE_TEST_CASE(server_state_sst_first_boostrap,
|
|
sst_first_server_fixture)
|
|
{
|
|
wsrep::id cluster_id("1");
|
|
wsrep::gtid state_id(cluster_id, wsrep::seqno(0));
|
|
std::vector<wsrep::view::member> members;
|
|
members.push_back(wsrep::view::member(wsrep::id("s1"), "name", ""));
|
|
wsrep::view bootstrap_view(state_id,
|
|
wsrep::seqno(1),
|
|
wsrep::view::primary,
|
|
0,
|
|
0,
|
|
1,
|
|
members);
|
|
BOOST_REQUIRE(ss.connect("cluster", "local", "0", false) == 0);
|
|
ss.on_connect(wsrep::gtid(cluster_id, wsrep::seqno(0)));
|
|
BOOST_REQUIRE(ss.state() == wsrep::server_state::s_connected);
|
|
ss.sync_point_enabled_ = "on_view_wait_initialized";
|
|
ss.sync_point_action_ = ss.spa_initialize;
|
|
ss.on_view(bootstrap_view);
|
|
ss.sync_point_enabled_ = "";
|
|
BOOST_REQUIRE(ss.state() == wsrep::server_state::s_joined);
|
|
ss.on_sync();
|
|
BOOST_REQUIRE(ss.state() == wsrep::server_state::s_synced);
|
|
}
|
|
|
|
BOOST_FIXTURE_TEST_CASE(server_state_init_first_boostrap,
|
|
init_first_server_fixture)
|
|
{
|
|
wsrep::id cluster_id("1");
|
|
wsrep::gtid state_id(cluster_id, wsrep::seqno(0));
|
|
std::vector<wsrep::view::member> members;
|
|
members.push_back(wsrep::view::member(wsrep::id("s1"), "name", ""));
|
|
wsrep::view bootstrap_view(state_id,
|
|
wsrep::seqno(1),
|
|
wsrep::view::primary,
|
|
0,
|
|
0,
|
|
1,
|
|
members);
|
|
ss.initialized();
|
|
BOOST_REQUIRE(ss.state() == wsrep::server_state::s_initialized);
|
|
BOOST_REQUIRE(ss.connect("cluster", "local", "0", false) == 0);
|
|
ss.on_connect(wsrep::gtid(cluster_id, wsrep::seqno(0)));
|
|
BOOST_REQUIRE(ss.state() == wsrep::server_state::s_connected);
|
|
ss.on_view(bootstrap_view);
|
|
BOOST_REQUIRE(ss.state() == wsrep::server_state::s_joined);
|
|
ss.on_sync();
|
|
BOOST_REQUIRE(ss.state() == wsrep::server_state::s_synced);
|
|
}
|