修复pool.Put开启 race 时被丢弃测试失败的问题

thinkeridea · Oct 14, 2019 · 44748b8 · 44748b8
1 parent af8987b
commit 44748b8
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pool/buffer_test.go b/pool/buffer_test.go
@@ -173,18 +173,19 @@ func TestPool_Put(t *testing.T) {
 
 	p.Put(b)
 
+	var bb *bytes.Buffer
 	// 开启 race 时有一定概率导致 Put 被丢弃
 	pp := (*sync.Pool)(p)
 	for i := 0; i < 10; i++ {
-		b = pp.Get().(*bytes.Buffer)
-		if b.String() == "xx" {
+		bb = pp.Get().(*bytes.Buffer)
+		if bb.String() == "xx" {
 			break
 		}
 
 		p.Put(b)
 	}
 
-	if b.String() != "xx" {
-		t.Errorf("b1.String():%s != xx", b.String())
+	if bb.String() != "xx" {
+		t.Errorf("b1.String():%s != xx", bb.String())
 	}
 }