35 lines
1.1 KiB
Go
35 lines
1.1 KiB
Go
package grpc
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/letsencrypt/boulder/test"
|
|
"google.golang.org/grpc/resolver"
|
|
)
|
|
|
|
func Test_parseResolverIPAddress(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
addr string
|
|
expectTarget *resolver.Address
|
|
wantErr bool
|
|
}{
|
|
{"valid, IPv4 address", "127.0.0.1:1337", &resolver.Address{Addr: "127.0.0.1:1337", ServerName: "127.0.0.1:1337"}, false},
|
|
{"valid, IPv6 address", "[::1]:1337", &resolver.Address{Addr: "[::1]:1337", ServerName: "[::1]:1337"}, false},
|
|
{"valid, port only", ":1337", &resolver.Address{Addr: "127.0.0.1:1337", ServerName: "127.0.0.1:1337"}, false},
|
|
{"invalid, hostname address", "localhost:1337", nil, true},
|
|
{"invalid, IPv6 address, no brackets", "::1:1337", nil, true},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := parseResolverIPAddress(tt.addr)
|
|
if tt.wantErr {
|
|
test.AssertError(t, err, "expected error, got nil")
|
|
} else {
|
|
test.AssertNotError(t, err, "unexpected error")
|
|
}
|
|
test.AssertDeepEquals(t, got, tt.expectTarget)
|
|
})
|
|
}
|
|
}
|