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"}