]> Sergey Matveev's repositories - btrtrc.git/blobdiff - misc_test.go
Drop support for go 1.20
[btrtrc.git] / misc_test.go
index 32e60fa956319a9b1b414a9f411e8ff63e0758d0..d8c0c7aab2850b0abc14c49e2f00102ac1397d16 100644 (file)
@@ -1,15 +1,18 @@
 package torrent
 
 import (
+       "reflect"
+       "strings"
        "testing"
 
-       "github.com/anacrolix/missinggo/bitmap"
        "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) {
+       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)
@@ -17,15 +20,28 @@ func TestTorrentOffsetRequest(t *testing.T) {
        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)
+       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)
-       skipCopy := skip.Copy()
-       assert.Equal(t, []interface{}{0, 3, 2}, iter.ToSlice(iterBitmapsDistinct(&skipCopy, first, second)))
-       assert.Equal(t, []int{1}, skip.ToSortedSlice())
+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)
 }