stores: add rocksdb tests

Cargo.toml

@@ -42,6 +42,9 @@ serde = { version = "1.0.204", features = ["derive"] }
 toml = "0.8.19"
 zipf = "7.0.1"
 
+[dev-dependencies]
+tempfile = "3.10.1"
+
 [features]
 chashmap = ["dep:chashmap"]
 contrie = ["dep:contrie"]

src/bench.rs

@@ -1393,6 +1393,20 @@ mod tests {
         ));
         example(OPT);
     }
+
+    #[test]
+    fn example_rocksdb() {
+        let tmp_dir = tempfile::tempdir().unwrap();
+        let opt = format!(
+            r#"
+            [map]
+            name = "rocksdb"
+            path = "{}"
+            "#,
+            tmp_dir.path().to_str().unwrap().to_string()
+        );
+        example(&opt);
+    }
 }
 
 // }}} tests

src/server.rs

@@ -824,6 +824,28 @@ mod tests {
         simple(map);
     }
 
+    #[test]
+    fn simple_mutex_btreemap() {
+        let map = BenchKVMap::Regular(Box::new(btreemap::MutexBTreeMap::new()));
+        simple(map);
+    }
+
+    #[test]
+    fn simple_rwlock_btreemap() {
+        let map = BenchKVMap::Regular(Box::new(btreemap::RwLockBTreeMap::new()));
+        simple(map);
+    }
+
+    #[test]
+    fn simple_rocksdb() {
+        let tmp_dir = tempfile::tempdir().unwrap();
+        let opt = rocksdb::RocksDBOpt {
+            path: tmp_dir.path().to_str().unwrap().to_string(),
+        };
+        let map = BenchKVMap::Regular(Box::new(rocksdb::RocksDB::new(&opt)));
+        simple(map);
+    }
+
     fn batch(map: BenchKVMap) {
         const NR_CLIENTS: usize = 8;
         const NR_BATCHES: usize = 1000;
@@ -977,4 +999,14 @@ mod tests {
         let map = BenchKVMap::Regular(Box::new(btreemap::RwLockBTreeMap::new()));
         batch(map);
     }
+
+    #[test]
+    fn batch_rocksdb() {
+        let tmp_dir = tempfile::tempdir().unwrap();
+        let opt = rocksdb::RocksDBOpt {
+            path: tmp_dir.path().to_str().unwrap().to_string(),
+        };
+        let map = BenchKVMap::Regular(Box::new(rocksdb::RocksDB::new(&opt)));
+        batch(map);
+    }
 }

src/stores.rs

@@ -231,4 +231,15 @@ mod tests {
         let mut map = scc::SccHashMap::new();
         map_test(&mut map);
     }
+
+    #[test]
+    #[cfg(feature = "rocksdb")]
+    fn rocksdb() {
+        let tmp_dir = tempfile::tempdir().unwrap();
+        let opt = rocksdb::RocksDBOpt {
+            path: tmp_dir.path().to_str().unwrap().to_string(),
+        };
+        let mut map = rocksdb::RocksDB::new(&opt);
+        map_test(&mut map);
+    }
 }

src/stores/rocksdb.rs

@@ -17,7 +17,7 @@ use serde::Deserialize;
 
 #[derive(Deserialize)]
 pub struct RocksDBOpt {
-    path: String,
+    pub path: String,
 }
 
 #[derive(Clone)]