def test_confirm_user_phone_number_cancel(dbsession, user_id):
"""SMS confirmation success."""
# Prepare confirmation
with transaction.manager:
user = dbsession.query(User).get(user_id)
confirmation = UserNewPhoneNumberConfirmation.require_confirmation(user, "+15551231234")
assert confirmation.created_at
assert confirmation.confirmation_type == ManualConfirmationType.sms
assert confirmation.state == ManualConfirmationState.pending
assert UserNewPhoneNumberConfirmation.get_pending_confirmation(user)
# Resolve confirmation and see user gets a phone number
with transaction.manager:
user = dbsession.query(User).get(user_id)
confirmation = dbsession.query(UserNewPhoneNumberConfirmation).first()
confirmation.cancel()
assert confirmation.action_taken_at
assert confirmation.state == ManualConfirmationState.cancelled
assert "phone_number" not in user.user_data
assert not UserNewPhoneNumberConfirmation.get_pending_confirmation(user)
def test_confirm_user_phone_number_success(dbsession, user_id):
"""SMS confirmation success."""
# Prepare confirmation
with transaction.manager:
user = dbsession.query(User).get(user_id)
confirmation = UserNewPhoneNumberConfirmation.require_confirmation(user, "+15551231234")
assert confirmation.created_at
assert confirmation.confirmation_type == ManualConfirmationType.sms
assert confirmation.state == ManualConfirmationState.pending
assert UserNewPhoneNumberConfirmation.get_pending_confirmation(user)
# Resolve confirmation and see user gets a phone number
with transaction.manager:
user = dbsession.query(User).get(user_id)
confirmation = dbsession.query(UserNewPhoneNumberConfirmation).first()
code = confirmation.other_data["sms_code"]
confirmation.resolve_sms(code, None)
assert confirmation.action_taken_at
assert confirmation.state == ManualConfirmationState.resolved
assert user.user_data["phone_number"] == "+15551231234"
assert not UserNewPhoneNumberConfirmation.get_pending_confirmation(user)
def test_ui_confirm_phone_number(require_phone_number,
logged_in_wallet_user_browser: DriverAPI,
dbsession: Session, mock_eth_service,
test_request):
"""User needs a confirmed phone number before entering the wallet."""
# Run functional tests against a Waitress web server running in another thread
b = logged_in_wallet_user_browser
b.find_by_css("#nav-wallet").click()
assert b.is_element_present_by_css("#heading-new-phone-number")
b.fill("phone_number", "+15551231234")
b.find_by_css("button[type='submit']").click()
# We arrived to SMS code verification page
assert b.is_element_present_by_css("#heading-confirm-phone-number")
# Peek into SMS code
with transaction.manager:
user = dbsession.query(User).first()
confirmation = UserNewPhoneNumberConfirmation.get_pending_confirmation(
user)
sms_code = confirmation.other_data["sms_code"]
# Get a dummy SMS backend that's configured in test fixtures
backend = get_sms_backend(test_request)
# Make sure code got out to the user
msg = backend.get_last_message()
assert sms_code in msg
# Enter the code
b.fill("code", sms_code)
b.find_by_css("button[type='submit']").click()
# We arrived to wallet overview
assert b.is_element_present_by_css("#heading-wallet-overview")
# We have a notification for phone number verified
assert b.is_element_present_by_css("#msg-phone-confirmed")
def test_ui_confirm_phone_number(require_phone_number, logged_in_wallet_user_browser: DriverAPI, dbsession: Session, mock_eth_service, test_request):
"""User needs a confirmed phone number before entering the wallet."""
# Run functional tests against a Waitress web server running in another thread
b = logged_in_wallet_user_browser
b.find_by_css("#nav-wallet").click()
assert b.is_element_present_by_css("#heading-new-phone-number")
b.fill("phone_number", "+15551231234")
b.find_by_css("button[type='submit']").click()
# We arrived to SMS code verification page
assert b.is_element_present_by_css("#heading-confirm-phone-number")
# Peek into SMS code
with transaction.manager:
user = dbsession.query(User).first()
confirmation = UserNewPhoneNumberConfirmation.get_pending_confirmation(user)
sms_code = confirmation.other_data["sms_code"]
# Get a dummy SMS backend that's configured in test fixtures
backend = get_sms_backend(test_request)
# Make sure code got out to the user
msg = backend.get_last_message()
assert sms_code in msg
# Enter the code
b.fill("code", sms_code)
b.find_by_css("button[type='submit']").click()
# We arrived to wallet overview
assert b.is_element_present_by_css("#heading-wallet-overview")
# We have a notification for phone number verified
assert b.is_element_present_by_css("#msg-phone-confirmed")
def manual_confirmation(self) -> UserNewPhoneNumberConfirmation:
wallet = self.context
return UserNewPhoneNumberConfirmation.get_pending_confirmation(wallet.user)
def manual_confirmation(self) -> UserNewPhoneNumberConfirmation:
wallet = self.context
return UserNewPhoneNumberConfirmation.get_pending_confirmation(wallet.user)
