Merge pull request #11 from paramsingh/debugtoolbar

paramsingh · web-flow · commit c4340f3f3683 · 2018-02-22T23:37:47.000+05:30
BU-7: Create a method to init DebugToolbar in CustomFlask
diff --git a/brainzutils/flask/__init__.py b/brainzutils/flask/__init__.py
@@ -8,7 +8,7 @@ class CustomFlask(Flask):
     """Custom version of Flask with our bells and whistles."""
 
     def __init__(self, import_name, config_file=None, debug=None,
-                 use_flask_uuid=False, use_debug_toolbar=False,
+                 use_flask_uuid=False,
                  *args, **kwargs):
         """Create an instance of Flask app.
 
@@ -20,8 +20,6 @@ def __init__(self, import_name, config_file=None, debug=None,
                 Should be in a form of Python module.
             debug (bool): Override debug value.
             use_flask_uuid (bool): Turn on Flask-UUID extension if set to True.
-            use_debug_toolbar (bool): Turn on Flask-DebugToolbar extension in
-                debug mode if set to True.
         """
         super(CustomFlask, self).__init__(import_name, *args, **kwargs)
         if config_file:
@@ -30,9 +28,19 @@ def __init__(self, import_name, config_file=None, debug=None,
             self.debug = debug
         if use_flask_uuid:
             FlaskUUID(self)
-        if use_debug_toolbar and self.debug:
+
+
+    def init_debug_toolbar(self):
+        """This method initializes the Flask-Debug extension toolbar for the
+        Flask app.
+
+        Note that the Flask-Debug extension requires app.debug be true
+        and the SECRET_KEY be defined in app.config.
+        """
+        if self.debug:
             DebugToolbarExtension(self)
 
+
     def init_loggers(self,
                      file_config=None,
                      email_config=None,
diff --git a/brainzutils/flask/test/test_main.py b/brainzutils/flask/test/test_main.py
@@ -1,9 +1,30 @@
 import unittest
-from brainzutils import flask
 
+from brainzutils import flask
 
 class FlaskTestCase(unittest.TestCase):
 
     def test_create_app(self):
         app = flask.CustomFlask(__name__)
         self.assertIsNotNone(app)
+
+    def test_debug_toolbar(self):
+        """ Tests that debug toolbar loads if initialized correctly
+        """
+
+        # create an app
+        app = flask.CustomFlask(__name__)
+        self.assertIsNotNone(app)
+        app.debug = True
+        app.config['SECRET_KEY'] = 'this is a totally secret key btw'
+        app.init_debug_toolbar()
+
+        # add a dummy route
+        @app.route('/')
+        def index():
+            return '<html><body>test</body></html>'
+
+        client = app.test_client()
+        response = client.get('/')
+        self.assertEqual(response.status_code, 200)
+        self.assertIn('flDebug', str(response.data))
