gorond test files

[btrtrc.git] / peer_protocol / pex_test.go

diff --git a/peer_protocol/pex_test.go b/peer_protocol/pex_test.go
index 374dfaa4be370b8944781991c53361c6fd93626d..5e5e96c14a3f0854f8c366bb63f3dd3b3cb9352a 100644 (file)
--- a/peer_protocol/pex_test.go
+++ b/peer_protocol/pex_test.go
@@ -6,9 +6,9 @@ import (
        "net"
        "testing"
 
+       "github.com/anacrolix/dht/v2/krpc"
        "github.com/stretchr/testify/require"
 
-       "github.com/anacrolix/dht/v2/krpc"
        "github.com/anacrolix/torrent/bencode"
 )
 
@@ -36,19 +36,19 @@ func TestMarshalPexMessage(t *testing.T) {
        pm.Added = append(pm.Added, addr)
        pm.AddedFlags = append(pm.AddedFlags, f)
 
-       b, err := bencode.Marshal(pm)
+       _, err := bencode.Marshal(pm)
        require.NoError(t, err)
 
        pexExtendedId := ExtensionNumber(7)
        msg := pm.Message(pexExtendedId)
        expected := []byte("\x00\x00\x00\x4c\x14\x07d5:added6:\x7f\x00\x00\x01\x55\xaa7:added.f1:\x116:added60:8:added6.f0:7:dropped0:8:dropped60:e")
-       b, err = msg.MarshalBinary()
+       b, err := msg.MarshalBinary()
        require.NoError(t, err)
        require.EqualValues(t, b, expected)
 
        msg = Message{}
        dec := Decoder{
-               R:         bufio.NewReader(bytes.NewBuffer(b)),
+               R:         bufio.NewReader(bytes.NewReader(b)),
                MaxLength: 128,
        }
        pmOut := PexMsg{}