adjust test function relates to #55

xtaci · Aug 28, 2019 · b71f58e · b71f58e
1 parent 67e00e8
commit b71f58e
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/alloc_test.go b/alloc_test.go
@@ -61,16 +61,10 @@ func TestAllocPut(t *testing.T) {
 func TestAllocPutThenGet(t *testing.T) {
 	alloc := NewAllocator()
 	data := alloc.Get(4)
-	for k := range data {
-		data[k] = 99
-	}
 	alloc.Put(data)
-
 	newData := alloc.Get(4)
-	for k := range newData {
-		if newData[k] != 99 {
-			t.Fatal("cannot fetch written []bytes from pool")
-		}
+	if cap(data) != cap(newData) {
+		t.Fatal("different cap while alloc.Get()")
 	}
 }