Clean up integration_tests/backend_helpers.go.

alphagov · Jan 21, 2024 · 015a353 · 015a353
1 parent 6e8e6da
commit 015a353
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/integration_tests/backend_helpers.go b/integration_tests/backend_helpers.go
@@ -35,18 +35,14 @@ func startTarpitBackend(delays ...time.Duration) *httptest.Server {
 		bodyDelay = delays[1]
 	}
 	return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		body := "Tarpit\n"
+		const body = "Tarpit\n"
 
-		if responseDelay > 0 {
-			time.Sleep(responseDelay)
-		}
+		time.Sleep(responseDelay)
 		w.Header().Add("Content-Length", strconv.Itoa(len(body)))
 		w.WriteHeader(http.StatusOK)
 		w.(http.Flusher).Flush()
 
-		if bodyDelay > 0 {
-			time.Sleep(bodyDelay)
-		}
+		time.Sleep(bodyDelay)
 		_, err := w.Write([]byte(body))
 		Expect(err).NotTo(HaveOccurred())
 	}))