Djang中的自定义用户测试

Question

我有一个Django应用程序，自定义用户想要测试它（进行单元测试）。我的自定义用户是emailuser，它由电子邮件和密码字段组成。我想建立这样的国家mplayer的东西，给它一个字段和一个密码。但我下面的代码不起作用。在

设置.py

AUTH_USER_MODEL = 'custom_user.EmailUser'

我的代码测试.py

^{pr2}$

Answer 1

好吧，试试这个：

from django.test import TestCase
from django.contrib.auth import get_user_model

User = get_user_model()

class CustomUserTestCase(TestCase):

    def test_create_user(self):
        # params = depends on your basemanager’s create_user methods.
        user = User.objects.create(**params)
        self.assertEqual(user.pk, user.id)

Answer 2

您应该使用create_user()方法，而不是仅仅使用create()。您还应该检查email字段，而不是user本身。在

class MyTest(TestCase):
    def setUp(self):
        self.user = settings.AUTH_USER_MODEL.objects.create_user(
                                              'testuser@test.pl', 'testpass')

    def test_user(self):
        self.assertEqual(self.user.email, 'testuser@test.pl')