Include PeerConn prefix for log messages

[btrtrc.git] / pexconn_test.go

diff --git a/pexconn_test.go b/pexconn_test.go
index caee9d849fd4d601661eb38ce8a1cff5b242dc57..603398c54cc93fb3927617b8f2938acbb960b8a7 100644 (file)
--- a/pexconn_test.go
+++ b/pexconn_test.go
@@ -17,7 +17,10 @@ func TestPexConnState(t *testing.T) {
        cl.initLogger()
        torrent := cl.newTorrent(metainfo.Hash{}, nil)
        addr := &net.TCPAddr{IP: net.IPv6loopback, Port: 4747}
-       c := cl.newConnection(nil, false, addr, addr.Network(), "")
+       c := cl.newConnection(nil, newConnectionOpts{
+               remoteAddr: addr,
+               network:    addr.Network(),
+       })
        c.PeerExtensionIDs = make(map[pp.ExtensionName]pp.ExtensionNumber)
        c.PeerExtensionIDs[pp.ExtensionNamePex] = pexExtendedId
        c.messageWriter.mu.Lock()
@@ -26,6 +29,7 @@ func TestPexConnState(t *testing.T) {
                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 +41,7 @@ func TestPexConnState(t *testing.T) {
                out = m
                return true
        }
-       <-c.messageWriter.writeCond.Signaled()
+       <-connWriteCond
        c.pex.Share(testWriter)
        require.True(t, writerCalled)
        require.EqualValues(t, pp.Extended, out.Type)