init push
This commit is contained in:
oceanbase-admin
201
unittest/clog/test_ob_log_broadcast_info_mgr.cpp
Normal file
201
unittest/clog/test_ob_log_broadcast_info_mgr.cpp
Normal file
@ -0,0 +1,201 @@
|
||||
/**
|
||||
* 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 "clog/ob_log_broadcast_info_mgr.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "clog_mock_container/mock_log_membership_mgr.h"
|
||||
|
||||
namespace oceanbase {
|
||||
using namespace common;
|
||||
using namespace clog;
|
||||
namespace unittest {
|
||||
class TestMockObLogMembershipMgr : public MockObLogMembershipMgr {
|
||||
public:
|
||||
void set_replica_num(const int64_t replica_num)
|
||||
{
|
||||
replica_num_ = replica_num;
|
||||
}
|
||||
int64_t get_replica_num() const
|
||||
{
|
||||
return replica_num_;
|
||||
}
|
||||
void set_member_list(const common::ObMemberList& member_list)
|
||||
{
|
||||
member_list_.deep_copy(member_list);
|
||||
}
|
||||
const common::ObMemberList& get_curr_member_list() const
|
||||
{
|
||||
return member_list_;
|
||||
}
|
||||
|
||||
private:
|
||||
int64_t replica_num_;
|
||||
common::ObMemberList member_list_;
|
||||
};
|
||||
|
||||
class ObLogBroadcastInfoMgrTest : public testing::Test {
|
||||
public:
|
||||
void SetUp();
|
||||
void TearDown();
|
||||
|
||||
TestMockObLogMembershipMgr mm_;
|
||||
ObAddr server1_;
|
||||
ObAddr server2_;
|
||||
ObAddr server3_;
|
||||
ObAddr server4_;
|
||||
ObAddr server5_;
|
||||
ObMemberList member_list_;
|
||||
|
||||
ObLogBroadcastInfoMgr broadcast_info_mgr_;
|
||||
};
|
||||
|
||||
void ObLogBroadcastInfoMgrTest::SetUp()
|
||||
{
|
||||
const int64_t replica_num = 5;
|
||||
|
||||
common::ObPartitionKey partition_key(1099511627777, 1, 1);
|
||||
|
||||
server1_.set_ip_addr("127.0.0.1", 1234);
|
||||
server2_.set_ip_addr("127.0.0.2", 1234);
|
||||
server3_.set_ip_addr("127.0.0.3", 1234);
|
||||
server4_.set_ip_addr("127.0.0.4", 1234);
|
||||
server5_.set_ip_addr("127.0.0.5", 1234);
|
||||
|
||||
member_list_.add_server(server1_);
|
||||
member_list_.add_server(server2_);
|
||||
member_list_.add_server(server3_);
|
||||
member_list_.add_server(server4_);
|
||||
member_list_.add_server(server5_);
|
||||
|
||||
mm_.set_replica_num(replica_num);
|
||||
mm_.set_member_list(member_list_);
|
||||
|
||||
broadcast_info_mgr_.init(partition_key, &mm_);
|
||||
}
|
||||
|
||||
void ObLogBroadcastInfoMgrTest::TearDown()
|
||||
{
|
||||
broadcast_info_mgr_.destroy();
|
||||
}
|
||||
|
||||
TEST_F(ObLogBroadcastInfoMgrTest, test1)
|
||||
{
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server1_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server2_, REPLICA_TYPE_FULL, 20));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server3_, REPLICA_TYPE_FULL, 30));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server4_, REPLICA_TYPE_LOGONLY, 5));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server5_, REPLICA_TYPE_LOGONLY, 50));
|
||||
|
||||
uint64_t log_id = OB_INVALID_ID;
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.get_recyclable_log_id(log_id));
|
||||
EXPECT_EQ(10, log_id);
|
||||
}
|
||||
|
||||
TEST_F(ObLogBroadcastInfoMgrTest, test2)
|
||||
{
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server1_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server2_, REPLICA_TYPE_FULL, 20));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server3_, REPLICA_TYPE_FULL, 30));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server4_, REPLICA_TYPE_FULL, 40));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server5_, REPLICA_TYPE_LOGONLY, 50));
|
||||
|
||||
uint64_t log_id = OB_INVALID_ID;
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.get_recyclable_log_id(log_id));
|
||||
EXPECT_EQ(20, log_id);
|
||||
}
|
||||
|
||||
TEST_F(ObLogBroadcastInfoMgrTest, test3)
|
||||
{
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server1_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server2_, REPLICA_TYPE_FULL, 20));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server3_, REPLICA_TYPE_LOGONLY, 30));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server4_, REPLICA_TYPE_LOGONLY, 40));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server5_, REPLICA_TYPE_LOGONLY, 50));
|
||||
|
||||
uint64_t log_id = OB_INVALID_ID;
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.get_recyclable_log_id(log_id));
|
||||
EXPECT_EQ(10, log_id);
|
||||
}
|
||||
|
||||
TEST_F(ObLogBroadcastInfoMgrTest, test4)
|
||||
{
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server1_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server2_, REPLICA_TYPE_FULL, 20));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server3_, REPLICA_TYPE_FULL, 30));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server4_, REPLICA_TYPE_FULL, 40));
|
||||
|
||||
uint64_t log_id = OB_INVALID_ID;
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.get_recyclable_log_id(log_id));
|
||||
EXPECT_EQ(20, log_id);
|
||||
}
|
||||
|
||||
TEST_F(ObLogBroadcastInfoMgrTest, test5)
|
||||
{
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server1_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server2_, REPLICA_TYPE_FULL, 20));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server3_, REPLICA_TYPE_FULL, 30));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server4_, REPLICA_TYPE_LOGONLY, 40));
|
||||
|
||||
uint64_t log_id = OB_INVALID_ID;
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.get_recyclable_log_id(log_id));
|
||||
EXPECT_EQ(10, log_id);
|
||||
}
|
||||
|
||||
TEST_F(ObLogBroadcastInfoMgrTest, test6)
|
||||
{
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server1_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server2_, REPLICA_TYPE_FULL, 20));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server3_, REPLICA_TYPE_LOGONLY, 30));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server4_, REPLICA_TYPE_LOGONLY, 40));
|
||||
|
||||
uint64_t log_id = OB_INVALID_ID;
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.get_recyclable_log_id(log_id));
|
||||
EXPECT_EQ(0, log_id);
|
||||
}
|
||||
|
||||
TEST_F(ObLogBroadcastInfoMgrTest, test7)
|
||||
{
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server1_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server2_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server3_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server4_, REPLICA_TYPE_FULL, 10));
|
||||
|
||||
uint64_t log_id = OB_INVALID_ID;
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.get_recyclable_log_id(log_id));
|
||||
EXPECT_EQ(10, log_id);
|
||||
}
|
||||
|
||||
TEST_F(ObLogBroadcastInfoMgrTest, test8)
|
||||
{
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server1_, REPLICA_TYPE_FULL, 10));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server2_, REPLICA_TYPE_FULL, 20));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server2_, REPLICA_TYPE_FULL, 25));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server3_, REPLICA_TYPE_FULL, 30));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server4_, REPLICA_TYPE_FULL, 40));
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.update_broadcast_info(server5_, REPLICA_TYPE_LOGONLY, 50));
|
||||
|
||||
uint64_t log_id = OB_INVALID_ID;
|
||||
EXPECT_EQ(OB_SUCCESS, broadcast_info_mgr_.get_recyclable_log_id(log_id));
|
||||
EXPECT_EQ(25, log_id);
|
||||
}
|
||||
|
||||
} // namespace unittest
|
||||
} // namespace oceanbase
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
OB_LOGGER.set_file_name("test_ob_log_broadcast_info_mgr.log", true);
|
||||
OB_LOGGER.set_log_level("INFO");
|
||||
CLOG_LOG(INFO, "begin unittest: test_ob_log_broadcast_info_mgr");
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
return RUN_ALL_TESTS();
|
||||
}
|
||||
Reference in New Issue
Block a user