X-Git-Url: http://www.git.stargrave.org/?a=blobdiff_plain;f=testing.go;h=6fb5411267e8c486d454ba384d823f59e9ccc6bd;hb=HEAD;hp=facf68a97a70e9ab03d1a83a1489c766185f925f;hpb=0d4858ba29a32c0a46ad38dfccc2328c8c04f57e;p=btrtrc.git

diff --git a/testing.go b/testing.go
index facf68a9..6fb54112 100644
--- a/testing.go
+++ b/testing.go
@@ -1,20 +1,29 @@
 package torrent
 
 import (
-	"github.com/anacrolix/torrent/internal/tmproot"
-)
+	"testing"
+	"time"
+
+	"github.com/anacrolix/log"
 
-var TestingTempDir tmproot.Dir
+	pp "github.com/anacrolix/torrent/peer_protocol"
+)
 
-func TestingConfig() *ClientConfig {
+func TestingConfig(t testing.TB) *ClientConfig {
 	cfg := NewDefaultClientConfig()
 	cfg.ListenHost = LoopbackListenHost
 	cfg.NoDHT = true
-	cfg.DataDir = TestingTempDir.NewSub()
+	cfg.DataDir = t.TempDir()
 	cfg.DisableTrackers = true
 	cfg.NoDefaultPortForwarding = true
 	cfg.DisableAcceptRateLimiting = true
 	cfg.ListenPort = 0
+	cfg.KeepAliveTimeout = time.Millisecond
+	cfg.MinPeerExtensions.SetBit(pp.ExtensionBitFast, true)
+	cfg.Logger = log.Default.WithContextText(t.Name())
+	// 2 would suffice for the greeting test, but 5 is needed for a few other tests. This should be
+	// something slightly higher than the usual chunk size, so it gets tickled in some tests.
+	cfg.MaxAllocPeerRequestDataPerConn = 5
 	//cfg.Debug = true
 	//cfg.Logger = cfg.Logger.WithText(func(m log.Msg) string {
 	//	t := m.Text()