Question

我试图用factory_boy为这些模型建立工厂：

class Course(models.Model):
    group = models.OneToOneField(Group)
    title = models.CharField(max_length=1024)


class CourseInfo(models.Model):
    course = models.OneToOneField(Course, related_name='info')
    starttime = models.TimeField(null=True, blank=True)
    endtime = models.TimeField(null=True, blank=True)

这是我的工厂：

class CourseInfoFactory(factory.DjangoModelFactory):
    class Meta:
        model = CourseInfo

    course = factory.SubFactory('courseaffils.tests.factories.CourseFactory')
    starttime = time(hour=4)
    endtime = time(hour=6)


class CourseFactory(factory.DjangoModelFactory):
    class Meta:
        model = Course

    title = factory.Sequence(lambda n: 'Example Course %d' % n)
    group = factory.SubFactory(GroupFactory)
    info = factory.RelatedFactory(CourseInfoFactory, 'course')

现在，当我致电CourseFactory()或CourseInfoFactory()时，我收到了IntegrityError：

IntegrityError：UNIQUE约束失败：courseaffils_courseinfo.course_id

我的工厂出了什么问题？我该如何解决这个问题？

如果我在CourseInfoFactory上执行course = factory.RelatedFactory('courseaffils.tests.factories.CourseFactory')，我会收到此错误，如预期的那样：

IntegrityError：NOT NULL约束失败：courseaffils_courseinfo.course_id

如果您需要更多详细信息，请参阅我正在使用的实际代码：

https://github.com/ccnmtl/django_courseaffils/blob/master/courseaffils/models.py

Answer 1

只需在 CourseInfoFactory 的Meta类中添加 django_get_or_create ：

class Meta:
    django_get_or_create = ('group',)

此解决方案可以避免组字段中的唯一约束问题获取对象，而不是在存在时创建它。

您可以在此处查看：https://github.com/FactoryBoy/factory_boy/blob/v2.5.2/factory/django.py#L133-L139

Answer 2

问题是CourseFactory()已经创建了一个CourseInfo对象。我已经通过删除info = factory.RelatedFactory(CourseInfoFactory, 'course')行解决了这个问题，现在不需要使用CourseInfoFactory - 我可以通过在course.info获取创建的对象来测试CourseInfo。

Django：带有factory_boy的一对一字段：UNIQUE约束失败

2 个答案: