158 lines
3.5 KiB
C++
158 lines
3.5 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 <gmock/gmock.h>
|
|
|
|
#define USING_LOG_PREFIX STORAGE
|
|
#define protected public
|
|
#define private public
|
|
|
|
#include "mtlenv/mock_tenant_module_env.h"
|
|
|
|
namespace oceanbase
|
|
{
|
|
using namespace share;
|
|
using namespace common;
|
|
using namespace storage;
|
|
using namespace compaction;
|
|
|
|
|
|
class TestCompactionMemCtx : public ::testing::Test
|
|
{
|
|
public:
|
|
TestCompactionMemCtx();
|
|
virtual ~TestCompactionMemCtx() = default;
|
|
virtual void SetUp() override;
|
|
virtual void TearDown() override;
|
|
static void SetUpTestCase();
|
|
static void TearDownTestCase();
|
|
|
|
|
|
private:
|
|
const uint64_t tenant_id_;
|
|
ObTenantBase tenant_base_;
|
|
};
|
|
|
|
|
|
TestCompactionMemCtx::TestCompactionMemCtx()
|
|
: tenant_id_(1),
|
|
tenant_base_(tenant_id_)
|
|
{
|
|
}
|
|
|
|
void TestCompactionMemCtx::SetUpTestCase()
|
|
{
|
|
EXPECT_EQ(OB_SUCCESS, MockTenantModuleEnv::get_instance().init());
|
|
}
|
|
|
|
void TestCompactionMemCtx::TearDownTestCase()
|
|
{
|
|
MockTenantModuleEnv::get_instance().destroy();
|
|
}
|
|
|
|
void TestCompactionMemCtx::SetUp()
|
|
{
|
|
int ret = OB_SUCCESS;
|
|
|
|
ObTenantEnv::set_tenant(&tenant_base_);
|
|
ASSERT_EQ(OB_SUCCESS, tenant_base_.init());
|
|
ASSERT_EQ(tenant_id_, MTL_ID());
|
|
}
|
|
|
|
void TestCompactionMemCtx::TearDown()
|
|
{
|
|
ObTenantEnv::set_tenant(nullptr);
|
|
}
|
|
|
|
|
|
namespace unittest
|
|
{
|
|
|
|
TEST_F(TestCompactionMemCtx, basic_test)
|
|
{
|
|
ObLocalArena arena("TestArena", OB_MALLOC_NORMAL_BLOCK_SIZE);
|
|
|
|
void *buf = arena.alloc(128);
|
|
ASSERT_TRUE(nullptr != buf);
|
|
EXPECT_EQ(144, arena.used());
|
|
EXPECT_EQ(true, arena.total() <= 8064);
|
|
|
|
arena.free(buf);
|
|
arena.clear();
|
|
EXPECT_EQ(0, arena.used());
|
|
EXPECT_EQ(0, arena.total());
|
|
}
|
|
|
|
|
|
TEST_F(TestCompactionMemCtx, monitor_arena_test)
|
|
{
|
|
ObArenaAllocator inner_arena;
|
|
ObTabletMergeDagParam param;
|
|
ObCompactionMemoryContext mem_ctx(param, inner_arena);
|
|
|
|
ObLocalArena arena("TestArena");
|
|
arena.bind_mem_ctx(mem_ctx);
|
|
EXPECT_EQ(&mem_ctx, arena.ref_mem_ctx_);
|
|
|
|
void *buf = arena.alloc(128);
|
|
ASSERT_TRUE(nullptr != buf);
|
|
EXPECT_EQ(144, arena.used());
|
|
EXPECT_EQ(true, arena.total() <= 8064);
|
|
|
|
mem_ctx.mem_click();
|
|
EXPECT_EQ(true, mem_ctx.get_total_mem_peak() <= 8064);
|
|
|
|
arena.free(buf);
|
|
arena.clear();
|
|
EXPECT_EQ(0, arena.used());
|
|
EXPECT_EQ(0, arena.total());
|
|
|
|
// only stat the peak val
|
|
mem_ctx.mem_click();
|
|
EXPECT_EQ(true, mem_ctx.get_total_mem_peak() <= 8064);
|
|
|
|
arena.~ObLocalArena();
|
|
EXPECT_EQ(nullptr, arena.ref_mem_ctx_);
|
|
}
|
|
|
|
|
|
TEST_F(TestCompactionMemCtx, mem_ctx_test)
|
|
{
|
|
ObArenaAllocator inner_arena;
|
|
ObTabletMergeDagParam param;
|
|
ObCompactionMemoryContext mem_ctx(param, inner_arena);
|
|
|
|
void *buf1 = mem_ctx.alloc(128);
|
|
ASSERT_TRUE(nullptr != buf1);
|
|
|
|
void *buf2 = mem_ctx.local_alloc(128);
|
|
ASSERT_TRUE(nullptr != buf2);
|
|
|
|
mem_ctx.mem_click();
|
|
ASSERT_TRUE(0 != mem_ctx.get_total_mem_peak());
|
|
}
|
|
|
|
|
|
} // namespace unittest
|
|
} // namespace oceanbase
|
|
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
system("rm -f test_compaction_memory_context.log*");
|
|
OB_LOGGER.set_file_name("test_compaction_memory_context.log", true);
|
|
OB_LOGGER.set_log_level("INFO");
|
|
OB_LOGGER.set_max_file_size(256*1024*1024);
|
|
return RUN_ALL_TESTS();
|
|
}
|