2
0
mirror of synced 2025-02-24 14:48:27 +00:00

Fix a few tests

This commit is contained in:
Matt Joiner 2015-01-11 21:42:57 +11:00
parent 4fbe138c8c
commit c303ce09d8
2 changed files with 17 additions and 4 deletions

View File

@ -17,6 +17,7 @@ import (
func TestClientDefault(t *testing.T) { func TestClientDefault(t *testing.T) {
cl, err := NewClient(&Config{ cl, err := NewClient(&Config{
NoDefaultBlocklist: true, NoDefaultBlocklist: true,
ListenAddr: ":0",
}) })
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
@ -178,3 +179,15 @@ func TestUTPRawConn(t *testing.T) {
t.Fatalf("messages received: %d", msgsReceived) t.Fatalf("messages received: %d", msgsReceived)
} }
} }
func TestTwoClientsArbitraryPorts(t *testing.T) {
for i := 0; i < 2; i++ {
cl, err := NewClient(&Config{
ListenAddr: ":0",
})
if err != nil {
t.Fatal(err)
}
defer cl.Stop()
}
}

View File

@ -65,7 +65,7 @@ func (ca cachedAddr) UDPAddr() *net.UDPAddr {
return ca.a.(*net.UDPAddr) return ca.a.(*net.UDPAddr)
} }
func newDHTAddr(addr *net.UDPAddr) dHTAddr { func newDHTAddr(addr net.Addr) dHTAddr {
return cachedAddr{addr, addr.String()} return cachedAddr{addr, addr.String()}
} }
@ -499,7 +499,7 @@ func (s *Server) serve() error {
logonce.Stderr.Printf("received dht packet exceeds buffer size") logonce.Stderr.Printf("received dht packet exceeds buffer size")
continue continue
} }
s.processPacket(b[:n], newDHTAddr(addr.(*net.UDPAddr))) s.processPacket(b[:n], newDHTAddr(addr))
} }
} }
@ -991,8 +991,8 @@ func (s *Server) bootstrap() (err error) {
var t *transaction var t *transaction
t, err = s.findNode(node.addr, s.id) t, err = s.findNode(node.addr, s.id)
if err != nil { if err != nil {
log.Printf("error sending find_node: %s", err) err = fmt.Errorf("error sending find_node: %s", err)
continue return
} }
outstanding.Add(1) outstanding.Add(1)
go func() { go func() {