mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-04-18 08:11:47 -04:00
fix tests
This commit is contained in:
@@ -612,7 +612,6 @@ class SetupUserProfileForm(StripWhitespaceForm):
|
||||
mobile_number = international_phone_number()
|
||||
|
||||
|
||||
|
||||
class RegisterUserFromInviteForm(RegisterUserForm):
|
||||
def __init__(self, invited_user):
|
||||
super().__init__(
|
||||
|
||||
@@ -1,4 +1,5 @@
|
||||
import os
|
||||
|
||||
from flask import abort, redirect, render_template, request, url_for
|
||||
from flask_login import current_user
|
||||
|
||||
|
||||
@@ -1,11 +1,18 @@
|
||||
from datetime import datetime, timedelta
|
||||
import uuid
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
from flask import abort, current_app, redirect, render_template, request, session, url_for
|
||||
from app.main.views import sign_in
|
||||
from app import user_api_client
|
||||
from flask import (
|
||||
abort,
|
||||
current_app,
|
||||
redirect,
|
||||
render_template,
|
||||
request,
|
||||
session,
|
||||
url_for,
|
||||
)
|
||||
from flask_login import current_user
|
||||
|
||||
from app import user_api_client
|
||||
from app.main import main
|
||||
from app.main.forms import (
|
||||
RegisterUserForm,
|
||||
@@ -13,6 +20,7 @@ from app.main.forms import (
|
||||
RegisterUserFromOrgInviteForm,
|
||||
SetupUserProfileForm,
|
||||
)
|
||||
from app.main.views import sign_in
|
||||
from app.main.views.verify import activate_user
|
||||
from app.models.user import InvitedOrgUser, InvitedUser, User
|
||||
from app.utils import hide_from_search_engines
|
||||
@@ -128,16 +136,12 @@ def registration_continue():
|
||||
raise Exception("Unexpected routing in registration_continue")
|
||||
|
||||
|
||||
@main.route("/set-up-your-profile", methods=["GET", "POST"])
|
||||
@main.route("/set-up-your-profile", methods=["GET", "POST"])
|
||||
@hide_from_search_engines
|
||||
def set_up_your_profile():
|
||||
print("ENTER set_up_your_profile")
|
||||
|
||||
|
||||
|
||||
form = SetupUserProfileForm()
|
||||
if form.validate_on_submit():
|
||||
print("VALIDATING FORM")
|
||||
|
||||
# start login.gov
|
||||
code = request.args.get("code")
|
||||
@@ -148,8 +152,6 @@ def set_up_your_profile():
|
||||
user_email, user_uuid = sign_in._get_user_email_and_uuid(access_token)
|
||||
redirect_url = request.args.get("next")
|
||||
|
||||
|
||||
|
||||
elif login_gov_error:
|
||||
current_app.logger.error(f"login.gov error: {login_gov_error}")
|
||||
raise Exception(f"Could not login with login.gov {login_gov_error}")
|
||||
|
||||
@@ -61,7 +61,6 @@ def _get_access_token(code, state):
|
||||
url = f"{base_url}{cli_assert}&{cli_assert_type}&{code_param}&grant_type=authorization_code"
|
||||
headers = {"Authorization": "Bearer %s" % token}
|
||||
response = requests.post(url, headers=headers)
|
||||
print(f"RESPONSE FROM LOGIN DOT GOV {response.json()}")
|
||||
access_token = response.json()["access_token"]
|
||||
return access_token
|
||||
|
||||
|
||||
@@ -30,6 +30,7 @@ def test_logged_in_user_redirects_to_account(
|
||||
)
|
||||
|
||||
|
||||
@pytest.mark.skip("Deprecated due to change to login-dot-gov-only registration")
|
||||
@pytest.mark.parametrize(
|
||||
"phone_number_to_register_with",
|
||||
[
|
||||
@@ -75,14 +76,14 @@ def test_register_creates_new_user_and_redirects_to_continue_page(
|
||||
== "An email has been sent to notfound@example.gsa.gov."
|
||||
)
|
||||
|
||||
mock_send_verify_email.assert_called_with(ANY, user_data["email_address"])
|
||||
mock_register_user.assert_called_with(
|
||||
user_data["name"],
|
||||
user_data["email_address"],
|
||||
user_data["mobile_number"],
|
||||
user_data["password"],
|
||||
user_data["auth_type"],
|
||||
)
|
||||
# mock_send_verify_email.assert_called_with(ANY, user_data["email_address"])
|
||||
# mock_register_user.assert_called_with(
|
||||
# user_data["name"],
|
||||
# user_data["email_address"],
|
||||
# user_data["mobile_number"],
|
||||
# user_data["password"],
|
||||
# user_data["auth_type"],
|
||||
# )
|
||||
|
||||
|
||||
def test_register_continue_handles_missing_session_sensibly(
|
||||
|
||||
@@ -166,6 +166,7 @@ EXCLUDED_ENDPOINTS = tuple(
|
||||
"send_one_off",
|
||||
"send_one_off_step",
|
||||
"send_one_off_to_myself",
|
||||
"set_up_your_profile",
|
||||
"service_add_email_reply_to",
|
||||
"service_add_sms_sender",
|
||||
"service_confirm_delete_email_reply_to",
|
||||
|
||||
Reference in New Issue
Block a user