From d0b26bf0370703d4d7ff3d9258b64de8bd26ae72 Mon Sep 17 00:00:00 2001
From: yihaoDeng <yhdeng@taosdata.com>
Date: Wed, 21 Feb 2024 14:07:46 +0800
Subject: [PATCH] add coveriry

---
 source/libs/index/test/CMakeLists.txt |  4 ++
 source/libs/index/test/fstTest.cc     | 53 +++++++++++++++++----------
 2 files changed, 38 insertions(+), 19 deletions(-)

diff --git a/source/libs/index/test/CMakeLists.txt b/source/libs/index/test/CMakeLists.txt
index 1ee835573b..eff74f93d2 100644
--- a/source/libs/index/test/CMakeLists.txt
+++ b/source/libs/index/test/CMakeLists.txt
@@ -175,4 +175,8 @@ IF(NOT TD_DARWIN)
     NAME idxFstUT 
     COMMAND idxFstUT 
   )
+  add_test(
+    NAME idxFstTest
+    COMMAND idxFstTest 
+  )
 ENDIF ()
diff --git a/source/libs/index/test/fstTest.cc b/source/libs/index/test/fstTest.cc
index b889a2209a..2b1744c28d 100644
--- a/source/libs/index/test/fstTest.cc
+++ b/source/libs/index/test/fstTest.cc
@@ -1,4 +1,5 @@
 
+#include <gtest/gtest.h>
 #include <algorithm>
 #include <iostream>
 #include <string>
@@ -14,6 +15,12 @@
 #include "tutil.h"
 void* callback(void* s) { return s; }
 
+class FstEnv : public ::testing::Test {
+ protected:
+  virtual void SetUp() {}
+  virtual void TearDown() {}
+};
+
 static std::string fileName = TD_TMP_DIR_PATH "tindex.tindex";
 class FstWriter {
  public:
@@ -154,7 +161,7 @@ class FstReadMemory {
   int32_t     _size;
 };
 
-#define L 100
+#define L 200
 #define M 100
 #define N 100
 
@@ -637,23 +644,31 @@ void iterTFileReader(char* path, char* uid, char* colName, char* ver) {
   tfileIteratorDestroy(iter);
 }
 
-int main(int argc, char* argv[]) {
-  // tool to check all kind of fst test
-  // if (argc > 1) { validateTFile(argv[1]); }
-  // if (argc > 4) {
-  // path suid colName ver
-  // iterTFileReader(argv[1], argv[2], argv[3], argv[4]);
-  //}
-  checkFstCheckIterator1();
-  // checkFstCheckIterator2();
-  // checkFstCheckIteratorPrefix();
-  // checkFstCheckIteratorRange1();
-  // checkFstCheckIteratorRange2();
-  // checkFstCheckIteratorRange3();
-  // checkFstLongTerm();
-  // checkFstPrefixSearch();
+// int main(int argc, char* argv[]) {
+//   // tool to check all kind of fst test
+//   // if (argc > 1) { validateTFile(argv[1]); }
+//   // if (argc > 4) {
+//   // path suid colName ver
+//   // iterTFileReader(argv[1], argv[2], argv[3], argv[4]);
+//   //}
+//   checkFstCheckIterator1();
+//   // checkFstCheckIterator2();
+//   // checkFstCheckIteratorPrefix();
+//   // checkFstCheckIteratorRange1();
+//   // checkFstCheckIteratorRange2();
+//   // checkFstCheckIteratorRange3();
+//   // checkFstLongTerm();
+//   // checkFstPrefixSearch();
 
-  // checkMillonWriteAndReadOfFst();
+//   // checkMillonWriteAndReadOfFst();
 
-  return 1;
-}
+//   return 1;
+// }
+TEST_F(FstEnv, checkIterator1) { checkFstCheckIterator1(); }
+TEST_F(FstEnv, checkItertor2) { checkFstCheckIterator2(); }
+TEST_F(FstEnv, checkPrefix) { checkFstCheckIteratorPrefix(); }
+TEST_F(FstEnv, checkRange1) { checkFstCheckIteratorRange1(); }
+TEST_F(FstEnv, checkRange2) { checkFstCheckIteratorRange2(); }
+TEST_F(FstEnv, checkRange3) { checkFstCheckIteratorRange3(); }
+TEST_F(FstEnv, checkLongTerm) { checkFstLongTerm(); }
+TEST_F(FstEnv, checkMillonWriteData) { checkMillonWriteAndReadOfFst(); }