Skip to content

Commit

Permalink
Don't show sms option if it's not defined in US_ENABLED_METHODS
Browse files Browse the repository at this point in the history
field `us_phone_number` was required and there was still option to use
SMS as an auth.
  • Loading branch information
fredipevcin committed Jul 5, 2024
1 parent 25ad68d commit fb3b0da
Show file tree
Hide file tree
Showing 2 changed files with 87 additions and 7 deletions.
22 changes: 15 additions & 7 deletions flask_security/unified_signin.py
Original file line number Diff line number Diff line change
Expand Up @@ -765,7 +765,9 @@ def us_setup() -> ResponseValue:
setup_methods = _compute_setup_methods()
active_methods = _compute_active_methods(current_user)
form.chosen_method.choices = [
c for c in form.setup_choices if c[0] not in active_methods
c
for c in form.setup_choices
if c[0] not in active_methods and c[0] in setup_methods
]
form.delete_method.choices = [
c for c in form.delete_choices if c[0] in active_methods
Expand Down Expand Up @@ -803,7 +805,8 @@ def us_setup() -> ResponseValue:
"US_CURRENT_METHODS", method_list=current_methods
)[0]
form.chosen_method.choices = [
c for c in form.setup_choices if c[0] not in active_methods
c for c in form.setup_choices
if c[0] not in active_methods and c[0] in setup_methods
]
form.delete_method.choices = [
c for c in form.delete_choices if c[0] in active_methods
Expand Down Expand Up @@ -900,6 +903,10 @@ def us_setup() -> ResponseValue:
**_security._run_ctx_processor("us_setup"),
)

phone_number = None
if "sms" in cv("US_ENABLED_METHODS"):
phone_number = current_user.us_phone_number

# Get here on initial new setup (GET)
# Or failure of POST
if _security._want_json(request):
Expand All @@ -908,12 +915,12 @@ def us_setup() -> ResponseValue:
"available_methods": cv("US_ENABLED_METHODS"),
"active_methods": active_methods,
"setup_methods": setup_methods,
"phone": current_user.us_phone_number,
"phone": phone_number,
}
return base_render_json(form, include_user=False, additional=payload)

# Show user existing phone number
form.phone.data = current_user.us_phone_number
form.phone.data = phone_number
form.chosen_method.data = None
form.delete_method.data = None
return _security.render_template(
Expand Down Expand Up @@ -970,12 +977,13 @@ def us_setup_validate(token: str) -> ResponseValue:
delete=False,
)
if _security._want_json(request):
phone_number = None
if "sms" in cv("US_ENABLED_METHODS"):
phone_number = current_user.us_phone_number
return base_render_json(
form,
include_user=False,
additional=dict(
chosen_method=method, phone=current_user.us_phone_number
),
additional=dict(chosen_method=method, phone=phone_number),
)
else:
do_flash(*get_message("US_SETUP_SUCCESSFUL"))
Expand Down
72 changes: 72 additions & 0 deletions tests/test_unified_signin.py
Original file line number Diff line number Diff line change
Expand Up @@ -2307,3 +2307,75 @@ def test_csrf_2fa_us_cookie(app, client):
)
assert response.status_code == 200
assert response.json["label"] == "label"


@pytest.mark.settings(us_enabled_methods=["password", "email"])
def test_us_setup_email(app, client, get_message):
set_email(app)
us_authenticate(client)
chosen_methods_choices = []

def recorder(template, us_setup_form, **kwargs):
chosen_methods_choices.extend(us_setup_form.chosen_method.choices)
return "ok"

app.security.render_template = recorder

client.get("/us-setup")

assert len(chosen_methods_choices) == 0


@pytest.mark.settings(us_enabled_methods=["email", "authenticator"])
def test_us_setup_authenticator(app, client, get_message):
set_email(app)
us_authenticate(client)
chosen_methods_choices = []

def recorder(template, us_setup_form, **kwargs):
chosen_methods_choices.extend(us_setup_form.chosen_method.choices)
return "ok"

app.security.render_template = recorder

client.get("/us-setup")

assert len(chosen_methods_choices) == 1
assert chosen_methods_choices[0][0] == "authenticator"


@pytest.mark.settings(us_enabled_methods=["email", "sms"])
def test_us_setup_sms(app, client, get_message):
set_email(app)
us_authenticate(client)
chosen_methods_choices = []

def recorder(template, us_setup_form, **kwargs):
chosen_methods_choices.extend(us_setup_form.chosen_method.choices)
return "ok"

app.security.render_template = recorder

client.get("/us-setup")

assert len(chosen_methods_choices) == 1
assert chosen_methods_choices[0][0] == "sms"


@pytest.mark.settings(us_enabled_methods=["email", "sms", "authenticator"])
def test_us_setup_authenticator_sms(app, client, get_message):
set_email(app)
us_authenticate(client)
chosen_methods_choices = []

def recorder(template, us_setup_form, **kwargs):
chosen_methods_choices.extend(us_setup_form.chosen_method.choices)
return "ok"

app.security.render_template = recorder

client.get("/us-setup")

assert len(chosen_methods_choices) == 2
assert chosen_methods_choices[0][0] == "authenticator"
assert chosen_methods_choices[1][0] == "sms"

0 comments on commit fb3b0da

Please sign in to comment.