From 21c61a25f6f2beda96131d791f729bbe9bd45554 Mon Sep 17 00:00:00 2001
From: Alex Chi <iskyzh@gmail.com>
Date: Tue, 30 Jan 2024 18:21:27 +0800
Subject: [PATCH] consider mvcc iterator in compaction test

Signed-off-by: Alex Chi <iskyzh@gmail.com>
---
 mini-lsm/src/tests/harness.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/mini-lsm/src/tests/harness.rs b/mini-lsm/src/tests/harness.rs
index 7987666..b41745b 100644
--- a/mini-lsm/src/tests/harness.rs
+++ b/mini-lsm/src/tests/harness.rs
@@ -295,6 +295,11 @@ pub fn check_compaction_ratio(storage: Arc<MiniLsm>) {
         };
         level_size.push(size);
     }
+    let extra_iterators = if TS_ENABLED {
+        1 /* txn local iterator for OCC */
+    } else {
+        0
+    };
     let num_iters = storage
         .scan(Bound::Unbounded, Bound::Unbounded)
         .unwrap()
@@ -326,7 +331,7 @@ pub fn check_compaction_ratio(storage: Arc<MiniLsm>) {
                 );
             }
             assert!(
-                num_iters <= l0_sst_num + num_memtables + max_levels,
+                num_iters <= l0_sst_num + num_memtables + max_levels + extra_iterators,
                 "we found {num_iters} iterators in your implementation, (l0_sst_num={l0_sst_num}, num_memtables={num_memtables}, max_levels={max_levels}) did you use concat iterators?"
             );
         }
@@ -354,7 +359,7 @@ pub fn check_compaction_ratio(storage: Arc<MiniLsm>) {
                 );
             }
             assert!(
-                num_iters <= l0_sst_num + num_memtables + max_levels,
+                num_iters <= l0_sst_num + num_memtables + max_levels + extra_iterators,
                 "we found {num_iters} iterators in your implementation, (l0_sst_num={l0_sst_num}, num_memtables={num_memtables}, max_levels={max_levels}) did you use concat iterators?"
             );
         }
@@ -396,7 +401,7 @@ pub fn check_compaction_ratio(storage: Arc<MiniLsm>) {
                 sum_size += this_size;
             }
             assert!(
-                num_iters <= num_memtables + num_tiers,
+                num_iters <= num_memtables + num_tiers + extra_iterators,
                 "we found {num_iters} iterators in your implementation, (num_memtables={num_memtables}, num_tiers={num_tiers}) did you use concat iterators?"
             );
         }