Fix another flaky test

valyala · Jun 2, 2024 · 2909827 · 2909827
1 parent e6d9374
commit 2909827
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/server_test.go b/server_test.go
@@ -1047,6 +1047,11 @@ func TestRejectedRequestsCount(t *testing.T) {
 			}
 		}
 
+		// The server's worker pool is a separate goroutine, give it
+		// a little bit of time to process the failed connection,
+		// otherwise the test may fail from time to time.
+		time.Sleep(time.Millisecond * 10)
+
 		if cnt := s.GetRejectedConnectionsCount(); cnt != uint32(expectedCount) {
 			t.Errorf("unexpected rejected connections count: %d. Expecting %d",
 				cnt, expectedCount)