Skip to content

Commit

Permalink
add test
Browse files Browse the repository at this point in the history
  • Loading branch information
sukunrt committed Sep 7, 2023
1 parent e388576 commit c9d082d
Show file tree
Hide file tree
Showing 2 changed files with 36 additions and 0 deletions.
8 changes: 8 additions & 0 deletions p2p/net/swarm/swarm_dial_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -277,6 +277,9 @@ func TestAddrsForDialFiltering(t *testing.T) {
t1 := ma.StringCast("/ip4/1.2.3.4/tcp/1")
ws1 := ma.StringCast("/ip4/1.2.3.4/tcp/1/ws")

unSpecQ := ma.StringCast("/ip4/0.0.0.0/udp/2/quic-v1")
unSpecT := ma.StringCast("/ip6/::/tcp/2/")

resolver, err := madns.NewResolver(madns.WithDefaultResolver(&madns.MockResolver{}))
require.NoError(t, err)
s := newTestSwarmWithResolver(t, resolver)
Expand Down Expand Up @@ -307,6 +310,11 @@ func TestAddrsForDialFiltering(t *testing.T) {
input: append([]ma.Multiaddr{q1}, ourAddrs...),
output: []ma.Multiaddr{q1},
},
{
name: "unspecified-filtered",
input: []ma.Multiaddr{q1v1, t1, unSpecQ, unSpecT},
output: []ma.Multiaddr{q1v1, t1},
},
}

ctx := context.Background()
Expand Down
28 changes: 28 additions & 0 deletions x.patch
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
diff --git a/p2p/transport/quicreuse/listener.go b/p2p/transport/quicreuse/listener.go
index 4ee20042..757cee13 100644
--- a/p2p/transport/quicreuse/listener.go
+++ b/p2p/transport/quicreuse/listener.go
@@ -127,6 +127,7 @@ func (l *quicListener) Run() error {
defer l.transport.DecreaseCount()
for {
conn, err := l.l.Accept(context.Background())
+ fmt.Println("received main conn")
if err != nil {
if errors.Is(err, quic.ErrServerClosed) || strings.Contains(err.Error(), "use of closed network connection") {
return transport.ErrListenerClosed
@@ -191,6 +192,7 @@ func (l *listener) Accept(ctx context.Context) (quic.Connection, error) {
case <-l.acceptLoopRunning:
return nil, transport.ErrListenerClosed
case c, ok := <-l.queue:
+ fmt.Println("received conn from queue")
if !ok {
return nil, transport.ErrListenerClosed
}
@@ -214,6 +216,7 @@ func (l *listener) Close() error {
for conn := range l.queue {
conn.CloseWithError(1, "closing")
}
+ fmt.Println("drained the queue")
})
return nil
}

0 comments on commit c9d082d

Please sign in to comment.