202 lines
7.7 KiB
C++
202 lines
7.7 KiB
C++
/**
|
|
* 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();
|
|
}
|