31 lines
862 B
Go
31 lines
862 B
Go
package torrent
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/anacrolix/missinggo/bitmap"
|
|
"github.com/anacrolix/missinggo/iter"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestTorrentOffsetRequest(t *testing.T) {
|
|
check := func(tl, ps, off int64, expected request, ok bool) {
|
|
req, _ok := torrentOffsetRequest(tl, ps, defaultChunkSize, off)
|
|
assert.Equal(t, _ok, ok)
|
|
assert.Equal(t, req, expected)
|
|
}
|
|
check(13, 5, 0, newRequest(0, 0, 5), true)
|
|
check(13, 5, 3, newRequest(0, 0, 5), true)
|
|
check(13, 5, 11, newRequest(2, 0, 3), true)
|
|
check(13, 5, 13, request{}, false)
|
|
}
|
|
|
|
func TestIterBitmapsDistinct(t *testing.T) {
|
|
var skip, first, second bitmap.Bitmap
|
|
skip.Add(1)
|
|
first.Add(1, 0, 3)
|
|
second.Add(1, 2, 0)
|
|
assert.Equal(t, []interface{}{0, 3, 2}, iter.ToSlice(iterBitmapsDistinct(skip.Copy(), first, second)))
|
|
assert.Equal(t, []int{1}, skip.ToSortedSlice())
|
|
}
|