在 Django 中的数据模型工厂的字典中使用 Factory Boy？ 或者，让 FactoryBoy 按需评估子工厂？

Question

我有以下模型工厂：

class ProjectDataModelFactory(DjangoModelFactory):
    project = factory.SubFactory("projects.factories.ProjectModelFactory")
    version = "1"
    content = {
        "id": {"value": fuzzy.FuzzyText(length=7, chars=string.ascii_letters).fuzz().upper(), "type": "string"},
        "win_probability": {"value": fuzzy.FuzzyInteger(0, 100).fuzz(), "type": "number"},
        "bid_value": {
            "value": fuzzy.FuzzyInteger(0, 10000000).fuzz(),
            "type": "number",
        },
        "status": {"value": random.choice(["Won", "Lost"]), "type": "string"},
        "sector": {
            "value": random.choice(
                ["Construction", "Technology", "Finance", "Consulting", "Agriculture", "Engineering"]
            ),
            "type": "string",
        },
    } 

在content["id"] ，我希望value project.external_id而不是fuzzy.FuzzyText(length=7, chars=string.ascii_letters).fuzz().upper(). external_id是ProjectModel.一部分ProjectModel. 但是当我把project.external_id放在project.external_id的位置时，我得到这个错误：

AttributeError: 'SubFactory' 对象没有属性 'external_id

这似乎只是字典的问题。 我们在其他情况下使用它没有问题。

我没有很多使用 FactoryBoy 的经验，所以它很容易成为微不足道的事情。 有任何想法吗？

Answer 1

您有第一个问题：在导入类时评估您放入content字典的所有对random或.fuzz()的调用：每个实例将使用完全相同的值。

为了执行延迟评估，您必须将它们包装在一些factory_boy声明中：

class ProjectDataModelFactory(DjangoModelFactory):
    project = factory.SubFactory("projects.factories.ProjectModelFactory")
    version = "1"
    content = factory.Dict(
        id=factory.Dict(
            value=fuzzy.FuzzyText(length=7, chars=string.ascii_uppercase),
            type="string",
        ),
        win_probability=factory.Dict(
            value=fuzzy.FuzzyInteger(0, 100),
            type="number",
        ),
        bid_value=factory.Dict(
            value=fuzzy.FuzzyInteger(0, 10_000_000),
            type="number",
        ),
        status=factory.Dict(
            value=fuzzy.FuzzyChoice(["Won", "Lost"]),
            type="string",
        ),
        sector=factory.Dict(
            value=fuzzy.FuzzyChoice(
                ["Construction", "Technology", "Finance", "Consulting", "Agriculture", "Engineering"],
            ),
            type="string",
        ),
    )

factory.Dict(**kwargs)基本上是factory.SubFactory(DictFactory, **kwargs)的别名； 然后，您可以像使用简单的SubFactory一样使用任何声明。

对于最初的目标，您现在可以轻松使用所有功能：

class ProjectDataModelFactory(DjangoModelFactory):
    project = factory.SubFactory("projects.factories.ProjectModelFactory")
    version = "1"
    content = factory.Dict(
        id=factory.Dict(
            # . is the {"value": x, "type": "string"} dict;
            # .. is {"id": {"value": x, ...}} dict;
            # ... is the ProjectData(content={"id": {...}}) object.
            value=factory.SelfAttribute('...project.external_id'),
            type="string",
        ),
        # Other fields here
    )