aboutsummaryrefslogtreecommitdiff
path: root/cmd/bolt/keys_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'cmd/bolt/keys_test.go')
-rw-r--r--cmd/bolt/keys_test.go10
1 files changed, 6 insertions, 4 deletions
diff --git a/cmd/bolt/keys_test.go b/cmd/bolt/keys_test.go
index c426836..ea530f6 100644
--- a/cmd/bolt/keys_test.go
+++ b/cmd/bolt/keys_test.go
@@ -11,7 +11,7 @@ import (
// Ensure that a list of keys can be retrieved for a given bucket.
func TestKeys(t *testing.T) {
SetTestMode(true)
- open(func(db *bolt.DB) {
+ open(func(db *bolt.DB, path string) {
db.Update(func(tx *bolt.Tx) error {
tx.CreateBucket("widgets")
tx.Bucket("widgets").Put([]byte("0002"), []byte(""))
@@ -19,7 +19,8 @@ func TestKeys(t *testing.T) {
tx.Bucket("widgets").Put([]byte("0003"), []byte(""))
return nil
})
- output := run("keys", db.Path(), "widgets")
+ db.Close()
+ output := run("keys", path, "widgets")
assert.Equal(t, "0001\n0002\n0003", output)
})
}
@@ -34,8 +35,9 @@ func TestKeysDBNotFound(t *testing.T) {
// Ensure that an error is reported if the bucket is not found.
func TestKeysBucketNotFound(t *testing.T) {
SetTestMode(true)
- open(func(db *bolt.DB) {
- output := run("keys", db.Path(), "widgets")
+ open(func(db *bolt.DB, path string) {
+ db.Close()
+ output := run("keys", path, "widgets")
assert.Equal(t, "bucket not found: widgets", output)
})
}