pexconn_test.go | 3 ++- diff --git a/pexconn_test.go b/pexconn_test.go index caee9d849fd4d601661eb38ce8a1cff5b242dc57..42c62cff69168ed788b849692f7626572d225c46 100644 --- a/pexconn_test.go +++ b/pexconn_test.go @@ -26,6 +26,7 @@ if err := torrent.addPeerConn(c); err != nil { t.Log(err) } + connWriteCond := c.messageWriter.writeCond.Signaled() c.pex.Init(c) require.True(t, c.pex.IsEnabled(), "should get enabled") defer c.pex.Close() @@ -37,7 +38,7 @@ writerCalled = true out = m return true } - <-c.messageWriter.writeCond.Signaled() + <-connWriteCond c.pex.Share(testWriter) require.True(t, writerCalled) require.EqualValues(t, pp.Extended, out.Type)