go-gost/x
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test(handler/router): add associate_test.go (handleAssociate, handlePacket, getRoute, getAddrforRoute, sdRenew)

Browse Source
This commit is contained in:
ginuerzh
2026-06-04 08:46:34 +08:00
parent ca83acc323
commit 7e9d14bb41
2 changed files with 402 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
handler/router/helpers_test.go
+17 -9
View File
@@ -328,15 +328,15 @@ func (r *mockRouter) GetRoute(ctx context.Context, dst string, opts ...router.Op
// ---------------------------------------------------------------------------
type mockSD struct {
getFn func(ctx context.Context, name string, opts ...sd.Option) ([]*sd.Service, error)
getFn func(ctx context.Context, name string) ([]*sd.Service, error)
registerFn func(ctx context.Context, svc *sd.Service, opts ...sd.Option) error
deregisterFn func(ctx context.Context, svc *sd.Service, opts ...sd.Option) error
renewFn func(ctx context.Context, svc *sd.Service, opts ...sd.Option) error
deregisterFn func(ctx context.Context, svc *sd.Service) error
renewFn func(ctx context.Context, svc *sd.Service) error
}
func (s *mockSD) Get(ctx context.Context, name string, opts ...sd.Option) ([]*sd.Service, error) {
func (s *mockSD) Get(ctx context.Context, name string) ([]*sd.Service, error) {
if s.getFn != nil {
return s.getFn(ctx, name, opts...)
return s.getFn(ctx, name)
}
return nil, nil
}
@@ -346,15 +346,15 @@ func (s *mockSD) Register(ctx context.Context, svc *sd.Service, opts ...sd.Optio
}
return nil
}
func (s *mockSD) Deregister(ctx context.Context, svc *sd.Service, opts ...sd.Option) error {
func (s *mockSD) Deregister(ctx context.Context, svc *sd.Service) error {
if s.deregisterFn != nil {
return s.deregisterFn(ctx, svc, opts...)
return s.deregisterFn(ctx, svc)
}
return nil
}
func (s *mockSD) Renew(ctx context.Context, svc *sd.Service, opts ...sd.Option) error {
func (s *mockSD) Renew(ctx context.Context, svc *sd.Service) error {
if s.renewFn != nil {
return s.renewFn(ctx, svc, opts...)
return s.renewFn(ctx, svc)
}
return nil
}
@@ -364,9 +364,17 @@ func (s *mockSD) Renew(ctx context.Context, svc *sd.Service, opts ...sd.Option)
// ---------------------------------------------------------------------------
type mockIngress struct {
setRuleFn func(ctx context.Context, rule *ingress.Rule, opts ...ingress.Option) bool
getRuleFn func(ctx context.Context, host string, opts ...ingress.Option) *ingress.Rule
}
func (ing *mockIngress) SetRule(ctx context.Context, rule *ingress.Rule, opts ...ingress.Option) bool {
if ing.setRuleFn != nil {
return ing.setRuleFn(ctx, rule, opts...)
}
return false
}
func (ing *mockIngress) GetRule(ctx context.Context, host string, opts ...ingress.Option) *ingress.Rule {
if ing.getRuleFn != nil {
return ing.getRuleFn(ctx, host, opts...)