summaryrefslogtreecommitdiff
path: root/vendor/github.com/oxtoacart/bpool/bufferpool_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/oxtoacart/bpool/bufferpool_test.go')
-rw-r--r--vendor/github.com/oxtoacart/bpool/bufferpool_test.go30
1 files changed, 30 insertions, 0 deletions
diff --git a/vendor/github.com/oxtoacart/bpool/bufferpool_test.go b/vendor/github.com/oxtoacart/bpool/bufferpool_test.go
new file mode 100644
index 0000000..b344bdc
--- /dev/null
+++ b/vendor/github.com/oxtoacart/bpool/bufferpool_test.go
@@ -0,0 +1,30 @@
+package bpool
+
+import (
+ "bytes"
+ "testing"
+)
+
+func TestBufferPool(t *testing.T) {
+ var size int = 4
+
+ bufPool := NewBufferPool(size)
+
+ // Test Get/Put
+ b := bufPool.Get()
+ bufPool.Put(b)
+
+ // Add some additional buffers beyond the pool size.
+ for i := 0; i < size*2; i++ {
+ bufPool.Put(bytes.NewBuffer([]byte{}))
+ }
+
+ // Close the channel so we can iterate over it.
+ close(bufPool.c)
+
+ // Check the size of the pool.
+ if len(bufPool.c) != size {
+ t.Fatalf("bufferpool size invalid: got %v want %v", len(bufPool.c), size)
+ }
+
+}