torrent/request-strategy/order_test.go

package request_strategy

import (
	"encoding/gob"
	"testing"

	"github.com/RoaringBitmap/roaring"
	qt "github.com/frankban/quicktest"
	"github.com/google/go-cmp/cmp"
)

func init() {
	gob.Register(chunkIterRange(0))
	gob.Register(sliceChunksIter{})
}

type chunkIterRange ChunkIndex

func (me chunkIterRange) Iter(f func(ChunkIndex)) {
	for offset := ChunkIndex(0); offset < ChunkIndex(me); offset += 1 {
		f(offset)
	}
}

type sliceChunksIter []ChunkIndex

func chunkIter(offsets ...ChunkIndex) ChunksIter {
	return sliceChunksIter(offsets)
}

func (offsets sliceChunksIter) Iter(f func(ChunkIndex)) {
	for _, offset := range offsets {
		f(offset)
	}
}

func requestSetFromSlice(rs ...RequestIndex) (ret roaring.Bitmap) {
	ret.AddMany(rs)
	return
}

func init() {
	gob.Register(intPeerId(0))
}

type intPeerId int

func (i intPeerId) Uintptr() uintptr {
	return uintptr(i)
}

var hasAllRequests = func() (all roaring.Bitmap) {
	all.AddRange(0, roaring.MaxRange)
	return
}()

func checkNumRequestsAndInterest(c *qt.C, next PeerNextRequestState, num uint64, interest bool) {
	addressableBm := next.Requests
	c.Check(addressableBm.GetCardinality(), qt.ContentEquals, num)
	c.Check(next.Interested, qt.Equals, interest)
}

func checkResultsRequestsLen(t *testing.T, reqs roaring.Bitmap, l uint64) {
	qt.Check(t, reqs.GetCardinality(), qt.Equals, l)
}

var peerNextRequestStateChecker = qt.CmpEquals(
	cmp.Transformer(
		"bitmap",
		func(bm roaring.Bitmap) []uint32 {
			return bm.ToArray()
		}))