From 1c57c95d93efdf6d26a795f1987a9da8a712bac2 Mon Sep 17 00:00:00 2001
From: Carlosokumu <carlosokumu254@gmail.com>
Date: Fri, 18 Oct 2024 17:04:02 +0300
Subject: [PATCH] use api structs

---
 internal/mocks/servicemock.go | 30 +++++++++++++++++++++++++++---
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/internal/mocks/servicemock.go b/internal/mocks/servicemock.go
index d828045..7796b77 100644
--- a/internal/mocks/servicemock.go
+++ b/internal/mocks/servicemock.go
@@ -1,6 +1,7 @@
 package mocks
 
 import (
+	"git.grassecon.net/urdt/ussd/internal/handlers/server"
 	"git.grassecon.net/urdt/ussd/internal/models"
 	"github.com/stretchr/testify/mock"
 )
@@ -10,9 +11,19 @@ type MockAccountService struct {
 	mock.Mock
 }
 
-func (m *MockAccountService) CreateAccount() (*models.AccountResponse, error) {
+func (m *MockAccountService) CreateAccount() (*server.OKResponse, *server.ErrResponse) {
 	args := m.Called()
-	return args.Get(0).(*models.AccountResponse), args.Error(1)
+	okResponse, ok := args.Get(0).(*server.OKResponse)
+	errResponse, err := args.Get(1).(*server.ErrResponse)
+
+	if ok {
+		return okResponse, nil
+	}
+
+	if err {
+		return nil, errResponse
+	}
+	return nil, nil
 }
 
 func (m *MockAccountService) CheckBalance(publicKey string) (*models.BalanceResponse, error) {
@@ -23,4 +34,17 @@ func (m *MockAccountService) CheckBalance(publicKey string) (*models.BalanceResp
 func (m *MockAccountService) CheckAccountStatus(trackingId string) (*models.TrackStatusResponse, error) {
 	args := m.Called(trackingId)
 	return args.Get(0).(*models.TrackStatusResponse), args.Error(1)
-}
\ No newline at end of file
+}
+
+func (m *MockAccountService) TrackAccountStatus(publicKey string) (*server.OKResponse, *server.ErrResponse) {
+	args := m.Called(publicKey)
+	okResponse, ok := args.Get(0).(*server.OKResponse)
+	errResponse, err := args.Get(1).(*server.ErrResponse)
+	if ok {
+		return okResponse, nil
+	}
+	if err {
+		return nil, errResponse
+	}
+	return nil, nil
+}