mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-08-18 18:02:24 +00:00
test(p2p/addrutil): add unit tests
This commit is contained in:
tianyeyouyou
parent
ae707445f5
commit
9dab60faf7
1 changed files with 191 additions and 0 deletions
191
p2p/netutil/addrutil_test.go
Normal file
191
p2p/netutil/addrutil_test.go
Normal file
|
|
@ -0,0 +1,191 @@
|
|||
package netutil
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"net"
|
||||
"net/netip"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// customNetAddr is a custom implementation of net.Addr for testing purposes.
|
||||
type customNetAddr struct{}
|
||||
|
||||
func (c *customNetAddr) Network() string { return "custom" }
|
||||
func (c *customNetAddr) String() string { return "custom" }
|
||||
|
||||
func TestAddrAddr(t *testing.T) {
|
||||
tempDir := t.TempDir()
|
||||
tests := []struct {
|
||||
name string
|
||||
addr net.Addr
|
||||
want netip.Addr
|
||||
}{
|
||||
{
|
||||
name: "IPAddr IPv4",
|
||||
addr: &net.IPAddr{IP: net.ParseIP("192.0.2.1")},
|
||||
want: netip.MustParseAddr("192.0.2.1"),
|
||||
},
|
||||
{
|
||||
name: "IPAddr IPv6",
|
||||
addr: &net.IPAddr{IP: net.ParseIP("2001:db8::1")},
|
||||
want: netip.MustParseAddr("2001:db8::1"),
|
||||
},
|
||||
{
|
||||
name: "TCPAddr IPv4",
|
||||
addr: &net.TCPAddr{IP: net.ParseIP("192.0.2.1"), Port: 8080},
|
||||
want: netip.MustParseAddr("192.0.2.1"),
|
||||
},
|
||||
{
|
||||
name: "TCPAddr IPv6",
|
||||
addr: &net.TCPAddr{IP: net.ParseIP("2001:db8::1"), Port: 8080},
|
||||
want: netip.MustParseAddr("2001:db8::1"),
|
||||
},
|
||||
{
|
||||
name: "UDPAddr IPv4",
|
||||
addr: &net.UDPAddr{IP: net.ParseIP("192.0.2.1"), Port: 8080},
|
||||
want: netip.MustParseAddr("192.0.2.1"),
|
||||
},
|
||||
{
|
||||
name: "UDPAddr IPv6",
|
||||
addr: &net.UDPAddr{IP: net.ParseIP("2001:db8::1"), Port: 8080},
|
||||
want: netip.MustParseAddr("2001:db8::1"),
|
||||
},
|
||||
{
|
||||
name: "Unsupported Addr type",
|
||||
addr: &net.UnixAddr{Name: filepath.Join(tempDir, "test.sock"), Net: "unix"},
|
||||
want: netip.Addr{},
|
||||
},
|
||||
{
|
||||
name: "Nil input",
|
||||
addr: nil,
|
||||
want: netip.Addr{},
|
||||
},
|
||||
{
|
||||
name: "Custom net.Addr implementation",
|
||||
addr: &customNetAddr{},
|
||||
want: netip.Addr{},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := AddrAddr(tt.addr); got != tt.want {
|
||||
t.Errorf("AddrAddr() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestIPToAddr(t *testing.T) {
|
||||
tests := []struct {
|
||||
name string
|
||||
ip net.IP
|
||||
want netip.Addr
|
||||
}{
|
||||
{
|
||||
name: "IPv4",
|
||||
ip: net.ParseIP("192.0.2.1"),
|
||||
want: netip.MustParseAddr("192.0.2.1"),
|
||||
},
|
||||
{
|
||||
name: "IPv6",
|
||||
ip: net.ParseIP("2001:db8::1"),
|
||||
want: netip.MustParseAddr("2001:db8::1"),
|
||||
},
|
||||
{
|
||||
name: "Invalid IP",
|
||||
ip: net.IP{1, 2, 3},
|
||||
want: netip.Addr{},
|
||||
},
|
||||
{
|
||||
name: "Invalid IP (5 octets)",
|
||||
ip: net.IP{192, 0, 2, 1, 1},
|
||||
want: netip.Addr{},
|
||||
},
|
||||
{
|
||||
name: "IPv4-mapped IPv6",
|
||||
ip: net.ParseIP("::ffff:192.0.2.1"),
|
||||
want: netip.MustParseAddr("192.0.2.1"),
|
||||
},
|
||||
{
|
||||
name: "Nil input",
|
||||
ip: nil,
|
||||
want: netip.Addr{},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
if got := IPToAddr(tt.ip); got != tt.want {
|
||||
t.Errorf("IPToAddr() = %v, want %v", got, tt.want)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestRandomAddr(t *testing.T) {
|
||||
// Use a fixed seed for reproducibility
|
||||
rng := rand.New(rand.NewSource(42))
|
||||
|
||||
// Test IPv4 generation
|
||||
t.Run("IPv4", func(t *testing.T) {
|
||||
addr := RandomAddr(rng, true)
|
||||
if !addr.Is4() {
|
||||
t.Errorf("Expected IPv4 address, got %v", addr)
|
||||
}
|
||||
})
|
||||
|
||||
// Test IPv6 generation
|
||||
t.Run("IPv6", func(t *testing.T) {
|
||||
addr := RandomAddr(rng, false)
|
||||
if !addr.Is6() {
|
||||
t.Errorf("Expected IPv6 address, got %v", addr)
|
||||
}
|
||||
})
|
||||
|
||||
// Test random choice between IPv4 and IPv6
|
||||
t.Run("Random choice", func(t *testing.T) {
|
||||
ipv4Count := 0
|
||||
ipv6Count := 0
|
||||
iterations := 1000
|
||||
|
||||
for i := 0; i < iterations; i++ {
|
||||
addr := RandomAddr(rng, false)
|
||||
if addr.Is4() {
|
||||
ipv4Count++
|
||||
} else if addr.Is6() {
|
||||
ipv6Count++
|
||||
} else {
|
||||
t.Errorf("Invalid address generated: %v", addr)
|
||||
}
|
||||
}
|
||||
|
||||
if ipv4Count == 0 || ipv6Count == 0 {
|
||||
t.Errorf("Expected mix of IPv4 and IPv6 addresses, got %d IPv4 and %d IPv6", ipv4Count, ipv6Count)
|
||||
}
|
||||
})
|
||||
|
||||
// Test randomness of generated addresses
|
||||
t.Run("Randomness check", func(t *testing.T) {
|
||||
addresses := make(map[string]bool)
|
||||
for i := 0; i < 1000; i++ {
|
||||
addr := RandomAddr(rng, false)
|
||||
addresses[addr.String()] = true
|
||||
}
|
||||
if len(addresses) < 990 {
|
||||
t.Errorf("Expected close to 1000 unique addresses, got %d", len(addresses))
|
||||
}
|
||||
})
|
||||
|
||||
// Test different seeds produce different results
|
||||
t.Run("Different seeds", func(t *testing.T) {
|
||||
rng1 := rand.New(rand.NewSource(42))
|
||||
rng2 := rand.New(rand.NewSource(43))
|
||||
addr1 := RandomAddr(rng1, false)
|
||||
addr2 := RandomAddr(rng2, false)
|
||||
if addr1 == addr2 {
|
||||
t.Errorf("Expected different addresses for different seeds, got %v and %v", addr1, addr2)
|
||||
}
|
||||
})
|
||||
}
|
||||
Loading…
Reference in a new issue