Fixed exception propagation which, when DEBUG==False, prevented a test from returning 401. Instead, it was just crashing with a JWTError exception.

jslvtr · jslvtr · commit 5fbc910b3d00 · 2017-04-15T15:22:58.000+01:00
diff --git a/section4/video_code/app.py b/section4/video_code/app.py
@@ -1,8 +1,8 @@
 import os
 
-from flask import Flask
+from flask import Flask, jsonify
 from flask_restful import Api
-from flask_jwt import JWT
+import flask_jwt
 
 from security import authenticate, identity
 from resources.user import UserRegister
@@ -15,10 +15,11 @@
 
 app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('DATABASE_URL', 'sqlite:///data.db')
 app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+app.config['PROPAGATE_EXCEPTIONS'] = True
 app.secret_key = 'jose'
 api = Api(app)
 
-jwt = JWT(app, authenticate, identity)  # /auth
+jwt = flask_jwt.JWT(app, authenticate, identity)  # /auth
 
 api.add_resource(Store, '/store/<string:name>')
 api.add_resource(Item, '/item/<string:name>')
@@ -27,6 +28,11 @@
 
 api.add_resource(UserRegister, '/register')
 
+
+@app.errorhandler(flask_jwt.JWTError)
+def auth_error(err):
+    return jsonify({'message': 'Could not authorize. Did you include a valid Authorization header?'}), 401
+
 if __name__ == '__main__':
     from db import db
 
diff --git a/section4/video_code/tests/base_test.py b/section4/video_code/tests/base_test.py
@@ -17,6 +17,7 @@ class BaseTest(TestCase):
     @classmethod
     def setUpClass(cls):
         app.config['SQLALCHEMY_DATABASE_URI'] = BaseTest.SQLALCHEMY_DATABASE_URI
+        app.config['DEBUG'] = False
         with app.app_context():
             db.init_app(app)
 
