From fea388b03d67a26a6bb13e9b9363b15e72b7f2c5 Mon Sep 17 00:00:00 2001 From: Ben Johnson Date: Thu, 27 Mar 2014 22:36:05 -0600 Subject: Refactor bolt commands into individual files. --- cmd/bolt/set_test.go | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 cmd/bolt/set_test.go (limited to 'cmd/bolt/set_test.go') diff --git a/cmd/bolt/set_test.go b/cmd/bolt/set_test.go new file mode 100644 index 0000000..d76b3c0 --- /dev/null +++ b/cmd/bolt/set_test.go @@ -0,0 +1,38 @@ +package main_test + +import ( + "testing" + + "github.com/boltdb/bolt" + . "github.com/boltdb/bolt/cmd/bolt" + "github.com/stretchr/testify/assert" +) + +// Ensure that a value can be set from the CLI. +func TestSet(t *testing.T) { + SetTestMode(true) + open(func(db *bolt.DB) { + db.Update(func(tx *bolt.Tx) error { + tx.CreateBucket("widgets") + return nil + }) + assert.Equal(t, "", run("set", db.Path(), "widgets", "foo", "bar")) + assert.Equal(t, "bar", run("get", db.Path(), "widgets", "foo")) + }) +} + +// Ensure that an error is reported if the database is not found. +func TestSetDBNotFound(t *testing.T) { + SetTestMode(true) + output := run("set", "no/such/db", "widgets", "foo", "bar") + assert.Equal(t, "stat no/such/db: no such file or directory", output) +} + +// Ensure that an error is reported if the bucket is not found. +func TestSetBucketNotFound(t *testing.T) { + SetTestMode(true) + open(func(db *bolt.DB) { + output := run("set", db.Path(), "widgets", "foo", "bar") + assert.Equal(t, "bucket not found: widgets", output) + }) +} -- cgit v1.2.3