edan_f0/tests/integration/customer/test_dispatcher.py

from django.core import mail
from django.test import TestCase

from oscar.apps.customer.models import CommunicationEventType, Email
from oscar.apps.customer.utils import Dispatcher, get_password_reset_url
from oscar.apps.order.models import CommunicationEvent
from oscar.core.compat import get_user_model
from oscar.test.factories import SiteFactory, create_order

User = get_user_model()


class TestDispatcher(TestCase):

    def _dispatch_order_messages(self, order_number, order, email=None):
        et = CommunicationEventType.objects.create(code="ORDER_PLACED",
                                                   name="Order Placed",
                                                   category=CommunicationEventType.ORDER_RELATED)

        messages = et.get_messages({
            'order': order,
            'lines': order.lines.all()
        })

        self.assertIn(order_number, messages['body'])
        self.assertIn(order_number, messages['html'])

        dispatcher = Dispatcher()
        dispatcher.dispatch_order_messages(order, messages, et)

        self.assertEqual(CommunicationEvent.objects.filter(order=order, event_type=et).count(), 1)

        self.assertEqual(len(mail.outbox), 1)

        message = mail.outbox[0]
        self.assertIn(order_number, message.body)

        # test sending messages to emails without account and text body
        messages['body'] = ''
        dispatcher.dispatch_direct_messages(email, messages)
        self.assertEqual(len(mail.outbox), 2)

    def _dispatch_user_messages(self, user, event_code, ctx, subject):
        msgs = CommunicationEventType.objects.get_and_render(
            code=event_code, context=ctx)
        Dispatcher().dispatch_user_messages(user, msgs)
        self.assertEqual(len(mail.outbox), 1)
        self.assertEqual(mail.outbox[0].subject, subject)
        self.assertEqual(Email.objects.count(), 1)
        email = Email.objects.last()
        self.assertEqual(email.user.id, user.id)
        self.assertEqual(email.email, 'testuser@example.com')

    def test_dispatch_order_messages(self):
        email = 'testuser@example.com'
        user = User.objects.create_user('testuser', email,
                                        'somesimplepassword')
        order = create_order(number='12345', user=user)
        self.assertFalse(order.is_anonymous)
        self._dispatch_order_messages(order_number='12345', order=order, email=email)

    def test_dispatch_anonymous_order_messages(self):
        order = create_order(number='12346', guest_email='testguest@example.com')
        self.assertTrue(order.is_anonymous)
        self._dispatch_order_messages(order_number='12346', order=order, email='testguest@example.com', )

    def test_dispatch_email_changed_user_message(self):
        user = User.objects.create_user('testuser', 'testuser@example.com', 'somesimplepassword')
        CommunicationEventType.objects.create(code='EMAIL_CHANGED',
                                              name='Email Changed',
                                              category=CommunicationEventType.USER_RELATED)
        ctx = {
            'user': user,
            'site': SiteFactory(name='Test Site'),
            'reset_url': get_password_reset_url(user),
            'new_email': 'newtestuser@example.com',
        }
        self._dispatch_user_messages(
            user, 'EMAIL_CHANGED', ctx, 'Your email address has changed at Test Site.'
        )

    def test_dispatch_registration_email_message(self):
        user = User.objects.create_user('testuser', 'testuser@example.com', 'somesimplepassword')
        CommunicationEventType.objects.create(code='REGISTRATION',
                                              name='Registration',
                                              category=CommunicationEventType.USER_RELATED)
        ctx = {'user': user,
               'site': SiteFactory()}
        self._dispatch_user_messages(user, 'REGISTRATION', ctx, 'Thank you for registering.')

    def test_dispatcher_uses_email_connection(self):
        connection = mail.get_connection()
        disp = Dispatcher(mail_connection=connection)
        disp.dispatch_direct_messages('test@example.com', {
            'subject': 'Test',
            'body': 'This is a test.',
            'html': '',
        })
        self.assertEqual(len(mail.outbox), 1)