mirror of
https://github.com/grafana/grafana.git
synced 2024-11-26 02:40:26 -06:00
16b1e285ea
* GetSignedInUser unexported (renamed to getSignedInUser) * GetSignedInUserWithCacheCtx renamed to GetSignedInUser * added a check for a nil cacheservice (as defensive programming / test convenience)
115 lines
3.6 KiB
Go
115 lines
3.6 KiB
Go
package usertest
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/services/user"
|
|
)
|
|
|
|
type FakeUserService struct {
|
|
ExpectedUser *user.User
|
|
ExpectedSignedInUser *user.SignedInUser
|
|
ExpectedError error
|
|
ExpectedSetUsingOrgError error
|
|
ExpectedSearchUsers user.SearchUserQueryResult
|
|
ExpectedUserProfileDTO *user.UserProfileDTO
|
|
ExpectedUserProfileDTOs []*user.UserProfileDTO
|
|
ExpectedUsageStats map[string]any
|
|
|
|
UpdateFn func(ctx context.Context, cmd *user.UpdateUserCommand) error
|
|
GetSignedInUserFn func(ctx context.Context, query *user.GetSignedInUserQuery) (*user.SignedInUser, error)
|
|
CreateFn func(ctx context.Context, cmd *user.CreateUserCommand) (*user.User, error)
|
|
BatchDisableUsersFn func(ctx context.Context, cmd *user.BatchDisableUsersCommand) error
|
|
GetByEmailFn func(ctx context.Context, query *user.GetUserByEmailQuery) (*user.User, error)
|
|
|
|
counter int
|
|
}
|
|
|
|
func NewUserServiceFake() *FakeUserService {
|
|
return &FakeUserService{}
|
|
}
|
|
|
|
func (f FakeUserService) GetUsageStats(ctx context.Context) map[string]any {
|
|
return f.ExpectedUsageStats
|
|
}
|
|
|
|
func (f *FakeUserService) Create(ctx context.Context, cmd *user.CreateUserCommand) (*user.User, error) {
|
|
if f.CreateFn != nil {
|
|
return f.CreateFn(ctx, cmd)
|
|
}
|
|
|
|
return f.ExpectedUser, f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) CreateServiceAccount(ctx context.Context, cmd *user.CreateUserCommand) (*user.User, error) {
|
|
return f.ExpectedUser, f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) Delete(ctx context.Context, cmd *user.DeleteUserCommand) error {
|
|
return f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) GetByID(ctx context.Context, query *user.GetUserByIDQuery) (*user.User, error) {
|
|
return f.ExpectedUser, f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) GetByLogin(ctx context.Context, query *user.GetUserByLoginQuery) (*user.User, error) {
|
|
return f.ExpectedUser, f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) GetByEmail(ctx context.Context, query *user.GetUserByEmailQuery) (*user.User, error) {
|
|
if f.GetByEmailFn != nil {
|
|
return f.GetByEmailFn(ctx, query)
|
|
}
|
|
return f.ExpectedUser, f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) Update(ctx context.Context, cmd *user.UpdateUserCommand) error {
|
|
if f.UpdateFn != nil {
|
|
return f.UpdateFn(ctx, cmd)
|
|
}
|
|
return f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) UpdateLastSeenAt(ctx context.Context, cmd *user.UpdateUserLastSeenAtCommand) error {
|
|
return f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) GetSignedInUser(ctx context.Context, query *user.GetSignedInUserQuery) (*user.SignedInUser, error) {
|
|
if f.GetSignedInUserFn != nil {
|
|
return f.GetSignedInUserFn(ctx, query)
|
|
}
|
|
if f.ExpectedSignedInUser == nil {
|
|
return &user.SignedInUser{}, f.ExpectedError
|
|
}
|
|
return f.ExpectedSignedInUser, f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) NewAnonymousSignedInUser(ctx context.Context) (*user.SignedInUser, error) {
|
|
return f.ExpectedSignedInUser, f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) Search(ctx context.Context, query *user.SearchUsersQuery) (*user.SearchUserQueryResult, error) {
|
|
return &f.ExpectedSearchUsers, f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) BatchDisableUsers(ctx context.Context, cmd *user.BatchDisableUsersCommand) error {
|
|
if f.BatchDisableUsersFn != nil {
|
|
return f.BatchDisableUsersFn(ctx, cmd)
|
|
}
|
|
return f.ExpectedError
|
|
}
|
|
|
|
func (f *FakeUserService) GetProfile(ctx context.Context, query *user.GetUserProfileQuery) (*user.UserProfileDTO, error) {
|
|
if f.ExpectedUserProfileDTO != nil {
|
|
return f.ExpectedUserProfileDTO, f.ExpectedError
|
|
}
|
|
|
|
if f.ExpectedUserProfileDTOs == nil {
|
|
return nil, f.ExpectedError
|
|
}
|
|
|
|
f.counter++
|
|
return f.ExpectedUserProfileDTOs[f.counter-1], f.ExpectedError
|
|
}
|