darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 09:51:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add unit tests for the Users DAO create_secret_code function

Browse Source
This commit is contained in:
Paul Wright
2018-08-23 12:18:47 +01:00
parent 47892168d6
commit 267a82758a
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -16,7 +16,8 @@ from app.dao.users_dao import (
get_user_by_email, get_user_by_email,
delete_codes_older_created_more_than_a_day_ago, delete_codes_older_created_more_than_a_day_ago,
update_user_password, update_user_password,
count_user_verify_codes) count_user_verify_codes,
create_secret_code)
from app.models import User, VerifyCode from app.models import User, VerifyCode
@@ -156,3 +157,14 @@ def test_count_user_verify_codes(sample_user):
[make_verify_code(sample_user) for i in range(5)] [make_verify_code(sample_user) for i in range(5)]
assert count_user_verify_codes(sample_user) == 5 assert count_user_verify_codes(sample_user) == 5
def test_create_secret_code_different_subsequent_codes():
code1 = create_secret_code()
code2 = create_secret_code()
assert code1 != code2
def test_create_secret_code_returns_5_digits():
code = create_secret_code()
assert len(str(code)) == 5