darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-04 10:21:14 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add a limit to the number of active 2fa codes that we create. At the moment that is set to 10.

Browse Source
This commit is contained in:
Rebecca Law
2017-02-15 16:18:05 +00:00
parent 9de88c50ba
commit 52342afe3f
5 changed files with 44 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/app/dao/test_users_dao.py
View File

@@ -14,8 +14,8 @@ from app.dao.users_dao import (
reset_failed_login_count,
get_user_by_email,
delete_codes_older_created_more_than_a_day_ago,
update_user_password
)
update_user_password,
count_user_verify_codes)
from app.models import User, VerifyCode
@@ -140,3 +140,8 @@ def test_update_user_password(notify_api, notify_db, notify_db_session, sample_u
assert not sample_user.check_password(password)
update_user_password(sample_user, password)
assert sample_user.check_password(password)
def test_count_user_verify_codes(sample_user):
[make_verify_code(sample_user) for i in range(5)]
assert count_user_verify_codes(sample_user) == 5