peerconn_test.go | 11 ++++------- diff --git a/peerconn_test.go b/peerconn_test.go index 304a3ec48ec266fca9adcbbb1be0fb550687c8b3..679648bbd098c9bcdc6a0e05620bac9f5ef89ec8 100644 --- a/peerconn_test.go +++ b/peerconn_test.go @@ -7,7 +7,6 @@ "net" "sync" "testing" - "github.com/anacrolix/missinggo/pubsub" "github.com/frankban/quicktest" "github.com/stretchr/testify/require" @@ -93,17 +92,15 @@ DownloadRateLimiter: unlimited, }) cl.initLogger() ts := &torrentStorage{} - t := &Torrent{ - cl: &cl, - storage: &storage.Torrent{TorrentImpl: storage.TorrentImpl{Piece: ts.Piece, Close: ts.Close}}, - pieceStateChanges: pubsub.NewPubSub(), - } - t.setChunkSize(defaultChunkSize) + t := cl.newTorrent(metainfo.Hash{}, nil) + t.initialPieceCheckDisabled = true require.NoError(b, t.setInfo(&metainfo.Info{ Pieces: make([]byte, 20), Length: 1 << 20, PieceLength: 1 << 20, })) + t.storage = &storage.Torrent{TorrentImpl: storage.TorrentImpl{Piece: ts.Piece, Close: ts.Close}} + t.onSetInfo() t._pendingPieces.Add(0) r, w := net.Pipe() cn := cl.newConnection(r, true, r.RemoteAddr(), r.RemoteAddr().Network(), regularNetConnPeerConnConnString(r))