data/file/file_test.go

   1 package file
   2
   3 import (
   4         "bytes"
   5         "io"
   6         "io/ioutil"
   7         "os"
   8         "path/filepath"
   9         "testing"
  10
  11         "github.com/stretchr/testify/assert"
  12         "github.com/stretchr/testify/require"
  13
  14         "github.com/anacrolix/torrent/metainfo"
  15 )
  16
  17 func TestShortFile(t *testing.T) {
  18         td, err := ioutil.TempDir("", "")
  19         require.NoError(t, err)
  20         defer os.RemoveAll(td)
  21         data := TorrentData(&metainfo.Info{
  22                 Name:   "a",
  23                 Length: 2,
  24         }, td)
  25         f, err := os.Create(filepath.Join(td, "a"))
  26         err = f.Truncate(1)
  27         f.Close()
  28         var buf bytes.Buffer
  29         n, err := io.Copy(&buf, io.NewSectionReader(data, 0, 2))
  30         assert.EqualValues(t, 1, n)
  31         assert.Equal(t, io.ErrUnexpectedEOF, err)
  32 }