mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-08 10:08:42 +08:00
67 lines
2.0 KiB
C++
67 lines
2.0 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 <thread>
|
|
#include <easylogging++.h>
|
|
|
|
#include "utils.h"
|
|
#include "db/DBMetaImpl.h"
|
|
#include "db/Factories.h"
|
|
|
|
using namespace zilliz::vecwise::engine;
|
|
|
|
TEST_F(MetaTest, GROUP_TEST) {
|
|
auto group_id = "meta_test_group";
|
|
|
|
meta::GroupSchema group;
|
|
group.group_id = group_id;
|
|
auto status = impl_->add_group(group);
|
|
ASSERT_TRUE(status.ok());
|
|
|
|
auto gid = group.id;
|
|
group.id = -1;
|
|
status = impl_->get_group(group);
|
|
ASSERT_TRUE(status.ok());
|
|
ASSERT_EQ(group.id, gid);
|
|
ASSERT_EQ(group.group_id, group_id);
|
|
|
|
group.group_id = "not_found";
|
|
status = impl_->get_group(group);
|
|
ASSERT_TRUE(!status.ok());
|
|
|
|
group.group_id = group_id;
|
|
status = impl_->add_group(group);
|
|
ASSERT_TRUE(!status.ok());
|
|
}
|
|
|
|
TEST_F(MetaTest, GROUP_FILE_TEST) {
|
|
auto group_id = "meta_test_group";
|
|
|
|
meta::GroupSchema group;
|
|
group.group_id = group_id;
|
|
auto status = impl_->add_group(group);
|
|
|
|
meta::GroupFileSchema group_file;
|
|
group_file.group_id = group.group_id;
|
|
status = impl_->add_group_file(group_file);
|
|
ASSERT_TRUE(status.ok());
|
|
ASSERT_EQ(group_file.file_type, meta::GroupFileSchema::NEW);
|
|
|
|
auto file_id = group_file.file_id;
|
|
|
|
auto new_file_type = meta::GroupFileSchema::INDEX;
|
|
group_file.file_type = new_file_type;
|
|
|
|
status = impl_->update_group_file(group_file);
|
|
ASSERT_TRUE(status.ok());
|
|
ASSERT_EQ(group_file.file_type, new_file_type);
|
|
|
|
/* group_file.file_type = meta::GroupFileSchema::NEW; */
|
|
/* status = impl_->get_group_file(group_file.group_id, group_file.file_id, group_file); */
|
|
/* ASSERT_TRUE(status.ok()); */
|
|
/* ASSERT_EQ(group_file.file_type, new_file_type); */
|
|
}
|