129 lines
		
	
	
		
			4.6 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			129 lines
		
	
	
		
			4.6 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 <gtest/gtest.h>
 | |
| #include "clog/ob_log_entry_header.h"
 | |
| #include "share/ob_proposal_id.h"
 | |
| 
 | |
| using namespace oceanbase::common;
 | |
| namespace oceanbase {
 | |
| using namespace clog;
 | |
| namespace unittest {
 | |
| class TestObLogEntryHeader : public ::testing::Test {
 | |
| public:
 | |
|   virtual void SetUp()
 | |
|   {}
 | |
|   virtual void TearDown()
 | |
|   {}
 | |
| };
 | |
| 
 | |
| TEST_F(TestObLogEntryHeader, test_ob_log_entry_header)
 | |
| {
 | |
|   const int64_t BUFSIZE = 1 << 11;
 | |
|   ObLogEntryHeader header;
 | |
|   ObLogEntryHeader header2;
 | |
|   ObLogType ltype = OB_LOG_SUBMIT;
 | |
|   ObPartitionKey partition_key;
 | |
|   uint64_t log_id = 4;
 | |
|   int64_t data_len = BUFSIZE >> 1;
 | |
|   int64_t gts = ::oceanbase::common::ObTimeUtility::current_time();
 | |
|   int64_t tts = ::oceanbase::common::ObTimeUtility::current_time();
 | |
|   ObProposalID rts;
 | |
|   rts.ts_ = ::oceanbase::common::ObTimeUtility::current_time();
 | |
|   uint64_t tid = 2;
 | |
|   int32_t pid = 3;
 | |
|   int32_t p_count = 10;
 | |
|   int64_t pos = 0;
 | |
|   char data[data_len];
 | |
|   char buf[BUFSIZE];
 | |
| 
 | |
|   memset(data, 'A', data_len);
 | |
|   partition_key.init(tid, pid, p_count);
 | |
| 
 | |
|   // test_generate_entry_header
 | |
|   EXPECT_EQ(OB_SUCCESS,
 | |
|       header.generate_header(
 | |
|           OB_LOG_SUBMIT, partition_key, log_id, data, data_len, gts, tts, rts, 0, ObVersion(0), true));
 | |
|   header2.shallow_copy(header);
 | |
|   EXPECT_EQ(header.get_data_checksum(), header2.get_data_checksum());
 | |
|   EXPECT_EQ(header.get_header_checksum(), header2.get_header_checksum());
 | |
|   EXPECT_TRUE(header == header2);
 | |
| 
 | |
|   // test_check
 | |
|   EXPECT_TRUE(header.check_data_checksum(data, data_len));
 | |
|   EXPECT_TRUE(header.check_integrity(data, data_len));
 | |
| 
 | |
|   EXPECT_FALSE(header.check_data_checksum(data, 0));
 | |
|   EXPECT_FALSE(header.check_data_checksum(NULL, data_len));
 | |
|   EXPECT_FALSE(header.check_integrity(data, -1));  // check data checksum fail
 | |
| 
 | |
|   // test_get_and_set
 | |
|   EXPECT_EQ(header2.get_magic_num(), header.get_magic_num());
 | |
|   EXPECT_EQ(header2.get_version(), header.get_version());
 | |
|   EXPECT_EQ(ltype, header.get_log_type());
 | |
|   EXPECT_EQ(tid, header.get_partition_key().table_id_);
 | |
|   EXPECT_EQ(pid, header.get_partition_key().get_partition_id());
 | |
|   EXPECT_EQ(log_id, header.get_log_id());
 | |
|   EXPECT_EQ(gts, header.get_generation_timestamp());
 | |
|   EXPECT_EQ(tts, header.get_epoch_id());
 | |
|   EXPECT_EQ(rts, header.get_proposal_id());
 | |
|   EXPECT_EQ(data_len, header.get_data_len());
 | |
|   EXPECT_EQ(data_len + header.get_serialize_size(), header.get_total_len());
 | |
| 
 | |
|   int64_t ts = ::oceanbase::common::ObTimeUtility::current_time();
 | |
|   rts.ts_ = ts;
 | |
|   header.set_epoch_id(ts);
 | |
|   header.set_proposal_id(rts);
 | |
|   EXPECT_EQ(ts, header.get_epoch_id());
 | |
|   EXPECT_EQ(ts, header.get_proposal_id().ts_);
 | |
| 
 | |
|   // test_serialize_and_deserialize
 | |
|   CLOG_LOG(INFO, "before serialize", "header", to_cstring(header2));
 | |
|   EXPECT_EQ(OB_INVALID_ARGUMENT, header2.serialize(buf, 0, pos));
 | |
|   EXPECT_EQ(OB_INVALID_ARGUMENT, header2.serialize(NULL, 0, pos));
 | |
|   EXPECT_EQ(OB_SERIALIZE_ERROR, header2.serialize(buf, 1, pos));  // buf is too small
 | |
|   EXPECT_EQ(OB_SUCCESS, header2.serialize(buf, BUFSIZE, pos));
 | |
|   header2.reset();
 | |
|   EXPECT_TRUE(header2.check_data_checksum(NULL, 0));
 | |
|   EXPECT_FALSE(header2.check_integrity(buf, data_len));  // magic = 0
 | |
|   EXPECT_NE(header.get_header_checksum(), header2.get_header_checksum());
 | |
| 
 | |
|   pos = 0;
 | |
|   CLOG_LOG(INFO, "after reset", "header", to_cstring(header2));
 | |
|   EXPECT_EQ(OB_INVALID_ARGUMENT, header2.deserialize(NULL, data_len, pos));
 | |
|   EXPECT_EQ(OB_INVALID_ARGUMENT, header2.deserialize(buf, -1, pos));
 | |
|   EXPECT_EQ(OB_DESERIALIZE_ERROR, header2.deserialize(buf, 1, pos));  // buf is too small
 | |
|   EXPECT_EQ(OB_SUCCESS, header2.deserialize(buf, data_len, pos));
 | |
|   CLOG_LOG(INFO, "after serialize", "header", to_cstring(header2));
 | |
| 
 | |
|   EXPECT_TRUE(header2.check_integrity(data, data_len));
 | |
| 
 | |
|   buf[3] = 'B';
 | |
|   pos = 0;
 | |
|   EXPECT_EQ(OB_SUCCESS, header2.deserialize(buf, data_len, pos));
 | |
|   EXPECT_FALSE(header2.check_integrity(data, data_len));  // data checksum fail
 | |
| 
 | |
|   header2.to_string(buf, data_len);
 | |
| }
 | |
| 
 | |
| }  // end namespace unittest
 | |
| }  // end namespace oceanbase
 | |
| 
 | |
| int main(int argc, char** argv)
 | |
| {
 | |
|   OB_LOGGER.set_file_name("test_ob_log_entry_header.log", true);
 | |
|   OB_LOGGER.set_log_level("INFO");
 | |
|   CLOG_LOG(INFO, "begin unittest::test_ob_log_entry_header");
 | |
|   testing::InitGoogleTest(&argc, argv);
 | |
|   return RUN_ALL_TESTS();
 | |
| }
 | 
