工厂男孩和相关对象的创建

Question

假设你有这些Django模型的关联方式：

class Service:
   restaurant = models.ForeignKey(Restaurant)
   hist_day_period = models.ForeignKey(DayPeriod)

class DayPeriod:
   restaurant = models.ForeignKey(Restaurant)

我想使用工厂创建一个Service对象。它应该创建所有3个模型，但使用相同的餐厅。在

使用此代码：

^{pr2}$

工厂男孩将创建两个不同的餐厅：

s1 = ServiceFactory.create()
s1.restaurant == s1.hist_day_period.restaurant
>>> False

有什么办法吗？我不清楚是否应该使用related factors而不是SubFactory来完成这一任务。在

Answer 1

与使用SubFactory不同，您可以将Iterator与查询集一起使用。在

例如：

class ServiceFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Service

    restaurant = factory.Iterator(models.Restaurant.object.all())

    hist_day_period = factory.SubFactory(
        'day_periods.factories.DayPeriodFactory', restaurant=restaurant)

不过，我不确定创建Restaurant的最佳位置。在我们的代码库中，我重写DiscoverRunner并在那里创建用于此目的的模型（在setup_databases中，这感觉像一个黑客）。在