Make opening a torrent in storage an explicit method

[btrtrc.git] / storage / file_test.go

diff --git a/storage/file_test.go b/storage/file_test.go
index 0297c389eaae839a328749332f5e4e9edbc1425f..6e1d055e7d65c13a6e9fa32078b8a2f1c6d839b4 100644 (file)
--- a/storage/file_test.go
+++ b/storage/file_test.go
@@ -19,7 +19,7 @@ func TestShortFile(t *testing.T) {
        td, err := ioutil.TempDir("", "")
        require.NoError(t, err)
        defer os.RemoveAll(td)
-       data := NewFile(td)
+       s := NewFile(td)
        info := &metainfo.InfoEx{
                Info: metainfo.Info{
                        Name:        "a",
@@ -27,12 +27,14 @@ func TestShortFile(t *testing.T) {
                        PieceLength: missinggo.MiB,
                },
        }
+       ts, err := s.OpenTorrent(info)
+       assert.NoError(t, err)
        f, err := os.Create(filepath.Join(td, "a"))
        err = f.Truncate(1)
        f.Close()
        var buf bytes.Buffer
        p := info.Piece(0)
-       n, err := io.Copy(&buf, io.NewSectionReader(data.Piece(p), 0, p.Length()))
+       n, err := io.Copy(&buf, io.NewSectionReader(ts.Piece(p), 0, p.Length()))
        assert.EqualValues(t, 1, n)
        assert.Equal(t, io.ErrUnexpectedEOF, err)
 }