Attempt holepunch after initial dial fails

[btrtrc.git] / test / transfer_test.go

diff --git a/test/transfer_test.go b/test/transfer_test.go
index 2c77e673f0f9c96c196695bed784bdf602c99701..fa159326a908f9526a1a730e45296adfe82d18ce 100644 (file)
--- a/test/transfer_test.go
+++ b/test/transfer_test.go
@@ -13,7 +13,7 @@ import (
        "github.com/anacrolix/torrent"
        "github.com/anacrolix/torrent/internal/testutil"
        "github.com/anacrolix/torrent/storage"
-       "github.com/frankban/quicktest"
+       qt "github.com/frankban/quicktest"
        "golang.org/x/time/rate"
 
        "github.com/stretchr/testify/assert"
@@ -138,6 +138,7 @@ func TestSeedAfterDownloading(t *testing.T) {
 
        cfg := torrent.TestingConfig(t)
        cfg.Seed = true
+       cfg.MaxAllocPeerRequestDataPerConn = 4
        cfg.DataDir = greetingTempDir
        seeder, err := torrent.NewClient(cfg)
        require.NoError(t, err)
@@ -159,6 +160,7 @@ func TestSeedAfterDownloading(t *testing.T) {
        cfg = torrent.TestingConfig(t)
        cfg.Seed = false
        cfg.DataDir = t.TempDir()
+       cfg.MaxAllocPeerRequestDataPerConn = 4
        leecherLeecher, _ := torrent.NewClient(cfg)
        require.NoError(t, err)
        defer leecherLeecher.Close()
@@ -186,7 +188,7 @@ func TestSeedAfterDownloading(t *testing.T) {
                defer wg.Done()
                r := llg.NewReader()
                defer r.Close()
-               quicktest.Check(t, iotest.TestReader(r, []byte(testutil.GreetingFileContents)), quicktest.IsNil)
+               qt.Check(t, iotest.TestReader(r, []byte(testutil.GreetingFileContents)), qt.IsNil)
        }()
        done := make(chan struct{})
        defer close(done)