152 lines
		
	
	
		
			5.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			152 lines
		
	
	
		
			5.4 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_define.h"
 | |
| #include "common/ob_member_list.h"
 | |
| 
 | |
| #include <gtest/gtest.h>
 | |
| 
 | |
| namespace oceanbase {
 | |
| using namespace common;
 | |
| namespace unittest {
 | |
| // TEST(ob_member_list_test, test1)
 | |
| //{
 | |
| //  const int64_t MEMBER_NUMBER = 3;
 | |
| //  const int64_t TIMESTAMP = 1;
 | |
| //  const int32_t BEGIN_PORT = 3000;
 | |
| //  const int64_t BUF_LEN = 1000;
 | |
| //
 | |
| //  ObMemberList ob_member_list;
 | |
| //  ob_member_list.reset();
 | |
| //  common::ObAddr tmp_server;
 | |
| //
 | |
| //  tmp_server.set_ip_addr("127.0.0.1", BEGIN_PORT);
 | |
| //  EXPECT_EQ(common::OB_ENTRY_NOT_EXIST, ob_member_list.remove_member(tmp_server));
 | |
| //  for (int64_t i = 0; i < clog::OB_MAX_MEMBER_NUMBER; ++i) {
 | |
| //    tmp_server.reset();
 | |
| //    tmp_server.set_ip_addr("127.0.0.1", BEGIN_PORT + static_cast<int32_t>(i));
 | |
| //    EXPECT_EQ(common::OB_SUCCESS, ob_member_list.add_member(tmp_server));
 | |
| //  }
 | |
| //  EXPECT_EQ(common::OB_SIZE_OVERFLOW, ob_member_list.add_member(tmp_server));
 | |
| //  for (int64_t i = MEMBER_NUMBER; i < clog::OB_MAX_MEMBER_NUMBER; ++i) {
 | |
| //    tmp_server.reset();
 | |
| //    tmp_server.set_ip_addr("127.0.0.1", BEGIN_PORT + static_cast<int32_t>(i));
 | |
| //    EXPECT_EQ(common::OB_SUCCESS, ob_member_list.remove_member(tmp_server));
 | |
| //  }
 | |
| //
 | |
| //  for (int64_t i = 0; i < clog::OB_MAX_MEMBER_NUMBER; ++i) {
 | |
| //    tmp_server.reset();
 | |
| //    tmp_server.set_ip_addr("127.0.0.1", BEGIN_PORT + static_cast<int32_t>(i));
 | |
| //    if (i < MEMBER_NUMBER) {
 | |
| //      EXPECT_TRUE(ob_member_list.contains(tmp_server));
 | |
| //    } else {
 | |
| //      EXPECT_FALSE(ob_member_list.contains(tmp_server));
 | |
| //    }
 | |
| //  }
 | |
| //
 | |
| //  tmp_server.reset();
 | |
| //  tmp_server.set_ip_addr("127.0.0.1", BEGIN_PORT + static_cast<int32_t>(0));
 | |
| //  EXPECT_EQ(common::OB_ENTRY_EXIST, ob_member_list.add_member(tmp_server));
 | |
| //
 | |
| //  tmp_server.reset();
 | |
| //  tmp_server.set_ip_addr("127.0.0.1", BEGIN_PORT + static_cast<int32_t>(MEMBER_NUMBER));
 | |
| //  EXPECT_EQ(common::OB_ENTRY_NOT_EXIST, ob_member_list.remove_member(tmp_server));
 | |
| //
 | |
| //  tmp_server.reset();
 | |
| //  tmp_server.set_ip_addr("127.0.0.1", BEGIN_PORT + static_cast<int32_t>(MEMBER_NUMBER - 2));
 | |
| //  EXPECT_EQ(common::OB_SUCCESS, ob_member_list.remove_member(tmp_server));
 | |
| //  EXPECT_EQ(MEMBER_NUMBER - 1, ob_member_list.get_member_number());
 | |
| //
 | |
| //  ob_member_list.set_mc_timestamp(TIMESTAMP);
 | |
| //
 | |
| //  char buf[BUF_LEN];
 | |
| //  int64_t pos = 0;
 | |
| //  ob_member_list.serialize(buf, BUF_LEN, pos);
 | |
| //
 | |
| //  ObMemberList ob_member_list2;
 | |
| //  pos = 0;
 | |
| //  EXPECT_EQ(common::OB_SUCCESS, ob_member_list2.deserialize(buf, BUF_LEN, pos));
 | |
| //
 | |
| //  ObMemberList ob_member_list3;
 | |
| //  ob_member_list3.deep_copy(ob_member_list2);
 | |
| //  EXPECT_EQ(33, ob_member_list3.get_serialize_size());
 | |
| //  EXPECT_EQ(MEMBER_NUMBER - 1, ob_member_list3.get_member_number());
 | |
| //
 | |
| //  tmp_server.reset();
 | |
| //  EXPECT_EQ(common::OB_SUCCESS, ob_member_list3.get_server_by_index(0, tmp_server));
 | |
| //  EXPECT_STREQ("\"127.0.0.1:3000\"", tmp_server.to_cstring());
 | |
| //
 | |
| //  tmp_server.reset();
 | |
| //  ob_member_list3.get_server_by_index(1, tmp_server);
 | |
| //  EXPECT_STREQ("\"127.0.0.1:3002\"", tmp_server.to_cstring());
 | |
| //
 | |
| //  for (int64_t i = 2; i != clog::OB_MAX_MEMBER_NUMBER; ++i) {
 | |
| //    tmp_server.reset();
 | |
| //    EXPECT_EQ(common::OB_SIZE_OVERFLOW, ob_member_list3.get_server_by_index(i, tmp_server));
 | |
| //    EXPECT_STREQ("\"0.0.0.0\"", tmp_server.to_cstring());
 | |
| //  }
 | |
| //  EXPECT_EQ(TIMESTAMP, ob_member_list3.get_mc_timestamp());
 | |
| //}
 | |
| 
 | |
| TEST(ob_member_list_test, testif_equal)
 | |
| {
 | |
|   common::ObMemberList memberlist1;
 | |
|   common::ObMemberList memberlist2;
 | |
|   memberlist1.reset();
 | |
|   memberlist2.reset();
 | |
|   EXPECT_TRUE(memberlist1.member_addr_equal(memberlist2));  // test if both null is equal
 | |
| 
 | |
|   common::ObAddr addr1;
 | |
|   common::ObAddr addr2;
 | |
|   common::ObAddr addr3;
 | |
|   common::ObAddr addr4;
 | |
|   addr1.reset();
 | |
|   addr2.reset();
 | |
|   addr3.reset();
 | |
|   addr4.reset();
 | |
|   addr1.parse_from_cstring("127.0.0.1:8080");
 | |
|   addr2.parse_from_cstring("127.0.0.2:8080");
 | |
|   addr3.parse_from_cstring("127.0.0.3:8080");
 | |
|   addr4.parse_from_cstring("127.0.0.4:8080");
 | |
| 
 | |
|   memberlist1.add_server(addr1);
 | |
|   memberlist1.add_server(addr2);
 | |
| 
 | |
|   EXPECT_FALSE(memberlist1.member_addr_equal(memberlist2));  // test only on memberlist is null
 | |
| 
 | |
|   memberlist2.add_server(addr3);
 | |
|   EXPECT_FALSE(memberlist1.member_addr_equal(memberlist2));  // test both the count and server addr are different
 | |
| 
 | |
|   memberlist2.add_server(addr4);
 | |
|   EXPECT_FALSE(memberlist1.member_addr_equal(memberlist2));  // test all server are different
 | |
| 
 | |
|   memberlist1.add_server(addr3);
 | |
|   memberlist1.add_server(addr4);
 | |
|   EXPECT_FALSE(memberlist1.member_addr_equal(
 | |
|       memberlist2));  // test both the count and server addr are different when no memberlist is null
 | |
| 
 | |
|   memberlist2.add_server(addr1);
 | |
|   memberlist2.add_server(addr2);
 | |
|   EXPECT_TRUE(memberlist1.member_addr_equal(memberlist2));  // not null. but equal
 | |
| }
 | |
| }  // namespace unittest
 | |
| }  // namespace oceanbase
 | |
| 
 | |
| int main(int argc, char** argv)
 | |
| {
 | |
|   OB_LOGGER.set_file_name("test_member_list.log", true);
 | |
|   OB_LOGGER.set_log_level("INFO");
 | |
|   CLOG_LOG(INFO, "begin unittest:test_member_list");
 | |
|   ::testing::InitGoogleTest(&argc, argv);
 | |
|   return RUN_ALL_TESTS();
 | |
| }
 | 
