Skip to content

Commit

Permalink
[skip ci] update test_db
Browse files Browse the repository at this point in the history
  • Loading branch information
@youny626
youny626 committed Feb 25, 2020
1 parent df49da4 commit e3557f7
Showing 1 changed file with 31 additions and 39 deletions.
70 changes: 31 additions & 39 deletions core/unittest/db/test_db.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -372,14 +372,12 @@ TEST_F(DBTest, SEARCH_TEST) {
ASSERT_TRUE(stat.ok());

FIU_ENABLE_FIU("SqliteMetaImpl.FilesToSearch.throw_exception");
stat = db_->QueryByFileID(dummy_context_, TABLE_NAME, file_ids, k, 10, xq, result_ids,
result_distances);
stat = db_->QueryByFileID(dummy_context_, TABLE_NAME, file_ids, k, 10, xq, result_ids, result_distances);
ASSERT_FALSE(stat.ok());
fiu_disable("SqliteMetaImpl.FilesToSearch.throw_exception");

FIU_ENABLE_FIU("DBImpl.QueryByFileID.empty_files_array");
stat = db_->QueryByFileID(dummy_context_, TABLE_NAME, file_ids, k, 10, xq, result_ids,
result_distances);
stat = db_->QueryByFileID(dummy_context_, TABLE_NAME, file_ids, k, 10, xq, result_ids, result_distances);
ASSERT_FALSE(stat.ok());
fiu_disable("DBImpl.QueryByFileID.empty_files_array");
}
Expand Down Expand Up @@ -472,7 +470,7 @@ TEST_F(DBTest, PRELOADTABLE_TEST) {
ASSERT_FALSE(stat.ok());
fiu_disable("SqliteMetaImpl.FilesToSearch.throw_exception");

//create a partition
// create a partition
stat = db_->CreatePartition(TABLE_NAME, "part0", "0");
ASSERT_TRUE(stat.ok());
stat = db_->PreloadTable(TABLE_NAME);
Expand Down Expand Up @@ -575,8 +573,8 @@ TEST_F(DBTest, SHUTDOWN_TEST) {
stat = db_->QueryByFileID(dummy_context_, table_info.table_id_, file_ids, 1, 1, xb, result_ids, result_distances);
ASSERT_FALSE(stat.ok());

stat = db_->Query(dummy_context_, table_info.table_id_, tags, 1, 1,
milvus::engine::VectorsData(), result_ids, result_distances);
stat = db_->Query(dummy_context_, table_info.table_id_, tags, 1, 1, milvus::engine::VectorsData(), result_ids,
result_distances);
ASSERT_FALSE(stat.ok());

stat = db_->DropTable(table_info.table_id_);
Expand All @@ -587,20 +585,19 @@ TEST_F(DBTest, BACK_TIMER_THREAD_1) {
fiu_init(0);
milvus::engine::meta::TableSchema table_info = BuildTableSchema();
milvus::Status stat;
//test background timer thread
// test background timer thread
{
FIU_ENABLE_FIU("DBImpl.StartMetricTask.InvalidTotalCache");
FIU_ENABLE_FIU("SqliteMetaImpl.FilesToMerge.throw_exception");
stat = db_->CreateTable(table_info);
ASSERT_TRUE(stat.ok());

//insert some vector to create some tablefiles
int64_t nb = VECTOR_COUNT;
milvus::engine::VectorsData xb;
BuildVectors(nb, 0, xb);

// insert some vector to create some tablefiles
int loop = 10;
for (auto i = 0; i < loop; ++i) {
int64_t nb = VECTOR_COUNT;
milvus::engine::VectorsData xb;
BuildVectors(nb, , xb);
db_->InsertVectors(TABLE_NAME, "", xb);
ASSERT_EQ(xb.id_array_.size(), nb);
}
Expand All @@ -626,18 +623,16 @@ TEST_F(DBTest, BACK_TIMER_THREAD_2) {
stat = db_->CreateTable(table_info);
ASSERT_TRUE(stat.ok());

//insert some vector to create some tablefiles
int64_t nb = VECTOR_COUNT;
milvus::engine::VectorsData xb;
BuildVectors(nb, 0, xb);

// insert some vector to create some tablefiles
int loop = 10;
for (auto i = 0; i < loop; ++i) {
int64_t nb = VECTOR_COUNT;
milvus::engine::VectorsData xb;
BuildVectors(nb, , xb);
db_->InsertVectors(TABLE_NAME, "", xb);
ASSERT_EQ(xb.id_array_.size(), nb);
}


FIU_ENABLE_FIU("SqliteMetaImpl.CreateTableFile.throw_exception");
std::this_thread::sleep_for(std::chrono::seconds(2));
db_->Stop();
Expand All @@ -652,13 +647,12 @@ TEST_F(DBTest, BACK_TIMER_THREAD_3) {
stat = db_->CreateTable(table_info);
ASSERT_TRUE(stat.ok());

//insert some vector to create some tablefiles
int64_t nb = VECTOR_COUNT;
milvus::engine::VectorsData xb;
BuildVectors(nb, 0, xb);

// insert some vector to create some tablefiles
int loop = 10;
for (auto i = 0; i < loop; ++i) {
int64_t nb = VECTOR_COUNT;
milvus::engine::VectorsData xb;
BuildVectors(nb, , xb);
db_->InsertVectors(TABLE_NAME, "", xb);
ASSERT_EQ(xb.id_array_.size(), nb);
}
Expand All @@ -678,13 +672,12 @@ TEST_F(DBTest, BACK_TIMER_THREAD_4) {
stat = db_->CreateTable(table_info);
ASSERT_TRUE(stat.ok());

//insert some vector to create some tablefiles
int64_t nb = VECTOR_COUNT;
milvus::engine::VectorsData xb;
BuildVectors(nb, 0, xb);

// insert some vector to create some tablefiles
int loop = 10;
for (auto i = 0; i < loop; ++i) {
int64_t nb = VECTOR_COUNT;
milvus::engine::VectorsData xb;
BuildVectors(nb, i, xb);
db_->InsertVectors(TABLE_NAME, "", xb);
ASSERT_EQ(xb.id_array_.size(), nb);
}
Expand Down Expand Up @@ -783,7 +776,7 @@ TEST_F(DBTest, PARTITION_TEST) {
db_->InsertVectors(table_name, partition_tag, xb);
ASSERT_EQ(vector_ids.size(), INSERT_BATCH);

//insert data into not existed partition
// insert data into not existed partition
stat = db_->InsertVectors(TABLE_NAME, "notexist", xb);
ASSERT_FALSE(stat.ok());
}
Expand Down Expand Up @@ -813,10 +806,10 @@ TEST_F(DBTest, PARTITION_TEST) {
ASSERT_FALSE(stat.ok());
fiu_disable("DBImpl.BuildTableIndexRecursively.fail_build_table_Index_for_partition");

FIU_ENABLE_FIU("DBImpl.BuildTableIndexRecursively.not_empty_err_msg");
FIU_ENABLE_FIU("DBImpl.BuildTableIndexRecursively.not_empty_err_msg");
stat = db_->CreateIndex(table_info.table_id_, index);
ASSERT_FALSE(stat.ok());
fiu_disable("DBImpl.BuildTableIndexRecursively.not_empty_err_msg");
fiu_disable("DBImpl.BuildTableIndexRecursively.not_empty_err_msg");

uint64_t row_count = 0;
stat = db_->GetTableRowCount(TABLE_NAME, row_count);
Expand Down Expand Up @@ -956,11 +949,11 @@ TEST_F(DBTest2, DELETE_TEST) {
milvus::engine::TableIndex index;
stat = db_->CreateIndex(TABLE_NAME, index);

//create partition, drop table will drop partition recursively
// create partition, drop table will drop partition recursively
stat = db_->CreatePartition(TABLE_NAME, "part0", "0");
ASSERT_TRUE(stat.ok());

//fail drop table
// fail drop table
fiu_init(0);
FIU_ENABLE_FIU("DBImpl.DropTableRecursively.failed");
stat = db_->DropTable(TABLE_NAME);
Expand Down Expand Up @@ -1062,10 +1055,9 @@ TEST_F(DBTestWAL, DB_STOP_TEST) {
ASSERT_TRUE(stat.ok());

uint64_t qb = 100;
milvus::engine::VectorsData qxb;
BuildVectors(qb, 0, qxb);

for (int i = 0; i < 5; i++) {
milvus::engine::VectorsData qxb;
BuildVectors(qb, i, qxb);
stat = db_->InsertVectors(table_info.table_id_, "", qxb);
ASSERT_TRUE(stat.ok());
}
Expand All @@ -1091,10 +1083,10 @@ TEST_F(DBTestWALRecovery, RECOVERY_WITH_NO_ERROR) {
ASSERT_TRUE(stat.ok());

uint64_t qb = 100;
milvus::engine::VectorsData qxb;
BuildVectors(qb, 0, qxb);

for (int i = 0; i < 5; i++) {
milvus::engine::VectorsData qxb;
BuildVectors(qb, i, qxb);
stat = db_->InsertVectors(table_info.table_id_, "", qxb);
ASSERT_TRUE(stat.ok());
}
Expand Down

0 comments on commit e3557f7

Please sign in to comment.