add unittest case

core/src/db/IndexFailedChecker.cpp

@@ -75,6 +75,9 @@ IndexFailedChecker::MarkSucceedIndexFile(const meta::TableFileSchema& file) {
     auto iter = index_failed_files_.find(file.table_id_);
     if (iter != index_failed_files_.end()) {
         iter->second.erase(file.file_id_);
+        if (iter->second.empty()) {
+            index_failed_files_.erase(file.table_id_);
+        }
     }
 
     return Status::OK();

core/src/db/OngoingFileChecker.cpp

@@ -89,6 +89,9 @@ OngoingFileChecker::UnmarkOngoingFileNoLock(const meta::TableFileSchema& table_f
     auto iter = ongoing_files_.find(table_file.table_id_);
     if (iter != ongoing_files_.end()) {
         iter->second.erase(table_file.file_id_);
+        if (iter->second.empty()) {
+            ongoing_files_.erase(table_file.table_id_);
+        }
     }
 
     return Status::OK();

core/unittest/db/test_misc.cpp

@@ -15,6 +15,8 @@
 // specific language governing permissions and limitations
 // under the License.
 
+#include "db/IndexFailedChecker.h"
+#include "db/OngoingFileChecker.h"
 #include "db/Options.h"
 #include "db/Utils.h"
 #include "db/engine/EngineFactory.h"
@@ -119,3 +121,65 @@ TEST(DBMiscTest, UTILS_TEST) {
     status = milvus::engine::utils::DeleteTableFilePath(options, file);
     ASSERT_TRUE(status.ok());
 }
+
+TEST(DBMiscTest, CHECKER_TEST) {
+    {
+        milvus::engine::IndexFailedChecker checker;
+        milvus::engine::meta::TableFileSchema schema;
+        schema.table_id_ = "aaa";
+        schema.file_id_ = "5000";
+        checker.MarkFailedIndexFile(schema);
+        schema.table_id_ = "bbb";
+        schema.file_id_ = "5001";
+        checker.MarkFailedIndexFile(schema);
+
+        std::vector<std::string> failed_files;
+        checker.GetFailedIndexFileOfTable("aaa", failed_files);
+        ASSERT_EQ(failed_files.size(), 1UL);
+
+        schema.table_id_ = "bbb";
+        schema.file_id_ = "5002";
+        checker.MarkFailedIndexFile(schema);
+        checker.MarkFailedIndexFile(schema);
+
+        milvus::engine::meta::TableFilesSchema table_files = {schema};
+        checker.IgnoreFailedIndexFiles(table_files);
+        ASSERT_TRUE(table_files.empty());
+
+        checker.GetFailedIndexFileOfTable("bbb", failed_files);
+        ASSERT_EQ(failed_files.size(), 2UL);
+
+        checker.MarkSucceedIndexFile(schema);
+        checker.GetFailedIndexFileOfTable("bbb", failed_files);
+        ASSERT_EQ(failed_files.size(), 1UL);
+    }
+
+    {
+        milvus::engine::OngoingFileChecker checker;
+        milvus::engine::meta::TableFileSchema schema;
+        schema.table_id_ = "aaa";
+        schema.file_id_ = "5000";
+        checker.MarkOngoingFile(schema);
+
+        auto ongoing_files = checker.GetOngoingFiles();
+        ASSERT_EQ(ongoing_files.size(), 1UL);
+
+        schema.table_id_ = "bbb";
+        schema.file_id_ = "5001";
+        milvus::engine::meta::TableFilesSchema table_files = {schema};
+        checker.MarkOngoingFiles(table_files);
+
+        ongoing_files = checker.GetOngoingFiles();
+        ASSERT_EQ(ongoing_files.size(), 2UL);
+
+        checker.UnmarkOngoingFile(schema);
+        ongoing_files = checker.GetOngoingFiles();
+        ASSERT_EQ(ongoing_files.size(), 1UL);
+
+        schema.table_id_ = "aaa";
+        schema.file_id_ = "5000";
+        checker.UnmarkOngoingFile(schema);
+        ongoing_files = checker.GetOngoingFiles();
+        ASSERT_EQ(ongoing_files.size(), 0UL);
+    }
+}