90 lines
3.2 KiB
C++
90 lines
3.2 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 "lib/checksum/ob_crc64.h"
|
|
#include "lib/checksum/ob_parity_check.h"
|
|
#include <gtest/gtest.h>
|
|
#include <random>
|
|
|
|
namespace oceanbase
|
|
{
|
|
namespace unittest
|
|
{
|
|
using namespace common;
|
|
|
|
TEST(TestLogChecksum, test_log_checksum)
|
|
{
|
|
int64_t buffer_size = 2 * 1024 * 1024l;
|
|
char *buffer = NULL;
|
|
buffer = (char *)ob_malloc(buffer_size, ObModIds::TEST);
|
|
// fill buffer by random string
|
|
std::mt19937 generator{std::random_device{}()};
|
|
std::uniform_int_distribution<int> distribution{'a', 'z'};
|
|
std::string rand_str(buffer_size, '\0');
|
|
for(auto& dis: rand_str) {
|
|
dis = distribution(generator);
|
|
}
|
|
memcpy(buffer, rand_str.c_str(), buffer_size);
|
|
|
|
// test ob_crc64()
|
|
for (int i = 0; i < 1000; ++i) {
|
|
int64_t checksum01 = 0;
|
|
int64_t checksum02 = 0;
|
|
int64_t data_len = ObRandom::rand(10, buffer_size);
|
|
EXPECT_TRUE(data_len >= 10);
|
|
int64_t data_len_part_1 = ObRandom::rand(1, data_len - 1);;
|
|
EXPECT_TRUE(data_len_part_1 > 0);
|
|
int64_t data_len_part_2 = data_len - data_len_part_1;
|
|
EXPECT_TRUE(data_len_part_1 > 0);
|
|
checksum01 = static_cast<int64_t>(ob_crc64(buffer, data_len));
|
|
checksum02 = static_cast<int64_t>(ob_crc64(buffer, data_len_part_1));
|
|
checksum02 = static_cast<int64_t>(ob_crc64(checksum02, buffer + data_len_part_1, data_len_part_2));
|
|
EXPECT_EQ(checksum01, checksum02);
|
|
PALF_LOG(INFO, "test round ", K(i), K(data_len), K(data_len_part_1), K(data_len_part_2), K(checksum01), K(checksum02));
|
|
}
|
|
|
|
// test ob_crc32()
|
|
// for (int i = 0; i < 1000; ++i) {
|
|
// int32_t checksum01 = 0;
|
|
// int32_t checksum02 = 0;
|
|
// int64_t data_len = ObRandom::rand(10, buffer_size);
|
|
// EXPECT_TRUE(data_len >= 10);
|
|
// int64_t data_len_part_1 = ObRandom::rand(1, data_len - 1);;
|
|
// EXPECT_TRUE(data_len_part_1 > 0);
|
|
// int64_t data_len_part_2 = data_len - data_len_part_1;
|
|
// EXPECT_TRUE(data_len_part_1 > 0);
|
|
// checksum01 = static_cast<int32_t>(ob_crc32(buffer, data_len));
|
|
// checksum02 = static_cast<int32_t>(ob_crc32(buffer, data_len_part_1));
|
|
// checksum02 = static_cast<int32_t>(ob_crc32(checksum02, buffer + data_len_part_1, data_len_part_2));
|
|
// EXPECT_EQ(checksum01, checksum02);
|
|
// PALF_LOG(INFO, "test round ", K(i), K(data_len), K(data_len_part_1), K(data_len_part_2), K(checksum01), K(checksum02));
|
|
// }
|
|
|
|
// test parity_check()
|
|
uint64_t v1 = (1 << 20) + (1 << 10) + 1;
|
|
uint64_t v2 = (1 << 11) + (1 << 5);
|
|
EXPECT_TRUE(parity_check(v1));
|
|
EXPECT_FALSE(parity_check(v2));
|
|
}
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
OB_LOGGER.set_file_name("test_log_checksum.log", true);
|
|
OB_LOGGER.set_log_level("INFO");
|
|
PALF_LOG(INFO, "begin unittest::test_log_checksum");
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|