reader_test.go

   1 package torrent
   2
   3 import (
   4         "context"
   5         "testing"
   6         "time"
   7
   8         "github.com/anacrolix/torrent/internal/testutil"
   9         "github.com/stretchr/testify/require"
  10 )
  11
  12 func TestReaderReadContext(t *testing.T) {
  13         cl, err := NewClient(TestingConfig())
  14         require.NoError(t, err)
  15         defer cl.Close()
  16         tt, err := cl.AddTorrent(testutil.GreetingMetaInfo())
  17         require.NoError(t, err)
  18         defer tt.Drop()
  19         ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(time.Millisecond))
  20         defer cancel()
  21         r := tt.Files()[0].NewReader()
  22         defer r.Close()
  23         _, err = r.ReadContext(ctx, make([]byte, 1))
  24         require.EqualValues(t, context.DeadlineExceeded, err)
  25 }