More Windows test fixes

[btrtrc.git] / tracker / udp / udp_test.go

diff --git a/tracker/udp/udp_test.go b/tracker/udp/udp_test.go
index 862aec73d572d0ebd0e360a674ed7b6c033e04a8..64aeb80ff7426c514f5f63e7c2eab71fb1b5851b 100644 (file)
--- a/tracker/udp/udp_test.go
+++ b/tracker/udp/udp_test.go
@@ -2,12 +2,19 @@ package udp
 
 import (
        "bytes"
+       "context"
+       "crypto/rand"
        "encoding/binary"
        "io"
        "net"
+       "sync"
        "testing"
+       "time"
 
        "github.com/anacrolix/dht/v2/krpc"
+       _ "github.com/anacrolix/envpprof"
+       "github.com/anacrolix/missinggo/v2/iter"
+       qt "github.com/frankban/quicktest"
        "github.com/stretchr/testify/require"
 )
 
@@ -50,8 +57,10 @@ func TestLongWriteUDP(t *testing.T) {
        for msgLen := 1; ; msgLen *= 2 {
                n, err := c.Write(make([]byte, msgLen))
                if err != nil {
-                       require.Contains(t, err.Error(), "message too long")
-                       return
+                       if isErrMessageTooLong(err) {
+                               return
+                       }
+                       t.Fatalf("expected message too long error: %v", err)
                }
                if n < msgLen {
                        t.FailNow()
@@ -73,3 +82,58 @@ func TestConvertInt16ToInt(t *testing.T) {
                t.FailNow()
        }
 }
+
+func TestConnClientLogDispatchUnknownTransactionId(t *testing.T) {
+       const network = "udp"
+       cc, err := NewConnClient(NewConnClientOpts{
+               Network: network,
+       })
+       c := qt.New(t)
+       c.Assert(err, qt.IsNil)
+       defer cc.Close()
+       pc, err := net.ListenPacket(network, ":0")
+       c.Assert(err, qt.IsNil)
+       defer pc.Close()
+       ccAddr := *cc.LocalAddr().(*net.UDPAddr)
+       ccAddr.IP = net.IPv6loopback
+       _, err = pc.WriteTo(make([]byte, 30), &ccAddr)
+       c.Assert(err, qt.IsNil)
+}
+
+func TestConnectionIdMismatch(t *testing.T) {
+       t.Skip("Server host returns consistent connection ID in limited tests and so isn't effective.")
+       cl, err := NewConnClient(NewConnClientOpts{
+               // This host seems to return `Connection ID missmatch.\x00` every 2 minutes or so under
+               // heavy use.
+               Host: "tracker.torrent.eu.org:451",
+               //Host:    "tracker.opentrackr.org:1337",
+               Network: "udp",
+       })
+       c := qt.New(t)
+       c.Assert(err, qt.IsNil)
+       defer cl.Close()
+       ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
+       defer cancel()
+       // Force every request to use a different connection ID. It's racey, but we want to get a
+       // different ID issued before a request can be sent with an old ID.
+       cl.Client.shouldReconnectOverride = func() bool { return true }
+       started := time.Now()
+       var wg sync.WaitGroup
+       for range iter.N(2) {
+               ar := AnnounceRequest{
+                       NumWant: -1,
+                       Event:   2,
+               }
+               rand.Read(ar.InfoHash[:])
+               rand.Read(ar.PeerId[:])
+               //spew.Dump(ar)
+               wg.Add(1)
+               go func() {
+                       defer wg.Done()
+                       _, _, err := cl.Announce(ctx, ar, Options{})
+                       // I'm looking for `error response: "Connection ID missmatch.\x00"`.
+                       t.Logf("announce error after %v: %v", time.Since(started), err)
+               }()
+       }
+       wg.Wait()
+}