python - 避免与 factory_boy 工厂重复

Question

我正在使用 factory_boy创建测试装置。我有两个简单的工厂，由 SQLAlchemy 模型支持(在下面进行了简化)。

我希望能够多次调用 AddressFactory.create()，并让它创建一个 Country(如果它尚不存在)，否则我希望它重新使用现有记录。

class CountryFactory(factory.Factory):
    FACTORY_FOR = Country

    cc = "US"
    name = "United States"


class AddressFactory(factory.Factory):
    FACTORY_FOR = Address

    name = "Joe User"
    city = "Seven Mile Beach"
    country = factory.SubFactory(CountryFactory, cc="KY", name="Cayman Islands")

我的问题是:如何设置这些工厂，使 factory_boy 不会在每次创建地址时都尝试创建一个新的国家/地区？

Answer 1

在最新的factory-boy==2.3.1中可以添加FACTORY_DJANGO_GET_OR_CREATE

class CountryFactory(factory.django.DjangoModelFactory):
    FACTORY_FOR = 'appname.Country'
    FACTORY_DJANGO_GET_OR_CREATE = ('cc',)

    cc = "US"
    name = "United States"

假设 cc 字段是唯一标识符。