使用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")

我的问题是：我如何建立这些工厂，使工厂男孩不会在每次创建地址时都试图创建一个新的国家？

Answer 1

虽然您对基于SQLAlchemy的工厂没有get_or_create函数是正确的，但是如果您想用作外键的对象已经存在，您可以迭代它们：

http://factoryboy.readthedocs.org/en/latest/recipes.html#choosing-from-a-populated-table

因此，可以想象的是，您可以通过使用lazy属性在您的工厂中构建一个解决方案，该属性首先检查数据库中是否存在该对象，如果确实存在，它将使用此方法迭代这些对象，但如果对象不存在，它将首先调用SubFactory来创建对象。在