fix test

Signed-off-by: NikitaSkrynnik <nikita.skrynnik@xored.com>
networkservicemesh · Nov 6, 2024 · 372da31 · 372da31
1 parent 36a1060
commit 372da31
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 11 deletions.
diff --git a/pkg/networkservice/chains/client/client.go b/pkg/networkservice/chains/client/client.go
@@ -44,6 +44,7 @@ func NewClient(ctx context.Context, clientOpts ...Option) networkservice.Network
 		authorizeClient: null.NewClient(),
 		healClient:      null.NewClient(),
 		refreshClient:   refresh.NewClient(ctx),
+		reselectFunc:    begin.DefaultReselectFunc,
 	}
 	for _, opt := range clientOpts {
 		opt(opts)

diff --git a/pkg/networkservice/common/begin/options.go b/pkg/networkservice/common/begin/options.go
@@ -18,13 +18,11 @@ package begin
 
 import (
 	"context"
-	"time"
 )
 
 type option struct {
-	cancelCtx      context.Context
-	reselect       bool
-	contextTimeout time.Duration
+	cancelCtx context.Context
+	reselect  bool
 }
 
 type clientOption struct {
@@ -51,13 +49,6 @@ func WithReselect() Option {
 	}
 }
 
-// WithContextTimeout - set a custom timeout for a context in begin.Close
-func WithContextTimeout(timeout time.Duration) Option {
-	return func(o *option) {
-		o.contextTimeout = timeout
-	}
-}
-
 // WithReselectFunc - sets a function for changing request parameters on reselect
 func WithReselectFunc(reselectFunc ReselectFunc) ClientOption {
 	return func(o *clientOption) {