Limit keep alives for rate limited download test

[btrtrc.git] / test / transfer_test.go

diff --git a/test/transfer_test.go b/test/transfer_test.go
index c32dfa740c709e3fa1523753dcd278776cd86c6b..71de58b44b7b53fe0cd1ebe61f62865ebaf06f79 100644 (file)
--- a/test/transfer_test.go
+++ b/test/transfer_test.go
@@ -250,6 +250,13 @@ func TestClientTransferRateLimitedUpload(t *testing.T) {
 func TestClientTransferRateLimitedDownload(t *testing.T) {
        testClientTransfer(t, testClientTransferParams{
                LeecherDownloadRateLimiter: rate.NewLimiter(512, 512),
+               ConfigureSeeder: ConfigureClient{
+                       Config: func(cfg *torrent.ClientConfig) {
+                               // If we send too many keep alives, we consume all the leechers available download
+                               // rate. The default isn't exposed, but a minute is pretty reasonable.
+                               cfg.KeepAliveTimeout = time.Minute
+                       },
+               },
        })
 }
 
@@ -427,6 +434,6 @@ func TestSeedAfterDownloading(t *testing.T) {
 }
 
 type ConfigureClient struct {
-       Config func(*torrent.ClientConfig)
-       Client func(*torrent.Client)
+       Config func(cfg *torrent.ClientConfig)
+       Client func(cl *torrent.Client)
 }