feat: add TestBufPool

internal/bufpool/bufpool_test.go

@@ -0,0 +1,39 @@
+// Copyright 2018-2021 Burak Sezer
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package bufpool
+
+import "testing"
+
+func TestBufPool(t *testing.T) {
+	p := New()
+
+	count := 1000
+	mebibyte := 1 << 20
+	total := 10 * mebibyte
+
+	b := make([]byte, total/count)
+
+	for i := 0; i < count; i++ {
+		buf := p.Get()
+		nr, err := buf.Write(b)
+		if err != nil {
+			t.Fatalf("Expected nil. Got: %v", err)
+		}
+		if nr != total/count {
+			t.Fatalf("Expected 10. Got: %d", nr)
+		}
+		p.Put(buf)
+	}
+}

internal/transport/client.go

@@ -111,7 +111,7 @@ func (c *Client) conn(addr string) (net.Conn, error) {
 	}
 
 	// TODO: Make this configurable
-	ctx, cancel := context.WithTimeout(context.Background(), 5 * time.Second)
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer cancel()
 
 	conn, err := p.Get(ctx)