From 9ca1d8bca7ab6297ee0ad88f339edf3e33a63781 Mon Sep 17 00:00:00 2001 From: colin-stuart Date: Thu, 27 Mar 2025 04:45:31 -0400 Subject: [PATCH] Tests: Add delete user mock field to fake auth info service (#102947) --- pkg/services/login/authinfotest/fake.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) 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 }