apply workaround to support testing with flask-login 0.6.2

- flask-login now uses flask.g to cache the logged-in user
- level12/keg#90 needs to be resolved before we can remove the workaround

keg_auth/core.py

@@ -5,7 +5,7 @@
 import sqlalchemy as sa
 from blazeutils import tolist
 from keg.db import db
-from keg.signals import db_init_post
+from keg.signals import db_init_post, init_complete
 from webgrid.renderers import render_html_attributes
 
 import keg_auth.cli
@@ -92,6 +92,8 @@ def init_app(self, app):
         self.init_loaders(app)
         self.init_permissions(app)
 
+        self.fix_testing_teardown(app)
+
     def init_config(self, app):
         """Provide app config defaults for crypto, mail, logins, etc."""
         _cc_kwargs = dict(schemes=DEFAULT_CRYPTO_SCHEMES, deprecated='auto')
@@ -390,6 +392,21 @@ def resend_verification_email(self, user_id):
         user.token_generate()
         self.mail_manager.send_new_user(user)
 
+    def fix_testing_teardown(self, app):
+        """ Needed until https://github.com/level12/keg/issues/90 is fixed & released. """
+        if not app.config.get('TESTING', False):
+            return
+
+        version = tuple(map(lambda x: int(x), flask_login.__version__.split('.')))
+
+        if version < (0, 6, 2):
+            return
+
+        @app.teardown_request
+        def cleanup_user_cache(error):
+            if '_login_user' in flask.g:
+                del flask.g._login_user
+
 
 # ensure that any manager-attached menus are reset for auth requirements on login/logout
 def refresh_session_menus(app, user):

keg_auth/tests/test_views.py

@@ -1,6 +1,5 @@
 from unittest import mock
 
-from blazeutils.datastructures import BlankObject
 import flask
 import freezegun
 import arrow