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))

如您所见，我已经用date覆盖了PurchaseFactory上的lazy_attribute字段。但是，在此特定测试中，我尝试自己设置日期。我以为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将使用timezone.now()中的值来override any provided value for date。

要选择退出该行为，请使用date = models.DateTimeField(default=timezone.now, editable=False)（或易于理解的default=lambda: timezone.now()）：默认为使用timezone.now()，但可让您提供自定义值

NB：设置editable=False保留了默认的Django行为，即防止用户编辑字段，这通常是在插入时自动设置该值所隐含的。

顺便说一句，您不需要在if not self.date定义中检查@lazy_attribute，FactoryBoy将自动使用提供的值。这使您可以使用以下简短说明：

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

Django / FactoryBoy-覆盖lazy_attributes

1 个答案: