143 lines
3.6 KiB
C++
143 lines
3.6 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>
|
|
#define private public
|
|
#include "storage/ob_handle_cache.h"
|
|
|
|
#define OK(s) ASSERT_EQ(OB_SUCCESS, (s));
|
|
#define NOT_EXIST(s) ASSERT_EQ(OB_ENTRY_NOT_EXIST, (s))
|
|
|
|
namespace oceanbase {
|
|
using namespace storage;
|
|
using namespace common;
|
|
namespace unittest {
|
|
class TestHandleCache : public ::testing::Test {};
|
|
|
|
class TestKey {
|
|
public:
|
|
TestKey()
|
|
{}
|
|
TestKey(const int key) : key_(key)
|
|
{}
|
|
uint64_t hash() const
|
|
{
|
|
return key_;
|
|
}
|
|
int64_t to_string(char* buf, int64_t buf_len) const
|
|
{
|
|
UNUSED(buf);
|
|
UNUSED(buf_len);
|
|
return 0;
|
|
}
|
|
bool operator==(const TestKey& key) const
|
|
{
|
|
return key_ == key.key_;
|
|
}
|
|
int key_;
|
|
};
|
|
|
|
class TestHandle {
|
|
public:
|
|
void reset()
|
|
{}
|
|
int v_;
|
|
};
|
|
|
|
TEST_F(TestHandleCache, basic)
|
|
{
|
|
typedef ObHandleCache<TestKey, TestHandle, 4> TestCache;
|
|
TestCache handle_cache;
|
|
TestKey key(0);
|
|
TestHandle handle;
|
|
handle.v_ = 0;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
OK(handle_cache.put_handle(key, handle));
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(0, handle.v_);
|
|
|
|
key.key_ = 1;
|
|
handle.v_ = 1;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
OK(handle_cache.put_handle(key, handle));
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(1, handle.v_);
|
|
|
|
key.key_ = 64;
|
|
handle.v_ = 64;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
OK(handle_cache.put_handle(key, handle));
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(64, handle.v_);
|
|
key.key_ = 0;
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(0, handle.v_);
|
|
|
|
key.key_ = 32;
|
|
handle.v_ = 32;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
OK(handle_cache.put_handle(key, handle));
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(32, handle.v_);
|
|
key.key_ = 0;
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(0, handle.v_);
|
|
key.key_ = 64;
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(64, handle.v_);
|
|
|
|
key.key_ = 2;
|
|
handle.v_ = 2;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
OK(handle_cache.put_handle(key, handle));
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(2, handle.v_);
|
|
key.key_ = 1;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
|
|
key.key_ = 3;
|
|
handle.v_ = 3;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
OK(handle_cache.put_handle(key, handle));
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(3, handle.v_);
|
|
key.key_ = 32;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
key.key_ = 0;
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(0, handle.v_);
|
|
key.key_ = 64;
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(64, handle.v_);
|
|
|
|
key.key_ = 32;
|
|
handle.v_ = 32;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
OK(handle_cache.put_handle(key, handle));
|
|
OK(handle_cache.get_handle(key, handle));
|
|
ASSERT_EQ(32, handle.v_);
|
|
key.key_ = 2;
|
|
NOT_EXIST(handle_cache.get_handle(key, handle));
|
|
|
|
ASSERT_EQ(4, handle_cache.lru_list_.get_size());
|
|
}
|
|
|
|
} // namespace unittest
|
|
} // namespace oceanbase
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
OB_LOGGER.set_log_level("INFO");
|
|
testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|