def test_default_get_jwt_data_callback(app): jwt_manager = JWTManager(app) with app.test_request_context(): result = jwt_manager._get_jwt_data(identity='foo') assert 'exp' in result assert 'iat' in result assert 'nbf' in result assert result['sub'] == 'foo'
def test_custom_get_jwt_data_callback(app): jwt_manager = JWTManager(app) @jwt_manager.jwt_data_loader def custom(identity): return {"foo": "bar"} with app.test_request_context(): result = jwt_manager._get_jwt_data(identity='foo') assert result == {"foo": "bar"}