gorond test files

[btrtrc.git] / tracker / udp-server_test.go

diff --git a/tracker/udp-server_test.go b/tracker/udp-server_test.go
index 824038ea276870c166e79fc052c909258890a6fd..84408bfd1c253567908b24b2a4f85fb9ce9b6b08 100644 (file)
--- a/tracker/udp-server_test.go
+++ b/tracker/udp-server_test.go
@@ -10,6 +10,7 @@ import (
 
        "github.com/anacrolix/dht/v2/krpc"
        "github.com/anacrolix/missinggo/v2"
+
        "github.com/anacrolix/torrent/tracker/udp"
 )
 
@@ -21,7 +22,7 @@ type torrent struct {
 
 type server struct {
        pc    net.PacketConn
-       conns map[int64]struct{}
+       conns map[udp.ConnectionId]struct{}
        t     map[[20]byte]torrent
 }
 
@@ -46,10 +47,10 @@ func (s *server) respond(addr net.Addr, rh udp.ResponseHeader, parts ...interfac
        return
 }
 
-func (s *server) newConn() (ret int64) {
-       ret = rand.Int63()
+func (s *server) newConn() (ret udp.ConnectionId) {
+       ret = rand.Uint64()
        if s.conns == nil {
-               s.conns = make(map[int64]struct{})
+               s.conns = make(map[udp.ConnectionId]struct{})
        }
        s.conns[ret] = struct{}{}
        return