status-im
/
go-libp2p
mirror of https://github.com/status-im/go-libp2p.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

SubtestPingPong: ensure connections are closed. 

https://github.com/libp2p/go-libp2p-transport/pull/51
This commit is contained in:
Raúl Kripalani 2019-05-23 18:45:35 +01:00
parent aacc91fe0c
commit 2857d54b80
1 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
p2p/transport/testsuite/transport_suite.go
View File

@ -169,21 +169,33 @@ func SubtestPingPong(t *testing.T, ta, tb transport.Transport, maddr ma.Multiadd
}
defer list.Close()
var (
connA, connB transport.CapableConn
)
defer func() {
if connA != nil {
connA.Close()
}
if connB != nil {
connB.Close()
}
}()
var wg sync.WaitGroup
wg.Add(1)
go func() {
defer wg.Done()
c, err := list.Accept()
var err error
connA, err = list.Accept()
if err != nil {
t.Error(err)
return
}
defer c.Close()
var sWg sync.WaitGroup
for i := 0; i < streams; i++ {
s, err := c.AcceptStream()
s, err := connA.AcceptStream()
if err != nil {
t.Error(err)
return
@ -225,14 +237,13 @@ func SubtestPingPong(t *testing.T, ta, tb transport.Transport, maddr ma.Multiadd
t.Error("CanDial should have returned true")
}
c, err := tb.Dial(ctx, list.Multiaddr(), peerA)
connB, err = tb.Dial(ctx, list.Multiaddr(), peerA)
if err != nil {
t.Fatal(err)
}
defer c.Close()
for i := 0; i < streams; i++ {
s, err := c.OpenStream()
s, err := connB.OpenStream()
if err != nil {
t.Error(err)
continue