diff --git a/handlers/application/menuhandler_test.go b/handlers/application/menuhandler_test.go index f0edf2b..3bd0e92 100644 --- a/handlers/application/menuhandler_test.go +++ b/handlers/application/menuhandler_test.go @@ -2670,3 +2670,26 @@ func TestViewTransactionStatement(t *testing.T) { }) } } + +func TestRetrieveBlockedNumber(t *testing.T) { + sessionId := "session123" + blockedNumber := "0712345678" + + ctx, userStore := InitializeTestStore(t) + ctx = context.WithValue(ctx, "SessionId", sessionId) + + h := &MenuHandlers{ + userdataStore: userStore, + } + + err := userStore.WriteEntry(ctx, sessionId, storedb.DATA_BLOCKED_NUMBER, []byte(blockedNumber)) + if err != nil { + t.Fatal(err) + } + + res, err := h.RetrieveBlockedNumber(ctx, "retrieve_blocked_number", []byte("")) + + assert.NoError(t, err) + + assert.Equal(t, blockedNumber, res.Content) +}