localstore: add tests for engine.MultiSeek

rjmcguire · Dec 21, 2015 · 56a19f4 · 56a19f4
1 parent 9bc1ea2
commit 56a19f4
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/store/localstore/boltdb/boltdb_test.go b/store/localstore/boltdb/boltdb_test.go
@@ -138,4 +138,14 @@ func (s *testSuite) TestDB(c *C) {
 	c.Assert(err, NotNil)
 	c.Assert(k, IsNil)
 	c.Assert(v, IsNil)
+
+	m := db.MultiSeek([][]byte{[]byte("z"), []byte("a"), []byte("a1")})
+	c.Assert(m, HasLen, 3)
+	c.Assert(m[0].Err, NotNil)
+	c.Assert(m[1].Err, IsNil)
+	c.Assert(m[1].Key, BytesEquals, []byte("a"))
+	c.Assert(m[1].Value, BytesEquals, []byte("1"))
+	c.Assert(m[2].Err, IsNil)
+	c.Assert(m[2].Key, BytesEquals, []byte("b"))
+	c.Assert(m[2].Value, BytesEquals, []byte("2"))
 }
diff --git a/store/localstore/goleveldb/goleveldb_test.go b/store/localstore/goleveldb/goleveldb_test.go
@@ -95,4 +95,14 @@ func (s *testSuite) TestDB(c *C) {
 	c.Assert(err, NotNil)
 	c.Assert(k, IsNil)
 	c.Assert(v, IsNil)
+
+	m := db.MultiSeek([][]byte{[]byte("z"), []byte("a"), []byte("a1")})
+	c.Assert(m, HasLen, 3)
+	c.Assert(m[0].Err, NotNil)
+	c.Assert(m[1].Err, IsNil)
+	c.Assert(m[1].Key, BytesEquals, []byte("a"))
+	c.Assert(m[1].Value, BytesEquals, []byte("2"))
+	c.Assert(m[2].Err, IsNil)
+	c.Assert(m[2].Key, BytesEquals, []byte("b"))
+	c.Assert(m[2].Value, BytesEquals, []byte("2"))
 }