added TestValidateBlockedNumber

handlers/application/menuhandler_test.go

@@ -2670,3 +2670,231 @@ 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)
+}
+
+func TestMaxAmount(t *testing.T) {
+	sessionId := "session123"
+	activeBal := "500"
+
+	tests := []struct {
+		name           string
+		sessionId      string
+		activeBal      string
+		expectedError  bool
+		expectedResult resource.Result
+	}{
+		{
+			name:           "Valid session ID and active balance",
+			sessionId:      sessionId,
+			activeBal:      activeBal,
+			expectedError:  false,
+			expectedResult: resource.Result{Content: activeBal},
+		},
+		{
+			name:           "Missing Session ID",
+			sessionId:      "",
+			activeBal:      activeBal,
+			expectedError:  true,
+			expectedResult: resource.Result{},
+		},
+		{
+			name:           "Failed to Read Active Balance",
+			sessionId:      sessionId,
+			activeBal:      "", // failure to read active balance
+			expectedError:  true,
+			expectedResult: resource.Result{},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			ctx, userStore := InitializeTestStore(t)
+			if tt.sessionId != "" {
+				ctx = context.WithValue(ctx, "SessionId", tt.sessionId)
+			}
+
+			h := &MenuHandlers{
+				userdataStore: userStore,
+			}
+
+			// Write active balance to the store only if it's not empty
+			if tt.activeBal != "" {
+				err := userStore.WriteEntry(ctx, tt.sessionId, storedb.DATA_ACTIVE_BAL, []byte(tt.activeBal))
+				if err != nil {
+					t.Fatal(err)
+				}
+			}
+
+			res, err := h.MaxAmount(ctx, "max_amount", []byte(""))
+
+			if tt.expectedError {
+				assert.Error(t, err)
+			} else {
+				assert.NoError(t, err)
+			}
+
+			assert.Equal(t, tt.expectedResult, res)
+		})
+	}
+}
+
+func TestQuitWithHelp(t *testing.T) {
+	fm, err := NewFlagManager(flagsPath)
+	if err != nil {
+		t.Logf(err.Error())
+	}
+	flag_account_authorized, _ := fm.GetFlag("flag_account_authorized")
+
+	sessionId := "session123"
+
+	ctx := context.WithValue(context.Background(), "SessionId", sessionId)
+
+	h := &MenuHandlers{
+		flagManager: fm,
+	}
+	tests := []struct {
+		name           string
+		input          []byte
+		status         string
+		expectedResult resource.Result
+	}{
+		{
+			name: "Test quit with help message",
+			expectedResult: resource.Result{
+				FlagReset: []uint32{flag_account_authorized},
+				Content:   "For more help, please call: 0757628885",
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			res, _ := h.QuitWithHelp(ctx, "quit_with_help", tt.input)
+			//Assert that the aflag_account_authorized has been reset to the result
+			assert.Equal(t, res, tt.expectedResult, "Expected result should be equal to the actual result")
+		})
+	}
+}
+
+func TestShowBlockedAccount(t *testing.T) {
+	sessionId := "session123"
+	ctx := context.WithValue(context.Background(), "SessionId", sessionId)
+
+	h := &MenuHandlers{}
+
+	tests := []struct {
+		name           string
+		input          []byte
+		status         string
+		expectedResult resource.Result
+	}{
+		{
+			name: "Test quit with Show Blocked Account",
+			expectedResult: resource.Result{
+				Content: "Your account has been locked. For help on how to unblock your account, contact support at: 0757628885",
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			res, _ := h.ShowBlockedAccount(ctx, "show_blocked_account", tt.input)
+			//Assert that the result is as expected
+			assert.Equal(t, res, tt.expectedResult, "Expected result should be equal to the actual result")
+		})
+	}
+}
+
+func TestValidateBlockedNumber(t *testing.T) {
+	sessionId := "session123"
+	validNumber := "+254712345678"
+	invalidNumber := "12343"              // Invalid phone number
+	unregisteredNumber := "+254734567890" // Valid but unregistered number
+	publicKey := "0X13242618721"
+
+	ctx, userStore := InitializeTestStore(t)
+	ctx = context.WithValue(ctx, "SessionId", sessionId)
+
+	fm, err := NewFlagManager(flagsPath)
+	if err != nil {
+		t.Fatal(err)
+	}
+	flag_unregistered_number, _ := fm.GetFlag("flag_unregistered_number")
+
+	h := &MenuHandlers{
+		userdataStore: userStore,
+		flagManager:   fm,
+	}
+
+	err = userStore.WriteEntry(ctx, validNumber, storedb.DATA_PUBLIC_KEY, []byte(publicKey))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	tests := []struct {
+		name           string
+		input          []byte
+		expectedResult resource.Result
+	}{
+		{
+			name:           "Valid and registered number",
+			input:          []byte(validNumber),
+			expectedResult: resource.Result{},
+		},
+		{
+			name:  "Invalid Phone Number",
+			input: []byte(invalidNumber),
+			expectedResult: resource.Result{
+				FlagSet: []uint32{flag_unregistered_number},
+			},
+		},
+		{
+			name:  "Unregistered Phone Number",
+			input: []byte(unregisteredNumber),
+			expectedResult: resource.Result{
+				FlagSet: []uint32{flag_unregistered_number},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			res, err := h.ValidateBlockedNumber(ctx, "validate_blocked_number", tt.input)
+
+			assert.NoError(t, err)
+
+			assert.Equal(t, tt.expectedResult, res)
+
+			if tt.name == "Valid and registered number" {
+				blockedNumber, err := userStore.ReadEntry(ctx, sessionId, storedb.DATA_BLOCKED_NUMBER)
+				if err != nil {
+					t.Fatal(err)
+				}
+
+				assert.Equal(t, validNumber, string(blockedNumber))
+			}
+		})
+	}
+}