async def test_authentication_login_count(tmpcwd, app):
auth = NaasAuthenticator(db=app.db)
infos = {'username': '******', 'password': '******'}
wrong_infos = {'username': '******', 'password': '******'}
auth.get_or_create_user(infos['username'], infos['password'])
UserInfo.change_authorization(app.db, 'johnsnow')
assert not auth.login_attempts
await auth.authenticate(app, wrong_infos)
assert auth.login_attempts['johnsnow']['count'] == 1
await auth.authenticate(app, wrong_infos)
assert auth.login_attempts['johnsnow']['count'] == 2
await auth.authenticate(app, infos)
assert not auth.login_attempts.get('johnsnow')
async def test_authentication_with_exceed_atempts_of_login(tmpcwd, app):
auth = NaasAuthenticator(db=app.db)
auth.allowed_failed_logins = 3
auth.secs_before_next_try = 10
infos = {'username': '******', 'password': '******'}
auth.get_or_create_user(infos['username'], 'password')
UserInfo.change_authorization(app.db, 'johnsnow')
for i in range(3):
response = await auth.authenticate(app, infos)
assert not response
infos['password'] = '******'
response = await auth.authenticate(app, infos)
assert not response
time.sleep(12)
response = await auth.authenticate(app, infos)
assert response
