fix tests

This commit is contained in:
Kenneth Kehl
2024-03-19 11:32:36 -07:00
parent 57fa8dad5d
commit 197ef11075
6 changed files with 24 additions and 21 deletions

View File

@@ -612,7 +612,6 @@ class SetupUserProfileForm(StripWhitespaceForm):
mobile_number = international_phone_number()
class RegisterUserFromInviteForm(RegisterUserForm):
def __init__(self, invited_user):
super().__init__(

View File

@@ -1,4 +1,5 @@
import os
from flask import abort, redirect, render_template, request, url_for
from flask_login import current_user

View File

@@ -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}")

View File

@@ -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

View File

@@ -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(

View File

@@ -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",