Expose handshake stuff in peer_protocol

[btrtrc.git] / torrent_test.go

diff --git a/torrent_test.go b/torrent_test.go
index 33eb49035cfc725c704722572cba0a3fc80f5baa..2e28f96a163f6a33a6cbaf3dee77f7f6bb276905 100644 (file)
--- a/torrent_test.go
+++ b/torrent_test.go
@@ -182,12 +182,12 @@ func TestTorrentMetainfoIncompleteMetadata(t *testing.T) {
        require.NoError(t, err)
        defer nc.Close()
 
-       var pex peerExtensionBytes
-       pex.SetBit(ExtensionBitExtended)
-       hr, ok, err := handshake(nc, &ih, [20]byte{}, pex)
+       var pex PeerExtensionBits
+       pex.SetBit(pp.ExtensionBitExtended)
+       hr, ok, err := pp.Handshake(nc, &ih, [20]byte{}, pex)
        require.NoError(t, err)
        assert.True(t, ok)
-       assert.True(t, hr.peerExtensionBytes.GetBit(ExtensionBitExtended))
+       assert.True(t, hr.PeerExtensionBits.GetBit(pp.ExtensionBitExtended))
        assert.EqualValues(t, cl.PeerID(), hr.PeerID)
        assert.Equal(t, ih, hr.Hash)