diff --git a/staging/src/k8s.io/kube-aggregator/pkg/apiserver/handler_proxy_test.go b/staging/src/k8s.io/kube-aggregator/pkg/apiserver/handler_proxy_test.go index f5a832bf7ed..3b469699c5d 100644 --- a/staging/src/k8s.io/kube-aggregator/pkg/apiserver/handler_proxy_test.go +++ b/staging/src/k8s.io/kube-aggregator/pkg/apiserver/handler_proxy_test.go @@ -921,8 +921,11 @@ type hookedListener struct { } func (wl *hookedListener) Accept() (net.Conn, error) { - wl.onAccept() - return wl.l.Accept() + conn, err := wl.l.Accept() + if err == nil { + wl.onAccept() + } + return conn, err } func (wl *hookedListener) Close() error { @@ -1015,10 +1018,13 @@ func TestFlowControlSignal(t *testing.T) { req := tc.Request req.URL = surl - _, err = server.Client().Do(&req) + res, err := server.Client().Do(&req) if err != nil { t.Fatalf("unexpected error: %v", err) } + if err := res.Body.Close(); err != nil { + t.Fatalf("unexpected error: %v", err) + } if fired := (atomic.LoadInt32(&signalCountOnAccept) > 0); tc.SignalExpected && !fired { t.Errorf("flow control signal expected but not fired")