From aea7cc1630de080d6ae55803ab5f179a5a42b91c Mon Sep 17 00:00:00 2001
From: yukun <kun.yu@zilliz.com>
Date: Tue, 28 Sep 2021 17:48:08 +0800
Subject: [PATCH] Add rocksdb_kv ut (#8769)

Signed-off-by: fishpenguin <kun.yu@zilliz.com>
---
 internal/kv/rocksdb/rocksdb_kv_test.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/internal/kv/rocksdb/rocksdb_kv_test.go b/internal/kv/rocksdb/rocksdb_kv_test.go
index da5c9859a1..2fccd868a3 100644
--- a/internal/kv/rocksdb/rocksdb_kv_test.go
+++ b/internal/kv/rocksdb/rocksdb_kv_test.go
@@ -64,6 +64,21 @@ func TestRocksdbKV(t *testing.T) {
 	assert.Equal(t, len(vals), len(keys))
 	assert.Equal(t, vals[0], "123")
 	assert.Equal(t, vals[1], "456")
+
+	err = rocksdbKV.MultiRemove(keys)
+	assert.NoError(t, err)
+
+	saves := map[string]string{
+		"s_1": "111",
+		"s_2": "222",
+		"s_3": "333",
+	}
+	removals := []string{"key_3"}
+	err = rocksdbKV.MultiSaveAndRemove(saves, removals)
+	assert.NoError(t, err)
+
+	err = rocksdbKV.DeleteRange("s_1", "s_3")
+	assert.NoError(t, err)
 }
 
 func TestRocksdbKV_Prefix(t *testing.T) {