48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package torrent
|
|
|
|
import (
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/anacrolix/missinggo/iter"
|
|
"github.com/anacrolix/missinggo/v2/bitmap"
|
|
"github.com/davecgh/go-spew/spew"
|
|
"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 BenchmarkIterBitmapsDistinct(t *testing.B) {
|
|
t.ReportAllocs()
|
|
for range iter.N(t.N) {
|
|
var skip, first, second bitmap.Bitmap
|
|
skip.Add(1)
|
|
first.Add(1, 0, 3)
|
|
second.Add(1, 2, 0)
|
|
skipCopy := skip.Copy()
|
|
t.StartTimer()
|
|
output := iter.ToSlice(iterBitmapsDistinct(&skipCopy, first, second))
|
|
t.StopTimer()
|
|
assert.Equal(t, []interface{}{0, 3, 2}, output)
|
|
assert.Equal(t, []int{1}, skip.ToSortedSlice())
|
|
}
|
|
}
|
|
|
|
func TestSpewConnStats(t *testing.T) {
|
|
s := spew.Sdump(ConnStats{})
|
|
t.Logf("\n%s", s)
|
|
lines := strings.Count(s, "\n")
|
|
assert.EqualValues(t, 2+reflect.ValueOf(ConnStats{}).NumField(), lines)
|
|
}
|