Merge remote-tracking branch 'refs/remotes/origin/preferred-node-id' into preferred-node-id

Author: asmyasnikov

internal/query/client.go

@@ -39,7 +39,7 @@ type (
 		closer.Closer
 
 		Stats() pool.Stats
-		With(ctx context.Context, f func(context.Context, *Session) error, opts ...retry.Option) error
+		With(ctx context.Context, f func(ctx context.Context, s *Session) error, opts ...retry.Option) error
 	}
 	Client struct {
 		config *config.Config

internal/query/client_test.go

@@ -1590,9 +1590,7 @@ func testPool(
 ) *pool.Pool[*Session, Session] {
 	return pool.New[*Session, Session](ctx,
 		pool.WithLimit[*Session, Session](1),
-		pool.WithCreateItemFunc(func(ctx context.Context) (*Session, error) {
-			return createSession(ctx)
-		}),
+		pool.WithCreateItemFunc(createSession),
 		pool.WithSyncCloseItem[*Session, Session](),
 	)
 }