Fix BenchmarkConnectionMainReadLoop

Broken by piece request order changes.
This commit is contained in:
Matt Joiner 2021-12-15 18:54:47 +11:00
parent 1668a18859
commit e7ee568c52
1 changed files with 4 additions and 7 deletions

View File

@ -7,7 +7,6 @@ import (
"sync"
"testing"
"github.com/anacrolix/missinggo/pubsub"
"github.com/frankban/quicktest"
"github.com/stretchr/testify/require"
@ -93,17 +92,15 @@ func BenchmarkConnectionMainReadLoop(b *testing.B) {
})
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))