attack: Improve reliability of TestClient

tsenart · Oct 10, 2018 · a3ce5a2 · a3ce5a2
1 parent f25ac6a
commit a3ce5a2
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lib/attack_test.go b/lib/attack_test.go
@@ -279,8 +279,8 @@ func TestClient(t *testing.T) {
 	client := &http.Client{
 		Timeout: time.Duration(1 * time.Nanosecond),
 		Transport: &http.Transport{
-			Proxy: http.ProxyFromEnvironment,
-			Dial:  dialer.Dial,
+			Proxy:                 http.ProxyFromEnvironment,
+			Dial:                  dialer.Dial,
 			ResponseHeaderTimeout: DefaultTimeout,
 			TLSClientConfig:       DefaultTLSConfig,
 			TLSHandshakeTimeout:   10 * time.Second,
@@ -289,7 +289,9 @@ func TestClient(t *testing.T) {
 	}
 
 	server := httptest.NewServer(
-		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {}),
+		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			select {}
+		}),
 	)
 	defer server.Close()