edan_f0/tests/integration/customer/test_custom_user_model.py

from django.test import TestCase

from oscar.apps.customer.forms import ProfileForm
from oscar.core.compat import existing_user_fields, get_user_model
from oscar.test.factories import ProductAlertFactory, UserFactory


class TestACustomUserModel(TestCase):
    def setUp(self):
        self.user_klass = get_user_model()

    def test_can_be_created_without_error(self):
        try:
            self.user_klass.objects.create_user("_", "a@a.com", "pa55w0rd")
        except Exception as e:
            self.fail("Unable to create user model: %s" % e)

    def test_extra_field_is_accessible(self):
        self.assertTrue("extra_field" in existing_user_fields(["extra_field"]))
        self.assertTrue(hasattr(self.user_klass(), "extra_field"))

    def test_profile_form_doesnt_expose_extra_field(self):
        form = ProfileForm(self.user_klass())
        expected_fields = set(["first_name", "last_name", "email"])
        self.assertTrue(expected_fields == set(form.fields))

    def test_migrate_alerts_to_user(self):
        user = UserFactory(email="a@a.com")
        ProductAlertFactory(email=user.email)
        user._migrate_alerts_to_user()
        assert user.alerts.count() == 1