mirror of
https://github.com/linka-cloud/grpc.git
synced 2024-06-26 15:59:52 +00:00
59 lines
965 B
Go
59 lines
965 B
Go
package addr
|
|
|
|
import (
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
func TestIsLocal(t *testing.T) {
|
|
testData := []struct {
|
|
addr string
|
|
expect bool
|
|
}{
|
|
{"localhost", true},
|
|
{"localhost:8080", true},
|
|
{"127.0.0.1", true},
|
|
{"127.0.0.1:1001", true},
|
|
{"80.1.1.1", false},
|
|
}
|
|
|
|
for _, d := range testData {
|
|
res := IsLocal(d.addr)
|
|
if res != d.expect {
|
|
t.Fatalf("expected %t got %t", d.expect, res)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestExtractor(t *testing.T) {
|
|
testData := []struct {
|
|
addr string
|
|
expect string
|
|
parse bool
|
|
}{
|
|
{"127.0.0.1", "127.0.0.1", false},
|
|
{"10.0.0.1", "10.0.0.1", false},
|
|
{"", "", true},
|
|
{"0.0.0.0", "", true},
|
|
{"[::]", "", true},
|
|
}
|
|
|
|
for _, d := range testData {
|
|
addr, err := Extract(d.addr)
|
|
if err != nil {
|
|
t.Errorf("Unexpected error %v", err)
|
|
}
|
|
|
|
if d.parse {
|
|
ip := net.ParseIP(addr)
|
|
if ip == nil {
|
|
t.Error("Unexpected nil IP")
|
|
}
|
|
|
|
} else if addr != d.expect {
|
|
t.Errorf("Expected %s got %s", d.expect, addr)
|
|
}
|
|
}
|
|
|
|
}
|