From 7ced11fc10767d28c6f7e1dbcdd925732b85aa6d Mon Sep 17 00:00:00 2001 From: Eno Compton Date: Thu, 4 Aug 2022 11:30:53 -0600 Subject: [PATCH] chore: increase wait time to ensure TCP connections (#1293) Fixes #1291 --- internal/proxy/proxy_test.go | 19 ++----------------- 1 file changed, 2 insertions(+), 17 deletions(-) diff --git a/internal/proxy/proxy_test.go b/internal/proxy/proxy_test.go index 16b1f5459..cff004213 100644 --- a/internal/proxy/proxy_test.go +++ b/internal/proxy/proxy_test.go @@ -354,32 +354,17 @@ func TestClientCloseWaitsForActiveConnections(t *testing.T) { Instances: []proxy.InstanceConnConfig{ {Name: "proj:region:pg"}, }, + WaitOnClose: 5 * time.Second, } - c, err := proxy.NewClient(context.Background(), &fakeDialer{}, logger, in) if err != nil { t.Fatalf("proxy.NewClient error: %v", err) } go c.Serve(context.Background(), func() {}) - conn := tryTCPDial(t, "127.0.0.1:5000") - _ = conn.Close() - - if err := c.Close(); err != nil { - t.Fatalf("c.Close error: %v", err) - } - - in.WaitOnClose = time.Second - in.Port = 5001 - c, err = proxy.NewClient(context.Background(), &fakeDialer{}, logger, in) - if err != nil { - t.Fatalf("proxy.NewClient error: %v", err) - } - go c.Serve(context.Background(), func() {}) - var open []net.Conn for i := 0; i < 5; i++ { - conn = tryTCPDial(t, "127.0.0.1:5001") + conn := tryTCPDial(t, "127.0.0.1:5000") open = append(open, conn) } defer func() {