diff --git a/app/main/views/webauthn_credentials.py b/app/main/views/webauthn_credentials.py index 5f097c713..a0e9ddb53 100644 --- a/app/main/views/webauthn_credentials.py +++ b/app/main/views/webauthn_credentials.py @@ -53,6 +53,7 @@ def webauthn_complete_register(): return cbor.encode(str(e)), 400 current_user.create_webauthn_credential(credential) + current_user.update(auth_type='webauthn_auth') flash(( 'Registration complete. Next time you sign in to Notify ' diff --git a/tests/app/main/views/test_webauthn_credentials.py b/tests/app/main/views/test_webauthn_credentials.py index 011200eaa..19386097b 100644 --- a/tests/app/main/views/test_webauthn_credentials.py +++ b/tests/app/main/views/test_webauthn_credentials.py @@ -112,6 +112,7 @@ def test_begin_register_stores_state_in_session( def test_complete_register_creates_credential( platform_admin_user, platform_admin_client, + mock_update_user_attribute, mocker, ): with platform_admin_client.session_transaction() as session: @@ -134,6 +135,10 @@ def test_complete_register_creates_credential( assert response.status_code == 200 credential_mock.assert_called_once_with('state', 'public_key_credential') user_api_mock.assert_called_once_with(platform_admin_user['id'], 'cred') + mock_update_user_attribute.assert_called_once_with( + platform_admin_user['id'], + auth_type='webauthn_auth', + ) def test_complete_register_clears_session(