/**
 * 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/alloc/ob_malloc_allocator.h"

using namespace oceanbase::lib;
using namespace oceanbase::common;

TEST(TestMallocAllocator, create_tenant_ctx_allocator)
{
  ObMallocAllocator* malloc_allocator = ObMallocAllocator::get_instance();
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->init());
  const int64_t limit = 10 * 1000 * 1000;
  const uint64_t small_tenant_id = 200;
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->create_tenant_ctx_allocator(small_tenant_id, 0));
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->set_tenant_limit(small_tenant_id, limit));
  ASSERT_EQ(limit, malloc_allocator->get_tenant_limit(small_tenant_id));
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->create_tenant_ctx_allocator(small_tenant_id, 0));
  ASSERT_EQ(limit, malloc_allocator->get_tenant_limit(small_tenant_id));

  const uint64_t big_tenant_id = 20000 + 10000;
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->create_tenant_ctx_allocator(big_tenant_id, 0));
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->set_tenant_limit(big_tenant_id, limit));
  ASSERT_EQ(limit, malloc_allocator->get_tenant_limit(big_tenant_id));
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->create_tenant_ctx_allocator(big_tenant_id, 0));
  ASSERT_EQ(limit, malloc_allocator->get_tenant_limit(big_tenant_id));

  const uint64_t tenant_id = 20000;
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->create_tenant_ctx_allocator(tenant_id, 0));
  ObTenantCtxAllocator* tenant_allocator = malloc_allocator->get_tenant_ctx_allocator(big_tenant_id, 0);
  ASSERT_TRUE(NULL != tenant_allocator);

  // dump meta
  system("rm -rf log");
  system("mkdir log");
  const static char* file_name = "log/MMETA";
  ASSERT_EQ(OB_INVALID_ARGUMENT, malloc_allocator->dump_meta(nullptr));
  int ret = malloc_allocator->dump_meta(file_name);
  ASSERT_EQ(ret, OB_SUCCESS);
  system("rm -rf log");
}

TEST(TestMallocAllocator, idle)
{
  ObMallocAllocator* malloc_allocator = ObMallocAllocator::get_instance();
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->init());
  const uint64_t tenant_id = 200;
  const uint64_t ctx_id = 1;

  ASSERT_EQ(OB_TENANT_NOT_EXIST, malloc_allocator->set_tenant_ctx_idle(tenant_id, ctx_id, OB_MALLOC_BIG_BLOCK_SIZE));

  ASSERT_EQ(OB_SUCCESS, malloc_allocator->create_tenant_ctx_allocator(tenant_id, ctx_id));
  ObTenantCtxAllocator* ta = malloc_allocator->get_tenant_ctx_allocator(tenant_id, ctx_id);
  ASSERT_TRUE(NULL != ta);
  ta->set_tenant_memory_mgr();
  ASSERT_EQ(OB_SUCCESS, malloc_allocator->set_tenant_limit(tenant_id, 1024 * 1024 * 1024));

  ASSERT_EQ(OB_SUCCESS, malloc_allocator->set_tenant_ctx_idle(tenant_id, ctx_id, OB_MALLOC_BIG_BLOCK_SIZE));
}

int main(int argc, char* argv[])
{
  signal(49, SIG_IGN);
  OB_LOGGER.set_file_name("t.log", true, true);
  OB_LOGGER.set_log_level("INFO");

  ::testing::InitGoogleTest(&argc, argv);
  return RUN_ALL_TESTS();
}