mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
d42201dbf4
* Move test to the db so we test the queries and not just testing the mock * Remove unused function and dependencies * Remove unused functions from the database * Add some integration tests
49 lines
1.4 KiB
Go
49 lines
1.4 KiB
Go
package logintest
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/services/login"
|
|
)
|
|
|
|
type AuthInfoServiceFake struct {
|
|
login.AuthInfoService
|
|
LatestUserID int64
|
|
ExpectedUserAuth *login.UserAuth
|
|
ExpectedExternalUser *login.ExternalUserInfo
|
|
ExpectedError error
|
|
ExpectedLabels map[int64]string
|
|
|
|
SetAuthInfoFn func(ctx context.Context, cmd *login.SetAuthInfoCommand) error
|
|
UpdateAuthInfoFn func(ctx context.Context, cmd *login.UpdateAuthInfoCommand) error
|
|
}
|
|
|
|
func (a *AuthInfoServiceFake) GetAuthInfo(ctx context.Context, query *login.GetAuthInfoQuery) (*login.UserAuth, error) {
|
|
a.LatestUserID = query.UserId
|
|
return a.ExpectedUserAuth, a.ExpectedError
|
|
}
|
|
|
|
func (a *AuthInfoServiceFake) GetUserLabels(ctx context.Context, query login.GetUserLabelsQuery) (map[int64]string, error) {
|
|
return a.ExpectedLabels, a.ExpectedError
|
|
}
|
|
|
|
func (a *AuthInfoServiceFake) SetAuthInfo(ctx context.Context, cmd *login.SetAuthInfoCommand) error {
|
|
if a.SetAuthInfoFn != nil {
|
|
return a.SetAuthInfoFn(ctx, cmd)
|
|
}
|
|
|
|
return a.ExpectedError
|
|
}
|
|
|
|
func (a *AuthInfoServiceFake) UpdateAuthInfo(ctx context.Context, cmd *login.UpdateAuthInfoCommand) error {
|
|
if a.UpdateAuthInfoFn != nil {
|
|
return a.UpdateAuthInfoFn(ctx, cmd)
|
|
}
|
|
|
|
return a.ExpectedError
|
|
}
|
|
|
|
func (a *AuthInfoServiceFake) DeleteUserAuthInfo(ctx context.Context, userID int64) error {
|
|
return a.ExpectedError
|
|
}
|