frostfs-api-go/rpc/client/options_test.go
Pavel Karpy 52c1c4c5ab [#290] client: Add WithURIAddress option
Add `WithURIAddress` option to client.
It parses passed address with
`url.ParseRequestURI` function and
use(or not) TLS over grpc connection
based on retrieved scheme('grpc' or
'grpcs').

Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
2021-06-02 15:55:20 +03:00

197 lines
3.8 KiB
Go

package client
import (
"crypto/tls"
"testing"
"github.com/stretchr/testify/require"
)
func TestWithNetworkURIAddress(t *testing.T) {
hostPort := "neofs.example.com:8080"
apiPort := "127.0.0.1:8080"
serverName := "testServer"
testCases := []struct {
uri string
tlsConfig *tls.Config
wantHost string
wantTLS bool
}{
{
uri: grpcScheme + "://" + hostPort,
tlsConfig: nil,
wantHost: "neofs.example.com:8080",
wantTLS: false,
},
{
uri: grpcScheme + "://" + hostPort,
tlsConfig: &tls.Config{},
wantHost: "neofs.example.com:8080",
wantTLS: false,
},
{
uri: grpcTLSScheme + "://" + hostPort,
tlsConfig: nil,
wantHost: "neofs.example.com:8080",
wantTLS: true,
},
{
uri: grpcTLSScheme + "://" + hostPort,
tlsConfig: &tls.Config{ServerName: serverName},
wantHost: "neofs.example.com:8080",
wantTLS: true,
},
{
uri: "wrongScheme://" + hostPort,
tlsConfig: nil,
wantHost: "",
wantTLS: false,
},
{
uri: "impossibleToParseIt",
tlsConfig: nil,
wantHost: "impossibleToParseIt",
wantTLS: false,
},
{
uri: hostPort,
tlsConfig: nil,
wantHost: hostPort,
wantTLS: false,
},
{
uri: apiPort,
tlsConfig: nil,
wantHost: apiPort,
wantTLS: false,
},
}
for _, test := range testCases {
cfg := &cfg{}
opts := WithNetworkURIAddress(test.uri, test.tlsConfig)
for _, opt := range opts {
opt(cfg)
}
require.Equal(t, test.wantHost, cfg.addr, test.uri)
require.Equal(t, test.wantTLS, cfg.tlsCfg != nil, test.uri)
// check if custom tlsConfig was applied
if test.tlsConfig != nil && test.wantTLS {
require.Equal(t, test.tlsConfig.ServerName, cfg.tlsCfg.ServerName, test.uri)
}
}
}
func Test_WithNetworkAddress_WithTLS_WithNetworkURIAddress(t *testing.T) {
addr1, addr2 := "example1.com:8080", "example2.com:8080"
testCases := []struct {
addr string
withTLS bool
uri string
wantHost string
wantTLS bool
}{
{
addr: addr1,
withTLS: true,
uri: grpcScheme + "://" + addr2,
wantHost: addr2,
wantTLS: false,
},
{
addr: addr1,
withTLS: false,
uri: grpcTLSScheme + "://" + addr2,
wantHost: addr2,
wantTLS: true,
},
}
for _, test := range testCases {
// order:
// 1. WithNetworkAddress
// 2. WithTLSCfg(if test.withTLS == true)
// 3. WithNetworkURIAddress
config := &cfg{}
opts := []Option{WithNetworkAddress(test.addr)}
if test.withTLS {
opts = append(opts, WithTLSCfg(&tls.Config{}))
}
opts = append(opts, WithNetworkURIAddress(test.uri, nil)...)
for _, opt := range opts {
opt(config)
}
require.Equal(t, test.wantHost, config.addr, test.addr)
require.Equal(t, test.wantTLS, config.tlsCfg != nil, test.addr)
}
}
func Test_WithNetworkURIAddress_WithTLS_WithNetworkAddress(t *testing.T) {
addr1, addr2 := "example1.com:8080", "example2.com:8080"
testCases := []struct {
addr string
withTLS bool
uri string
wantHost string
wantTLS bool
}{
{
uri: grpcScheme + "://" + addr1,
addr: addr2,
withTLS: true,
wantHost: addr2,
wantTLS: true,
},
{
uri: grpcTLSScheme + "://" + addr1,
addr: addr2,
withTLS: false,
wantHost: addr2,
wantTLS: true,
},
}
for _, test := range testCases {
// order:
// 1. WithNetworkURIAddress
// 2. WithNetworkAddress
// 3. WithTLSCfg(if test.withTLS == true)
config := &cfg{}
opts := WithNetworkURIAddress(test.uri, nil)
opts = append(opts, WithNetworkAddress(test.addr))
if test.withTLS {
opts = append(opts, WithTLSCfg(&tls.Config{}))
}
for _, opt := range opts {
opt(config)
}
require.Equal(t, test.wantHost, config.addr, test.uri)
require.Equal(t, test.wantTLS, config.tlsCfg != nil, test.uri)
}
}