77from django .conf import settings as django_settings
88from django .core import mail
99
10- from django_yubin import models , constants , queue_email_message , settings , mail_admins , mail_managers
10+ from django_yubin import models , constants , settings , mail_admins , mail_managers
1111
1212from .base import MailerTestCase , RFC_6532_SUPPORT
1313
1414
1515class TestBackend (MailerTestCase ):
1616 """
1717 Backend tests for the django_yubin app.
18-
19- For Django versions less than 1.2, these tests are still run but they just
20- use the queue_email_message funciton rather than directly sending messages.
21-
2218 """
2319
2420 def setUp (self ):
2521 super (TestBackend , self ).setUp ()
26- if constants .EMAIL_BACKEND_SUPPORT :
27- if hasattr (django_settings , 'EMAIL_BACKEND' ):
28- self .old_email_backend = django_settings .EMAIL_BACKEND
29- else :
30- self .old_email_backend = None
31- django_settings .EMAIL_BACKEND = 'django_yubin.smtp_queue.' \
32- 'EmailBackend'
22+ if hasattr (django_settings , 'EMAIL_BACKEND' ):
23+ self .old_email_backend = django_settings .EMAIL_BACKEND
24+ else :
25+ self .old_email_backend = None
26+ django_settings .EMAIL_BACKEND = 'django_yubin.smtp_queue.' \
27+ 'EmailBackend'
3328
3429 def tearDown (self ):
3530 super (TestBackend , self ).tearDown ()
36- if constants .EMAIL_BACKEND_SUPPORT :
37- if self .old_email_backend :
38- django_settings .EMAIL_BACKEND = self .old_email_backend
39- else :
40- delattr (django_settings , 'EMAIL_BACKEND' )
41-
42- def send_message (self , msg ):
43- if constants .EMAIL_BACKEND_SUPPORT :
44- msg .send ()
31+ if self .old_email_backend :
32+ django_settings .EMAIL_BACKEND = self .old_email_backend
4533 else :
46- queue_email_message ( msg )
34+ delattr ( django_settings , 'EMAIL_BACKEND' )
4735
4836 def testQueuedMessagePriorities (self ):
4937 # now_not_queued priority message
5038 msg = mail .EmailMessage (subject = 'subject' , body = 'body' ,
51395240 headers = {'X-Mail-Queue-Priority' : 'now-not-queued' })
53- self . send_message ( msg )
41+ msg . send ( )
5442
5543 # now priority message
5644 msg = mail .EmailMessage (subject = 'subject' , body = 'body' ,
57455846 headers = {'X-Mail-Queue-Priority' : 'now' })
59- self . send_message ( msg )
47+ msg . send ( )
6048
6149 # high priority message
6250 msg = mail .EmailMessage (subject = 'subject' , body = 'body' ,
63516452 headers = {'X-Mail-Queue-Priority' : 'high' })
65- self . send_message ( msg )
53+ msg . send ( )
6654
6755 # low priority message
6856 msg = mail .EmailMessage (subject = 'subject' , body = 'body' ,
69577058 headers = {'X-Mail-Queue-Priority' : 'low' })
71- self . send_message ( msg )
59+ msg . send ( )
7260
7361 # normal priority message
7462 msg = mail .EmailMessage (subject = 'subject' , body = 'body' ,
75637664 headers = {'X-Mail-Queue-Priority' : 'normal' })
77- self . send_message ( msg )
65+ msg . send ( )
7866
7967 # normal priority message (no explicit priority header)
8068 msg = mail .EmailMessage (subject = 'subject' , body = 'body' ,
816982- self . send_message ( msg )
70+ msg . send ( )
8371
8472 qs = models .Message .objects .all ()
8573 self .assertEqual (qs .count (), 5 )
@@ -113,9 +101,9 @@ def testUnicodeQueuedMessage(self):
113101 """
114102 from django .core .management import call_command
115103 msg = mail .EmailMessage (subject = 'subject' , body = 'body' ,
116- 117- headers = {'X-Mail-Queue-Priority' : 'normal' })
118- self . send_message ( msg )
104+ 105+ headers = {'X-Mail-Queue-Priority' : 'normal' })
106+ msg . send ( )
119107 queued_messages = models .QueuedMessage .objects .all ()
120108 self .assertEqual (queued_messages .count (), 1 )
121109 call_command ('send_mail' , verbosity = '0' )
@@ -131,9 +119,9 @@ def testUnicodePriorityMessage(self):
131119 """
132120 from django .core .management import call_command
133121 msg = mail .EmailMessage (subject = u'á subject' , body = 'body' ,
134- from_email = u'juan.ló[email protected] ' ,
to = [
u'únñ[email protected] ' ],
135- headers = {'X-Mail-Queue-Priority' : 'now' })
136- self . send_message ( msg )
122+ from_email = u'juan.ló[email protected] ' ,
to = [
u'únñ[email protected] ' ],
123+ headers = {'X-Mail-Queue-Priority' : 'now' })
124+ msg . send ( )
137125 queued_messages = models .QueuedMessage .objects .all ()
138126 self .assertEqual (queued_messages .count (), 1 )
139127 call_command ('send_mail' , verbosity = '0' )
@@ -143,9 +131,9 @@ def testUnicodePriorityMessage(self):
143131 def testSendMessageNowPriority (self ):
144132 # NOW priority message
145133 msg = mail .EmailMessage (subject = 'subject' , body = 'body' ,
146- 147- headers = {'X-Mail-Queue-Priority' : 'now' })
148- self . send_message ( msg )
134+ 135+ headers = {'X-Mail-Queue-Priority' : 'now' })
136+ msg . send ( )
149137
150138 queued_messages = models .QueuedMessage .objects .all ()
151139 self .assertEqual (queued_messages .count (), 0 )
@@ -156,7 +144,7 @@ def testSendMessageTestMode(self):
156144 settings .
MAILER_TEST_EMAIL = '[email protected] ' 157145 msg = mail .EmailMessage (subject = 'subject' , body = 'body' ,
158146159- self . send_message ( msg )
147+ msg . send ( )
160148
161149 queued_messages = models .QueuedMessage .objects .all ()
162150
@@ -193,7 +181,6 @@ def testSendMessageManagersTestMode(self):
193181 self .assertTrue ('X-Yubin-Test-Original: {}' .format (',' .join (recipient_list )) in
194182 queued_messages [0 ].message .encoded_message )
195183
196-
197184 def testHighPriority (self ):
198185 self .assertEqual (models .QueuedMessage .objects .all ().count (), 0 )
199186 self .assertEqual (models .Message .objects .all ().count (), 0 )
@@ -202,7 +189,7 @@ def testHighPriority(self):
202189 # Test with default priority
203190 msg = mail .EmailMessage (subject = 'subject 1' , body = 'body' ,
204191205- self . send_message ( msg )
192+ msg . send ( )
206193
207194 self .assertEqual (models .Message .objects .all ().count (), 1 )
208195 self .assertEqual (models .QueuedMessage .objects .all ().count (), 1 )
@@ -212,7 +199,7 @@ def testHighPriority(self):
212199 msg = mail .EmailMessage (subject = 'subject 2' , body = 'body' ,
213200214201 headers = {'X-Mail-Queue-Priority' : 'now' })
215- self . send_message ( msg )
202+ msg . send ( )
216203
217204 self .assertEqual (models .QueuedMessage .objects .all ().count (), 1 )
218205 self .assertEqual (models .Message .objects .all ().count (), 2 )
0 commit comments