You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.2 KiB
Go
59 lines
1.2 KiB
Go
package internal
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"go.etcd.io/etcd/client/v3/mock/mockserver"
|
|
"google.golang.org/grpc/resolver"
|
|
"google.golang.org/grpc/serviceconfig"
|
|
)
|
|
|
|
func TestDiscovBuilder_Scheme(t *testing.T) {
|
|
var b discovBuilder
|
|
assert.Equal(t, DiscovScheme, b.Scheme())
|
|
}
|
|
|
|
func TestDiscovBuilder_Build(t *testing.T) {
|
|
servers, err := mockserver.StartMockServers(2)
|
|
assert.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
servers.Stop()
|
|
})
|
|
|
|
var addrs []string
|
|
for _, server := range servers.Servers {
|
|
addrs = append(addrs, server.Address)
|
|
}
|
|
u, err := url.Parse(fmt.Sprintf("%s://%s", DiscovScheme, strings.Join(addrs, ",")))
|
|
assert.NoError(t, err)
|
|
|
|
var b discovBuilder
|
|
_, err = b.Build(resolver.Target{
|
|
URL: *u,
|
|
}, mockClientConn{}, resolver.BuildOptions{})
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
type mockClientConn struct{}
|
|
|
|
func (m mockClientConn) UpdateState(_ resolver.State) error {
|
|
return nil
|
|
}
|
|
|
|
func (m mockClientConn) ReportError(_ error) {
|
|
}
|
|
|
|
func (m mockClientConn) NewAddress(_ []resolver.Address) {
|
|
}
|
|
|
|
func (m mockClientConn) NewServiceConfig(_ string) {
|
|
}
|
|
|
|
func (m mockClientConn) ParseServiceConfig(_ string) *serviceconfig.ParseResult {
|
|
return nil
|
|
}
|