使用没有“决定者”model 字段的 FactoryBoy 的 Maybe 声明

Question

是否可以使用FactoryBoy.Maybe ，其中“决定者”没有 map 到实际的 model 字段？

考虑以下 model：

class MyModel(models.Model):
  is_heavy = models.BooleanField()
  weight = models.PositiveIntegerField()

这家工厂工作：

class MyModelFactory(factory.django.DjangoModelFactory):
  class Meta:
    model = MyModel

  is_heavy = factory.Faker("boolean", chance_of_getting_true=50)
  weight = factory.Maybe(
    "is_heavy",
    yes_declaration=factory.Faker("pyint", min_value=0, max_value="10"),
    no_declaration=factory.Faker("pyint", min_value=10, max_value="20"),
  )

但是，如果我不希望 map 的“决定者”到MyModel的实际字段怎么办？ 就像是：

class MyModelFactory(factory.django.DjangoModelFactory):
  class Meta:
    model = MyModel

  is_heavy = factory.Faker("boolean", chance_of_getting_true=50)
  weight = factory.Maybe(
    factory.Faker("boolean", chance_of_getting_true=50),
    yes_declaration=factory.Faker("pyint", min_value=0, max_value="10"),
    no_declaration=factory.Faker("pyint", min_value=10, max_value="20"),
  )

那个工厂不行。

（请注意，它曾经在此版本和此 PR之前工作。）

---

编辑

尽管@Xelnor 的建议有效，但我实际上正在编写一个单独的 function 以像这样使用：

def optional_declaration(declaration, chance=50, default=None):

  return factory.Maybe(
    factory.Faker("boolean", chance_of_getting_true=chance),
    yes_declaration=declaration,
    no_declaration=default,
  )

class MyModelFactory(factory.django.DjangoModelFactory):
  class Meta:
    model = MyModel

  is_heavy = factory.Faker("boolean", chance_of_getting_true=50)
  weight = optional_declaration(
    factory.Faker("pyint", min_value=0, max_value="10"),
    chance=50,
    default=factory.Faker("pyint", min_value=10, max_value="20"),
  )

所以我无法访问工厂的Params 。

Answer 1

最简单的方法是使用class Params ：

class MyModelFactory(factory.django.DjangoModelFactory):
  class Meta:
    model = MyModel

  class Params:
    is_heavy = factory.Faker("boolean", chance_of_getting_true=50)

  weight = factory.Maybe(
    "is_heavy",
    ...
  )

这样，用户也可以调用MyModelFactory(is_heavy=True) ，但is_heavy永远不会传递给底层 class。

注意：关于您发现的“添加factory.Faker作为决策者不起作用”的问题，这看起来像一个错误； 您能否将其作为 github 上的问题提交给项目？

Answer 2

哦。 它永远不会失败——当我在 StackOverflow 上提出一个问题时，我想出了一个解决方案。

这应该可以解决问题：

from functools import partial
from faker import Faker

fake = Faker()

class MyModelFactory(factory.django.DjangoModelFactory):
  class Meta:
    model = MyModel

  is_heavy = factory.Faker("boolean", chance_of_getting_true=50)
  weight = factory.Maybe(
    factory.LazyFunction(partial(fake.boolean, chance_of_getting_true=50)),
    yes_declaration=factory.Faker("pyint", min_value=0, max_value="10"),
    no_declaration=factory.Faker("pyint", min_value=10, max_value="20"),
  )

（如果您将变量传递给充当“决策者”的伪造者，则只需要使用partial ）