Django 自定义用户管理器单元测试 - NoneType 对象不可调用

Question

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

以下是完整代码。

models.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”函数，以下是我的代码：

test.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”对象不可调用。

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

一些在线研究表明，可能是因为Manager不是直接调用的设计。那是不是这种情况呢？如果是，如何处理单元测试？

- Jon Hrovat

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))

- Staccato