diff options
Diffstat (limited to 'db_test.go')
| -rw-r--r-- | db_test.go | 126 |
1 files changed, 55 insertions, 71 deletions
@@ -2,7 +2,6 @@ package bolt import ( "errors" - "io" "io/ioutil" "math/rand" "os" @@ -37,38 +36,34 @@ func TestOpenBadPath(t *testing.T) { // Ensure that a database can be opened without error. func TestDBOpen(t *testing.T) { - withDB(func(db *DB, path string) { - err := db.Open(path, 0666) + withTempPath(func(path string) { + db, err := Open(path, 0666) + assert.NotNil(t, db) assert.NoError(t, err) assert.Equal(t, db.Path(), path) - }) -} - -// Ensure that the database returns an error if already open. -func TestDBReopen(t *testing.T) { - withDB(func(db *DB, path string) { - db.Open(path, 0666) - err := db.Open(path, 0666) - assert.Equal(t, err, ErrDatabaseOpen) + assert.NoError(t, db.Close()) }) } // Ensure that a re-opened database is consistent. func TestOpenCheck(t *testing.T) { - withDB(func(db *DB, path string) { - assert.NoError(t, db.Open(path, 0666)) + withTempPath(func(path string) { + db, err := Open(path, 0666) + assert.NoError(t, err) assert.NoError(t, db.Check()) db.Close() - assert.NoError(t, db.Open(path, 0666)) + db, err = Open(path, 0666) + assert.NoError(t, err) assert.NoError(t, db.Check()) + db.Close() }) } // Ensure that the database returns an error if the file handle cannot be open. func TestDBOpenFileError(t *testing.T) { - withDB(func(db *DB, path string) { - err := db.Open(path+"/youre-not-my-real-parent", 0666) + withTempPath(func(path string) { + _, err := Open(path+"/youre-not-my-real-parent", 0666) if err, _ := err.(*os.PathError); assert.Error(t, err) { assert.Equal(t, path+"/youre-not-my-real-parent", err.Path) assert.Equal(t, "open", err.Op) @@ -78,33 +73,27 @@ func TestDBOpenFileError(t *testing.T) { // Ensure that write errors to the meta file handler during initialization are returned. func TestDBMetaInitWriteError(t *testing.T) { - withDB(func(db *DB, path string) { - // Mock the file system. - db.ops.metaWriteAt = func(p []byte, offset int64) (n int, err error) { return 0, io.ErrShortWrite } - - // Open the database. - err := db.Open(path, 0666) - assert.Equal(t, err, io.ErrShortWrite) - }) + t.Skip("pending") } // Ensure that a database that is too small returns an error. func TestDBFileTooSmall(t *testing.T) { - withDB(func(db *DB, path string) { - assert.NoError(t, db.Open(path, 0666)) + withTempPath(func(path string) { + db, err := Open(path, 0666) + assert.NoError(t, err) db.Close() // corrupt the database assert.NoError(t, os.Truncate(path, int64(os.Getpagesize()))) - err := db.Open(path, 0666) + db, err = Open(path, 0666) assert.Equal(t, errors.New("file size too small"), err) }) } // Ensure that corrupt meta0 page errors get returned. func TestDBCorruptMeta0(t *testing.T) { - withDB(func(db *DB, path string) { + withTempPath(func(path string) { var m meta m.magic = magic m.version = version @@ -121,18 +110,17 @@ func TestDBCorruptMeta0(t *testing.T) { assert.NoError(t, err) // Open the database. - err = db.Open(path, 0666) + _, err = Open(path, 0666) assert.Equal(t, err, errors.New("meta error: invalid database")) }) } // Ensure that a database cannot open a transaction when it's not open. func TestDBTxErrDatabaseNotOpen(t *testing.T) { - withDB(func(db *DB, path string) { - tx, err := db.Begin(false) - assert.Nil(t, tx) - assert.Equal(t, err, ErrDatabaseNotOpen) - }) + var db DB + tx, err := db.Begin(false) + assert.Nil(t, tx) + assert.Equal(t, err, ErrDatabaseNotOpen) } // Ensure that a read-write transaction can be retrieved. @@ -149,11 +137,10 @@ func TestDBBeginRW(t *testing.T) { // Ensure that opening a transaction while the DB is closed returns an error. func TestDBRWTxOpenWithClosedDB(t *testing.T) { - withDB(func(db *DB, path string) { - tx, err := db.Begin(true) - assert.Equal(t, err, ErrDatabaseNotOpen) - assert.Nil(t, tx) - }) + var db DB + tx, err := db.Begin(true) + assert.Equal(t, err, ErrDatabaseNotOpen) + assert.Nil(t, tx) } // Ensure a database can provide a transactional block. @@ -179,29 +166,27 @@ func TestDBTxBlock(t *testing.T) { // Ensure a closed database returns an error while running a transaction block func TestDBTxBlockWhileClosed(t *testing.T) { - withDB(func(db *DB, path string) { - err := db.Update(func(tx *Tx) error { - tx.CreateBucket("widgets") - return nil - }) - assert.Equal(t, err, ErrDatabaseNotOpen) + var db DB + err := db.Update(func(tx *Tx) error { + tx.CreateBucket("widgets") + return nil }) + assert.Equal(t, err, ErrDatabaseNotOpen) } // Ensure a panic occurs while trying to commit a managed transaction. func TestDBTxBlockWithManualCommitAndRollback(t *testing.T) { - withOpenDB(func(db *DB, path string) { - db.Update(func(tx *Tx) error { - tx.CreateBucket("widgets") - assert.Panics(t, func() { tx.Commit() }) - assert.Panics(t, func() { tx.Rollback() }) - return nil - }) - db.View(func(tx *Tx) error { - assert.Panics(t, func() { tx.Commit() }) - assert.Panics(t, func() { tx.Rollback() }) - return nil - }) + var db DB + db.Update(func(tx *Tx) error { + tx.CreateBucket("widgets") + assert.Panics(t, func() { tx.Commit() }) + assert.Panics(t, func() { tx.Rollback() }) + return nil + }) + db.View(func(tx *Tx) error { + assert.Panics(t, func() { tx.Commit() }) + assert.Panics(t, func() { tx.Rollback() }) + return nil }) } @@ -217,8 +202,8 @@ func TestDBCopyFile(t *testing.T) { assert.NoError(t, os.RemoveAll("/tmp/bolt.copyfile.db")) assert.NoError(t, db.CopyFile("/tmp/bolt.copyfile.db", 0666)) - var db2 DB - assert.NoError(t, db2.Open("/tmp/bolt.copyfile.db", 0666)) + db2, err := Open("/tmp/bolt.copyfile.db", 0666) + assert.NoError(t, err) defer db2.Close() db2.View(func(tx *Tx) error { @@ -272,11 +257,10 @@ func TestDBStat(t *testing.T) { // Ensure the getting stats on a closed database returns an error. func TestDBStatWhileClosed(t *testing.T) { - withDB(func(db *DB, path string) { - stat, err := db.Stat() - assert.Equal(t, err, ErrDatabaseNotOpen) - assert.Nil(t, stat) - }) + var db DB + stat, err := db.Stat() + assert.Equal(t, err, ErrDatabaseNotOpen) + assert.Nil(t, stat) } // Ensure that an error is returned when a database write fails. @@ -379,22 +363,22 @@ func BenchmarkDBPutRandom(b *testing.B) { }) } -// withDB executes a function with a database reference. -func withDB(fn func(*DB, string)) { +// withTempPath executes a function with a database reference. +func withTempPath(fn func(string)) { f, _ := ioutil.TempFile("", "bolt-") path := f.Name() f.Close() os.Remove(path) defer os.RemoveAll(path) - var db DB - fn(&db, path) + fn(path) } // withOpenDB executes a function with an already opened database. func withOpenDB(fn func(*DB, string)) { - withDB(func(db *DB, path string) { - if err := db.Open(path, 0666); err != nil { + withTempPath(func(path string) { + db, err := Open(path, 0666) + if err != nil { panic("cannot open db: " + err.Error()) } defer db.Close() |