mirror of
https://github.com/zitadel/zitadel.git
synced 2025-08-11 15:17:33 +00:00
context: Add WithMockFeatures to context mock
This commit is contained in:
@@ -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})
|
||||
|
||||
|
Reference in New Issue
Block a user