storage/file_test.go

   1 package storage
   2
   3 import (
   4         "bytes"
   5         "io"
   6         "io/ioutil"
   7         "os"
   8         "path/filepath"
   9         "testing"
  10
  11         "github.com/anacrolix/missinggo"
  12         "github.com/stretchr/testify/assert"
  13         "github.com/stretchr/testify/require"
  14
  15         "github.com/anacrolix/torrent/metainfo"
  16 )
  17
  18 func TestShortFile(t *testing.T) {
  19         td, err := ioutil.TempDir("", "")
  20         require.NoError(t, err)
  21         defer os.RemoveAll(td)
  22         s := NewFile(td)
  23         info := &metainfo.InfoEx{
  24                 Info: metainfo.Info{
  25                         Name:        "a",
  26                         Length:      2,
  27                         PieceLength: missinggo.MiB,
  28                 },
  29         }
  30         ts, err := s.OpenTorrent(info)
  31         assert.NoError(t, err)
  32         f, err := os.Create(filepath.Join(td, "a"))
  33         err = f.Truncate(1)
  34         f.Close()
  35         var buf bytes.Buffer
  36         p := info.Piece(0)
  37         n, err := io.Copy(&buf, io.NewSectionReader(ts.Piece(p), 0, p.Length()))
  38         assert.EqualValues(t, 1, n)
  39         assert.Equal(t, io.ErrUnexpectedEOF, err)
  40 }