在 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"]，我希望用valuebeproject.external_id代替fuzzy.FuzzyText(length=7, chars=string.ascii_letters).fuzz().upper(). external_idis 的一部分，ProjectModel.但是当我放入project.external_id模糊器的位置时，出现此错误：

AttributeError：“SubFactory”对象没有属性“external_id”

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

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

Answer 1

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

为了执行惰性求值，您必须将它们包装在一些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
    )