Don't time out proxy handler

gravitational · May 1, 2024 · 3bd6cc4 · 3bd6cc4
1 parent c1f9af7
commit 3bd6cc4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/api/testhelpers/proxy.go b/api/testhelpers/proxy.go
@@ -21,7 +21,6 @@ import (
 	"net/http/httptest"
 	"sync"
 	"testing"
-	"time"
 
 	"github.com/gravitational/trace"
 	"github.com/stretchr/testify/require"
@@ -88,9 +87,9 @@ func (p *ProxyHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	go replicate(sconn, dconn)
 	go replicate(dconn, io.MultiReader(buf, sconn))
 
-	// Wait until done, error, or 10 second.
+	// Wait until done.
 	select {
-	case <-time.After(10 * time.Second):
+	case <-r.Context().Done():
 	case <-errc:
 	}
 }