Django / FactoryBoy - 覆盖 lazy_attributes

Question

我正在开发一个 Django 项目，为此我创建了以下工厂：

class PurchaseFactory(factory.DjangoModelFactory):
    class Meta:
        model = 'core.Purchase'

    amount = decimal.Decimal(random.randrange(100, 100000) / 100)
    ( .... )
    user = factory.SubFactory(UserFactory)

    @factory.lazy_attribute
    def date(self):
        if not self.date:
            return timezone.now()
        else:
            return self.date

为此我运行进行了以下测试：

class TestGetBalanceForPeriod(TestCase):
    def setUp(self) -> None:
        self.user: User = UserFactory()
        self.purchase_1: Purchase = PurchaseFactory(
            user=self.user, date=timezone.datetime(2019, 11, 1), amount=10
        )

    def test_test_setup(self) -> None:
        self.assertEqual(self.purchase_1.date, timezone.datetime(2019, 11, 1))

如您所见，我用 lazy_attribute 覆盖了 PurchaseFactory 上的 date 字段。但是，在这个特定的测试中，我试图自己设置日期。我想象 factoryboy 会用用户提供的值覆盖所有值，但测试失败并出现以下错误：

AssertionError: datetime.datetime(2019, 11, 22, 16, 15, 56, 311882, tzinfo=<UTC>) != datetime.datetime(2019, 11, 1, 0, 0)

第一个日期是 timezone.now() 调用的结果，而不是我作为输入提供的日期。到目前为止，我已经能够毫无问题地覆盖我们项目中许多工厂的所有值 - 但我无法解决这个问题。

有谁知道我做错了什么吗？

编辑

根据要求，我在下面包含了 Purchase 模型的代码：

class Purchase(models.Model):
    amount = models.DecimalField(default=0, decimal_places=2, max_digits=8)
    ( .... )
    date = models.DateTimeField(auto_now_add=True)
    user = models.ForeignKey(
        User,
        related_name='purchases',
        on_delete=models.CASCADE,
    )

Answer 1

问题来自在您的 Django 模型定义中使用 auto_now_add=True：在这种情况下，Django 将 override any provided value for date 使用 timezone.now().

中的值

为了选择退出该行为，请使用 date = models.DateTimeField(default=timezone.now, editable=False)（或 default=lambda: timezone.now()，更容易理解）：它将默认使用 timezone.now()，但使您能够提供自定义值。

注意：设置 editable=False 会保留防止用户编辑字段的默认 Django 行为，这通常通过在插入时自动设置值来暗示。

顺便说一下，您不需要在 @lazy_attribute 定义中检查 if not self.date，FactoryBoy 会自动使用提供的值。这允许您使用这个更短的描述：

class PurchaseFactory(factory.django.DjangoModelFactory):
    ...
    date = factory.LazyFunction(timezone.now)

Django / FactoryBoy - 覆盖 lazy_attributes

Django / FactoryBoy - Overriding lazy_attributes

python

django

factory-boy