1 files changed, 62 insertions, 0 deletions
diff --git a/bucket_test.go b/bucket_test.go
new file mode 100644
index 0000000..5e33189
--- /dev/null
+++ b/bucket_test.go
@@ -0,0 +1,62 @@
+package bolt
+
+import (
+	"strconv"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+// Ensure a bucket can calculate stats.
+func TestBucketStat(t *testing.T) {
+	withOpenDB(func(db *DB, path string) {
+		db.Do(func(txn *RWTransaction) error {
+			// Add bucket with lots of keys.
+			txn.CreateBucket("widgets")
+			for i := 0; i < 100000; i++ {
+				txn.Put("widgets", []byte(strconv.Itoa(i)), []byte(strconv.Itoa(i)))
+			}
+
+			// Add bucket with fewer keys but one big value.
+			txn.CreateBucket("woojits")
+			for i := 0; i < 500; i++ {
+				txn.Put("woojits", []byte(strconv.Itoa(i)), []byte(strconv.Itoa(i)))
+			}
+			txn.Put("woojits", []byte("really-big-value"), []byte(strings.Repeat("*", 10000)))
+
+			// Add a bucket that fits on a single root leaf.
+			txn.CreateBucket("whozawhats")
+			txn.Put("whozawhats", []byte("foo"), []byte("bar"))
+
+			return nil
+		})
+		db.With(func(txn *Transaction) error {
+			b := txn.Bucket("widgets")
+			stat := b.Stat()
+			assert.Equal(t, stat.BranchPageCount, 15)
+			assert.Equal(t, stat.LeafPageCount, 1281)
+			assert.Equal(t, stat.OverflowPageCount, 0)
+			assert.Equal(t, stat.KeyCount, 100000)
+			assert.Equal(t, stat.MaxDepth, 3)
+
+			b = txn.Bucket("woojits")
+			stat = b.Stat()
+			assert.Equal(t, stat.BranchPageCount, 1)
+			assert.Equal(t, stat.LeafPageCount, 6)
+			assert.Equal(t, stat.OverflowPageCount, 2)
+			assert.Equal(t, stat.KeyCount, 501)
+			assert.Equal(t, stat.MaxDepth, 2)
+
+			b = txn.Bucket("whozawhats")
+			stat = b.Stat()
+			assert.Equal(t, stat.BranchPageCount, 0)
+			assert.Equal(t, stat.LeafPageCount, 1)
+			assert.Equal(t, stat.OverflowPageCount, 0)
+			assert.Equal(t, stat.KeyCount, 1)
+			assert.Equal(t, stat.MaxDepth, 1)
+
+			return nil
+		})
+	})
+}