diff options
Diffstat (limited to 'cmd/bolt/buckets_test.go')
-rw-r--r-- | cmd/bolt/buckets_test.go | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/cmd/bolt/buckets_test.go b/cmd/bolt/buckets_test.go new file mode 100644 index 0000000..771c8d8 --- /dev/null +++ b/cmd/bolt/buckets_test.go @@ -0,0 +1,31 @@ +package main_test + +import ( + "testing" + + "github.com/boltdb/bolt" + . "github.com/boltdb/bolt/cmd/bolt" + "github.com/stretchr/testify/assert" +) + +// Ensure that a list of buckets can be retrieved. +func TestBuckets(t *testing.T) { + SetTestMode(true) + open(func(db *bolt.DB) { + db.Update(func(tx *bolt.Tx) error { + tx.CreateBucket("woojits") + tx.CreateBucket("widgets") + tx.CreateBucket("whatchits") + return nil + }) + output := run("buckets", db.Path()) + assert.Equal(t, "whatchits\nwidgets\nwoojits", output) + }) +} + +// Ensure that an error is reported if the database is not found. +func TestBucketsDBNotFound(t *testing.T) { + SetTestMode(true) + output := run("buckets", "no/such/db") + assert.Equal(t, "stat no/such/db: no such file or directory", output) +} |