1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
import base64
from django.test import TestCase
from django.test.utils import override_settings
from django.conf import settings
from django.contrib.sites.models import Site
from django.core import mail
from django.utils.six.moves import cPickle as pickle # pylint: disable-msg=F
from ..models import NoticeType, NoticeSetting, NoticeQueueBatch
from ..models import LanguageStoreNotAvailable
from ..models import get_notification_language, create_notice_type, send_now, send, queue
from ..compat import get_user_model
from .models import Language
from . import get_backend_id
class BaseTest(TestCase):
def setUp(self):
self.user = get_user_model().objects.create_user("test_user", "test@user.com", "123456")
self.user2 = get_user_model().objects.create_user("test_user2", "test2@user.com", "123456")
create_notice_type("label", "display", "description")
self.notice_type = NoticeType.objects.get(label="label")
def tearDown(self):
self.user.delete()
self.user2.delete()
self.notice_type.delete()
class TestNoticeType(TestCase):
def test_create_notice_type(self):
label = "friends_invite"
create_notice_type(label, "Invitation Received", "you received an invitation")
n = NoticeType.objects.get(label=label)
self.assertEqual(str(n), label)
def test_create(self):
label = "friends_invite"
NoticeType.create(label, "Invitation Received", "you received an invitation", default=2,
verbosity=2)
n = NoticeType.objects.get(label=label)
self.assertEqual(str(n), label)
# update
NoticeType.create(label, "Invitation for you", "you got an invitation", default=1,
verbosity=2)
n = NoticeType.objects.get(pk=n.pk)
self.assertEqual(n.display, "Invitation for you")
self.assertEqual(n.description, "you got an invitation")
self.assertEqual(n.default, 1)
class TestNoticeSetting(BaseTest):
def test_for_user(self):
email_id = get_backend_id("email")
notice_setting = NoticeSetting.objects.create(user=self.user, notice_type=self.notice_type,
medium=email_id, send=False)
self.assertEqual(NoticeSetting.for_user(self.user, self.notice_type, email_id),
notice_setting)
# test default fallback
NoticeSetting.for_user(self.user2, self.notice_type, email_id)
ns2 = NoticeSetting.objects.get(user=self.user2, notice_type=self.notice_type,
medium=email_id)
self.assertTrue(ns2.send)
class TestProcedures(BaseTest):
def setUp(self):
super(TestProcedures, self).setUp()
self.lang = Language.objects.create(user=self.user, language="en_US")
mail.outbox = []
def tearDown(self):
super(TestProcedures, self).tearDown()
self.lang.delete()
NoticeQueueBatch.objects.all().delete()
@override_settings(NOTIFICATION_LANGUAGE_MODULE="tests.Language")
def test_get_notification_language(self):
self.assertEqual(get_notification_language(self.user), "en_US")
self.assertRaises(LanguageStoreNotAvailable, get_notification_language, self.user2)
del settings.NOTIFICATION_LANGUAGE_MODULE
self.assertRaises(LanguageStoreNotAvailable, get_notification_language, self.user)
@override_settings(SITE_ID=1, NOTIFICATION_LANGUAGE_MODULE="tests.Language")
def test_send_now(self):
Site.objects.create(domain="localhost", name="localhost")
users = [self.user, self.user2]
send_now(users, "label")
self.assertEqual(len(mail.outbox), 2)
self.assertIn(self.user.email, mail.outbox[0].to)
self.assertIn(self.user2.email, mail.outbox[1].to)
@override_settings(SITE_ID=1)
def test_send(self):
self.assertRaises(AssertionError, send, queue=True, now=True)
users = [self.user, self.user2]
send(users, "label", now=True)
self.assertEqual(len(mail.outbox), 2)
self.assertIn(self.user.email, mail.outbox[0].to)
self.assertIn(self.user2.email, mail.outbox[1].to)
send(users, "label", queue=True)
self.assertEqual(NoticeQueueBatch.objects.count(), 1)
batch = NoticeQueueBatch.objects.all()[0]
notices = pickle.loads(base64.b64decode(batch.pickled_data))
self.assertEqual(len(notices), 2)
@override_settings(SITE_ID=1)
def test_send_default(self):
# default behaviout, send_now
users = [self.user, self.user2]
send(users, "label")
self.assertEqual(len(mail.outbox), 2)
self.assertEqual(NoticeQueueBatch.objects.count(), 0)
@override_settings(SITE_ID=1)
def test_queue_queryset(self):
users = get_user_model().objects.all()
queue(users, "label")
self.assertEqual(len(mail.outbox), 0)
self.assertEqual(NoticeQueueBatch.objects.count(), 1)
|
