如何使用Faker避免生成重复的Django Month对象？

问题场景 #

我定义了如下MonthFactory工厂类：

class MonthFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Month

    year = factory.Faker("random_int", min=2014, max=2022)
    month = factory.Faker("random_int", min=1, max=12)

Django的Month模型中，year和month设置了联合唯一约束：

class Meta:
    unique_together = (
        "month",
        "year",
    )

当执行以下代码创建测试对象时：

month1 = MonthFactory()
month2 = MonthFactory()

偶尔会触发错误：

django.db.utils.IntegrityError: duplicate key value violates unique constraint

原因是Faker随机生成了相同的年月组合，请问如何避免这种情况？

解决方案 #

方法1：使用factory.Sequence生成有序的年月组合 #

如果测试不需要完全随机的年月，可以用序列生成绝对唯一的组合，比如从起始年份开始按月份递增：

class MonthFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Month

    @factory.Sequence
    def year(n):
        start_year = 2014
        return start_year + (n // 12)

    @factory.Sequence
    def month(n):
        return (n % 12) + 1

生成顺序为2014-1、2014-2……2014-12、2015-1，不会出现重复。

方法2：自定义生成器确保唯一随机组合 #

如果需要保留随机特性，可以自定义逻辑，生成前检查已存在的组合，确保不重复：

import random
from django.db.models import Q

class MonthFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Month

    @factory.lazy_attribute
    def year(obj):
        # 获取所有已存在的年月组合
        existing = set(Month.objects.values_list('year', 'month'))
        possible_years = range(2014, 2023)
        possible_months = range(1, 13)
        
        # 循环生成直到找到未使用的组合
        while True:
            year = random.choice(possible_years)
            month = random.choice(possible_months)
            if (year, month) not in existing:
                obj.month = month  # 绑定生成的月份，避免重复生成
                return year

该方法在生成年份时同步确定唯一的月份组合，兼顾随机性与唯一性。

方法3：添加重试机制处理重复 #

保留完全随机特性，遇到唯一约束错误时自动重试生成：

from django.db import IntegrityError

class MonthFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Month

    year = factory.Faker("random_int", min=2014, max=2022)
    month = factory.Faker("random_int", min=1, max=12)

    @classmethod
    def create(cls, **kwargs):
        max_retries = 5
        for _ in range(max_retries):
            try:
                return super().create(**kwargs)
            except IntegrityError:
                continue
        # 多次重试失败后抛出错误
        raise IntegrityError("Failed to create unique Month after multiple retries")

设置合适的重试次数，可覆盖绝大多数重复场景，适合对随机性要求高的测试。

内容的提问来源于stack exchange，提问作者user2712231