diff options
Diffstat (limited to 'transaction_test.go')
| -rw-r--r-- | transaction_test.go | 36 |
1 files changed, 10 insertions, 26 deletions
diff --git a/transaction_test.go b/transaction_test.go index 4a7170c..8cf9dfa 100644 --- a/transaction_test.go +++ b/transaction_test.go @@ -53,8 +53,7 @@ func TestTransactionCursorEmptyBucket(t *testing.T) { withOpenDB(func(db *DB, path string) { db.CreateBucket("widgets") txn, _ := db.Transaction() - c, err := txn.Cursor("widgets") - assert.NoError(t, err) + c := txn.Bucket("widgets").Cursor() k, v := c.First() assert.Nil(t, k) assert.Nil(t, v) @@ -62,18 +61,6 @@ func TestTransactionCursorEmptyBucket(t *testing.T) { }) } -// Ensure that a Transaction returns a nil when a bucket doesn't exist. -func TestTransactionCursorMissingBucket(t *testing.T) { - withOpenDB(func(db *DB, path string) { - db.CreateBucket("widgets") - txn, _ := db.Transaction() - c, err := txn.Cursor("woojits") - assert.Nil(t, c) - assert.Equal(t, err, ErrBucketNotFound) - txn.Close() - }) -} - // Ensure that a Transaction cursor can iterate over a single root with a couple elements. func TestTransactionCursorLeafRoot(t *testing.T) { withOpenDB(func(db *DB, path string) { @@ -82,8 +69,7 @@ func TestTransactionCursorLeafRoot(t *testing.T) { db.Put("widgets", []byte("foo"), []byte{0}) db.Put("widgets", []byte("bar"), []byte{1}) txn, _ := db.Transaction() - c, err := txn.Cursor("widgets") - assert.NoError(t, err) + c := txn.Bucket("widgets").Cursor() k, v := c.First() assert.Equal(t, string(k), "bar") @@ -117,8 +103,7 @@ func TestTransactionCursorLeafRootReverse(t *testing.T) { db.Put("widgets", []byte("foo"), []byte{0}) db.Put("widgets", []byte("bar"), []byte{1}) txn, _ := db.Transaction() - c, err := txn.Cursor("widgets") - assert.NoError(t, err) + c := txn.Bucket("widgets").Cursor() k, v := c.Last() assert.Equal(t, string(k), "foo") @@ -152,8 +137,7 @@ func TestTransactionCursorRestart(t *testing.T) { db.Put("widgets", []byte("foo"), []byte{}) txn, _ := db.Transaction() - c, err := txn.Cursor("widgets") - assert.NoError(t, err) + c := txn.Bucket("widgets").Cursor() k, _ := c.First() assert.Equal(t, string(k), "bar") @@ -178,8 +162,9 @@ func TestTransactionCursorIterate(t *testing.T) { // Bulk insert all values. db.CreateBucket("widgets") rwtxn, _ := db.RWTransaction() + b := rwtxn.Bucket("widgets") for _, item := range items { - assert.NoError(t, rwtxn.Put("widgets", item.Key, item.Value)) + assert.NoError(t, b.Put(item.Key, item.Value)) } assert.NoError(t, rwtxn.Commit()) @@ -189,8 +174,7 @@ func TestTransactionCursorIterate(t *testing.T) { // Iterate over all items and check consistency. var index = 0 txn, _ := db.Transaction() - c, err := txn.Cursor("widgets") - assert.NoError(t, err) + c := txn.Bucket("widgets").Cursor() for k, v := c.First(); k != nil && index < len(items); k, v = c.Next() { assert.Equal(t, k, items[index].Key) assert.Equal(t, v, items[index].Value) @@ -215,8 +199,9 @@ func TestTransactionCursorIterateReverse(t *testing.T) { // Bulk insert all values. db.CreateBucket("widgets") rwtxn, _ := db.RWTransaction() + b := rwtxn.Bucket("widgets") for _, item := range items { - assert.NoError(t, rwtxn.Put("widgets", item.Key, item.Value)) + assert.NoError(t, b.Put(item.Key, item.Value)) } assert.NoError(t, rwtxn.Commit()) @@ -226,8 +211,7 @@ func TestTransactionCursorIterateReverse(t *testing.T) { // Iterate over all items and check consistency. var index = 0 txn, _ := db.Transaction() - c, err := txn.Cursor("widgets") - assert.NoError(t, err) + c := txn.Bucket("widgets").Cursor() for k, v := c.Last(); k != nil && index < len(items); k, v = c.Prev() { assert.Equal(t, k, items[index].Key) assert.Equal(t, v, items[index].Value) |