时间戳

首页 > 解决方案 > factory_boy：为一个相关工厂传递模型实例

django - factory_boy：为一个相关工厂传递模型实例

问题描述

我认为与其将数据以 的形式传递给工厂的 RelatedFactory，不如ATTR__SUBATTR直接传递已经存在的实例。除非我遗漏了一些非常明显的东西，否则这似乎不起作用。看一看：

class Owner(models.Model):
    name = models.CharField()

class Item(models.Model):
    name = models.CharField()
    owner = models.ForeignKey(Owner, null = True, related_name = 'items')

class ItemFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Item

class OwnerFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Owner
    items = factory.RelatedFactory(ItemFactory, 'owner')

item = Item.objects.create(name='Foo')
alice = OwnerFactory(name='Alice', items__name='Bar')
alice.items.all()
<QuerySet [<Item: Bar>]>
bob = OwnerFactory(name='Bob', items=item) # or items = [item] doesn't matter
bob.items.all()
<QuerySet []>

一直在努力让我的工厂变得漂亮和干燥，并遇到了这个障碍。写了我自己的改编版RelatedFactory，允许一次处理多个值，如果您在该过程中创建新对象，这可以正常工作 - 但如果您正在使用已经存在的对象，则不会。

有效的例子：OwnerFactory(items__name=['Foo','Bar'])=> Foo and Bar in owner.items.
不起作用的示例：OwnerFactory(items=[foo,bar])=> owner.items is empty
请注意，我在顶部的大示例中使用了默认的 RelatedFactory。

我一整天都在翻阅 factory_boy 的文档，但找不到解决方案，现在狭隘的视野已经接管了，禁止任何新的见解。

标签： djangofactory-boy

解决方案

您正在寻找http://factoryboy.readthedocs.io/en/latest/recipes.html#simple-many-to-many-relationship

class ItemFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Item

class OwnerFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Owner

    @factory.post_generation
    def items(self, create, extracted, **kwargs):
        if not create:
            # Simple build, do nothing.
            return

        if extracted:
            # A list of items were passed in, use them
            for item in extracted:
                self.items.add(item)

推荐阅读