diff --git a/client/hijack_test.go b/client/hijack_test.go index ac551d594f..c1ddd386f3 100644 --- a/client/hijack_test.go +++ b/client/hijack_test.go @@ -1,6 +1,7 @@ package client import ( + "context" "fmt" "io" "net" @@ -17,6 +18,8 @@ import ( func TestTLSCloseWriter(t *testing.T) { t.Parallel() + ctx, cancel := context.WithCancel(t.Context()) + defer cancel() var chErr chan error ts := &httptest.Server{Config: &http.Server{ @@ -88,7 +91,7 @@ func TestTLSCloseWriter(t *testing.T) { client, err := New(WithHost("tcp://"+serverURL.Host), WithHTTPClient(ts.Client())) assert.NilError(t, err) - resp, err := client.postHijacked(t.Context(), "/asdf", url.Values{}, nil, map[string][]string{"Content-Type": {"text/plain"}}) + resp, err := client.postHijacked(ctx, "/asdf", url.Values{}, nil, map[string][]string{"Content-Type": {"text/plain"}}) assert.NilError(t, err) defer resp.Close()