Files
doris/be/test/util/lru_cache_util_test.cpp
Zhengguo Yang 5a44eeaf62 [refactor] Unify all unit tests into one binary file (#8958)
1. solved the previous delayed unit test file size is too large (1.7G+) and the unit test link time is too long problem problems
2. Unify all unit tests into one file to significantly reduce unit test execution time to less than 3 mins
3. temporarily disable stream_load_test.cpp, metrics_action_test.cpp, load_channel_mgr_test.cpp because it will re-implement part of the code and affect other tests
2022-04-12 15:30:40 +08:00

94 lines
2.5 KiB
C++

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
#include <gtest/gtest.h>
#include <memory>
#include "common/config.h"
#include "util/logging.h"
#include "util/lru_cache.hpp"
namespace doris {
class LruCacheTest : public testing::Test {};
struct Foo {
Foo(int num_param) : num(num_param) {}
int num;
};
TEST_F(LruCacheTest, NormalTest) {
LruCache<int, std::shared_ptr<Foo>> cache(10);
for (int i = 0; i < 10; ++i) {
cache.put(i, std::shared_ptr<Foo>(new Foo(i * 10)));
}
EXPECT_EQ(10, cache.size());
std::shared_ptr<Foo> ptr;
for (int i = 0; i < 10; ++i) {
EXPECT_TRUE(cache.get(i, &ptr));
EXPECT_EQ(i * 10, ptr->num);
}
}
TEST_F(LruCacheTest, IteratorTest) {
LruCache<int, std::shared_ptr<Foo>> cache(10);
int keys = 0;
int values = 0;
for (int i = 0; i < 10; ++i) {
keys += i;
values += i * 10;
cache.put(i, std::shared_ptr<Foo>(new Foo(i * 10)));
}
EXPECT_EQ(10, cache.size());
int key_sum = 0;
int value_sum = 0;
for (auto& it : cache) {
key_sum += it.first;
value_sum += it.second->num;
}
EXPECT_EQ(keys, key_sum);
EXPECT_EQ(values, value_sum);
std::shared_ptr<Foo> ptr;
for (int i = 0; i < 10; ++i) {
EXPECT_TRUE(cache.get(i, &ptr));
EXPECT_EQ(i * 10, ptr->num);
}
}
TEST_F(LruCacheTest, OverSize) {
LruCache<int, std::shared_ptr<Foo>> cache(10);
for (int i = 0; i < 110; ++i) {
cache.put(i, std::shared_ptr<Foo>(new Foo(i * 10)));
}
EXPECT_EQ(10, cache.size());
std::shared_ptr<Foo> ptr;
for (int i = 100; i < 110; ++i) {
EXPECT_TRUE(cache.get(i, &ptr));
EXPECT_EQ(i * 10, ptr->num);
}
}
} // namespace doris