sentry/tests/sentry/services/smtp_test.py

import os.path

from sentry.models import Activity
from sentry.services.smtp import STATUS, SentrySMTPServer
from sentry.testutils import TestCase
from sentry.utils.email import email_to_group_id, group_id_to_email

with open(os.path.join(os.path.dirname(__file__), "email.txt")) as f:
    fixture = f.read()


class SentrySMTPTest(TestCase):
    def setUp(self):
        self.address = ("0.0.0.0", 0)
        self.server = SentrySMTPServer(*self.address)
        self.mailto = group_id_to_email(self.group.pk)
        self.event  # side effect of generating an event

    def test_decode_email_address(self):
        self.assertEqual(email_to_group_id(self.mailto), self.group.pk)

    def test_process_message(self):
        with self.tasks():
            self.assertEqual(
                self.server.process_message("", self.user.email, [self.mailto], fixture),
                STATUS[200],
            )
        self.assertEqual(Activity.objects.filter(type=Activity.NOTE)[0].data, {"text": "sup"})

    def test_process_message_no_recipients(self):
        with self.tasks():
            self.assertEqual(
                self.server.process_message("", self.user.email, [], fixture), STATUS[550]
            )

    def test_process_message_too_long(self):
        with self.tasks():
            self.assertEqual(
                self.server.process_message("", self.user.email, [self.mailto], fixture * 100),
                STATUS[552],
            )
        self.assertEqual(Activity.objects.count(), 0)

    def test_process_message_invalid_email(self):
        with self.tasks():
            self.assertEqual(
                self.server.process_message("", self.user.email, ["lol@localhost"], fixture),
                STATUS[550],
            )