예제 #1
0
def test_default_invalid_token_callback(app):
    jwt_manager = JWTManager(app)

    with app.test_request_context():
        err = "Test error"
        result = jwt_manager._invalid_token_callback(err)
        status_code, data = _parse_callback(result)

        assert status_code == 422
        assert data == {'msg': err}
예제 #2
0
def test_custom_invalid_token_callback(app):
    jwt_manager = JWTManager(app)

    @jwt_manager.invalid_token_loader
    def custom(err):
        return jsonify({"foo": "bar"}), 200

    with app.test_request_context():
        err = "Test error"
        result = jwt_manager._invalid_token_callback(err)
        status_code, data = _parse_callback(result)

        assert status_code == 200
        assert data == {'foo': 'bar'}