]> Sergey Matveev's repositories - btrtrc.git/blobdiff - file_test.go
Drop support for go 1.20
[btrtrc.git] / file_test.go
index 0443cf3bd5cc63a9e9766f45a9d4a489f709c43f..2f57bcf46ae3d747f499598eeb1116dcb2015df7 100644 (file)
@@ -3,7 +3,7 @@ package torrent
 import (
        "testing"
 
-       "github.com/anacrolix/missinggo/v2/bitmap"
+       "github.com/RoaringBitmap/roaring"
        "github.com/stretchr/testify/assert"
 )
 
@@ -28,19 +28,20 @@ type testFileBytesLeft struct {
        endPieceIndex   int
        fileOffset      int64
        fileLength      int64
-       completedPieces bitmap.Bitmap
+       completedPieces roaring.Bitmap
        expected        int64
        name            string
 }
 
 func (me testFileBytesLeft) Run(t *testing.T) {
        t.Run(me.name, func(t *testing.T) {
-               assert.EqualValues(t, me.expected, fileBytesLeft(me.usualPieceSize, me.firstPieceIndex, me.endPieceIndex, me.fileOffset, me.fileLength, me.completedPieces))
+               assert.EqualValues(t, me.expected, fileBytesLeft(me.usualPieceSize, me.firstPieceIndex, me.endPieceIndex, me.fileOffset, me.fileLength, &me.completedPieces, func(pieceIndex int) int64 {
+                       return 0
+               }))
        })
 }
 
 func TestFileBytesLeft(t *testing.T) {
-
        testFileBytesLeft{
                usualPieceSize:  3,
                firstPieceIndex: 1,
@@ -108,7 +109,7 @@ func TestFileBytesLeft(t *testing.T) {
                fileOffset:      5,
                fileLength:      7,
                expected:        0,
-               completedPieces: func() (ret bitmap.Bitmap) {
+               completedPieces: func() (ret roaring.Bitmap) {
                        ret.AddRange(0, 5)
                        return
                }(),