Skip to content

Commit e06f3aa

Browse files
committed
reuseport: use DialContext instead of Dial to fail quickly
1 parent 1153b1b commit e06f3aa

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

p2p/net/reuseport/transport_test.go

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,11 @@
11
package reuseport
22

33
import (
4+
"context"
45
"net"
56
"runtime"
67
"testing"
8+
"time"
79

810
ma "github.com/multiformats/go-multiaddr"
911
manet "github.com/multiformats/go-multiaddr/net"
@@ -64,8 +66,11 @@ func acceptOne(t *testing.T, listener manet.Listener) <-chan manet.Conn {
6466
func dialOne(t *testing.T, tr *Transport, listener manet.Listener, expected ...int) int {
6567
t.Helper()
6668

69+
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
70+
defer cancel()
71+
6772
connChan := acceptOne(t, listener)
68-
c, err := tr.Dial(listener.Multiaddr())
73+
c, err := tr.DialContext(ctx, listener.Multiaddr())
6974
if err != nil {
7075
t.Fatal(err)
7176
}

0 commit comments

Comments
 (0)