python - Django 中的自定义用户测试

Question

我有一个 Django 应用程序和自定义用户 - 想要测试它(进行单元测试)。我的自定义用户是 emailuser，它由电子邮件和密码字段组成。我想设置一些这样的国家级 mplayer，给它一个字段和一个密码。但是我下面的代码不起作用。

settings.py

AUTH_USER_MODEL = 'custom_user.EmailUser'

我在 test.py 中的代码

from django.test import TestCase
from django.conf import settings

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

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

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)