mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
8cb351e54a
* AuthN: Add new client extension interface that allows for custom logout logic * AuthN: Add tests for oauth client logout * Call authn.Logout Co-authored-by: Gabriel MABILLE <gamab@users.noreply.github.com>
124 lines
3.6 KiB
Go
124 lines
3.6 KiB
Go
package authntest
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/models/usertoken"
|
|
"github.com/grafana/grafana/pkg/services/auth/identity"
|
|
"github.com/grafana/grafana/pkg/services/authn"
|
|
"github.com/grafana/grafana/pkg/services/login"
|
|
)
|
|
|
|
var _ authn.Service = new(MockService)
|
|
var _ authn.IdentitySynchronizer = new(MockService)
|
|
|
|
type MockService struct {
|
|
SyncIdentityFunc func(ctx context.Context, identity *authn.Identity) error
|
|
RegisterPostAuthHookFunc func(hook authn.PostAuthHookFn, priority uint)
|
|
}
|
|
|
|
func (m *MockService) Authenticate(ctx context.Context, r *authn.Request) (*authn.Identity, error) {
|
|
panic("unimplemented")
|
|
}
|
|
|
|
func (m *MockService) Login(ctx context.Context, client string, r *authn.Request) (*authn.Identity, error) {
|
|
panic("unimplemented")
|
|
}
|
|
|
|
func (m *MockService) RedirectURL(ctx context.Context, client string, r *authn.Request) (*authn.Redirect, error) {
|
|
panic("unimplemented")
|
|
}
|
|
|
|
func (m *MockService) RegisterClient(c authn.Client) {
|
|
panic("unimplemented")
|
|
}
|
|
|
|
func (m *MockService) RegisterPostAuthHook(hook authn.PostAuthHookFn, priority uint) {
|
|
if m.RegisterPostAuthHookFunc != nil {
|
|
m.RegisterPostAuthHookFunc(hook, priority)
|
|
}
|
|
}
|
|
|
|
func (m *MockService) RegisterPostLoginHook(hook authn.PostLoginHookFn, priority uint) {
|
|
panic("unimplemented")
|
|
}
|
|
|
|
func (*MockService) Logout(_ context.Context, _ identity.Requester, _ *usertoken.UserToken) (*authn.Redirect, error) {
|
|
panic("unimplemented")
|
|
}
|
|
|
|
func (m *MockService) SyncIdentity(ctx context.Context, identity *authn.Identity) error {
|
|
if m.SyncIdentityFunc != nil {
|
|
return m.SyncIdentityFunc(ctx, identity)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var _ authn.HookClient = new(MockClient)
|
|
var _ authn.LogoutClient = new(MockClient)
|
|
var _ authn.ContextAwareClient = new(MockClient)
|
|
|
|
type MockClient struct {
|
|
NameFunc func() string
|
|
AuthenticateFunc func(ctx context.Context, r *authn.Request) (*authn.Identity, error)
|
|
TestFunc func(ctx context.Context, r *authn.Request) bool
|
|
PriorityFunc func() uint
|
|
HookFunc func(ctx context.Context, identity *authn.Identity, r *authn.Request) error
|
|
LogoutFunc func(ctx context.Context, user identity.Requester, info *login.UserAuth) (*authn.Redirect, bool)
|
|
}
|
|
|
|
func (m MockClient) Name() string {
|
|
if m.NameFunc != nil {
|
|
return m.NameFunc()
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (m MockClient) Authenticate(ctx context.Context, r *authn.Request) (*authn.Identity, error) {
|
|
if m.AuthenticateFunc != nil {
|
|
return m.AuthenticateFunc(ctx, r)
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (m MockClient) Test(ctx context.Context, r *authn.Request) bool {
|
|
if m.TestFunc != nil {
|
|
return m.TestFunc(ctx, r)
|
|
}
|
|
return false
|
|
}
|
|
|
|
func (m MockClient) Priority() uint {
|
|
if m.PriorityFunc != nil {
|
|
return m.PriorityFunc()
|
|
}
|
|
return 0
|
|
}
|
|
|
|
func (m MockClient) Hook(ctx context.Context, identity *authn.Identity, r *authn.Request) error {
|
|
if m.HookFunc != nil {
|
|
return m.HookFunc(ctx, identity, r)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *MockClient) Logout(ctx context.Context, user identity.Requester, info *login.UserAuth) (*authn.Redirect, bool) {
|
|
if m.LogoutFunc != nil {
|
|
return m.LogoutFunc(ctx, user, info)
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
var _ authn.ProxyClient = new(MockProxyClient)
|
|
|
|
type MockProxyClient struct {
|
|
AuthenticateProxyFunc func(ctx context.Context, r *authn.Request, username string, additional map[string]string) (*authn.Identity, error)
|
|
}
|
|
|
|
func (m MockProxyClient) AuthenticateProxy(ctx context.Context, r *authn.Request, username string, additional map[string]string) (*authn.Identity, error) {
|
|
if m.AuthenticateProxyFunc != nil {
|
|
return m.AuthenticateProxyFunc(ctx, r, username, additional)
|
|
}
|
|
return nil, nil
|
|
}
|