mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-31 16:05:30 +08:00
77 lines
2.9 KiB
C++
77 lines
2.9 KiB
C++
////////////////////////////////////////////////////////////////////////////////
|
|
// Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
|
|
// Unauthorized copying of this file, via any medium is strictly prohibited.
|
|
// Proprietary and confidential.
|
|
////////////////////////////////////////////////////////////////////////////////
|
|
#include <gtest/gtest.h>
|
|
|
|
#include "utils/ValidationUtil.h"
|
|
#include "utils/Error.h"
|
|
#include "db/ExecutionEngine.h"
|
|
|
|
#include <string>
|
|
|
|
using namespace zilliz::milvus;
|
|
using namespace zilliz::milvus::server;
|
|
|
|
TEST(ValidationUtilTest, TableNameTest) {
|
|
std::string table_name = "Normal123_";
|
|
ServerError res = ValidationUtil::ValidateTableName(table_name);
|
|
ASSERT_EQ(res, SERVER_SUCCESS);
|
|
|
|
table_name = "12sds";
|
|
res = ValidationUtil::ValidateTableName(table_name);
|
|
ASSERT_EQ(res, SERVER_INVALID_TABLE_NAME);
|
|
|
|
table_name = "";
|
|
res = ValidationUtil::ValidateTableName(table_name);
|
|
ASSERT_EQ(res, SERVER_INVALID_TABLE_NAME);
|
|
|
|
table_name = "_asdasd";
|
|
res = ValidationUtil::ValidateTableName(table_name);
|
|
ASSERT_EQ(res, SERVER_SUCCESS);
|
|
|
|
table_name = "!@#!@";
|
|
res = ValidationUtil::ValidateTableName(table_name);
|
|
ASSERT_EQ(res, SERVER_INVALID_TABLE_NAME);
|
|
|
|
table_name = "_!@#!@";
|
|
res = ValidationUtil::ValidateTableName(table_name);
|
|
ASSERT_EQ(res, SERVER_INVALID_TABLE_NAME);
|
|
|
|
table_name = "中文";
|
|
res = ValidationUtil::ValidateTableName(table_name);
|
|
ASSERT_EQ(res, SERVER_INVALID_TABLE_NAME);
|
|
|
|
|
|
table_name = std::string(10000, 'a');
|
|
res = ValidationUtil::ValidateTableName(table_name);
|
|
ASSERT_EQ(res, SERVER_INVALID_TABLE_NAME);
|
|
}
|
|
|
|
|
|
TEST(ValidationUtilTest, TableDimensionTest) {
|
|
ASSERT_EQ(ValidationUtil::ValidateTableDimension(-1), SERVER_INVALID_VECTOR_DIMENSION);
|
|
ASSERT_EQ(ValidationUtil::ValidateTableDimension(0), SERVER_INVALID_VECTOR_DIMENSION);
|
|
ASSERT_EQ(ValidationUtil::ValidateTableDimension(16385), SERVER_INVALID_VECTOR_DIMENSION);
|
|
ASSERT_EQ(ValidationUtil::ValidateTableDimension(16384), SERVER_SUCCESS);
|
|
ASSERT_EQ(ValidationUtil::ValidateTableDimension(1), SERVER_SUCCESS);
|
|
}
|
|
|
|
TEST(ValidationUtilTest, TableIndexTypeTest) {
|
|
ASSERT_EQ(ValidationUtil::ValidateTableIndexType((int)engine::EngineType::INVALID), SERVER_INVALID_INDEX_TYPE);
|
|
for(int i = 1; i <= (int)engine::EngineType::MAX_VALUE; i++) {
|
|
ASSERT_EQ(ValidationUtil::ValidateTableIndexType(i), SERVER_SUCCESS);
|
|
}
|
|
ASSERT_EQ(ValidationUtil::ValidateTableIndexType((int)engine::EngineType::MAX_VALUE + 1), SERVER_INVALID_INDEX_TYPE);
|
|
}
|
|
|
|
TEST(ValidationUtilTest, ValidateGpuTest) {
|
|
ASSERT_EQ(ValidationUtil::ValidateGpuIndex(0), SERVER_SUCCESS);
|
|
ASSERT_NE(ValidationUtil::ValidateGpuIndex(100), SERVER_SUCCESS);
|
|
|
|
size_t memory = 0;
|
|
ASSERT_EQ(ValidationUtil::GetGpuMemory(0, memory), SERVER_SUCCESS);
|
|
ASSERT_NE(ValidationUtil::GetGpuMemory(100, memory), SERVER_SUCCESS);
|
|
}
|