f055abe2fc
In the native file-based storage, mark pieces incomplete if the necessary file data is missing, or there's a read error on a piece.
38 lines
866 B
Go
38 lines
866 B
Go
package storage
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/anacrolix/torrent/metainfo"
|
|
)
|
|
|
|
func testMarkedCompleteMissingOnRead(t *testing.T, csf func(string) Client) {
|
|
td, err := ioutil.TempDir("", "")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(td)
|
|
cs := csf(td)
|
|
info := &metainfo.InfoEx{
|
|
Info: metainfo.Info{
|
|
PieceLength: 1,
|
|
Files: []metainfo.FileInfo{{Path: []string{"a"}, Length: 1}},
|
|
},
|
|
}
|
|
ts, err := cs.OpenTorrent(info)
|
|
require.NoError(t, err)
|
|
p := ts.Piece(info.Piece(0))
|
|
require.NoError(t, p.MarkComplete())
|
|
require.False(t, p.GetIsComplete())
|
|
n, err := p.ReadAt(make([]byte, 1), 0)
|
|
require.Error(t, err)
|
|
require.EqualValues(t, 0, n)
|
|
require.False(t, p.GetIsComplete())
|
|
}
|
|
|
|
func TestMarkedCompleteMissingOnReadFile(t *testing.T) {
|
|
testMarkedCompleteMissingOnRead(t, NewFile)
|
|
}
|