Merge pull request #1378 from CTFd/bulk-clear-sessions

* Add function to mass clear the cached user/team attributes for all users/teams

CTFd/cache/__init__.py

@@ -58,7 +58,19 @@ def clear_user_session(user_id):
     cache.delete_memoized(get_user_attrs, user_id=user_id)
 
 
+def clear_all_user_sessions():
+    from CTFd.utils.user import get_user_attrs
+
+    cache.delete_memoized(get_user_attrs)
+
+
 def clear_team_session(team_id):
     from CTFd.utils.user import get_team_attrs
 
     cache.delete_memoized(get_team_attrs, team_id=team_id)
+
+
+def clear_all_team_sessions():
+    from CTFd.utils.user import get_team_attrs
+
+    cache.delete_memoized(get_team_attrs)

tests/cache/test_cache.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
+from CTFd.cache import clear_user_session, clear_all_user_sessions
 from CTFd.models import Users
 from CTFd.utils.user import is_admin, get_current_user
 from CTFd.utils.security.auth import login_user
@@ -21,17 +22,45 @@ def test_clear_user_session():
             assert user.type == "user"
             assert is_admin() is False
 
-        # Set the user's updated type
-        user = Users.query.filter_by(id=2).first()
-        user.type = "admin"
-        app.db.session.commit()
+            # Set the user's updated type
+            user = Users.query.filter_by(id=2).first()
+            user.type = "admin"
+            app.db.session.commit()
 
-        # The user's type should now be admin
+            # Should still return False because this is still cached
+            assert is_admin() is False
+
+            clear_user_session(user_id=2)
+
+            # Should now return True after clearing cache
+            assert is_admin() is True
+    destroy_ctfd(app)
+
+
+def test_clear_all_user_sessions():
+    app = create_ctfd()
+    with app.app_context():
+        register_user(app)
+
+        # Users by default should have a non-admin type
         user = Users.query.filter_by(id=2).first()
         with app.test_request_context("/"):
             login_user(user)
             user = get_current_user()
             assert user.id == 2
-            assert user.type == "admin"
+            assert user.type == "user"
+            assert is_admin() is False
+
+            # Set the user's updated type
+            user = Users.query.filter_by(id=2).first()
+            user.type = "admin"
+            app.db.session.commit()
+
+            # Should still return False because this is still cached
+            assert is_admin() is False
+
+            clear_all_user_sessions()
+
+            # Should now return True after clearing cache
             assert is_admin() is True
     destroy_ctfd(app)