72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package torrent
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/anacrolix/missinggo/v2/bitmap"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestFileExclusivePieces(t *testing.T) {
|
|
for _, _case := range []struct {
|
|
off, size, pieceSize int64
|
|
begin, end int
|
|
}{
|
|
{0, 2, 2, 0, 1},
|
|
{1, 2, 2, 1, 1},
|
|
{1, 4, 2, 1, 2},
|
|
} {
|
|
begin, end := byteRegionExclusivePieces(_case.off, _case.size, _case.pieceSize)
|
|
assert.EqualValues(t, _case.begin, begin)
|
|
assert.EqualValues(t, _case.end, end)
|
|
}
|
|
}
|
|
|
|
type testFileBytesLeft struct {
|
|
usualPieceSize int64
|
|
firstPieceIndex int
|
|
endPieceIndex int
|
|
fileOffset int64
|
|
fileLength int64
|
|
completedPieces bitmap.Bitmap
|
|
expected int64
|
|
name string
|
|
}
|
|
|
|
func (me testFileBytesLeft) Run(t *testing.T) {
|
|
t.Run(me.name, func(t *testing.T) {
|
|
assert.EqualValues(t, me.expected, fileBytesLeft(me.usualPieceSize, me.firstPieceIndex, me.endPieceIndex, me.fileOffset, me.fileLength, me.completedPieces))
|
|
})
|
|
}
|
|
|
|
func TestFileBytesLeft(t *testing.T) {
|
|
testFileBytesLeft{
|
|
usualPieceSize: 2,
|
|
firstPieceIndex: 1,
|
|
endPieceIndex: 1,
|
|
fileOffset: 1,
|
|
fileLength: 1,
|
|
expected: 1,
|
|
}.Run(t)
|
|
|
|
testFileBytesLeft{
|
|
usualPieceSize: 3,
|
|
firstPieceIndex: 0,
|
|
endPieceIndex: 0,
|
|
fileOffset: 1,
|
|
fileLength: 1,
|
|
expected: 1,
|
|
name: "FileInFirstPiece",
|
|
}.Run(t)
|
|
|
|
testFileBytesLeft{
|
|
usualPieceSize: 3,
|
|
firstPieceIndex: 0,
|
|
endPieceIndex: 0,
|
|
fileOffset: 1,
|
|
fileLength: 1,
|
|
expected: 1,
|
|
name: "LandLocked",
|
|
}.Run(t)
|
|
}
|