chore: import upstream snapshot with attribution
This commit is contained in:
wehub-resource-sync
@@ -0,0 +1,32 @@
|
||||
include(${PROJECT_ROOT_DIR}/cmake/bazel.cmake)
|
||||
include(${PROJECT_ROOT_DIR}/cmake/option.cmake)
|
||||
|
||||
if(APPLE)
|
||||
set(APPLE_FRAMEWORK_LIBS
|
||||
-framework CoreFoundation
|
||||
-framework CoreGraphics
|
||||
-framework CoreData
|
||||
-framework CoreText
|
||||
-framework Security
|
||||
-framework Foundation
|
||||
-Wl,-U,_MallocExtension_ReleaseFreeMemory
|
||||
-Wl,-U,_ProfilerStart
|
||||
-Wl,-U,_ProfilerStop
|
||||
-Wl,-U,_RegisterThriftProtocol
|
||||
)
|
||||
endif()
|
||||
|
||||
file(GLOB ALL_TEST_SRCS *_test.cc)
|
||||
foreach(CC_SRCS ${ALL_TEST_SRCS})
|
||||
get_filename_component(CC_TARGET ${CC_SRCS} NAME_WE)
|
||||
cc_gmock(
|
||||
NAME ${CC_TARGET} STRICT
|
||||
LIBS zvec_common
|
||||
${CMAKE_THREAD_LIBS_INIT}
|
||||
${CMAKE_DL_LIBS}
|
||||
SRCS ${CC_SRCS}
|
||||
INCS .. ../../src
|
||||
LDFLAGS ${APPLE_FRAMEWORK_LIBS}
|
||||
)
|
||||
cc_test_suite(zvec_common ${CC_TARGET})
|
||||
endforeach()
|
||||
@@ -0,0 +1,246 @@
|
||||
// Copyright 2025-present the zvec project
|
||||
//
|
||||
// Licensed 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 "zvec/db/config.h"
|
||||
#include <gtest/gtest.h>
|
||||
#include "zvec/db/status.h"
|
||||
|
||||
using namespace zvec;
|
||||
|
||||
class ConfigTest : public ::testing::Test {
|
||||
protected:
|
||||
void SetUp() override {
|
||||
// Reset GlobalConfig for each test
|
||||
// Note: Since GlobalConfig is a singleton and uses atomic flag,
|
||||
// we cannot easily reset it. In a real test environment, you might
|
||||
// need to use a testing framework that supports fixture reset or
|
||||
// modify the GlobalConfig to support reset for testing purposes.
|
||||
}
|
||||
};
|
||||
|
||||
TEST_F(ConfigTest, InitializeWithDefaultConfig) {
|
||||
GlobalConfig::ConfigData config;
|
||||
|
||||
// Test initialization with default config
|
||||
auto status = GlobalConfig::Instance().Initialize(config);
|
||||
ASSERT_TRUE(status.ok()) << "Initialization failed: " << status.message();
|
||||
|
||||
// Verify default values
|
||||
ASSERT_GT(GlobalConfig::Instance().memory_limit_bytes(), 0);
|
||||
ASSERT_EQ(GlobalConfig::Instance().log_level(),
|
||||
GlobalConfig::LogLevel::kWarn);
|
||||
ASSERT_EQ(GlobalConfig::Instance().log_type(), "ConsoleLogger");
|
||||
ASSERT_GT(GlobalConfig::Instance().query_thread_count(), 0);
|
||||
ASSERT_EQ(GlobalConfig::Instance().invert_to_forward_scan_ratio(), 0.9f);
|
||||
ASSERT_EQ(GlobalConfig::Instance().brute_force_by_keys_ratio(), 0.1f);
|
||||
ASSERT_EQ(GlobalConfig::Instance().fts_brute_force_by_keys_ratio(), 0.05f);
|
||||
ASSERT_GT(GlobalConfig::Instance().optimize_thread_count(), 0);
|
||||
}
|
||||
|
||||
TEST_F(ConfigTest, InitializeWithCustomConsoleLogConfig) {
|
||||
GlobalConfig::ConfigData config;
|
||||
config.log_config = std::make_shared<GlobalConfig::ConsoleLogConfig>(
|
||||
GlobalConfig::LogLevel::kDebug);
|
||||
config.memory_limit_bytes = 1024 * 1024 * 1024; // 1GB
|
||||
config.query_thread_count = 4;
|
||||
config.optimize_thread_count = 2;
|
||||
|
||||
auto status = GlobalConfig::Instance().Initialize(config);
|
||||
// First initialization should succeed
|
||||
if (status.code() == StatusCode::INVALID_ARGUMENT &&
|
||||
status.message().find("already initialized") != std::string::npos) {
|
||||
// If already initialized, skip this test
|
||||
GTEST_SKIP() << "GlobalConfig already initialized";
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(ConfigTest, InitializeWithCustomFileLogConfig) {
|
||||
GlobalConfig::ConfigData config;
|
||||
auto file_config = std::make_shared<GlobalConfig::FileLogConfig>(
|
||||
GlobalConfig::LogLevel::kInfo, "/tmp/logs", "test.log", 1024, 14);
|
||||
config.log_config = file_config;
|
||||
config.memory_limit_bytes = 2 * 1024 * 1024 * 1024ULL; // 2GB
|
||||
config.query_thread_count = 8;
|
||||
config.optimize_thread_count = 4;
|
||||
|
||||
auto status = GlobalConfig::Instance().Initialize(config);
|
||||
// First initialization should succeed
|
||||
if (status.code() == StatusCode::INVALID_ARGUMENT &&
|
||||
status.message().find("already initialized") != std::string::npos) {
|
||||
// If already initialized, skip this test
|
||||
GTEST_SKIP() << "GlobalConfig already initialized";
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(ConfigTest, DoubleInitializationSilentlyFails) {
|
||||
GlobalConfig::ConfigData config;
|
||||
|
||||
auto status1 = GlobalConfig::Instance().Initialize(config);
|
||||
// If first initialization failed due to already being initialized
|
||||
if (status1.code() == StatusCode::INVALID_ARGUMENT &&
|
||||
status1.message().find("already initialized") != std::string::npos) {
|
||||
// Try again with a fresh config
|
||||
auto status2 = GlobalConfig::Instance().Initialize(config);
|
||||
ASSERT_FALSE(status2.ok());
|
||||
ASSERT_EQ(status2.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status2.message().find("already initialized"), std::string::npos);
|
||||
} else {
|
||||
// First initialization succeeded, second should fail
|
||||
ASSERT_TRUE(status1.ok());
|
||||
|
||||
// The second initialization is allowed but becomes a no-op
|
||||
auto status2 = GlobalConfig::Instance().Initialize(config);
|
||||
ASSERT_TRUE(status2.ok());
|
||||
}
|
||||
}
|
||||
|
||||
TEST_F(ConfigTest, ValidateConfigWithInvalidMemoryLimit) {
|
||||
GlobalConfig::ConfigData config;
|
||||
config.memory_limit_bytes = 0; // Invalid value
|
||||
|
||||
GlobalConfig
|
||||
config_instance; // Create a local instance for testing validation
|
||||
auto status = config_instance.Validate(config);
|
||||
ASSERT_FALSE(status.ok());
|
||||
ASSERT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status.message().find("memory_limit_bytes must be greater than"),
|
||||
std::string::npos);
|
||||
}
|
||||
|
||||
TEST_F(ConfigTest, ValidateConfigWithInvalidQueryThreadCount) {
|
||||
GlobalConfig::ConfigData config;
|
||||
config.query_thread_count = 0; // Invalid value
|
||||
|
||||
GlobalConfig config_instance;
|
||||
auto status = config_instance.Validate(config);
|
||||
ASSERT_FALSE(status.ok());
|
||||
ASSERT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status.message().find("query_thread_count must be greater than 0"),
|
||||
std::string::npos);
|
||||
}
|
||||
|
||||
TEST_F(ConfigTest, ValidateConfigWithInvalidRatios) {
|
||||
GlobalConfig::ConfigData config;
|
||||
|
||||
// Test invalid invert_to_forward_scan_ratio
|
||||
config.invert_to_forward_scan_ratio = -0.1f;
|
||||
GlobalConfig config_instance;
|
||||
auto status = config_instance.Validate(config);
|
||||
ASSERT_FALSE(status.ok());
|
||||
ASSERT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status.message().find(
|
||||
"invert_to_forward_scan_ratio must be between 0 and 1"),
|
||||
std::string::npos);
|
||||
|
||||
// Test invalid brute_force_by_keys_ratio
|
||||
config.invert_to_forward_scan_ratio = 0.9f; // Reset to valid value
|
||||
config.brute_force_by_keys_ratio = 1.5f; // Invalid value
|
||||
status = config_instance.Validate(config);
|
||||
ASSERT_FALSE(status.ok());
|
||||
ASSERT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status.message().find(
|
||||
"brute_force_by_keys_ratio must be between 0 and 1"),
|
||||
std::string::npos);
|
||||
|
||||
// Test invalid fts_brute_force_by_keys_ratio
|
||||
config.brute_force_by_keys_ratio = 0.1f; // Reset to valid value
|
||||
config.fts_brute_force_by_keys_ratio = -0.5f; // Invalid value
|
||||
status = config_instance.Validate(config);
|
||||
ASSERT_FALSE(status.ok());
|
||||
ASSERT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status.message().find(
|
||||
"fts_brute_force_by_keys_ratio must be between 0 and 1"),
|
||||
std::string::npos);
|
||||
}
|
||||
|
||||
TEST_F(ConfigTest, ValidateConfigWithInvalidFileLogSettings) {
|
||||
GlobalConfig::ConfigData config;
|
||||
|
||||
// Test with empty log directory
|
||||
auto file_config = std::make_shared<GlobalConfig::FileLogConfig>();
|
||||
file_config->dir = "";
|
||||
config.log_config = file_config;
|
||||
|
||||
GlobalConfig config_instance;
|
||||
auto status = config_instance.Validate(config);
|
||||
ASSERT_FALSE(status.ok());
|
||||
ASSERT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status.message().find("log_dir cannot be empty"),
|
||||
std::string::npos);
|
||||
|
||||
// Test with empty basename
|
||||
file_config->dir = "/tmp/logs";
|
||||
file_config->basename = "";
|
||||
status = config_instance.Validate(config);
|
||||
ASSERT_FALSE(status.ok());
|
||||
ASSERT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status.message().find("log_file basename cannot be empty"),
|
||||
std::string::npos);
|
||||
|
||||
// Test with invalid file size
|
||||
file_config->basename = "test.log";
|
||||
file_config->file_size = 0;
|
||||
status = config_instance.Validate(config);
|
||||
ASSERT_FALSE(status.ok());
|
||||
ASSERT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status.message().find("log file_size must be greater than"),
|
||||
std::string::npos);
|
||||
|
||||
// Test with invalid overdue days
|
||||
file_config->file_size = 1024;
|
||||
file_config->overdue_days = 0;
|
||||
status = config_instance.Validate(config);
|
||||
ASSERT_FALSE(status.ok());
|
||||
ASSERT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
ASSERT_NE(status.message().find("log_overdue_days must be greater than 0"),
|
||||
std::string::npos);
|
||||
}
|
||||
|
||||
TEST_F(ConfigTest, LogLevelEnumValues) {
|
||||
ASSERT_EQ(static_cast<int>(GlobalConfig::LogLevel::kDebug), 0);
|
||||
ASSERT_EQ(static_cast<int>(GlobalConfig::LogLevel::kInfo), 1);
|
||||
ASSERT_EQ(static_cast<int>(GlobalConfig::LogLevel::kWarn), 2);
|
||||
ASSERT_EQ(static_cast<int>(GlobalConfig::LogLevel::kError), 3);
|
||||
ASSERT_EQ(static_cast<int>(GlobalConfig::LogLevel::kFatal), 4);
|
||||
}
|
||||
|
||||
TEST_F(ConfigTest, LogConfigPolymorphism) {
|
||||
auto console_config = std::make_shared<GlobalConfig::ConsoleLogConfig>();
|
||||
auto file_config = std::make_shared<GlobalConfig::FileLogConfig>();
|
||||
|
||||
ASSERT_EQ(console_config->GetLoggerType(), CONSOLE_LOG_TYPE_NAME);
|
||||
ASSERT_EQ(file_config->GetLoggerType(), FILE_LOG_TYPE_NAME);
|
||||
}
|
||||
|
||||
// jieba_dict_dir is the only ConfigData field that can be written outside
|
||||
// of Initialize() — language SDKs call set_default_jieba_dict_dir() at
|
||||
// module-load to register the dict path they bundled. The setter is
|
||||
// independent of the Initialize() one-shot lifecycle.
|
||||
TEST_F(ConfigTest, JiebaDictDirSetterIsIndependentOfInitialize) {
|
||||
auto saved = GlobalConfig::Instance().jieba_dict_dir();
|
||||
|
||||
// Setter works regardless of whether Initialize was called.
|
||||
GlobalConfig::Instance().set_default_jieba_dict_dir("/tmp/zvec/dict-A");
|
||||
ASSERT_EQ(GlobalConfig::Instance().jieba_dict_dir(), "/tmp/zvec/dict-A");
|
||||
|
||||
// Last writer wins.
|
||||
GlobalConfig::Instance().set_default_jieba_dict_dir("/tmp/zvec/dict-B");
|
||||
ASSERT_EQ(GlobalConfig::Instance().jieba_dict_dir(), "/tmp/zvec/dict-B");
|
||||
|
||||
// Empty clears.
|
||||
GlobalConfig::Instance().set_default_jieba_dict_dir("");
|
||||
ASSERT_EQ(GlobalConfig::Instance().jieba_dict_dir(), "");
|
||||
|
||||
GlobalConfig::Instance().set_default_jieba_dict_dir(saved);
|
||||
}
|
||||
@@ -0,0 +1,160 @@
|
||||
// Copyright 2025-present the zvec project
|
||||
//
|
||||
// Licensed 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 "zvec/db/status.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
using namespace zvec;
|
||||
|
||||
TEST(StatusTest, DefaultConstructor) {
|
||||
Status status;
|
||||
EXPECT_TRUE(status.ok());
|
||||
EXPECT_EQ(status.code(), StatusCode::OK);
|
||||
EXPECT_EQ(status.message(), "");
|
||||
}
|
||||
|
||||
TEST(StatusTest, ConstructorWithCodeAndMessage) {
|
||||
std::string msg = "Test error message";
|
||||
Status status(StatusCode::INVALID_ARGUMENT, msg);
|
||||
|
||||
EXPECT_FALSE(status.ok());
|
||||
EXPECT_EQ(status.code(), StatusCode::INVALID_ARGUMENT);
|
||||
EXPECT_EQ(status.message(), msg);
|
||||
}
|
||||
|
||||
TEST(StatusTest, ConstructorWithRvalueMessage) {
|
||||
std::string msg = "Test error message";
|
||||
Status status(StatusCode::NOT_FOUND, std::move(msg));
|
||||
|
||||
EXPECT_FALSE(status.ok());
|
||||
EXPECT_EQ(status.code(), StatusCode::NOT_FOUND);
|
||||
EXPECT_EQ(status.message(), "Test error message");
|
||||
}
|
||||
|
||||
TEST(StatusTest, CopyConstructor) {
|
||||
Status original(StatusCode::INTERNAL_ERROR, "Copy test");
|
||||
Status copy(original);
|
||||
|
||||
EXPECT_FALSE(copy.ok());
|
||||
EXPECT_EQ(copy.code(), StatusCode::INTERNAL_ERROR);
|
||||
EXPECT_EQ(copy.message(), "Copy test");
|
||||
EXPECT_EQ(original.code(), copy.code());
|
||||
EXPECT_EQ(original.message(), copy.message());
|
||||
}
|
||||
|
||||
TEST(StatusTest, CopyAssignment) {
|
||||
Status original(StatusCode::PERMISSION_DENIED, "Assignment test");
|
||||
Status assigned;
|
||||
assigned = original;
|
||||
|
||||
EXPECT_FALSE(assigned.ok());
|
||||
EXPECT_EQ(assigned.code(), StatusCode::PERMISSION_DENIED);
|
||||
EXPECT_EQ(assigned.message(), "Assignment test");
|
||||
}
|
||||
|
||||
TEST(StatusTest, MoveConstructor) {
|
||||
Status original(StatusCode::RESOURCE_EXHAUSTED, "Move test");
|
||||
Status moved(std::move(original));
|
||||
|
||||
EXPECT_FALSE(moved.ok());
|
||||
EXPECT_EQ(moved.code(), StatusCode::RESOURCE_EXHAUSTED);
|
||||
EXPECT_EQ(moved.message(), "Move test");
|
||||
}
|
||||
|
||||
TEST(StatusTest, MoveAssignment) {
|
||||
Status original(StatusCode::UNAVAILABLE, "Move assignment test");
|
||||
Status moved;
|
||||
moved = std::move(original);
|
||||
|
||||
EXPECT_FALSE(moved.ok());
|
||||
EXPECT_EQ(moved.code(), StatusCode::UNAVAILABLE);
|
||||
EXPECT_EQ(moved.message(), "Move assignment test");
|
||||
}
|
||||
|
||||
TEST(StatusTest, ComparisonOperators) {
|
||||
Status status1(StatusCode::INVALID_ARGUMENT, "Error 1");
|
||||
Status status2(StatusCode::INVALID_ARGUMENT, "Error 1");
|
||||
Status status3(StatusCode::NOT_FOUND, "Error 2");
|
||||
Status ok1;
|
||||
Status ok2;
|
||||
|
||||
EXPECT_TRUE(status1 == status2);
|
||||
EXPECT_FALSE(status1 == status3);
|
||||
EXPECT_TRUE(ok1 == ok2);
|
||||
EXPECT_FALSE(status1 == ok1);
|
||||
|
||||
EXPECT_FALSE(status1 != status2);
|
||||
EXPECT_TRUE(status1 != status3);
|
||||
EXPECT_FALSE(ok1 != ok2);
|
||||
EXPECT_TRUE(status1 != ok1);
|
||||
}
|
||||
|
||||
TEST(StatusTest, FactoryMethods) {
|
||||
auto invalid_arg = Status::InvalidArgument("Invalid arg: ", 42);
|
||||
EXPECT_FALSE(invalid_arg.ok());
|
||||
EXPECT_EQ(invalid_arg.code(), StatusCode::INVALID_ARGUMENT);
|
||||
EXPECT_FALSE(invalid_arg.message().empty());
|
||||
|
||||
auto not_found = Status::NotFound("Not found: ", "key");
|
||||
EXPECT_FALSE(not_found.ok());
|
||||
EXPECT_EQ(not_found.code(), StatusCode::NOT_FOUND);
|
||||
EXPECT_FALSE(not_found.message().empty());
|
||||
|
||||
auto already_exists = Status::AlreadyExists("Already exists: ", "item");
|
||||
EXPECT_FALSE(already_exists.ok());
|
||||
EXPECT_EQ(already_exists.code(), StatusCode::ALREADY_EXISTS);
|
||||
EXPECT_FALSE(already_exists.message().empty());
|
||||
|
||||
auto internal_error = Status::InternalError("Internal error: ", "details");
|
||||
EXPECT_FALSE(internal_error.ok());
|
||||
EXPECT_EQ(internal_error.code(), StatusCode::INTERNAL_ERROR);
|
||||
EXPECT_FALSE(internal_error.message().empty());
|
||||
|
||||
auto permission_denied =
|
||||
Status::PermissionDenied("Permission denied for: ", "resource");
|
||||
EXPECT_FALSE(permission_denied.ok());
|
||||
EXPECT_EQ(permission_denied.code(), StatusCode::PERMISSION_DENIED);
|
||||
EXPECT_FALSE(permission_denied.message().empty());
|
||||
}
|
||||
|
||||
TEST(StatusTest, OKFactory) {
|
||||
auto ok = Status::OK();
|
||||
EXPECT_TRUE(ok.ok());
|
||||
EXPECT_EQ(ok.code(), StatusCode::OK);
|
||||
EXPECT_EQ(ok.message(), "");
|
||||
}
|
||||
|
||||
TEST(StatusTest, CStringConversion) {
|
||||
Status status(StatusCode::UNKNOWN, "C string test");
|
||||
EXPECT_STREQ(status.c_str(), "C string test");
|
||||
|
||||
Status ok_status;
|
||||
EXPECT_STREQ(ok_status.c_str(), "");
|
||||
}
|
||||
|
||||
TEST(StatusTest, OutputStreamOperator) {
|
||||
Status status(StatusCode::INVALID_ARGUMENT, "Stream test");
|
||||
std::ostringstream oss;
|
||||
oss << status;
|
||||
EXPECT_FALSE(oss.str().empty());
|
||||
EXPECT_NE(oss.str().find(GetDefaultMessage(StatusCode::INVALID_ARGUMENT)),
|
||||
std::string::npos);
|
||||
EXPECT_NE(oss.str().find("Stream test"), std::string::npos);
|
||||
|
||||
Status ok_status;
|
||||
std::ostringstream oss2;
|
||||
oss2 << ok_status;
|
||||
EXPECT_FALSE(oss2.str().empty());
|
||||
EXPECT_NE(oss2.str().find("OK"), std::string::npos);
|
||||
}
|
||||
Reference in New Issue
Block a user