From 701d7c324fa5f1ac637d820a4611cb3589122ffd Mon Sep 17 00:00:00 2001 From: Marco Munizaga Date: Tue, 20 Aug 2024 15:14:03 -0700 Subject: [PATCH] test: Try to fix test timeout (#2930) --- p2p/transport/webrtc/transport_test.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/p2p/transport/webrtc/transport_test.go b/p2p/transport/webrtc/transport_test.go index d603d610da..b2c06d3374 100644 --- a/p2p/transport/webrtc/transport_test.go +++ b/p2p/transport/webrtc/transport_test.go @@ -1014,13 +1014,16 @@ func TestConnectionClosedWhenRemoteCloses(t *testing.T) { listenT, p := getTransport(t) listener, err := listenT.Listen(ma.StringCast("/ip4/127.0.0.1/udp/0/webrtc-direct")) require.NoError(t, err) + defer listener.Close() + accepted := make(chan struct{}) dialer, _ := getTransport(t) var wg sync.WaitGroup wg.Add(1) go func() { defer wg.Done() c, err := listener.Accept() + close(accepted) if !assert.NoError(t, err) { return } @@ -1031,6 +1034,7 @@ func TestConnectionClosedWhenRemoteCloses(t *testing.T) { c, err := dialer.Dial(context.Background(), listener.Multiaddr(), p) require.NoError(t, err) + <-accepted c.Close() wg.Wait() }