8 "github.com/stretchr/testify/assert"
9 "github.com/stretchr/testify/require"
11 "github.com/anacrolix/torrent/metainfo"
14 func TestBoltPieceCompletion(t *testing.T) {
15 td, err := ioutil.TempDir("", "")
16 require.NoError(t, err)
17 defer os.RemoveAll(td)
19 pc, err := NewBoltPieceCompletion(td)
20 require.NoError(t, err)
23 pk := metainfo.PieceKey{}
26 require.NoError(t, err)
29 require.NoError(t, pc.Set(pk, false))
32 require.NoError(t, err)
33 assert.Equal(t, Completion{Complete: false, Ok: true}, b)
35 require.NoError(t, pc.Set(pk, true))
38 require.NoError(t, err)
39 assert.Equal(t, Completion{Complete: true, Ok: true}, b)