Break out peerConnWriter

[btrtrc.git] / pexconn_test.go

diff --git a/pexconn_test.go b/pexconn_test.go
index a3ff1d02dbbe3b36ea772b18183f96a94abe2b82..df9aa44ed670d45c362853ca93def22d2e19454a 100644 (file)
--- a/pexconn_test.go
+++ b/pexconn_test.go
@@ -21,7 +21,7 @@ func TestPexConnState(t *testing.T) {
        c := cl.newConnection(nil, false, addr, addr.Network(), "")
        c.PeerExtensionIDs = make(map[pp.ExtensionName]pp.ExtensionNumber)
        c.PeerExtensionIDs[pp.ExtensionNamePex] = pexExtendedId
-       c.writerCond.L.Lock()
+       c.messageWriter.mu.Lock()
        c.setTorrent(torrent)
        torrent.addPeerConn(c)
 
@@ -36,7 +36,7 @@ func TestPexConnState(t *testing.T) {
                out = m
                return true
        }
-       c.writerCond.Wait()
+       <-c.messageWriter.writeCond.WaitChan()
        c.pex.Share(testWriter)
        require.True(t, writerCalled)
        require.EqualValues(t, pp.Extended, out.Type)