使用 Factory Boy SelfAttribute + relativedelta

Question

我在測試中使用 Factory Boy 並希望實現以下目標：

• 使first_period_end_date依賴於first_period_date並添加 12 個月。

我正在嘗試將SelfAttribute與relativedelta結合使用，但我目前應用它的方式不起作用。 我的代碼：

import datetime

import factory
from dateutil import relativedelta

from somewhere.models import Contract

class ContractFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Contract

    start_date = factory.LazyFunction(datetime.date.today)
    first_period_date = factory.SelfAttribute('start_date')
    first_period_end_date = (
        factory.SelfAttribute('first_period_date')
        + relativedelta.relativedelta(months=12)
    )

但是在運行時我收到以下錯誤：

TypeError: unsupported operand type(s) for +: 'SelfAttribute' and 'relativedelta'

所以這顯然不是它的完成方式。 但是我該怎么做呢？

Answer 1

答案是LazyAttribute ； SelfAttribute僅對復制字段有幫助。

你應該做：

class ContractFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Contract

    start_date = factory.LazyFunction(datetime.date.today)
    first_period_date = factory.SelfAttribute('start_date')
    first_period_end_date = factory.LazyAttribute(
        lambda self: self.first_period_date + relativedelta.relativedelta(months=12)
    )