mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
ba41954854
* Add email and email_verified to id token if identity is a user * Add endpoint to trigger email verification for user * Add function to clear stored id tokens and use it when email verification is completed
41 lines
1.0 KiB
Go
41 lines
1.0 KiB
Go
package idtest
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/grafana/pkg/services/auth"
|
|
"github.com/grafana/grafana/pkg/services/auth/identity"
|
|
)
|
|
|
|
var _ auth.IDService = (*MockService)(nil)
|
|
|
|
type MockService struct {
|
|
SignIdentityFn func(ctx context.Context, identity identity.Requester) (string, error)
|
|
RemoveIDTokenFn func(ctx context.Context, identity identity.Requester) error
|
|
}
|
|
|
|
func (m *MockService) SignIdentity(ctx context.Context, identity identity.Requester) (string, error) {
|
|
if m.SignIdentityFn != nil {
|
|
return m.SignIdentityFn(ctx, identity)
|
|
}
|
|
return "", nil
|
|
}
|
|
|
|
func (m *MockService) RemoveIDToken(ctx context.Context, identity identity.Requester) error {
|
|
if m.RemoveIDTokenFn != nil {
|
|
return m.RemoveIDTokenFn(ctx, identity)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type MockSigner struct {
|
|
SignIDTokenFn func(ctx context.Context, claims *auth.IDClaims) (string, error)
|
|
}
|
|
|
|
func (s *MockSigner) SignIDToken(ctx context.Context, claims *auth.IDClaims) (string, error) {
|
|
if s.SignIDTokenFn != nil {
|
|
return s.SignIDTokenFn(ctx, claims)
|
|
}
|
|
return "", nil
|
|
}
|