From a64291bb3131a72488227560153a62c16fe953e2 Mon Sep 17 00:00:00 2001
From: Carlosokumu <carlosokumu254@gmail.com>
Date: Thu, 29 Aug 2024 19:54:26 +0300
Subject: [PATCH] setup test for successful account creation

---
 internal/handlers/ussd/menuhandler_test.go | 48 ++++++++++++++++++++--
 1 file changed, 45 insertions(+), 3 deletions(-)

diff --git a/internal/handlers/ussd/menuhandler_test.go b/internal/handlers/ussd/menuhandler_test.go
index 94ef85d..98b8dc4 100644
--- a/internal/handlers/ussd/menuhandler_test.go
+++ b/internal/handlers/ussd/menuhandler_test.go
@@ -1,11 +1,53 @@
 package ussd
 
-import "testing"
+import (
+	"encoding/json"
+	"testing"
 
+	"git.grassecon.net/urdt/ussd/internal/handlers/ussd/mocks"
+	"git.grassecon.net/urdt/ussd/internal/models"
+	"github.com/alecthomas/assert/v2"
+	"github.com/stretchr/testify/mock"
+)
 
+func TestCreateAccount_Success(t *testing.T) {
+	mockAccountFileHandler := new(mocks.MockAccountFileHandler)
+	mockCreateAccountService := new(mocks.MockAccountService)
 
+	mockAccountFileHandler.On("EnsureFileExists").Return(nil)
 
+	// Mock that no account data exists
+	mockAccountFileHandler.On("ReadAccountData").Return(nil, nil)
+
+	// Define expected account response after api call
+	expectedAccountResp := &models.AccountResponse{
+		Ok: true,
+		Result: struct {
+			CustodialId json.Number `json:"custodialId"`
+			PublicKey   string      `json:"publicKey"`
+			TrackingId  string      `json:"trackingId"`
+		}{
+			CustodialId: "12",
+			PublicKey:   "some-public-key",
+			TrackingId:  "some-tracking-id",
+		},
+	}
+	mockCreateAccountService.On("CreateAccount").Return(expectedAccountResp, nil)
+
+	// Mock WriteAccountData to not error
+	mockAccountFileHandler.On("WriteAccountData", mock.Anything).Return(nil)
+
+	handlers := &Handlers{
+		accountService:     mockCreateAccountService,
+	}
 
-func CreateAccountTest(t *testing.T){
 	
-}
\ No newline at end of file
+	actualResponse, err := handlers.accountService.CreateAccount()
+
+	// Assert results
+	assert.NoError(t, err)
+	assert.Equal(t,expectedAccountResp.Ok,true)
+	assert.Equal(t,expectedAccountResp,actualResponse)
+	
+
+}