forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-19 10:22:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

p2p/enode: more quic tests

Browse source
This commit is contained in:
lightclient 2024-08-18 16:44:37 -06:00
parent 299eff3eb9
commit 3b4a63934c
No known key found for this signature in database
GPG key ID: 75C916AFEE20183E
1 changed files with 59 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
p2p/enode/node_test.go
View file

@ -73,6 +73,7 @@ func TestNodeEndpoints(t *testing.T) {
wantIP netip.Addr wantIP netip.Addr
wantUDP int wantUDP int
wantTCP int wantTCP int
wantQUIC int
} }
tests := []endpointTest{ tests := []endpointTest{
{ {
@ -106,6 +107,14 @@ func TestNodeEndpoints(t *testing.T) {
return SignNull(&r, id) return SignNull(&r, id)
}(), }(),
}, },
{
name: "quic6-only",
node: func() *Node {
var r enr.Record
r.Set(enr.QUIC6(9000))
return SignNull(&r, id)
}(),
},
{ {
name: "ipv4-only-loopback", name: "ipv4-only-loopback",
node: func() *Node { node: func() *Node {
@ -217,6 +226,48 @@ func TestNodeEndpoints(t *testing.T) {
wantIP: netip.MustParseAddr("192.168.2.2"), wantIP: netip.MustParseAddr("192.168.2.2"),
wantUDP: 30304, wantUDP: 30304,
}, },
{
name: "ipv4-quic",
node: func() *Node {
var r enr.Record
r.Set(enr.IPv4Addr(netip.MustParseAddr("99.22.33.1")))
r.Set(enr.QUIC(9001))
return SignNull(&r, id)
}(),
wantIP: netip.MustParseAddr("99.22.33.1"),
wantQUIC: 9001,
},
{ // Because the node is IPv4, the quic6 entry won't be loaded.
name: "ipv4-quic6",
node: func() *Node {
var r enr.Record
r.Set(enr.IPv4Addr(netip.MustParseAddr("99.22.33.1")))
r.Set(enr.QUIC6(9001))
return SignNull(&r, id)
}(),
wantIP: netip.MustParseAddr("99.22.33.1"),
},
{
name: "ipv6-quic",
node: func() *Node {
var r enr.Record
r.Set(enr.IPv6Addr(netip.MustParseAddr("2001::ff00:0042:8329")))
r.Set(enr.QUIC(9001))
return SignNull(&r, id)
}(),
wantIP: netip.MustParseAddr("2001::ff00:0042:8329"),
},
{
name: "ipv6-quic6",
node: func() *Node {
var r enr.Record
r.Set(enr.IPv6Addr(netip.MustParseAddr("2001::ff00:0042:8329")))
r.Set(enr.QUIC6(9001))
return SignNull(&r, id)
}(),
wantIP: netip.MustParseAddr("2001::ff00:0042:8329"),
wantQUIC: 9001,
},
} }
for _, test := range tests { for _, test := range tests {
@ -230,6 +281,9 @@ func TestNodeEndpoints(t *testing.T) {
if test.wantTCP != test.node.TCP() { if test.wantTCP != test.node.TCP() {
t.Errorf("node has wrong TCP port %d, want %d", test.node.TCP(), test.wantTCP) t.Errorf("node has wrong TCP port %d, want %d", test.node.TCP(), test.wantTCP)
} }
if quic, _ := test.node.QUICEndpoint(); test.wantQUIC != int(quic.Port()) {
t.Errorf("node has wrong QUIC port %d, want %d", quic.Port(), test.wantQUIC)
}
}) })
} }
} }