From 984fe586011dfd78e7fe795abb51afff7b0e8614 Mon Sep 17 00:00:00 2001 From: Marco Ardizzone Date: Wed, 6 Aug 2025 18:29:20 +0200 Subject: [PATCH] context: Add WithMockFeatures to context mock --- internal/api/authz/context_mock.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/internal/api/authz/context_mock.go b/internal/api/authz/context_mock.go index d26b371bc6..4ff6f47bc9 100644 --- a/internal/api/authz/context_mock.go +++ b/internal/api/authz/context_mock.go @@ -4,6 +4,8 @@ import ( "context" "golang.org/x/text/language" + + "github.com/zitadel/zitadel/internal/feature" ) type MockContextInstanceOpts func(i *instance) @@ -14,6 +16,12 @@ func WithMockDefaultLanguage(lang language.Tag) MockContextInstanceOpts { } } +func WithMockFeatures(features feature.Features) MockContextInstanceOpts { + return func(i *instance) { + i.features = features + } +} + func NewMockContext(instanceID, orgID, userID string, opts ...MockContextInstanceOpts) context.Context { ctx := context.WithValue(context.Background(), dataKey, CtxData{UserID: userID, OrgID: orgID})