TheArchive/zitadel
1
0
Fork 0
mirror of https://github.com/zitadel/zitadel.git synced 2025-08-11 15:37:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

context: Add WithMockFeatures to context mock

Browse Source
This commit is contained in:
Marco Ardizzone
2025-08-06 18:29:20 +02:00
parent 3bb7c16782
commit 984fe58601
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/api/authz/context_mock.go
View File

@@ -4,6 +4,8 @@ import (
"context" "context"
"golang.org/x/text/language" "golang.org/x/text/language"
"github.com/zitadel/zitadel/internal/feature"
) )
type MockContextInstanceOpts func(i *instance) 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 { func NewMockContext(instanceID, orgID, userID string, opts ...MockContextInstanceOpts) context.Context {
ctx := context.WithValue(context.Background(), dataKey, CtxData{UserID: userID, OrgID: orgID}) ctx := context.WithValue(context.Background(), dataKey, CtxData{UserID: userID, OrgID: orgID})