urdt/ussd
5
0
Fork 0
Code Issues 50 Pull Requests 7 Actions Packages Projects Releases Wiki Activity

update service mock method signatures

Browse Source
This commit is contained in:
Carlosokumu 2024-10-24 10:10:03 +03:00
parent 308ca99fb0
commit d74a4cc33b
Signed by: carlos
GPG Key ID: 7BD6BC8160A5C953
1 changed files with 4 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
internal/mocks/servicemock.go
View File

@ -11,19 +11,9 @@ type MockAccountService struct {
mock.Mock mock.Mock
} }
func (m *MockAccountService) CreateAccount() (*api.OKResponse, *api.ErrResponse) { func (m *MockAccountService) CreateAccount() (*api.OKResponse, error) {
args := m.Called() args := m.Called()
okResponse, ok := args.Get(0).(*api.OKResponse) return args.Get(0).(*api.OKResponse), args.Error(1)
errResponse, err := args.Get(1).(*api.ErrResponse)
if ok {
return okResponse, nil
}
if err {
return nil, errResponse
}
return nil, nil
} }
func (m *MockAccountService) CheckBalance(publicKey string) (*models.BalanceResponse, error) { func (m *MockAccountService) CheckBalance(publicKey string) (*models.BalanceResponse, error) {
@ -36,15 +26,7 @@ func (m *MockAccountService) CheckAccountStatus(trackingId string) (*models.Trac
return args.Get(0).(*models.TrackStatusResponse), args.Error(1) return args.Get(0).(*models.TrackStatusResponse), args.Error(1)
} }
func (m *MockAccountService) TrackAccountStatus(publicKey string) (*api.OKResponse, *api.ErrResponse) { func (m *MockAccountService) TrackAccountStatus(publicKey string) (*api.OKResponse, error) {
args := m.Called(publicKey) args := m.Called(publicKey)
okResponse, ok := args.Get(0).(*api.OKResponse) return args.Get(0).(*api.OKResponse), args.Error(1)
errResponse, err := args.Get(1).(*api.ErrResponse)
if ok {
return okResponse, nil
}
if err {
return nil, errResponse
}
return nil, nil
} }