init push
This commit is contained in:
oceanbase-admin
266
unittest/clog/test_log_task.cpp
Normal file
266
unittest/clog/test_log_task.cpp
Normal file
@ -0,0 +1,266 @@
|
||||
/**
|
||||
* 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_task.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "lib/allocator/ob_concurrent_fifo_allocator.h"
|
||||
#include "share/allocator/ob_tenant_mutil_allocator.h"
|
||||
#include "clog/ob_clog_config.h"
|
||||
#include "clog/ob_clog_mgr.h"
|
||||
#include "clog/ob_partition_log_service.h"
|
||||
#include "clog_mock_container/mock_submit_log_cb.h"
|
||||
#include "clog_mock_container/mock_log_sliding_window.h"
|
||||
namespace oceanbase {
|
||||
using namespace common;
|
||||
namespace clog {
|
||||
class MockSlidingCallBack : public MockObSlidingCallBack {
|
||||
public:
|
||||
MockSlidingCallBack() : err_no_(OB_SUCCESS)
|
||||
{}
|
||||
virtual ~MockSlidingCallBack()
|
||||
{}
|
||||
int sliding_cb(const ObLogTask& log_task)
|
||||
{
|
||||
UNUSED(log_task);
|
||||
return err_no_;
|
||||
}
|
||||
void set_err_no(const int err_no)
|
||||
{
|
||||
err_no_ = err_no;
|
||||
}
|
||||
|
||||
private:
|
||||
int err_no_;
|
||||
};
|
||||
} // namespace clog
|
||||
namespace unittest {
|
||||
class ObLogTaskTest : public ::testing::Test {
|
||||
protected:
|
||||
ObLogTaskTest() : log_task_(NULL), alloc_mgr_(NULL)
|
||||
{}
|
||||
virtual ~ObLogTaskTest()
|
||||
{}
|
||||
|
||||
protected:
|
||||
virtual void SetUp()
|
||||
{
|
||||
const uint64_t TABLE_ID = 15;
|
||||
const uint32_t PARTITION_IDX = 5;
|
||||
const int32_t PARTITION_CNT = 3;
|
||||
partition_key_.init(TABLE_ID, PARTITION_IDX, PARTITION_CNT);
|
||||
const uint64_t tenant_id = OB_SERVER_TENANT_ID;
|
||||
|
||||
ObMemAttr attr(tenant_id, ObModIds::OB_TENANT_MUTIL_ALLOCATOR);
|
||||
void* buf = ob_malloc(sizeof(common::ObTenantMutilAllocator), attr);
|
||||
if (NULL == buf) {
|
||||
CLOG_LOG(WARN, "alloc memory failed");
|
||||
OB_ASSERT(FALSE);
|
||||
}
|
||||
alloc_mgr_ = new (buf) common::ObTenantMutilAllocator(tenant_id);
|
||||
if (NULL == alloc_mgr_) {
|
||||
CLOG_LOG(WARN, "alloc_mgr_ construct failed");
|
||||
OB_ASSERT(FALSE);
|
||||
}
|
||||
if (NULL != (log_task_ = static_cast<clog::ObLogTask*>(alloc_mgr_->alloc_log_task_buf()))) {
|
||||
new (log_task_) clog::ObLogTask();
|
||||
} else {
|
||||
CLOG_LOG(INFO, "log_task_ init fail");
|
||||
OB_ASSERT(false);
|
||||
}
|
||||
}
|
||||
virtual void TearDown()
|
||||
{
|
||||
ob_free(alloc_mgr_);
|
||||
alloc_mgr_ = NULL;
|
||||
}
|
||||
|
||||
protected:
|
||||
clog::ObLogTask* log_task_;
|
||||
clog::MockSlidingCallBack mock_sliding_cb_;
|
||||
common::ObTenantMutilAllocator* alloc_mgr_;
|
||||
common::ObPartitionKey partition_key_;
|
||||
common::ObMemberList curr_member_list_;
|
||||
common::ObVersion freeze_version_;
|
||||
};
|
||||
|
||||
// test for init, reset and destroy
|
||||
TEST_F(ObLogTaskTest, init_reset_destroy_test)
|
||||
{
|
||||
// test for init and reset
|
||||
const int64_t REPLICA_NUM = 3;
|
||||
const bool need_replay = true;
|
||||
clog::ObISubmitLogCb* submit_cb = NULL;
|
||||
|
||||
EXPECT_EQ(OB_SUCCESS, log_task_->init(submit_cb, REPLICA_NUM, need_replay));
|
||||
|
||||
// log_task_ -> reset();
|
||||
// EXPECT_EQ(OB_SUCCESS, log_task_ -> init(alloc_mgr_, &mock_sliding_cb_, submit_cb,REPLICA_NUM, curr_member_list_));
|
||||
|
||||
OB_ASSERT(log_task_ != NULL);
|
||||
log_task_->destroy();
|
||||
}
|
||||
// set_log
|
||||
TEST_F(ObLogTaskTest, set_log_test)
|
||||
{
|
||||
clog::ObLogEntryHeader header;
|
||||
clog::ObConfirmedInfo confirmed_info;
|
||||
const ObProposalID PROPOSAL_ID;
|
||||
const uint64_t LOG_ID = 150;
|
||||
const char* BUF = "abc";
|
||||
const int64_t DATA_LEN = 18;
|
||||
const int64_t GENERATION_TIMESTAMP = 1000;
|
||||
const int64_t EPOCH_ID = 800;
|
||||
const int64_t SUBMIT_TS = 1300;
|
||||
const int64_t REPLICA_NUM = 3;
|
||||
const bool need_replay = true;
|
||||
clog::ObISubmitLogCb* submit_cb = NULL;
|
||||
ASSERT_EQ(OB_SUCCESS, log_task_->init(submit_cb, REPLICA_NUM, need_replay));
|
||||
|
||||
const char* buff = "This is a test log";
|
||||
ASSERT_EQ(OB_SUCCESS,
|
||||
header.generate_header(clog::OB_LOG_SUBMIT,
|
||||
partition_key_,
|
||||
LOG_ID,
|
||||
BUF,
|
||||
DATA_LEN,
|
||||
GENERATION_TIMESTAMP,
|
||||
EPOCH_ID,
|
||||
PROPOSAL_ID,
|
||||
SUBMIT_TS,
|
||||
freeze_version_,
|
||||
true));
|
||||
EXPECT_EQ(OB_SUCCESS, log_task_->set_log(header, buff, true));
|
||||
log_task_->set_log_confirmed();
|
||||
log_task_->set_confirmed_info(confirmed_info);
|
||||
|
||||
EXPECT_TRUE(log_task_->is_submit_log_exist());
|
||||
EXPECT_TRUE(log_task_->is_confirmed_info_exist());
|
||||
EXPECT_TRUE(log_task_->is_submit_log_body_exist());
|
||||
log_task_->destroy();
|
||||
}
|
||||
|
||||
// log_cursor
|
||||
TEST_F(ObLogTaskTest, log_cursor_test)
|
||||
{
|
||||
const int64_t REPLICA_NUM = 3;
|
||||
const bool need_replay = true;
|
||||
clog::ObISubmitLogCb* submit_cb = NULL;
|
||||
ASSERT_EQ(OB_SUCCESS, log_task_->init(submit_cb, REPLICA_NUM, need_replay));
|
||||
clog::ObLogCursor log_cursor;
|
||||
log_cursor.file_id_ = 1;
|
||||
log_cursor.offset_ = 1000;
|
||||
log_task_->set_log_cursor(log_cursor);
|
||||
log_task_->get_log_cursor();
|
||||
log_task_->destroy();
|
||||
}
|
||||
|
||||
TEST_F(ObLogTaskTest, cb_test)
|
||||
{
|
||||
const int64_t REPLICA_NUM = 3;
|
||||
const bool need_replay = true;
|
||||
clog::ObISubmitLogCb* submit_cb = NULL;
|
||||
ASSERT_EQ(OB_SUCCESS, log_task_->init(submit_cb, REPLICA_NUM, need_replay));
|
||||
common::ObPartitionKey pkey;
|
||||
uint64_t log_id = 0;
|
||||
const bool batch_committed = false;
|
||||
const bool batch_last_succeed = false;
|
||||
EXPECT_EQ(OB_SUCCESS, log_task_->submit_log_succ_cb(pkey, log_id, batch_committed, batch_last_succeed));
|
||||
clog::MockObSubmitLogCb submit_cb_obj;
|
||||
// log_task_ -> reset();
|
||||
// ASSERT_EQ(OB_SUCCESS, log_task_ -> init(alloc_mgr_, &mock_sliding_cb_, &submit_cb_obj,
|
||||
// REPLICA_NUM,curr_member_list_)); EXPECT_EQ(OB_SUCCESS, log_task_ -> submit_log_succ_cb()); EXPECT_TRUE(log_task_ ->
|
||||
// is_on_success_cb_called()); log_task_ -> destroy();
|
||||
}
|
||||
|
||||
TEST_F(ObLogTaskTest, ack_majority_test)
|
||||
{
|
||||
common::ObAddr member1;
|
||||
common::ObAddr member2;
|
||||
common::ObAddr member3;
|
||||
common::ObAddr member4; // not include
|
||||
member1.parse_from_cstring("127.0.0.1:8000");
|
||||
member2.parse_from_cstring("127.0.0.1:8001");
|
||||
member3.parse_from_cstring("127.0.0.1:8002");
|
||||
curr_member_list_.add_server(member1);
|
||||
curr_member_list_.add_server(member2);
|
||||
curr_member_list_.add_server(member3);
|
||||
|
||||
EXPECT_EQ(0, log_task_->ack_log(member3));
|
||||
const int64_t REPLICA_NUM = 3;
|
||||
const bool need_replay = true;
|
||||
clog::ObISubmitLogCb* submit_cb = NULL;
|
||||
ASSERT_EQ(OB_SUCCESS, log_task_->init(submit_cb, REPLICA_NUM, need_replay));
|
||||
EXPECT_EQ(OB_SUCCESS, log_task_->ack_log(member1));
|
||||
EXPECT_EQ(OB_SUCCESS, log_task_->ack_log(member2));
|
||||
EXPECT_EQ(OB_SUCCESS, log_task_->ack_log(member3));
|
||||
|
||||
log_task_->set_flush_local_finished();
|
||||
log_task_->set_flush_local_finished();
|
||||
EXPECT_TRUE(log_task_->try_set_majority_finished());
|
||||
ASSERT_TRUE(log_task_->is_majority_finished());
|
||||
log_task_->destroy();
|
||||
}
|
||||
|
||||
TEST_F(ObLogTaskTest, on_pop_test)
|
||||
{
|
||||
mock_sliding_cb_.set_err_no(OB_SUCCESS);
|
||||
const int64_t REPLICA_NUM = 3;
|
||||
const bool need_replay = true;
|
||||
clog::ObISubmitLogCb* submit_cb = NULL;
|
||||
ASSERT_EQ(OB_SUCCESS, log_task_->init(submit_cb, REPLICA_NUM, need_replay));
|
||||
// EXPECT_EQ(OB_SUCCESS, log_task_ -> on_pop());
|
||||
log_task_->destroy();
|
||||
}
|
||||
|
||||
TEST_F(ObLogTaskTest, misc_test)
|
||||
{
|
||||
const int64_t REPLICA_NUM = 3;
|
||||
const bool need_replay = false;
|
||||
clog::ObISubmitLogCb* submit_cb = NULL;
|
||||
clog::ObConfirmedInfo confirmed_info;
|
||||
ASSERT_EQ(OB_SUCCESS, log_task_->init(submit_cb, REPLICA_NUM, need_replay));
|
||||
|
||||
// CONFIRM_SUBMITTED
|
||||
log_task_->set_log_confirmed();
|
||||
ASSERT_TRUE(log_task_->is_log_confirmed());
|
||||
|
||||
// can be removed = CONFIRM_SUBMITTED + LOCAL_FLUSHED
|
||||
EXPECT_FALSE(log_task_->can_be_removed());
|
||||
log_task_->set_confirmed_info(confirmed_info);
|
||||
log_task_->set_index_log_submitted();
|
||||
EXPECT_TRUE(log_task_->can_be_removed());
|
||||
|
||||
clog::ObILogExtRingBufferData* log_task = NULL;
|
||||
ASSERT_FALSE(log_task_->can_overwrite(log_task));
|
||||
|
||||
ASSERT_FALSE(log_task_->need_replay());
|
||||
log_task_->try_set_need_replay(false);
|
||||
ASSERT_FALSE(log_task_->need_replay());
|
||||
log_task_->try_set_need_replay(true);
|
||||
ASSERT_TRUE(log_task_->need_replay());
|
||||
|
||||
// reset_state
|
||||
log_task_->reset_state(false);
|
||||
|
||||
// reset_log_cursor
|
||||
log_task_->reset_log_cursor();
|
||||
log_task_->destroy();
|
||||
}
|
||||
|
||||
} // namespace unittest
|
||||
} // namespace oceanbase
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
oceanbase::common::ObLogger::get_logger().set_log_level("DEBUG");
|
||||
::testing::InitGoogleTest(&argc, argv);
|
||||
return RUN_ALL_TESTS();
|
||||
}
|
||||
Reference in New Issue
Block a user