python - Django Cusom 用户管理器单元测试 - NoneType 对象不可调用

Question

这一切都始于我下载coverage.py 并开始测试我的django 项目。一切都很好，但是我在 Models.py 中设置的自定义用户管理器缺乏覆盖

这是完整的代码。

模型.py

from django.contrib.auth.models import AbstractUser, BaseUserManager
from django.db import models
from django.utils.translation import ugettext_lazy as _

class UserManager(BaseUserManager):
    """Define a model manager for User model with no username field."""

    use_in_migrations = True

    def _create_user(self, email, password, **extra_fields):
        """Create and save a User with the given email and password."""
        if not email:
            raise ValueError('The given email must be set')
        email = self.normalize_email(email)

        user = self.model(email=email, **extra_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_user(self, email, password=None, **extra_fields):
        """Create and save a regular User with the given email and password."""
        extra_fields.setdefault('is_staff', False)
        extra_fields.setdefault('is_superuser', False)
        return self._create_user(email, password, **extra_fields)

    def create_superuser(self, email, password, **extra_fields):
        """Create and save a SuperUser with the given email and password."""
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)

        if extra_fields.get('is_staff') is not True:
            raise ValueError('Superuser must have is_staff=True.')
        if extra_fields.get('is_superuser') is not True:
            raise ValueError('Superuser must have is_superuser=True.')

        return self._create_user(email, password, **extra_fields)


class CustomUser(AbstractUser):
    first_name = models.CharField(max_length=200)
    last_name = models.CharField(max_length=200)
    # profile_image = models.ImageField(upload_to="users/profile_pic", blank=True)
    is_pastor = models.BooleanField(default=True)
    home_church = models.ForeignKey(
        'church.Church',
        on_delete="CASCADE",
        blank=True,
        null=True)

    username = None
    email = models.EmailField(_('email address'), unique=True)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = []

    objects = UserManager() ## This is the new line in the User model. ##

我正在尝试在 UserManager 中测试“_create_user”函​​数。这是我的代码

测试.py

class UserManagerTestCase(TestCase):

def test_main_create_user(self):
    manager = UserManager()
    user = manager._create_user('[e-mail]', 'password123')
    self.assertTrue(isinstance(user, User))

不幸的是，当我调用“_create_user”函​​数时，我得到一个类型错误：“NoneType”对象不可调用。

这很奇怪，因为模型管理器在创建用户时按预期工作，并且我通过打印语句确认 email 和 **extra_fields 参数都不是 None。

网上的一些研究说这可能是因为 Manager 不是为了直接调用而设计的。是这样吗？如果是这样，你应该如何处理单元测试？

标签： pythondjango

Answer 1

使用值进行实例化CustomUser()，以便可以使用它来访问要测试的方法。 UserManager不能直接使用，只能通过为其创建的模型访问：CustomUser。

#tests.py
class UserManagerTestCase(TestCase):
    def test_main_create_user(self):
        user = CustomUser.objects.create_user('[e-mail]', 'password123')
        self.assertTrue(isinstance(user, User))