40 lines
781 B
Go
40 lines
781 B
Go
package storage
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/anacrolix/torrent/metainfo"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBoltPieceCompletion(t *testing.T) {
|
|
td, err := ioutil.TempDir("", "")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(td)
|
|
|
|
pc, err := NewBoltPieceCompletion(td)
|
|
require.NoError(t, err)
|
|
defer pc.Close()
|
|
|
|
pk := metainfo.PieceKey{}
|
|
|
|
b, err := pc.Get(pk)
|
|
require.NoError(t, err)
|
|
assert.False(t, b.Ok)
|
|
|
|
require.NoError(t, pc.Set(pk, false))
|
|
|
|
b, err = pc.Get(pk)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, Completion{Complete: false, Ok: true}, b)
|
|
|
|
require.NoError(t, pc.Set(pk, true))
|
|
|
|
b, err = pc.Get(pk)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, Completion{Complete: true, Ok: true}, b)
|
|
}
|