From f8f718dff8a5c74b0b69608fdc7ac8cc1971fe0d Mon Sep 17 00:00:00 2001 From: Chris Hill-Scott Date: Mon, 7 Jun 2021 13:53:33 +0100 Subject: [PATCH] Set user to sign in with newly-added key --- app/main/views/webauthn_credentials.py | 1 + tests/app/main/views/test_webauthn_credentials.py | 5 +++++ 2 files changed, 6 insertions(+) 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(