From c8af29f648b92f574194411a6cdef0b1c368c6f8 Mon Sep 17 00:00:00 2001 From: Eno Compton Date: Tue, 13 Sep 2022 10:37:53 -0600 Subject: [PATCH] chore: increase wait time to ensure TCP connections This is port of https://github.com/GoogleCloudPlatform/cloudsql-proxy/pull/1293. --- 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 f53d1acf..fd33bf02 100644 --- a/internal/proxy/proxy_test.go +++ b/internal/proxy/proxy_test.go @@ -325,32 +325,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() {