用于单元测试的Django工厂：“不能分配”'1'“：”User.group“必须是”品牌“实例。”

Question

当我为user_profile / model.py运行单元测试时出现错误：“无法分配”'1'“：”User.group“必须是”品牌“实例。”

我相信这个错误是因为，在我的单元测试工厂（factories.py）中，当我创建UserFactory（）的实例时，我没有正确地分配组。我的小组应该映射到Brand类/工厂，但我不知道如何做到这一点..

这是我的user_profile/model.py：

class Brand(models.Model):

    name = models.CharField(max_length=20)

    def __unicode__(self):
        return self.name


class User(AbstractBaseUser, PermissionsMixin):

    username = models.CharField(
        'username',
        max_length=50,
        unique=True,
        db_index=True
    )
    email = models.EmailField('email address', unique=True)
    group = models.ForeignKey(Brand, null=True, blank=True)
    is_active = models.BooleanField(default=True)
    is_admin = models.BooleanField(default=False)
    is_staff = models.BooleanField(default=False)
    is_approved = models.BooleanField(default=True)

在我的user_profile/test/factories.py：

from user_profile.models import User, Brand

class UserFactory(factory.django.DjangoModelFactory):

    class Meta:

        model = User
        django_get_or_create = (
            'username',
            'email',
            'password',
            'is_approved',
            'is_active',
            'is_staff',
            'is_admin',
            'group'
        )


class BrandFactory(factory.django.DjangoModelFactory):

    class Meta:

        model = Brand
        django_get_or_create = ('name',)

BrandFactory(name='BRAND1')
BrandFactory(name='BRAND2')
BrandFactory(name='BRAND3')
BrandFactory(name='BRAND4')


UserFactory(
    username='[email protected]',
    email='[email protected]',
    password=12345,
    is_approved=True,
    is_active=True,
    is_staff=True,
    is_admin=True,
    group="1" <--- HOW DO I MAP THIS TO THE 'BRAND1' FACTORY/CLASS?!
)

UserFactory(
    username='[email protected]',
    email='[email protected]',
    password=12345,
    is_approved=True,
    is_active=True,
    is_staff=True,
    is_admin=True,
    group="2"  <--- HOW DO I MAP THIS TO THE 'BRAND2' FACTORY/CLASS?!
)

Answer 1

您必须将BrandFactory的返回值分配给某个东西。例如

b1 = BrandFactory(name='BRAND1')
b2 = BrandFactory(name='BRAND2')

然后在构造函数中使用对模型实例的引用

UserFactory(
    username='[email protected]',
    email='[email protected]',
    password=12345,
    is_approved=True,
    is_active=True,
    is_staff=True,
    is_admin=True,
    group=b1
)

UserFactory(
    username='[email protected]',
    email='[email protected]',
    password=12345,
    is_approved=True,
    is_active=True,
    is_staff=True,
    is_admin=True,
    group=b2
)