From 28185fc2c51bb757c40e35185562367274ef4b5b Mon Sep 17 00:00:00 2001 From: alfred-mk Date: Mon, 27 Jan 2025 01:05:32 +0300 Subject: [PATCH] added TestRetrieveBlockedNumber --- handlers/application/menuhandler_test.go | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) 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) +}