如何在不访问数据库的情况下测试含ModelChoiceField的Django表单？

我之前也遇到过一模一样的情况——用ModelChoiceField的表单在调用form.is_valid()时总会触发数据库查询，想Mock却找不对路径。你之前尝试patchMyModel._default_manager.get_queryset没成功，核心原因是这个方法并不是表单验证时实际调用的对象。

问题根源 #

ModelChoiceField在初始化时就会获取指定的queryset（比如MyModel.objects.all()），而is_valid()过程中，to_python()方法调用的是这个已存在的queryset的get()方法，不是模型管理器的get_queryset()。所以你patch的方法根本没被触发到。

两种可行的解决方案 #

方案1：直接Mock模型管理器的get()方法

既然验证时最终会调用queryset.get()，而默认queryset来自MyModel.objects，那我们直接patchMyModel.objects.get即可：

from django.test import TestCase
from unittest.mock import patch, Mock
from myapp.forms import MyForm
from myapp.models import MyModel

class MyFormValidationTest(TestCase):
    def test_form_valid_without_db(self):
        # 创建一个模拟的MyModel实例，spec参数确保它符合模型的结构
        mock_mymodel = Mock(spec=MyModel)
        mock_mymodel.pk = 1

        # Patch MyModel.objects.get，让它返回我们的模拟对象
        with patch('myapp.models.MyModel.objects.get') as mock_get:
            mock_get.return_value = mock_mymodel

            # 提交表单数据（这里的'1'是模拟的主键值）
            form = MyForm({'my_model': '1'})
            
            # 验证表单有效
            self.assertTrue(form.is_valid())
            # 确认get方法被正确调用
            mock_get.assert_called_once_with(pk='1')

方案2：替换表单字段的queryset为Mock对象

如果你的表单里ModelChoiceField用了自定义的queryset（比如带过滤条件的），直接替换字段的queryset会更灵活：

def test_form_valid_custom_queryset(self):
        mock_mymodel = Mock(spec=MyModel)
        mock_mymodel.pk = 2

        # 创建一个模拟的QuerySet，让它的get方法返回模拟对象
        mock_queryset = Mock()
        mock_queryset.get.return_value = mock_mymodel

        # 初始化表单后，替换字段的queryset
        form = MyForm()
        form.fields['my_model'].queryset = mock_queryset

        # 提交数据并验证
        form.data = {'my_model': '2'}
        self.assertTrue(form.is_valid())
        mock_queryset.get.assert_called_once_with(pk='2')

注意事项 #

• 用spec=MyModel来创建模拟对象，能保证模拟对象拥有模型的必要属性，避免后续代码因属性缺失报错。
• 如果你的ModelChoiceField设置了to_field_name（不是默认的pk），记得在assert_called_once_with里改成对应的字段名，比如slug='test-slug'。

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