Updated tests to include the mockState

This commit is contained in:
Alfred Kamanda 2025-02-06 15:10:15 +03:00
parent 1cdd5a37ae
commit 14df16098c
Signed by: Alfred-mk
GPG Key ID: 7EA3D01708908703

View File

@ -2823,6 +2823,7 @@ func TestValidateBlockedNumber(t *testing.T) {
invalidNumber := "12343" // Invalid phone number invalidNumber := "12343" // Invalid phone number
unregisteredNumber := "+254734567890" // Valid but unregistered number unregisteredNumber := "+254734567890" // Valid but unregistered number
publicKey := "0X13242618721" publicKey := "0X13242618721"
mockState := state.NewState(128)
ctx, userStore := InitializeTestStore(t) ctx, userStore := InitializeTestStore(t)
ctx = context.WithValue(ctx, "SessionId", sessionId) ctx = context.WithValue(ctx, "SessionId", sessionId)
@ -2835,6 +2836,7 @@ func TestValidateBlockedNumber(t *testing.T) {
h := &MenuHandlers{ h := &MenuHandlers{
userdataStore: userStore, userdataStore: userStore,
st: mockState,
flagManager: fm, flagManager: fm,
} }
@ -2992,6 +2994,7 @@ func TestCheckBlockedNumPinMisMatch(t *testing.T) {
sessionId := "session123" sessionId := "session123"
blockedNumber := "+254712345678" blockedNumber := "+254712345678"
testPin := "1234" testPin := "1234"
mockState := state.NewState(128)
ctx, userStore := InitializeTestStore(t) ctx, userStore := InitializeTestStore(t)
ctx = context.WithValue(ctx, "SessionId", sessionId) ctx = context.WithValue(ctx, "SessionId", sessionId)
@ -3010,6 +3013,7 @@ func TestCheckBlockedNumPinMisMatch(t *testing.T) {
h := &MenuHandlers{ h := &MenuHandlers{
userdataStore: userStore, userdataStore: userStore,
st: mockState,
flagManager: fm, flagManager: fm,
} }