Reintroduce torrent-wide PEX throttling

[btrtrc.git] / issue211_test.go

diff --git a/issue211_test.go b/issue211_test.go
index c12a9c9e303714a81de7d603cbe0cd6af978b336..7794f8a50137974993485f57d1a14f30f3d5a913 100644 (file)
--- a/issue211_test.go
+++ b/issue211_test.go
@@ -1,3 +1,6 @@
+//go:build !wasm
+// +build !wasm
+
 package torrent
 
 import (
@@ -5,15 +8,16 @@ import (
        "io/ioutil"
        "testing"
 
-       "github.com/anacrolix/torrent/internal/testutil"
-       "github.com/anacrolix/torrent/storage"
        "github.com/stretchr/testify/assert"
        "github.com/stretchr/testify/require"
        "golang.org/x/time/rate"
+
+       "github.com/anacrolix/torrent/internal/testutil"
+       "github.com/anacrolix/torrent/storage"
 )
 
 func TestDropTorrentWithMmapStorageWhileHashing(t *testing.T) {
-       cfg := TestingConfig()
+       cfg := TestingConfig(t)
        // Ensure the data is present when the torrent is added, and not obtained
        // over the network as the test runs.
        cfg.DownloadRateLimiter = rate.NewLimiter(0, 0)
@@ -22,8 +26,10 @@ func TestDropTorrentWithMmapStorageWhileHashing(t *testing.T) {
        defer cl.Close()
 
        td, mi := testutil.GreetingTestTorrent()
+       mms := storage.NewMMap(td)
+       defer mms.Close()
        tt, new, err := cl.AddTorrentSpec(&TorrentSpec{
-               Storage:   storage.NewMMap(td),
+               Storage:   mms,
                InfoHash:  mi.HashInfoBytes(),
                InfoBytes: mi.InfoBytes,
        })