Я пишу тесты для большого приложения 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 возвращает 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 ниже представлен как SubFactory на ProfileFactory выше.
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])
Опять же, большинство утверждений впроходить тест без сбоев все отдельные фабрики могут инициализировать значения выборки из их непосредственных отношений с иностранными ключами.Однако, как только фабрики / модели будут удалены друг от друга на три шага, вызов вернет 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)