diff --git a/pkg/services/login/authinfotest/fake.go b/pkg/services/login/authinfotest/fake.go index 730af8b8f73..9a4d7843a3c 100644 --- a/pkg/services/login/authinfotest/fake.go +++ b/pkg/services/login/authinfotest/fake.go @@ -14,8 +14,9 @@ type FakeService struct { ExpectedError error ExpectedLabels map[int64]string - SetAuthInfoFn func(ctx context.Context, cmd *login.SetAuthInfoCommand) error - UpdateAuthInfoFn func(ctx context.Context, cmd *login.UpdateAuthInfoCommand) error + SetAuthInfoFn func(ctx context.Context, cmd *login.SetAuthInfoCommand) error + UpdateAuthInfoFn func(ctx context.Context, cmd *login.UpdateAuthInfoCommand) error + DeleteUserAuthInfoFn func(ctx context.Context, userID int64) error } func (a *FakeService) GetAuthInfo(ctx context.Context, query *login.GetAuthInfoQuery) (*login.UserAuth, error) { @@ -44,5 +45,9 @@ func (a *FakeService) UpdateAuthInfo(ctx context.Context, cmd *login.UpdateAuthI } func (a *FakeService) DeleteUserAuthInfo(ctx context.Context, userID int64) error { + if a.DeleteUserAuthInfoFn != nil { + return a.DeleteUserAuthInfoFn(ctx, userID) + } + return a.ExpectedError }