diff --git a/p2p/net/swarm/swarm_notif_test.go b/p2p/net/swarm/swarm_notif_test.go index d9e9df62a5..8ebcccb973 100644 --- a/p2p/net/swarm/swarm_notif_test.go +++ b/p2p/net/swarm/swarm_notif_test.go @@ -17,8 +17,10 @@ func streamsSame(a, b inet.Stream) bool { } func TestNotifications(t *testing.T) { + const swarmSize = 5 + ctx := context.Background() - swarms := makeSwarms(ctx, t, 5) + swarms := makeSwarms(ctx, t, swarmSize) defer func() { for _, s := range swarms { s.Close() @@ -30,7 +32,7 @@ func TestNotifications(t *testing.T) { // signup notifs notifiees := make([]*netNotifiee, len(swarms)) for i, swarm := range swarms { - n := newNetNotifiee() + n := newNetNotifiee(swarmSize) swarm.Notify(n) notifiees[i] = n } @@ -184,14 +186,14 @@ type netNotifiee struct { closedStream chan inet.Stream } -func newNetNotifiee() *netNotifiee { +func newNetNotifiee(buffer int) *netNotifiee { return &netNotifiee{ - listen: make(chan ma.Multiaddr), - listenClose: make(chan ma.Multiaddr), - connected: make(chan inet.Conn), - disconnected: make(chan inet.Conn), - openedStream: make(chan inet.Stream), - closedStream: make(chan inet.Stream), + listen: make(chan ma.Multiaddr, buffer), + listenClose: make(chan ma.Multiaddr, buffer), + connected: make(chan inet.Conn, buffer), + disconnected: make(chan inet.Conn, buffer), + openedStream: make(chan inet.Stream, buffer), + closedStream: make(chan inet.Stream, buffer), } }