def test_current_user(self, user_class, db, default_guard):
"""
This test verifies that the current user can be successfully
determined based on jwt token data that has been added to the current
flask app's context.
"""
jwt_data = {}
add_jwt_data_to_app_context(jwt_data)
with pytest.raises(PraetorianError) as err_info:
current_user()
assert 'Could not fetch an id' in str(err_info.value)
jwt_data = {'id': 31}
add_jwt_data_to_app_context(jwt_data)
with pytest.raises(PraetorianError) as err_info:
current_user()
assert 'Could not identify the current user' in str(err_info.value)
the_dude = user_class(
id=13,
username='******',
)
db.session.add(the_dude)
db.session.commit()
jwt_data = {'id': 13}
add_jwt_data_to_app_context(jwt_data)
assert current_user() is the_dude
def test_current_user_id(self, user_class, db, default_guard):
"""
This test verifies that the current user id can be successfully
determined based on jwt token data that has been added to the current
flask app's context.
"""
jwt_data = {}
add_jwt_data_to_app_context(jwt_data)
with pytest.raises(PraetorianError) as err_info:
current_user()
assert 'Could not fetch an id' in str(err_info.value)
jwt_data = {'id': 31}
add_jwt_data_to_app_context(jwt_data)
assert current_user_id() == 31