112 lines
3.8 KiB
C++
112 lines
3.8 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.
|
|
*/
|
|
|
|
#define USING_LOG_PREFIX SHARE
|
|
|
|
#include "share/backup/ob_tenant_name_mgr.h"
|
|
#include <gtest/gtest.h>
|
|
|
|
using namespace oceanbase;
|
|
using namespace common;
|
|
using namespace share;
|
|
|
|
TEST(ObTenantNameSimpleMgr, basic)
|
|
{
|
|
ObTenantNameSimpleMgr mgr;
|
|
ObArenaAllocator allocator;
|
|
char* buf = nullptr;
|
|
int64_t buf_size = 0;
|
|
int64_t pos = 0;
|
|
int64_t schema_version = 10;
|
|
uint64_t tenant_id = 0;
|
|
|
|
ASSERT_EQ(OB_SUCCESS, mgr.init());
|
|
// create 1001 t1
|
|
ASSERT_EQ(OB_SUCCESS, mgr.add("t1", 1, 1001));
|
|
// create 1002 t2
|
|
ASSERT_EQ(OB_SUCCESS, mgr.add("t2", 2, 1002));
|
|
// create 1003 t3
|
|
ASSERT_EQ(OB_SUCCESS, mgr.add("t3", 3, 1003));
|
|
ASSERT_EQ(OB_SUCCESS, mgr.complete(schema_version));
|
|
|
|
pos = 0;
|
|
buf_size = mgr.get_write_buf_size();
|
|
buf = (char*)allocator.alloc(buf_size);
|
|
ASSERT_TRUE(buf != nullptr);
|
|
ASSERT_EQ(OB_SUCCESS, mgr.write_buf(buf, buf_size, pos));
|
|
|
|
ObTenantNameSimpleMgr mgr2;
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.init());
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.read_buf(buf, buf_size));
|
|
ASSERT_EQ(schema_version, mgr2.get_schema_version());
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.get_tenant_id("t1", 10, tenant_id));
|
|
ASSERT_EQ(1001, tenant_id);
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.get_tenant_id("t2", 10, tenant_id));
|
|
ASSERT_EQ(1002, tenant_id);
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.get_tenant_id("t3", 3, tenant_id));
|
|
ASSERT_EQ(1003, tenant_id);
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.get_tenant_id("t3", 10, tenant_id));
|
|
ASSERT_EQ(1003, tenant_id);
|
|
ASSERT_EQ(OB_TENANT_NOT_EXIST, mgr2.get_tenant_id("t3", 2, tenant_id));
|
|
|
|
// drop t1 and rename 1002 to t1
|
|
schema_version = 20;
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.add("t1", 20, 1002));
|
|
ASSERT_EQ(OB_NOT_INIT, mgr2.get_tenant_id("t1", 20, tenant_id));
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.complete(schema_version));
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.get_tenant_id("t1", 20, tenant_id));
|
|
ASSERT_EQ(1002, tenant_id);
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.get_tenant_id("t1", 19, tenant_id));
|
|
ASSERT_EQ(1001, tenant_id);
|
|
pos = 0;
|
|
buf_size = mgr2.get_write_buf_size();
|
|
buf = (char*)allocator.alloc(buf_size);
|
|
ASSERT_TRUE(buf != nullptr);
|
|
ASSERT_EQ(OB_SUCCESS, mgr2.write_buf(buf, buf_size, pos));
|
|
|
|
ObTenantNameSimpleMgr mgr3;
|
|
ASSERT_EQ(OB_SUCCESS, mgr3.init());
|
|
ASSERT_EQ(OB_SUCCESS, mgr3.read_buf(buf, buf_size));
|
|
ASSERT_EQ(schema_version, mgr3.get_schema_version());
|
|
ASSERT_EQ(OB_SUCCESS, mgr3.get_tenant_id("t1", 20, tenant_id));
|
|
ASSERT_EQ(1002, tenant_id);
|
|
ASSERT_EQ(OB_SUCCESS, mgr3.get_tenant_id("t1", 19, tenant_id));
|
|
ASSERT_EQ(1001, tenant_id);
|
|
}
|
|
|
|
TEST(ObTenantNameSimpleMgr, add_same_tenant)
|
|
{
|
|
ObTenantNameSimpleMgr mgr;
|
|
ObArenaAllocator allocator;
|
|
char* buf = nullptr;
|
|
int64_t buf_size = 0;
|
|
int64_t pos = 0;
|
|
int64_t schema_version = 10;
|
|
uint64_t tenant_id = 0;
|
|
|
|
ASSERT_EQ(OB_SUCCESS, mgr.init());
|
|
ASSERT_EQ(OB_SUCCESS, mgr.add("t1", 100, 1001));
|
|
ASSERT_EQ(OB_ERR_SYS, mgr.add("t1", 100, 1002));
|
|
ASSERT_EQ(OB_SUCCESS, mgr.add("t1", 3, 1003));
|
|
ASSERT_EQ(OB_SUCCESS, mgr.add("t1", 30, 1004));
|
|
ASSERT_EQ(OB_SUCCESS, mgr.add("t1", 300, 1001));
|
|
ASSERT_EQ(OB_SUCCESS, mgr.complete(schema_version));
|
|
ASSERT_EQ(OB_SUCCESS, mgr.get_tenant_id("t1", 200, tenant_id));
|
|
ASSERT_EQ(1001, tenant_id);
|
|
}
|
|
int main(int argc, char** argv)
|
|
{
|
|
OB_LOGGER.set_log_level("INFO");
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|