Update to multiple-blobs-per-value squirrel

[btrtrc.git] / storage / issue96_test.go

diff --git a/storage/issue96_test.go b/storage/issue96_test.go
index b7267ba4ca76aaffae289ef1dbd58f33302772d8..726c11c803148e80bffa2df3166e605505bf3b0b 100644 (file)
--- a/storage/issue96_test.go
+++ b/storage/issue96_test.go
@@ -1,8 +1,6 @@
 package storage
 
 import (
-       "io/ioutil"
-       "os"
        "testing"
 
        "github.com/stretchr/testify/require"
@@ -10,11 +8,11 @@ import (
        "github.com/anacrolix/torrent/metainfo"
 )
 
-func testMarkedCompleteMissingOnRead(t *testing.T, csf func(string) ClientImpl) {
-       td, err := ioutil.TempDir("", "")
-       require.NoError(t, err)
-       defer os.RemoveAll(td)
-       cs := NewClient(csf(td))
+func testMarkedCompleteMissingOnRead(t *testing.T, csf func(string) ClientImplCloser) {
+       td := t.TempDir()
+       cic := csf(td)
+       defer cic.Close()
+       cs := NewClient(cic)
        info := &metainfo.Info{
                PieceLength: 1,
                Files:       []metainfo.FileInfo{{Path: []string{"a"}, Length: 1}},
@@ -27,7 +25,7 @@ func testMarkedCompleteMissingOnRead(t *testing.T, csf func(string) ClientImpl)
        n, err := p.ReadAt(make([]byte, 1), 0)
        require.Error(t, err)
        require.EqualValues(t, 0, n)
-       require.False(t, p.GetIsComplete())
+       require.False(t, p.Completion().Complete)
 }
 
 func TestMarkedCompleteMissingOnReadFile(t *testing.T) {