- return storage.NewResourcePieces(fc.AsResourceProvider())
-}
-
-func TestClientTransferSmallCache(t *testing.T) {
- testClientTransfer(t, testClientTransferParams{
- LeecherStorage: NewFileCacheClientStorageFactory(FileCacheClientStorageFactoryParams{
- SetCapacity: true,
- // Going below the piece length means it can't complete a piece so
- // that it can be hashed.
- Capacity: 5,
- Wrapper: fileCachePieceResourceStorage,
- }),
- SetReadahead: true,
- // Can't readahead too far or the cache will thrash and drop data we
- // thought we had.
- Readahead: 0,
- ExportClientStatus: true,
- })
-}
-
-func TestClientTransferVarious(t *testing.T) {
- // Leecher storage
- for _, ls := range []storageFactory{
- NewFileCacheClientStorageFactory(FileCacheClientStorageFactoryParams{
- Wrapper: fileCachePieceResourceStorage,
- }),
- storage.NewBoltDB,
- } {
- // Seeder storage
- for _, ss := range []func(string) storage.ClientImpl{
- storage.NewFile,
- storage.NewMMap,
- } {
- for _, responsive := range []bool{false, true} {
- testClientTransfer(t, testClientTransferParams{
- Responsive: responsive,
- SeederStorage: ss,
- LeecherStorage: ls,
- })
- for _, readahead := range []int64{-1, 0, 1, 2, 3, 4, 5, 6, 9, 10, 11, 12, 13, 14, 15, 20} {
- testClientTransfer(t, testClientTransferParams{
- SeederStorage: ss,
- Responsive: responsive,
- SetReadahead: true,
- Readahead: readahead,
- LeecherStorage: ls,
- })
- }
- }
- }
- }
-}
-
-type testClientTransferParams struct {
- Responsive bool
- Readahead int64
- SetReadahead bool
- ExportClientStatus bool
- LeecherStorage func(string) storage.ClientImpl
- SeederStorage func(string) storage.ClientImpl
- SeederUploadRateLimiter *rate.Limiter
- LeecherDownloadRateLimiter *rate.Limiter
-}
-
-// Creates a seeder and a leecher, and ensures the data transfers when a read
-// is attempted on the leecher.
-func testClientTransfer(t *testing.T, ps testClientTransferParams) {
- greetingTempDir, mi := testutil.GreetingTestTorrent()
- defer os.RemoveAll(greetingTempDir)
- // Create seeder and a Torrent.
- cfg := TestingConfig()
- cfg.Seed = true
- cfg.UploadRateLimiter = ps.SeederUploadRateLimiter
- // cfg.ListenAddr = "localhost:4000"
- if ps.SeederStorage != nil {
- cfg.DefaultStorage = ps.SeederStorage(greetingTempDir)
- defer cfg.DefaultStorage.Close()
- } else {
- cfg.DataDir = greetingTempDir
- }
- seeder, err := NewClient(cfg)
- require.NoError(t, err)
- if ps.ExportClientStatus {
- testutil.ExportStatusWriter(seeder, "s")
- }
- seederTorrent, _, _ := seeder.AddTorrentSpec(TorrentSpecFromMetaInfo(mi))
- // Run a Stats right after Closing the Client. This will trigger the Stats
- // panic in #214 caused by RemoteAddr on Closed uTP sockets.
- defer seederTorrent.Stats()
- defer seeder.Close()
- seederTorrent.VerifyData()
- // Create leecher and a Torrent.
- leecherDataDir, err := ioutil.TempDir("", "")
- require.NoError(t, err)
- defer os.RemoveAll(leecherDataDir)
- if ps.LeecherStorage == nil {
- cfg.DataDir = leecherDataDir
- } else {
- cfg.DefaultStorage = ps.LeecherStorage(leecherDataDir)
- }
- cfg.DownloadRateLimiter = ps.LeecherDownloadRateLimiter
- cfg.Seed = false
- leecher, err := NewClient(cfg)
- require.NoError(t, err)
- defer leecher.Close()
- if ps.ExportClientStatus {
- testutil.ExportStatusWriter(leecher, "l")
- }
- leecherTorrent, new, err := leecher.AddTorrentSpec(func() (ret *TorrentSpec) {
- ret = TorrentSpecFromMetaInfo(mi)
- ret.ChunkSize = 2
- return
- }())
- require.NoError(t, err)
- assert.True(t, new)
- // Now do some things with leecher and seeder.
- leecherTorrent.AddClientPeer(seeder)
- // The Torrent should not be interested in obtaining peers, so the one we
- // just added should be the only one.
- assert.False(t, leecherTorrent.Seeding())
- assert.EqualValues(t, 1, leecherTorrent.Stats().PendingPeers)
- r := leecherTorrent.NewReader()
- defer r.Close()
- if ps.Responsive {
- r.SetResponsive()
- }
- if ps.SetReadahead {
- r.SetReadahead(ps.Readahead)
- }
- assertReadAllGreeting(t, r)
- assert.True(t, 13 <= seederTorrent.Stats().BytesWrittenData)
- assert.True(t, 8 <= seederTorrent.Stats().ChunksWritten)
- assert.True(t, 13 <= leecherTorrent.Stats().BytesReadData)
- assert.True(t, 8 <= leecherTorrent.Stats().ChunksRead)
- // Try reading through again for the cases where the torrent data size
- // exceeds the size of the cache.
- assertReadAllGreeting(t, r)
-}
-
-func assertReadAllGreeting(t *testing.T, r io.ReadSeeker) {
- pos, err := r.Seek(0, io.SeekStart)
- assert.NoError(t, err)
- assert.EqualValues(t, 0, pos)
- _greeting, err := ioutil.ReadAll(r)
- assert.NoError(t, err)
- assert.EqualValues(t, testutil.GreetingFileContents, _greeting)
-}
-
-// Check that after completing leeching, a leecher transitions to a seeding
-// correctly. Connected in a chain like so: Seeder <-> Leecher <-> LeecherLeecher.
-func TestSeedAfterDownloading(t *testing.T) {
- greetingTempDir, mi := testutil.GreetingTestTorrent()
- defer os.RemoveAll(greetingTempDir)
- cfg := TestingConfig()
- cfg.Seed = true
- cfg.DataDir = greetingTempDir
- seeder, err := NewClient(cfg)
- require.NoError(t, err)
- defer seeder.Close()
- testutil.ExportStatusWriter(seeder, "s")
- seederTorrent, _, _ := seeder.AddTorrentSpec(TorrentSpecFromMetaInfo(mi))
- seederTorrent.VerifyData()
- cfg.DataDir, err = ioutil.TempDir("", "")
- require.NoError(t, err)
- defer os.RemoveAll(cfg.DataDir)
- leecher, err := NewClient(cfg)
- require.NoError(t, err)
- defer leecher.Close()
- testutil.ExportStatusWriter(leecher, "l")
- cfg.Seed = false
- cfg.DataDir, err = ioutil.TempDir("", "")
- require.NoError(t, err)
- defer os.RemoveAll(cfg.DataDir)
- leecherLeecher, _ := NewClient(cfg)
- require.NoError(t, err)
- defer leecherLeecher.Close()
- testutil.ExportStatusWriter(leecherLeecher, "ll")
- leecherGreeting, _, _ := leecher.AddTorrentSpec(func() (ret *TorrentSpec) {
- ret = TorrentSpecFromMetaInfo(mi)
- ret.ChunkSize = 2
- return
- }())
- llg, _, _ := leecherLeecher.AddTorrentSpec(func() (ret *TorrentSpec) {
- ret = TorrentSpecFromMetaInfo(mi)
- ret.ChunkSize = 3
- return
- }())
- // Simultaneously DownloadAll in Leecher, and read the contents
- // consecutively in LeecherLeecher. This non-deterministically triggered a
- // case where the leecher wouldn't unchoke the LeecherLeecher.
- var wg sync.WaitGroup
- wg.Add(1)
- go func() {
- defer wg.Done()
- r := llg.NewReader()
- defer r.Close()
- b, err := ioutil.ReadAll(r)
- require.NoError(t, err)
- assert.EqualValues(t, testutil.GreetingFileContents, b)
- }()
- leecherGreeting.AddClientPeer(seeder)
- leecherGreeting.AddClientPeer(leecherLeecher)
- wg.Add(1)
- go func() {
- defer wg.Done()
- leecherGreeting.DownloadAll()
- leecher.WaitAll()
- }()
- wg.Wait()