|
@@ -1,25 +1,21 @@
|
|
|
|
+from unittest.mock import Mock
|
|
|
|
+
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.auth import get_user_model
|
|
from django.test import TestCase
|
|
from django.test import TestCase
|
|
from django.utils import timezone
|
|
from django.utils import timezone
|
|
|
|
|
|
-from misago.acl import add_acl
|
|
|
|
|
|
+from misago.acl import add_acl, useracl
|
|
from misago.categories.models import Category
|
|
from misago.categories.models import Category
|
|
from misago.threads.events import record_event
|
|
from misago.threads.events import record_event
|
|
from misago.threads.models import Thread
|
|
from misago.threads.models import Thread
|
|
|
|
|
|
-
|
|
|
|
-UserModel = get_user_model()
|
|
|
|
-
|
|
|
|
-
|
|
|
|
-class MockRequest(object):
|
|
|
|
- def __init__(self, user):
|
|
|
|
- self.user = user
|
|
|
|
- self.user_ip = '123.14.15.222'
|
|
|
|
|
|
+User = get_user_model()
|
|
|
|
+cache_versions = {"acl": "abcdefgh"}
|
|
|
|
|
|
|
|
|
|
class EventsApiTests(TestCase):
|
|
class EventsApiTests(TestCase):
|
|
def setUp(self):
|
|
def setUp(self):
|
|
- self.user = UserModel.objects.create_user("Bob", "bob@bob.com", "Pass.123")
|
|
|
|
|
|
+ self.user = User.objects.create_user("Bob", "bob@bob.com", "Pass.123")
|
|
|
|
|
|
datetime = timezone.now()
|
|
datetime = timezone.now()
|
|
|
|
|
|
@@ -37,12 +33,13 @@ class EventsApiTests(TestCase):
|
|
self.thread.set_title("Test thread")
|
|
self.thread.set_title("Test thread")
|
|
self.thread.save()
|
|
self.thread.save()
|
|
|
|
|
|
- add_acl(self.user, self.category)
|
|
|
|
- add_acl(self.user, self.thread)
|
|
|
|
|
|
+ user_acl = useracl.get_user_acl(self.user, cache_versions)
|
|
|
|
+ add_acl(user_acl, self.category)
|
|
|
|
+ add_acl(user_acl, self.thread)
|
|
|
|
|
|
def test_record_event_with_context(self):
|
|
def test_record_event_with_context(self):
|
|
"""record_event registers event with context in thread"""
|
|
"""record_event registers event with context in thread"""
|
|
- request = MockRequest(self.user)
|
|
|
|
|
|
+ request = Mock(user=self.user, user_ip="123.14.15.222")
|
|
context = {'user': 'Lorem ipsum'}
|
|
context = {'user': 'Lorem ipsum'}
|
|
event = record_event(request, self.thread, 'announcement', context)
|
|
event = record_event(request, self.thread, 'announcement', context)
|
|
|
|
|
|
@@ -59,7 +56,7 @@ class EventsApiTests(TestCase):
|
|
|
|
|
|
def test_record_event_is_read(self):
|
|
def test_record_event_is_read(self):
|
|
"""record_event makes recorded event read to its author"""
|
|
"""record_event makes recorded event read to its author"""
|
|
- request = MockRequest(self.user)
|
|
|
|
|
|
+ request = Mock(user=self.user, user_ip="123.14.15.222")
|
|
event = record_event(request, self.thread, 'announcement')
|
|
event = record_event(request, self.thread, 'announcement')
|
|
|
|
|
|
self.user.postread_set.get(
|
|
self.user.postread_set.get(
|