mirror of https://github.com/etcd-io/bbolt.git
33 lines
786 B
Go
33 lines
786 B
Go
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, path string) {
|
|
db.Update(func(tx *bolt.Tx) error {
|
|
tx.CreateBucket("woojits")
|
|
tx.CreateBucket("widgets")
|
|
tx.CreateBucket("whatchits")
|
|
return nil
|
|
})
|
|
db.Close()
|
|
output := run("buckets", 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)
|
|
}
|