hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity
Files
generic
oceanbase/unittest/election/test_election_priority.cpp
gm 4a92b6d7df reformat source code 
according to code styles, 'AccessModifierOffset' should be -2.
2021-06-17 10:40:36 +08:00

480 lines
14 KiB
C++
Raw Permalink Blame History

/**
* Copyright (c) 2021 OceanBase
* OceanBase CE is licensed under Mulan PubL v2.
* You can use this software according to the terms and conditions of the Mulan PubL v2.
* You may obtain a copy of Mulan PubL v2 at:
* http://license.coscl.org.cn/MulanPubL-2.0
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PubL v2 for more details.
*/
#include <gtest/gtest.h>
#include "share/ob_errno.h"
#include "lib/oblog/ob_log.h"
#include "share/ob_define.h"
#include "lib/container/ob_array.h"
#include "election/ob_election_msg.h"
#include "election/ob_election_priority.h"
#include "election/ob_election_msg_pool.h"
#include "election/ob_election_time_def.h"
#include "common/ob_clock_generator.h"
#include "lib/container/ob_array_serialization.h"
#include "mock_election.h"
namespace oceanbase {
namespace unittest {
using namespace election;
using namespace common;
class TestObElectionPriority : public ::testing::Test {
public:
TestObElectionPriority()
{}
~TestObElectionPriority()
{}
virtual void SetUp()
{
init();
}
virtual void TearDown()
{}
private:
int init();
public:
int create_deprepare_msg(common::ObSArray<ObElectionPriority>& priority_array, ObElectionVoteMsgPool& msg_pool);
ObElectionVoteMsgPool msg_pool_;
MockElection election_;
int64_t t1_;
int64_t cur_ts_;
int64_t lease_time_;
};
int TestObElectionPriority::init()
{
t1_ = ((ObClockGenerator::getClock() + T_ELECT2) / T_ELECT2) * T_ELECT2;
cur_ts_ = t1_ + 10;
;
lease_time_ = T_LEADER_LEASE_EXTENDS * T_ELECT2;
return OB_SUCCESS;
}
int TestObElectionPriority::create_deprepare_msg(
ObSArray<ObElectionPriority>& priority_array, ObElectionVoteMsgPool& msg_pool)
{
msg_pool.reset();
msg_pool_.init(ObAddr(ObAddr::IPV4, "127.0.0.1", 123456), &election_);
for (int32_t i = 0; i < priority_array.count(); i++) {
ObAddr self;
EXPECT_EQ(true, self.set_ip_addr("127.0.0.1", 34506 + i));
ObElectionMsgDEPrepare msg(priority_array.at(i), t1_, cur_ts_, self, T_LEADER_LEASE_EXTENDS * T_ELECT2);
msg_pool.store(msg);
}
return OB_SUCCESS;
}
TEST_F(TestObElectionPriority, normal_decentrialize_msg)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(true, 1002, 102, 0);
ObElectionPriority p3;
p3.init(true, 1003, 103, 0);
ObElectionPriority p4;
p4.init(true, 1004, 104, 0);
ObElectionPriority p5;
p5.init(true, 1005, 105, 0);
ObElectionPriority p6;
p6.init(true, 1006, 106, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p5));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p6));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
EXPECT_EQ(0, priority.compare_with_accurate_logid(p6));
}
}
TEST_F(TestObElectionPriority, candidate_is_false)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(false, 1002, 102, 0);
ObElectionPriority p3;
p3.init(true, 1003, 103, 0);
ObElectionPriority p4;
p4.init(false, 1004, 104, 0);
ObElectionPriority p5;
p5.init(true, 1005, 105, 0);
ObElectionPriority p6;
p6.init(false, 1006, 106, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p5));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p6));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
EXPECT_EQ(0, priority.compare_with_accurate_logid(p5));
}
}
TEST_F(TestObElectionPriority, log_id_compare)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(false, 1002, 102, 0);
ObElectionPriority p3;
p3.init(true, 1003, 103, 0);
ObElectionPriority p4;
p4.init(true, 1005, 104, 0);
ObElectionPriority p5;
p5.init(true, 1005, 105, 0);
ObElectionPriority p6;
p6.init(false, 1006, 106, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p5));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p6));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
EXPECT_EQ(0, priority.compare_with_accurate_logid(p5));
}
}
TEST_F(TestObElectionPriority, data_version_compare)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(true, 1002, 102, 0);
ObElectionPriority p3;
p3.init(true, 1005, 103, 0);
ObElectionPriority p4;
p4.init(true, 1005, 104, 0);
ObElectionPriority p5;
p5.init(true, 1005, 105, 0);
ObElectionPriority p6;
p6.init(true, 1005, 106, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p5));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p6));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
EXPECT_EQ(0, priority.compare_with_accurate_logid(p6));
}
}
TEST_F(TestObElectionPriority, no_majority)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(false, 1002, 102, 0);
ObElectionPriority p3;
p3.init(false, 1003, 103, 0);
ObElectionPriority p4;
p4.init(false, 1005, 104, 0);
ObElectionPriority p5;
p5.init(false, 1005, 105, 0);
ObElectionPriority p6;
p6.init(false, 1006, 106, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p5));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p6));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
}
}
TEST_F(TestObElectionPriority, no_all_message)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(false, 1002, 102, 0);
ObElectionPriority p3;
p3.init(false, 1003, 103, 0);
ObElectionPriority p4;
p4.init(false, 1005, 104, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
}
}
TEST_F(TestObElectionPriority, no_all_message2)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(true, 1002, 102, 0);
ObElectionPriority p3;
p3.init(false, 1003, 103, 0);
ObElectionPriority p4;
p4.init(false, 1005, 104, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
}
}
TEST_F(TestObElectionPriority, no_enough_message)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(true, 1002, 102, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
}
}
TEST_F(TestObElectionPriority, no_all_message_but_false)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(true, 1002, 102, 0);
ObElectionPriority p3;
p3.init(false, 1003, 103, 0);
ObElectionPriority p4;
p4.init(false, 1005, 104, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
}
}
TEST_F(TestObElectionPriority, no_all_message_candidate)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(false, 1002, 102, 0);
ObElectionPriority p3;
p3.init(true, 1003, 103, 0);
ObElectionPriority p4;
p4.init(false, 1005, 104, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
EXPECT_EQ(0, priority.compare_with_accurate_logid(p3));
}
}
TEST_F(TestObElectionPriority, no_all_message_candidate_log_id)
{
ObElectionPriority priority;
ObAddr addr;
ObSArray<ObElectionPriority> priority_array;
priority_array.reset();
{
ObElectionPriority p0;
p0.init(true, 1000, 100, 0);
ObElectionPriority p1;
p1.init(true, 1001, 101, 0);
ObElectionPriority p2;
p2.init(false, 1002, 102, 0);
ObElectionPriority p3;
p3.init(true, 1005, 103, 0);
ObElectionPriority p4;
p4.init(true, 1005, 104, 0);
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p0));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p1));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p2));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p3));
EXPECT_EQ(OB_SUCCESS, priority_array.push_back(p4));
create_deprepare_msg(priority_array, msg_pool_);
EXPECT_EQ(
OB_SUCCESS, msg_pool_.get_decentralized_candidate(addr, priority, priority_array.count(), t1_, lease_time_));
EXPECT_EQ(0, priority.compare_with_accurate_logid(p4));
}
}
TEST_F(TestObElectionPriority, locality)
{
ObElectionPriority p0;
p0.init(true, 1000, 10000, 1);
ObElectionPriority p1;
p1 = p0;
EXPECT_EQ(0, p1.compare_with_accurate_logid(p0));
}
} // namespace unittest
} // namespace oceanbase
int main(int argc, char** argv)
{
int ret = -1;
oceanbase::election::ASYNC_LOG_INIT("test_election_priority.log", OB_LOG_LEVEL_INFO, true);
if (OB_FAIL(oceanbase::common::ObClockGenerator::init())) {
ELECT_LOG(WARN, "clock generator init error.", K(ret));
} else {
testing::InitGoogleTest(&argc, argv);
ret = RUN_ALL_TESTS();
}
oceanbase::election::ASYNC_LOG_DESTROY();
(void)oceanbase::common::ObClockGenerator::destroy();
return ret;
}
Reference in New Issue View Git Blame Copy Permalink