131 lines
3.9 KiB
C++
131 lines
3.9 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/ob_define.h"
|
|
#include "logservice/palf/palf_callback.h"
|
|
#include "logservice/palf/palf_env.h"
|
|
#include "logservice/palf/palf_options.h"
|
|
#include "logservice/palf/palf_handle.h"
|
|
#include "logservice/palf_handle_guard.h"
|
|
#include "rpc/frame/ob_req_transport.h"
|
|
#include "share/allocator/ob_tenant_mutil_allocator.h"
|
|
#include "lib/file/file_directory_utils.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
namespace palf
|
|
{
|
|
using namespace common;
|
|
using namespace palf;
|
|
|
|
class MockLogCtx : public logservice::AppendCb
|
|
{
|
|
public:
|
|
explicit MockLogCtx()
|
|
{}
|
|
~MockLogCtx() {}
|
|
int on_success() override {
|
|
PALF_LOG(INFO, "on_success");
|
|
return OB_SUCCESS;
|
|
}
|
|
// 日志未形成多数派时会调用此函数，调用此函数后对象不再使用
|
|
int on_failure() override {
|
|
PALF_LOG(INFO, "on_failure");
|
|
return OB_SUCCESS;
|
|
}
|
|
};
|
|
|
|
class TestCreatePalfEnv : public ::testing::Test
|
|
{
|
|
public:
|
|
TestCreatePalfEnv() : self_(ObAddr::VER::IPV4, "127.0.0.1", 4096), palf_env_(), transport_(NULL, NULL), allocator_(500)
|
|
{
|
|
}
|
|
~TestCreatePalfEnv() {palf::PalfEnv::destroy_palf_env(palf_env_);}
|
|
int create_palf_handle(const int64_t id, palf::PalfRoleChangeCb *cb, palf::PalfHandle &handle)
|
|
{
|
|
return palf_env_->create(id, cb, handle);
|
|
}
|
|
void SetUp()
|
|
{
|
|
std::snprintf(log_dir_, OB_MAX_FILE_NAME_LENGTH, "%s_%ld", "unittest", ob_gettid());
|
|
common::FileDirectoryUtils::delete_directory_rec(log_dir_);
|
|
common::FileDirectoryUtils::create_directory(log_dir_);
|
|
PalfDiskOptions opts;
|
|
palf::PalfEnv::create_palf_env(opts, log_dir_, self_, &transport_, &allocator_, palf_env_);
|
|
}
|
|
void TearDown()
|
|
{
|
|
palf::PalfEnv::destroy_palf_env(palf_env_);
|
|
}
|
|
protected:
|
|
char log_dir_[OB_MAX_FILE_NAME_LENGTH];
|
|
common::ObAddr self_;
|
|
PalfEnv *palf_env_;
|
|
rpc::frame::ObReqTransport transport_;
|
|
ObTenantMutilAllocator allocator_;
|
|
};
|
|
|
|
class MockPalfRCCb : public PalfRoleChangeCb
|
|
{
|
|
public:
|
|
virtual int on_leader_revoke(int64_t id)
|
|
{ UNUSED(id); return OB_SUCCESS;}
|
|
virtual int is_leader_revoke_done(int64_t id, bool &is_done) const
|
|
{ UNUSED(id); UNUSED(is_done); return OB_SUCCESS;}
|
|
virtual int on_leader_takeover(int64_t id)
|
|
{ UNUSED(id); return OB_SUCCESS;}
|
|
virtual int is_leader_takeover_done(int64_t id, bool &is_done) const
|
|
{ UNUSED(id); UNUSED(is_done); return OB_SUCCESS;}
|
|
virtual int on_leader_active(int64_t id)
|
|
{ UNUSED(id); return OB_SUCCESS;}
|
|
};
|
|
|
|
TEST_F(TestCreatePalfEnv, creat)
|
|
{
|
|
PalfHandle handle;
|
|
int64_t base_ts = 1;
|
|
PalfAppendOptions opts;
|
|
const char *buffer = "123123123";
|
|
int64_t buf_len = strlen(buffer);
|
|
MockLogCtx cb;
|
|
MockPalfRCCb cb1;
|
|
LSN lsn;
|
|
int64_t ts;
|
|
palf_env_->create(1, &cb1, handle);
|
|
handle.register_role_change_cb(&cb1);
|
|
PalfHandle handle1;
|
|
handle.register_role_change_cb(&cb1);
|
|
palf_env_->create(2, &cb1, handle1);
|
|
handle1.register_role_change_cb(&cb1);
|
|
EXPECT_EQ(OB_SUCCESS, handle.append(opts, buffer, buf_len, base_ts, &cb, lsn, ts));
|
|
EXPECT_EQ(OB_SUCCESS, handle1.append(opts, buffer, buf_len, base_ts, &cb, lsn, ts));
|
|
palf_env_->close(handle);
|
|
palf_env_->close(handle1);
|
|
sleep(1);
|
|
}
|
|
}
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
rmdir("unit_01");
|
|
mkdir("unit_01", 0777);
|
|
unlink("test_palf_env.log");
|
|
OB_LOGGER.set_file_name("test_palf_env.log", true);
|
|
OB_LOGGER.set_log_level("TRACE");
|
|
PALF_LOG(INFO, "begin unittest::test_palf_env");
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|