如何在Factory-boy中实现无数据库的查询操作用于测试？

实现思路 #

• 给Factory类添加类级别的内存存储，记录所有通过Factory创建的实例
• 重写Factory的_create方法，跳过数据库保存步骤，直接将实例存入内存
• 自定义管理器类，实现all()等检索方法，从内存存储中返回模拟的Django QuerySet，保持与原生ORM接口的一致性

完整代码实现 #

修改StudentFactory类 #

import factory
from django.db.models.query import QuerySet
from .models import Student

class StudentFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Student

    # 类级别的内存容器，保存所有创建的实例
    _instances = []

    @classmethod
    def _create(cls, model_class, *args, **kwargs):
        # 直接创建模型实例，不调用save()避免数据库操作
        instance = model_class(*args, **kwargs)
        # 将实例加入内存存储
        cls._instances.append(instance)
        return instance

    # 自定义内存检索管理器
    class MemoryManager:
        @classmethod
        def all(cls):
            # 将内存实例包装为Django QuerySet，兼容原有业务代码的调用方式
            queryset = QuerySet(model=Student)
            queryset._result_cache = StudentFactory._instances.copy()
            queryset._prefetch_done = True
            return queryset

        # 可选：扩展filter方法实现简单的内存过滤逻辑
        @classmethod
        def filter(cls, **kwargs):
            filtered_instances = [
                inst for inst in StudentFactory._instances
                if all(getattr(inst, k) == v for k, v in kwargs.items())
            ]
            queryset = QuerySet(model=Student)
            queryset._result_cache = filtered_instances
            queryset._prefetch_done = True
            return queryset

    # 将管理器挂载到Factory的objects属性
    objects = MemoryManager

使用示例 #

# 创建内存中的学生实例
StudentFactory(name="Alice")
StudentFactory(name="Bob")
StudentFactory(name="Charlie")

# 获取所有内存实例
all_students = StudentFactory.objects.all()
print([s.name for s in all_students])  # 输出: ['Alice', 'Bob', 'Charlie']

# 过滤指定实例
bob = StudentFactory.objects.filter(name="Bob").first()
print(bob.name)  # 输出: Bob

注意事项 #

• 该方案完全脱离数据库，所有实例仅存在于内存中，测试结束后会随进程销毁
• 模拟的QuerySet仅实现了基础的all()和filter()，如果需要更多QuerySet方法（如exclude、order_by），可以在MemoryManager中自行扩展
• 如果需要在测试中切换数据库/内存模式，可以给Factory添加开关控制逻辑：

# 新增类属性控制是否使用数据库
use_database = False

@classmethod
def _create(cls, model_class, *args, **kwargs):
    instance = model_class(*args, **kwargs)
    if cls.use_database:
        instance.save()
    else:
        cls._instances.append(instance)
    return instance

内容的提问来源于stack exchange，提问作者Omar Sarmiento Rolo