context: Add WithMockFeatures to context mock

This commit is contained in:
Marco Ardizzone
2025-08-06 18:29:20 +02:00
parent 3bb7c16782
commit 984fe58601

View File

@@ -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})