From e68bc0b4530475503afb34956dfb915673af919c Mon Sep 17 00:00:00 2001
From: Ben Johnson <benbjohnson@yahoo.com>
Date: Sat, 15 Feb 2014 10:23:00 -0700
Subject: Add bucket sequence.

---
 rwtransaction_test.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

(limited to 'rwtransaction_test.go')

diff --git a/rwtransaction_test.go b/rwtransaction_test.go
index 1ce3f70..fa254d8 100644
--- a/rwtransaction_test.go
+++ b/rwtransaction_test.go
@@ -72,6 +72,32 @@ func TestRWTransactionDeleteBucket(t *testing.T) {
 	t.Skip("pending") // TODO(benbjohnson)
 }
 
+// Ensure that a bucket can return an autoincrementing sequence.
+func TestRWTransactionNextSequence(t *testing.T) {
+	withOpenDB(func(db *DB, path string) {
+		db.CreateBucket("widgets")
+		db.CreateBucket("woojits")
+
+		// Make sure sequence increments.
+		seq, err := db.NextSequence("widgets")
+		assert.NoError(t, err)
+		assert.Equal(t, seq, 1)
+		seq, err = db.NextSequence("widgets")
+		assert.NoError(t, err)
+		assert.Equal(t, seq, 2)
+
+		// Buckets should be separate.
+		seq, err = db.NextSequence("woojits")
+		assert.NoError(t, err)
+		assert.Equal(t, seq, 1)
+
+		// Missing buckets return an error.
+		seq, err = db.NextSequence("no_such_bucket")
+		assert.Equal(t, err, BucketNotFoundError)
+		assert.Equal(t, seq, 0)
+	})
+}
+
 // Ensure that an error is returned when inserting into a bucket that doesn't exist.
 func TestRWTransactionPutBucketNotFound(t *testing.T) {
 	t.Skip("pending") // TODO(benbjohnson)
-- 
cgit v1.2.3