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

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

编辑

根据要求，我已包含以下购买模型的代码:

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)