diff options
Diffstat (limited to 'cmd/bolt/import_test.go')
| -rw-r--r-- | cmd/bolt/import_test.go | 27 |
1 files changed, 12 insertions, 15 deletions
diff --git a/cmd/bolt/import_test.go b/cmd/bolt/import_test.go index 3d4f275..086bf03 100644 --- a/cmd/bolt/import_test.go +++ b/cmd/bolt/import_test.go @@ -6,7 +6,6 @@ import ( "github.com/boltdb/bolt" . "github.com/boltdb/bolt/cmd/bolt" - "github.com/stretchr/testify/assert" ) // Ensure that a database can be imported. @@ -15,32 +14,30 @@ func TestImport(t *testing.T) { // Write input file. input := tempfile() - assert.NoError(t, ioutil.WriteFile(input, []byte(`[{"type":"bucket","key":"ZW1wdHk=","value":[]},{"type":"bucket","key":"d2lkZ2V0cw==","value":[{"key":"YmFy","value":""},{"key":"Zm9v","value":"MDAwMA=="}]},{"type":"bucket","key":"d29vaml0cw==","value":[{"key":"YmF6","value":"WFhYWA=="},{"type":"bucket","key":"d29vaml0cy9zdWJidWNrZXQ=","value":[{"key":"YmF0","value":"QQ=="}]}]}]`), 0600)) + ok(t, ioutil.WriteFile(input, []byte(`[{"type":"bucket","key":"ZW1wdHk=","value":[]},{"type":"bucket","key":"d2lkZ2V0cw==","value":[{"key":"YmFy","value":""},{"key":"Zm9v","value":"MDAwMA=="}]},{"type":"bucket","key":"d29vaml0cw==","value":[{"key":"YmF6","value":"WFhYWA=="},{"type":"bucket","key":"d29vaml0cy9zdWJidWNrZXQ=","value":[{"key":"YmF0","value":"QQ=="}]}]}]`), 0600)) // Import database. path := tempfile() output := run("import", path, "--input", input) - assert.Equal(t, ``, output) + equals(t, ``, output) // Open database and verify contents. db, err := bolt.Open(path, 0600, nil) - assert.NoError(t, err) + ok(t, err) db.View(func(tx *bolt.Tx) error { - assert.NotNil(t, tx.Bucket([]byte("empty"))) + assert(t, tx.Bucket([]byte("empty")) != nil, "") b := tx.Bucket([]byte("widgets")) - if assert.NotNil(t, b) { - assert.Equal(t, []byte("0000"), b.Get([]byte("foo"))) - assert.Equal(t, []byte(""), b.Get([]byte("bar"))) - } + assert(t, b != nil, "") + equals(t, []byte("0000"), b.Get([]byte("foo"))) + equals(t, []byte(""), b.Get([]byte("bar"))) b = tx.Bucket([]byte("woojits")) - if assert.NotNil(t, b) { - assert.Equal(t, []byte("XXXX"), b.Get([]byte("baz"))) + assert(t, b != nil, "") + equals(t, []byte("XXXX"), b.Get([]byte("baz"))) - b = b.Bucket([]byte("woojits/subbucket")) - assert.Equal(t, []byte("A"), b.Get([]byte("bat"))) - } + b = b.Bucket([]byte("woojits/subbucket")) + equals(t, []byte("A"), b.Get([]byte("bat"))) return nil }) @@ -51,5 +48,5 @@ func TestImport(t *testing.T) { func TestImport_NotFound(t *testing.T) { SetTestMode(true) output := run("import", "path/to/db", "--input", "no/such/file") - assert.Equal(t, "open no/such/file: no such file or directory", output) + equals(t, "open no/such/file: no such file or directory", output) } |