def test_auth_mfa(next_action: str):
"""Authentication by MFA."""
global mock_cnt
application = "testapplication"
account = "testaccount"
user = "******"
password = "******"
# success test case
mock_cnt = 0
rest = _init_rest(application,
_create_mock_auth_mfs_rest_response(next_action))
auth = Auth(rest)
auth_instance = AuthByDefault(password)
auth.authenticate(auth_instance, account, user)
assert not rest._connection.errorhandler.called # not error
assert rest.token == "TOKEN"
assert rest.master_token == "MASTER_TOKEN"
# failure test case
mock_cnt = 0
rest = _init_rest(application, _mock_auth_mfa_rest_response_failure)
auth = Auth(rest)
auth_instance = AuthByDefault(password)
auth.authenticate(auth_instance, account, user)
assert rest._connection.errorhandler.called # error
# timeout 1 second
mock_cnt = 0
rest = _init_rest(application, _mock_auth_mfa_rest_response_timeout)
auth = Auth(rest)
auth_instance = AuthByDefault(password)
auth.authenticate(auth_instance, account, user, timeout=1)
assert rest._connection.errorhandler.called # error
def test_auth_mfa():
"""
Authentication by MFA
"""
global mock_cnt
application = 'testapplication'
account = 'testaccount'
user = '******'
password = '******'
# success test case
mock_cnt = 0
rest = _init_rest(application, _mock_auth_mfa_rest_response)
auth = Auth(rest)
auth_instance = AuthByDefault(password)
auth.authenticate(auth_instance, account, user)
assert not rest._connection.errorhandler.called # not error
assert rest.token == 'TOKEN'
assert rest.master_token == 'MASTER_TOKEN'
# failure test case
mock_cnt = 0
rest = _init_rest(application, _mock_auth_mfa_rest_response_failure)
auth = Auth(rest)
auth_instance = AuthByDefault(password)
auth.authenticate(auth_instance, account, user)
assert rest._connection.errorhandler.called # error
# timeout 1 second
mock_cnt = 0
rest = _init_rest(application, _mock_auth_mfa_rest_response_timeout)
auth = Auth(rest)
auth_instance = AuthByDefault(password)
auth.authenticate(auth_instance, account, user, timeout=1)
assert rest._connection.errorhandler.called # error
def test_auth_password_change():
"""Tests password change."""
global mock_cnt
def _password_callback():
return "NEW_PASSWORD"
application = 'testapplication'
account = 'testaccount'
user = '******'
password = '******'
# success test case
mock_cnt = 0
rest = _init_rest(application, _mock_auth_password_change_rest_response)
auth = Auth(rest)
auth_instance = AuthByDefault(password)
auth.authenticate(auth_instance, account, user,
password_callback=_password_callback)
assert not rest._connection.errorhandler.called # not error
