From d6199efe75068646a97a441c3a8dc188cebe46ad Mon Sep 17 00:00:00 2001
From: Wesley <10225102463@stu.ecnu.edu.cn>
Date: Wed, 6 Nov 2024 23:42:08 +0800
Subject: [PATCH] modify new test

---
 test/ttl_test.cc | 86 +++++++++++++++++++++++++++++++-------------------------
 1 file changed, 48 insertions(+), 38 deletions(-)

diff --git a/test/ttl_test.cc b/test/ttl_test.cc
index db6e757..fdf882d 100644
--- a/test/ttl_test.cc
+++ b/test/ttl_test.cc
@@ -40,41 +40,41 @@ void GetData(DB *db, int size = (1 << 30)) {
   }
 }
 
-TEST(TestTTL, ReadTTL) {
-    DB *db;
-    if(OpenDB("testdb", &db).ok() == false) {
-        std::cerr << "open db failed" << std::endl;
-        abort();
-    }
-
-    uint64_t ttl = 20;
-
-    InsertData(db, ttl);
-
-    ReadOptions readOptions;
-    Status status;
-    int key_num = data_size / value_size;
-    srand(0);
-    for (int i = 0; i < 100; i++) {
-        int key_ = rand() % key_num+1;
-        std::string key = std::to_string(key_);
-        std::string value;
-        status = db->Get(readOptions, key, &value);
-        ASSERT_TRUE(status.ok());
-    }
-
-    Env::Default()->SleepForMicroseconds(ttl * 1000000);
-
-    for (int i = 0; i < 100; i++) {
-        int key_ = rand() % key_num+1;
-        std::string key = std::to_string(key_);
-        std::string value;
-        status = db->Get(readOptions, key, &value);
-        ASSERT_FALSE(status.ok());
-    }
-
-    delete db;
-}
+// TEST(TestTTL, ReadTTL) {
+//     DB *db;
+//     if(OpenDB("testdb", &db).ok() == false) {
+//         std::cerr << "open db failed" << std::endl;
+//         abort();
+//     }
+
+//     uint64_t ttl = 20;
+
+//     InsertData(db, ttl);
+
+//     ReadOptions readOptions;
+//     Status status;
+//     int key_num = data_size / value_size;
+//     srand(0);
+//     for (int i = 0; i < 100; i++) {
+//         int key_ = rand() % key_num+1;
+//         std::string key = std::to_string(key_);
+//         std::string value;
+//         status = db->Get(readOptions, key, &value);
+//         ASSERT_TRUE(status.ok());
+//     }
+
+//     Env::Default()->SleepForMicroseconds(ttl * 1000000);
+
+//     for (int i = 0; i < 100; i++) {
+//         int key_ = rand() % key_num+1;
+//         std::string key = std::to_string(key_);
+//         std::string value;
+//         status = db->Get(readOptions, key, &value);
+//         ASSERT_FALSE(status.ok());
+//     }
+
+//     delete db;
+// }
 
 TEST(TestTTL, CompactionTTL) {
     DB *db;
@@ -124,11 +124,21 @@ TEST(TestTTL, LastLevelCompaction) {
   db->GetApproximateSizes(ranges, 1, sizes);
   ASSERT_GT(sizes[0], 0);
 
-  std::string last_level_file_num;
+  int last_level_file_num;
+  std::string file_num;
   std::string last_level = "2";
-  db->GetProperty("leveldb.num-files-at-level" + last_level, &last_level_file_num);
+  for (int level = 0; level < 7; level++){
+    db->GetProperty("leveldb.num-files-at-level" + std::to_string(level), &file_num);
+    if (std::atoi(file_num.c_str()) > 0){
+      last_level = std::to_string(level);
+      last_level_file_num = std::atoi(file_num.c_str());
+    }
+    else{
+      break;
+    }
+  }
   std::cout << "File nums in last level: " << last_level_file_num << std::endl;
-  ASSERT_GT(std::atoi(last_level_file_num.c_str()), 0);
+  ASSERT_GT(last_level_file_num, 0);
 
   Env::Default()->SleepForMicroseconds(ttl * 1000000);