44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
"""Tests of email marketing signal handlers."""
|
|
import logging
|
|
import ddt
|
|
from django.test import TestCase
|
|
from mock import patch
|
|
|
|
from student.tests.factories import UserFactory
|
|
from openedx.features.enterprise_support.tests.factories import EnterpriseCustomerFactory, EnterpriseCustomerUserFactory
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
LOGGER_NAME = "enterprise_support.signals"
|
|
|
|
TEST_EMAIL = "test@edx.org"
|
|
|
|
|
|
@ddt.ddt
|
|
class EnterpriseSupportSignals(TestCase):
|
|
"""
|
|
Tests for the enterprise support signals.
|
|
"""
|
|
|
|
def setUp(self):
|
|
self.user = UserFactory.create(username='test', email=TEST_EMAIL)
|
|
super(EnterpriseSupportSignals, self).setUp()
|
|
|
|
@patch('openedx.features.enterprise_support.signals.update_user.delay')
|
|
def test_register_user(self, mock_update_user):
|
|
"""
|
|
make sure marketing enterprise user call invokes update_user
|
|
"""
|
|
enterprise_customer = EnterpriseCustomerFactory()
|
|
EnterpriseCustomerUserFactory(
|
|
user_id=self.user.id,
|
|
enterprise_customer=enterprise_customer
|
|
)
|
|
mock_update_user.assert_called_with(
|
|
sailthru_vars={
|
|
'is_enterprise_learner': True,
|
|
'enterprise_name': enterprise_customer.name,
|
|
},
|
|
email=self.user.email
|
|
)
|