test_floodprotection_middleware.py 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. from datetime import timedelta
  2. from django.utils import timezone
  3. from misago.acl.testutils import override_acl
  4. from misago.threads.api.postingendpoint import PostingInterrupt
  5. from misago.threads.api.postingendpoint.floodprotection import (
  6. MIN_POSTING_PAUSE, FloodProtectionMiddleware)
  7. from misago.users.testutils import AuthenticatedUserTestCase
  8. class FloodProtectionMiddlewareTests(AuthenticatedUserTestCase):
  9. def test_flood_protection_middleware_on_no_posts(self):
  10. """middleware sets last_posted_on on user"""
  11. self.user.update_fields = []
  12. self.assertIsNone(self.user.last_posted_on)
  13. middleware = FloodProtectionMiddleware(user=self.user)
  14. middleware.interrupt_posting(None)
  15. self.assertIsNotNone(self.user.last_posted_on)
  16. def test_flood_protection_middleware_old_posts(self):
  17. """middleware is not interrupting if previous post is old"""
  18. self.user.update_fields = []
  19. original_last_posted_on = timezone.now() - timedelta(days=1)
  20. self.user.last_posted_on = original_last_posted_on
  21. middleware = FloodProtectionMiddleware(user=self.user)
  22. middleware.interrupt_posting(None)
  23. self.assertTrue(self.user.last_posted_on > original_last_posted_on)
  24. def test_flood_protection_middleware_on_flood(self):
  25. """middleware is interrupting flood"""
  26. self.user.last_posted_on = timezone.now()
  27. with self.assertRaises(PostingInterrupt):
  28. middleware = FloodProtectionMiddleware(user=self.user)
  29. middleware.interrupt_posting(None)
  30. def test_flood_permission(self):
  31. """middleware is respects permission to flood for team members"""
  32. override_acl(self.user, {
  33. 'can_omit_flood_protection': True
  34. })
  35. middleware = FloodProtectionMiddleware(user=self.user)
  36. self.assertFalse(middleware.use_this_middleware())