如何使用 FactoryBoy 实例化我的 Django AbstractBaseUser 的子类

Question

我一直在尝试使用Factory Boy来简化单元测试的编写。但是，当我尝试实例化自定义用户类的实例时，库不太合作

自定义类继承自AbstractBaseUser和PermissionsMixin，按照此处输入链接描述的指南。

每次我都尝试像这样实例化这个类：

class LogoutTests(TestCase):
def setUp(self):
self.employee_password = "foo"
self.manager_password = "bar"
self.test_employee_user = EmployeeFactory.create(self.employee_password)
self.test_manager_user = ManagerFactory.create(self.manager_password)

但是，我不断收到此错误消息：

factory.errors.FactoryError: Cannot generate instances of abstract factory ManagerFactory; Ensure ManagerFactory.Meta.model is set and ManagerFactory.Meta.abstract is either not set or False.

我已经用其他模型对此进行了测试，只有我的自定义用户模型会生成此错误，因为它可能继承自AbstractBaseUser。对此有任何帮助吗？

作为参考，我的自定义用户类：

class Employee(AbstractBaseUser, PermissionsMixin):
class Meta:
abstract = False
first_name = models.CharField(max_length=250, default="Thibault", blank=False)
last_name = models.CharField(max_length=250, default="Dupont", blank=False)
email = models.EmailField(max_length=250, unique=True, blank=False, null=True)
employee_number = models.OneToOneField(EmployeeNumber)
is_manager = models.BooleanField(default=False, blank=False)
is_employed = models.BooleanField(default=True, blank=False)
is_active = models.BooleanField(default=False)
objects = EmployeeManager()
USERNAME_FIELD = 'email'
REQUIRED_FIELDS = []
def __str__(self):
return self.get_full_name()
def get_short_name(self):
return '{0}. {1}'.format(self.first_name[0], self.last_name)
def get_full_name(self):
return '{0} {1}'.format(self.first_name, self.last_name)
def email_user(self, subject, message, from_email=None, **kwargs):
send_mail(subject, message, from_email, [self.email], **kwargs

Answer 1

该消息说您尚未告知应由ManagerFactory创建哪个模型：未设置ManagerFactory.Meta.model。

工厂应如下所示：

import factory
import factory.django
from myapp import models
class ManagerFactory(factory.django.DjangoModelFactory):
class Meta:
model = models.Manager
# Your declarations here