use test_db_prefix_scan for all 4 database types

romanz · Sep 11, 2024 · dcdc18b · dcdc18b
1 parent 8cfdae1
commit dcdc18b
Show file tree

Hide file tree

Showing 5 changed files with 56 additions and 21 deletions.
diff --git a/src/db/lmdb.rs b/src/db/lmdb.rs
@@ -259,3 +259,13 @@ impl DBStore {
         Ok(())
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::DBStore;
+
+    #[test]
+    fn test_db_prefix_scan() {
+        super::super::test_db_prefix_scan::<DBStore>();
+    }
+}
diff --git a/src/db/mod.rs b/src/db/mod.rs
@@ -88,3 +88,28 @@ pub(crate) fn hash_prefix_range(prefix: HashPrefix) -> impl RangeBounds<Serializ
 
     lower..=upper
 }
+
+#[cfg(test)]
+pub(crate) fn test_db_prefix_scan<D: Database>() {
+    let dir = tempfile::tempdir().unwrap();
+    let store = D::open(dir.path(), None, true).unwrap();
+
+    let items = [
+        *b"ab          ",
+        *b"abcdefgh    ",
+        *b"abcdefghj   ",
+        *b"abcdefghjk  ",
+        *b"abcdefghxyz ",
+        *b"abcdefgi    ",
+        *b"b           ",
+        *b"c           ",
+    ];
+
+    store.write(&WriteBatch {
+        txid_rows: items.to_vec(),
+        ..Default::default()
+    });
+
+    let rows = store.iter_txid(*b"abcdefgh");
+    assert_eq!(rows.collect::<Vec<_>>(), items[1..5]);
+}
diff --git a/src/db/redb.rs b/src/db/redb.rs
@@ -271,3 +271,13 @@ impl DBStore {
         Ok(())
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::DBStore;
+
+    #[test]
+    fn test_db_prefix_scan() {
+        super::super::test_db_prefix_scan::<DBStore>();
+    }
+}
diff --git a/src/db/rocksdb.rs b/src/db/rocksdb.rs
@@ -502,27 +502,7 @@ mod tests {
 
     #[test]
     fn test_db_prefix_scan() {
-        let dir = tempfile::tempdir().unwrap();
-        let store = DBStore::open(dir.path(), None, true).unwrap();
-
-        let items = [
-            *b"ab          ",
-            *b"abcdefgh    ",
-            *b"abcdefghj   ",
-            *b"abcdefghjk  ",
-            *b"abcdefghxyz ",
-            *b"abcdefgi    ",
-            *b"b           ",
-            *b"c           ",
-        ];
-
-        store.write(&WriteBatch {
-            txid_rows: items.to_vec(),
-            ..Default::default()
-        });
-
-        let rows = store.iter_txid(*b"abcdefgh");
-        assert_eq!(rows.collect::<Vec<_>>(), items[1..5]);
+        super::super::test_db_prefix_scan::<DBStore>();
     }
 
     #[test]

diff --git a/src/db/sled.rs b/src/db/sled.rs
@@ -198,3 +198,13 @@ impl DBStore {
         Ok(())
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::DBStore;
+
+    #[test]
+    fn test_db_prefix_scan() {
+        super::super::test_db_prefix_scan::<DBStore>();
+    }
+}