尽管FACTORY_DJANGO_GET_OR_CREATE,为什么FactoryBoy会从SubFactory创建一个新对象

2024年1月2日 233次阅读

我刚刚开始在Django上使用
factory boy.它有一个设置FACTORY_DJANGO_GET_OR_CREATE,这意味着它不会创建一个新对象(如果已存在).但是当我要求现有的具有现有SubFactory对象的对象时,尽管有此设置,它仍会创建一个未使用的对象.

例如,在一个全新的项目中,我试过：

# models.py
from django.db import models

class A(models.Model):
    name = models.CharField(max_length=10)

class B(models.Model):
    name = models.CharField(max_length=10)
    a = models.ForeignKey(A)

和

# factories.py
import factory

from . import models

class AFactory(factory.DjangoModelFactory):
    FACTORY_FOR = models.A
    FACTORY_DJANGO_GET_OR_CREATE = ('name',)

    name = factory.Sequence(lambda n: 'A-{0}'.format(n))

class BFactory(factory.DjangoModelFactory):
    FACTORY_FOR = models.B
    FACTORY_DJANGO_GET_OR_CREATE = ('name',)

    name = factory.Sequence(lambda n: 'B-{0}'.format(n))
    a = factory.SubFactory(AFactory)

现在：

from factories import *

a = AFactory(name="Apple")
models.A.objects.all()
# one object
b = BFactory(a__name="Apple", name="Beetle")
models.B.objects.all()
models.A.objects.all()
# one A object, one B object
b = BFactory(name="Beetle")
models.B.objects.all()
models.A.objects.all()
# still one B object, but now a new, unused A object too

然后对BFactory的最终调用已经成为A类的新对象,即使名为Beetle的B对象已经存在(并且不会被重新创建).为什么,以及如何停止创建这个新的A对象？

(我知道我可以打电话来解决这个问题：

b = BFactory(name="Beetle", a__name="Apple")

但在我的实际使用案例中,我有几个依赖关系和层次结构级别,以这种方式提供额外的冗余参数很麻烦 – 我似乎无法获得正确的参数组合.)

谢谢！

最佳答案 当我还有一个具有B外键的C类时,上面是一个问题.当我创建一个新的CFactory对象时：

c = CFactory(name="Cat", b__name="Beetle")

它还会创建一个新的未使用的A对象.

我发现这个解决方案很简单：相反,像这样制作新的CFactory对象：

c = CFactory(name="Cat", b=b)

其中b是B对象或BFactory对象.

点赞