87 lines
3.5 KiB
Python
87 lines
3.5 KiB
Python
import datetime
|
|
|
|
from django.urls import reverse as urlreverse
|
|
|
|
from ietf.utils.test_utils import TestCase, unicontent
|
|
from ietf.utils.mail import outbox
|
|
|
|
from ietf.message.models import Message, SendQueue
|
|
from ietf.person.models import Person
|
|
from ietf.group.factories import GroupFactory
|
|
from ietf.message.utils import send_scheduled_message_from_send_queue
|
|
|
|
class MessageTests(TestCase):
|
|
def test_message_view(self):
|
|
nomcom = GroupFactory(name="nomcom%s" % datetime.date.today().year, type_id="nomcom")
|
|
msg = Message.objects.create(
|
|
by=Person.objects.get(name="(System)"),
|
|
subject="This is a test",
|
|
to="test@example.com",
|
|
frm="nomcomchair@example.com",
|
|
body="Hello World!",
|
|
content_type="",
|
|
)
|
|
msg.related_groups.add(nomcom)
|
|
|
|
r = self.client.get(urlreverse("ietf.message.views.message", kwargs=dict(message_id=msg.id)))
|
|
self.assertEqual(r.status_code, 200)
|
|
self.assertTrue(msg.subject in unicontent(r))
|
|
self.assertTrue(msg.to in unicontent(r))
|
|
self.assertTrue(msg.frm in unicontent(r))
|
|
self.assertTrue("Hello World!" in unicontent(r))
|
|
|
|
|
|
class SendScheduledAnnouncementsTests(TestCase):
|
|
def test_send_plain_announcement(self):
|
|
msg = Message.objects.create(
|
|
by=Person.objects.get(name="(System)"),
|
|
subject="This is a test",
|
|
to="test@example.com",
|
|
frm="testmonkey@example.com",
|
|
cc="cc.a@example.com, cc.b@example.com",
|
|
bcc="bcc@example.com",
|
|
body=u"Hello World!",
|
|
content_type="",
|
|
)
|
|
|
|
q = SendQueue.objects.create(
|
|
by=Person.objects.get(name="(System)"),
|
|
message=msg,
|
|
send_at=datetime.datetime.now() + datetime.timedelta(hours=12)
|
|
)
|
|
|
|
mailbox_before = len(outbox)
|
|
|
|
send_scheduled_message_from_send_queue(q)
|
|
|
|
self.assertEqual(len(outbox), mailbox_before + 1)
|
|
self.assertTrue("This is a test" in outbox[-1]["Subject"])
|
|
self.assertTrue(SendQueue.objects.get(id=q.id).sent_at)
|
|
|
|
def test_send_mime_announcement(self):
|
|
msg = Message.objects.create(
|
|
by=Person.objects.get(name="(System)"),
|
|
subject="This is a test",
|
|
to="test@example.com",
|
|
frm="testmonkey@example.com",
|
|
cc="cc.a@example.com, cc.b@example.com",
|
|
bcc="bcc@example.com",
|
|
body=u'--NextPart\r\n\r\nA New Internet-Draft is available from the on-line Internet-Drafts directories.\r\n--NextPart\r\nContent-Type: Message/External-body;\r\n\tname="draft-huang-behave-bih-01.txt";\r\n\tsite="ftp.ietf.org";\r\n\taccess-type="anon-ftp";\r\n\tdirectory="internet-drafts"\r\n\r\nContent-Type: text/plain\r\nContent-ID: <2010-07-30001541.I-D@ietf.org>\r\n\r\n--NextPart--',
|
|
content_type='Multipart/Mixed; Boundary="NextPart"',
|
|
)
|
|
|
|
q = SendQueue.objects.create(
|
|
by=Person.objects.get(name="(System)"),
|
|
message=msg,
|
|
send_at=datetime.datetime.now() + datetime.timedelta(hours=12)
|
|
)
|
|
|
|
mailbox_before = len(outbox)
|
|
|
|
send_scheduled_message_from_send_queue(q)
|
|
|
|
self.assertEqual(len(outbox), mailbox_before + 1)
|
|
self.assertTrue("This is a test" in outbox[-1]["Subject"])
|
|
self.assertTrue("--NextPart" in outbox[-1].as_string())
|
|
self.assertTrue(SendQueue.objects.get(id=q.id).sent_at)
|