Fix flakiness for TestDoDeadlineRetry

valyala · Nov 8, 2023 · f7d4f61 · f7d4f61
1 parent dfce853
commit f7d4f61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/client_test.go b/client_test.go
@@ -149,7 +149,7 @@ func TestHostClientNegativeTimeout(t *testing.T) {
 func TestDoDeadlineRetry(t *testing.T) {
 	t.Parallel()
 
-	tries := 0
+	var tries int32
 	done := make(chan struct{})
 
 	ln := fasthttputil.NewInmemoryListener()
@@ -160,7 +160,7 @@ func TestDoDeadlineRetry(t *testing.T) {
 				close(done)
 				break
 			}
-			tries++
+			atomic.AddInt32(&tries, 1)
 			br := bufio.NewReader(c)
 			(&RequestHeader{}).Read(br)                      //nolint:errcheck
 			(&Request{}).readBodyStream(br, 0, false, false) //nolint:errcheck