add timeout

libp2p · Jan 15, 2025 · 9f641ae · 9f641ae
1 parent b183d5c
commit 9f641ae
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/p2p/transport/websocket/websocket_test.go b/p2p/transport/websocket/websocket_test.go
@@ -615,7 +615,11 @@ func TestHTTPSProxyDoesSocks(t *testing.T) {
 	_, err = tpt.Dial(context.Background(), maToDial, "")
 	require.Error(t, err, "This should error as we don't have a real socks server")
 
-	if err := <-proxyServerErr; err != nil {
-		t.Fatal(err)
+	select {
+	case <-time.After(1 * time.Second):
+	case err := <-proxyServerErr:
+		if err != nil {
+			t.Fatal(err)
+		}
 	}
 }