36 lines
1,023 B
Go
36 lines
1,023 B
Go
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
package netutil
|
|
|
|
import "testing"
|
|
|
|
func TestPublicIPMatchesPythonCases(t *testing.T) {
|
|
for _, address := range []string{"8.8.8.8", "1.1.1.1"} {
|
|
if !IsPublicIP(address) {
|
|
t.Fatalf("expected public: %s", address)
|
|
}
|
|
}
|
|
for _, address := range []string{
|
|
"10.0.0.5", "192.168.1.1", "172.16.0.1", "127.0.0.1",
|
|
"169.254.1.1", "100.64.0.1", "::1", "not-an-ip", "",
|
|
} {
|
|
if IsPublicIP(address) {
|
|
t.Fatalf("expected non-public: %s", address)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestCIDRAndEndpointHelpers(t *testing.T) {
|
|
if !IPInCIDRs("203.0.113.55", []string{"203.0.113.0/24"}) ||
|
|
IPInCIDRs("8.8.8.8", []string{"203.0.113.0/24"}) {
|
|
t.Fatal("CIDR matching failed")
|
|
}
|
|
host, port := SplitHostPort("[2001:db8::1]:22")
|
|
if host != "2001:db8::1" || port != "22" {
|
|
t.Fatalf("unexpected IPv6 split: %q %q", host, port)
|
|
}
|
|
host, port = SplitHostPort("10.2.0.2%proton0:61877")
|
|
if host != "10.2.0.2%proton0" || port != "61877" {
|
|
t.Fatalf("unexpected zone split: %q %q", host, port)
|
|
}
|
|
}
|