FactoryBoy - 嵌套工厂/最大深度？

Question

我正在为大型 Django 应用程序编写测试，作为此过程的一部分，我正在逐步为 Django 项目中不同应用程序的所有模型创建工厂。

但是，我运行对 FactoryBoy 的一些令人困惑的行为感到困惑，似乎 SubFactories 有一个最大深度，超过该深度就不会生成任何实例。

当我尝试运行以下测试时出现错误：

    def test_subfactories(self):
        """ Verify that the factory is able to initialize """
        user = UserFactory()
        self.assertTrue(user)
        self.assertTrue(user.profile)
        self.assertTrue(user.profile.tenant)

        order = OrderFactory()
        self.assertTrue(order)
        self.assertTrue(order.user.profile.tenant)

最后一行将失败 (AssertionError: None is not true)，运行通过调试器进行此测试表明确实 order.user.profile.tenant returns None 而不是预期的 Tenant 实例。

这里涉及的工厂/模型比较多，但是布局比较简单。

User（django 默认值）和 Profile 模型通过 OneToOneField 链接，后者（经过一些麻烦）由 UserFactory 和 ProfileFactory 表示

@factory.django.mute_signals(post_save)
class ProfileFactory(factory.django.DjangoModelFactory):

    class Meta:
        model = yuza_models.Profile
        django_get_or_create = ('user',)

    user = factory.SubFactory('yuza.factories.UserFactory')
    birth_date = factory.Faker('date_of_birth')
    street = factory.Faker('street_name')
    house_number = factory.Faker('building_number')
    city = factory.Faker('city')
    country = factory.Faker('country')
    avatar_file = factory.django.ImageField(color='blue')
    tenant = factory.SubFactory(TenantFactory)

@factory.django.mute_signals(post_save)
class UserFactory(factory.django.DjangoModelFactory):

    class Meta:
        model = auth_models.User

    username = factory.Sequence(lambda n: "user_%d" % n)
    first_name = factory.Faker('first_name')
    last_name = factory.Faker('last_name')

    email = factory.Faker('email')
    is_staff = False
    is_superuser = False
    is_active = True
    last_login = factory.LazyFunction(timezone.now)

    @factory.post_generation
    def profile(self, create, extracted):
        if not create:
            return
        if extracted is None:
            ProfileFactory(user=self)

下面的TenantFactory表示为上面ProfileFactory上的SubFactory。

class TenantFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = elearning_models.Tenant

    name = factory.Faker('company')
    slug = factory.LazyAttribute(lambda obj: text.slugify(obj.name))
    name_manager = factory.Faker('name')
    title_manager = factory.Faker('job')
    street = factory.Faker('street_name')
    house_number = factory.Faker('building_number')
    house_number_addition = factory.Faker('secondary_address')

Order 链接到 User，但它的许多方法调用其 self.user.profile.tenant

的字段

class OrderFactory(factory.DjangoModelFactory):
    class Meta:
        model = Order

    user = factory.SubFactory(UserFactory)
    order_date = factory.LazyFunction(timezone.now)
    price = factory.LazyFunction(lambda: Decimal(random.uniform(1, 100)))
    site_tenant = factory.SubFactory(TenantFactory)
    no_tax = fuzzy.FuzzyChoice([True, False])

同样，测试中的大多数断言都通过了，没有失败，所有单独的工厂都能够从它们的直接外键关系中初始化获取值。但是，只要 factories/models 彼此相隔三个步骤，调用就会 return None 而不是预期的 Tenant 实例。

由于我无法在 FactoryBoy 文档中找到任何对此行为的引用，这可能是我这边的错误，但到目前为止我无法确定其来源。有谁知道我做错了什么？

post_save方法

def create_user_profile(sender, instance, created, **kwargs):
    if created:
        profile = Profile.objects.create(user=instance)
        resume = profile.get_resume()
        resume.initialize()


post_save.connect(create_user_profile, sender=User)

Answer 1

正如我在评论中提到的，我发现了问题的根源：链接到 UserProfile 的 post-save 方法（我已将代码包含在我的 post).

此 post-save 方法在 User 创建时创建了一个 Profile。我通过在 UserFactory 和 ProfileFactory 上使用 @factory.django.mute_signals 装饰器来解释这个信号。

我假设对 Order.user 的任何调用都会触发 UserFactory ，它已经包含在装饰器中，但这并不是假设被证明是错误的。只有当我将装饰也应用到 OrderFactory 时，测试才通过。

因此 @factory.django.mute_signals 装饰器不应该只用在受这些信号影响的工厂上，也应该用在任何将这些工厂用作 SubFactory!

的工厂上

FactoryBoy - 嵌套工厂/最大深度？

FactoryBoy - nested factories / max depth?

python

django

factory-boy