You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
2.3 KiB
105 lines
2.3 KiB
package discovery
|
|
|
|
import (
|
|
"context"
|
|
"net/url"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"google.golang.org/grpc/resolver"
|
|
"google.golang.org/grpc/serviceconfig"
|
|
|
|
"github.com/go-kratos/kratos/v2/registry"
|
|
)
|
|
|
|
func TestWithInsecure(t *testing.T) {
|
|
b := &builder{}
|
|
WithInsecure(true)(b)
|
|
if !b.insecure {
|
|
t.Errorf("expected insecure to be true")
|
|
}
|
|
}
|
|
|
|
func TestWithTimeout(t *testing.T) {
|
|
o := &builder{}
|
|
v := time.Duration(123)
|
|
WithTimeout(v)(o)
|
|
if !reflect.DeepEqual(v, o.timeout) {
|
|
t.Errorf("expected %v, got %v", v, o.timeout)
|
|
}
|
|
}
|
|
|
|
func TestDisableDebugLog(t *testing.T) {
|
|
o := &builder{}
|
|
DisableDebugLog()(o)
|
|
if !o.debugLogDisabled {
|
|
t.Errorf("expected debugLogDisabled true, got %v", o.debugLogDisabled)
|
|
}
|
|
}
|
|
|
|
type mockDiscovery struct{}
|
|
|
|
func (m *mockDiscovery) GetService(ctx context.Context, serviceName string) ([]*registry.ServiceInstance, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (m *mockDiscovery) Watch(ctx context.Context, serviceName string) (registry.Watcher, error) {
|
|
time.Sleep(time.Microsecond * 500)
|
|
return &testWatch{}, nil
|
|
}
|
|
|
|
func TestBuilder_Scheme(t *testing.T) {
|
|
b := NewBuilder(&mockDiscovery{})
|
|
if !reflect.DeepEqual("discovery", b.Scheme()) {
|
|
t.Errorf("expected %v, got %v", "discovery", b.Scheme())
|
|
}
|
|
}
|
|
|
|
type mockConn struct{}
|
|
|
|
func (m *mockConn) UpdateState(resolver.State) error {
|
|
return nil
|
|
}
|
|
|
|
func (m *mockConn) ReportError(error) {}
|
|
|
|
func (m *mockConn) NewAddress(addresses []resolver.Address) {}
|
|
|
|
func (m *mockConn) NewServiceConfig(serviceConfig string) {}
|
|
|
|
func (m *mockConn) ParseServiceConfig(serviceConfigJSON string) *serviceconfig.ParseResult {
|
|
return nil
|
|
}
|
|
|
|
func TestBuilder_Build(t *testing.T) {
|
|
b := NewBuilder(&mockDiscovery{}, DisableDebugLog())
|
|
_, err := b.Build(
|
|
resolver.Target{
|
|
URL: url.URL{
|
|
Scheme: resolver.GetDefaultScheme(),
|
|
Path: "grpc://authority/endpoint",
|
|
},
|
|
},
|
|
&mockConn{},
|
|
resolver.BuildOptions{},
|
|
)
|
|
if err != nil {
|
|
t.Errorf("expected no error, got %v", err)
|
|
return
|
|
}
|
|
timeoutBuilder := NewBuilder(&mockDiscovery{}, WithTimeout(0))
|
|
_, err = timeoutBuilder.Build(
|
|
resolver.Target{
|
|
URL: url.URL{
|
|
Scheme: resolver.GetDefaultScheme(),
|
|
Path: "grpc://authority/endpoint",
|
|
},
|
|
},
|
|
&mockConn{},
|
|
resolver.BuildOptions{},
|
|
)
|
|
if err == nil {
|
|
t.Errorf("expected error, got %v", err)
|
|
}
|
|
}
|
|
|