Fix panic logging unknown holepunch error code

[btrtrc.git] / torrent_test.go

diff --git a/torrent_test.go b/torrent_test.go
index d2c7ba72057ac0af9d0f22eb94a30894c4cfb9f6..808947e974a23fe9df13e8fb5714eac781445305 100644 (file)
--- a/torrent_test.go
+++ b/torrent_test.go
@@ -9,7 +9,7 @@ import (
        "sync"
        "testing"
 
-       "github.com/anacrolix/generics"
+       g "github.com/anacrolix/generics"
        "github.com/anacrolix/log"
        "github.com/anacrolix/missinggo/v2"
        "github.com/anacrolix/missinggo/v2/bitmap"
@@ -184,10 +184,10 @@ func TestTorrentMetainfoIncompleteMetadata(t *testing.T) {
        defer nc.Close()
 
        var pex PeerExtensionBits
-       pex.SetBit(pp.ExtensionBitExtended, true)
+       pex.SetBit(pp.ExtensionBitLtep, true)
        hr, err := pp.Handshake(nc, &ih, [20]byte{}, pex)
        require.NoError(t, err)
-       assert.True(t, hr.PeerExtensionBits.GetBit(pp.ExtensionBitExtended))
+       assert.True(t, hr.PeerExtensionBits.GetBit(pp.ExtensionBitLtep))
        assert.EqualValues(t, cl.PeerID(), hr.PeerID)
        assert.EqualValues(t, ih, hr.Hash)
 
@@ -232,12 +232,12 @@ func TestRelativeAvailabilityHaveNone(t *testing.T) {
                gotMetainfoC: make(chan struct{}),
        }
        tt.setChunkSize(2)
-       generics.MakeMapIfNil(&tt.conns)
+       g.MakeMapIfNil(&tt.conns)
        pc := PeerConn{}
        pc.t = &tt
        pc.peerImpl = &pc
        pc.initRequestState()
-       generics.InitNew(&pc.callbacks)
+       g.InitNew(&pc.callbacks)
        tt.conns[&pc] = struct{}{}
        err = pc.peerSentHave(0)
        c.Assert(err, qt.IsNil)