edan_f0/tests/integration/order/test_order_dispatcher.py

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

from oscar.core.compat import get_user_model
from oscar.core.loading import get_class, get_model
from oscar.test.factories import create_order

User = get_user_model()

CommunicationEventType = get_model('communication', 'CommunicationEventType')
CommunicationEvent = get_model('order', 'CommunicationEvent')
Email = get_model('communication', 'Email')
Dispatcher = get_class('communication.utils', 'Dispatcher')
OrderDispatcher = get_class('order.utils', 'OrderDispatcher')


class TestDispatcher(TestCase):

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

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

        assert order_number in messages['body']
        assert order_number in messages['html']

        order_dispatcher = OrderDispatcher()
        order_dispatcher.dispatch_order_messages(order, messages, event_code)

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

        assert len(mail.outbox) == 1

        message = mail.outbox[0]
        assert order_number in message.body

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

    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)
        assert not 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')
        assert order.is_anonymous
        self._dispatch_order_messages(order_number='12346', order=order, email='testguest@example.com', )