FedP2P/misc_test.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 i := 0; i < t.N; i += 1 {
		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, []bitmap.BitIndex{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)
}