如何使用Mock在调用Django模型create_user时抛出异常？

你的mock失败核心原因是patch路径未匹配视图代码中实际调用的create_user对象，Django模型的objects是管理器的实例，需精准定位patch目标。以下是两种可行的实现方案：

方法一：直接patch视图中引用的CustomUser.objects.create_user #

假设视图位于myapp/views.py，模型位于myapp/models.py，直接patch视图代码里实际调用的create_user方法：

from django.test import TestCase
from django.urls import reverse
from unittest.mock import patch
from myapp.views import YourFormView

class TestYourFormView(TestCase):
    def test_form_valid_handles_create_user_exception(self):
        # 构造测试表单数据
        form_data = {
            'username': 'testuser',
            'email': 'test@example.com',
            'password': 'testpass123'
            # 补充你的表单其他必填字段
        }
        
        # 关键：patch视图中实际调用的CustomUser.objects.create_user
        with patch('myapp.views.CustomUser.objects.create_user') as mock_create_user:
            # 设置要触发的异常（比如数据库唯一约束冲突的IntegrityError）
            mock_create_user.side_effect = Exception("创建用户失败")
            
            # 模拟POST请求提交表单
            response = self.client.post(reverse('your-view-url-name'), data=form_data)
            
            # 验证异常被正确捕获：根据你的业务逻辑调整断言
            # 示例：假设捕获异常后返回表单页面并显示错误
            self.assertEqual(response.status_code, 200)
            self.assertFormError(response, 'form', None, '创建用户失败')
            
            # 验证create_user确实被调用
            mock_create_user.assert_called_once_with(
                username='testuser',
                email='test@example.com',
                password='testpass123'
                # 匹配视图中调用create_user的实际参数
            )

方法二：patch CustomUserManager类的create_user方法 #

如果你的CustomUserManager定义在myapp/models.py，可以直接patch管理器类的create_user方法，确保覆盖所有实例的调用：

from django.test import TestCase
from django.urls import reverse
from unittest.mock import patch
from myapp.models import CustomUserManager

class TestYourFormView(TestCase):
    def test_form_valid_handles_create_user_exception(self):
        form_data = {
            'username': 'testuser',
            'email': 'test@example.com',
            'password': 'testpass123'
        }
        
        # patch管理器类的create_user方法
        with patch.object(CustomUserManager, 'create_user') as mock_create_user:
            mock_create_user.side_effect = Exception("创建用户失败")
            
            response = self.client.post(reverse('your-view-url-name'), data=form_data)
            
            # 验证业务逻辑
            self.assertEqual(response.status_code, 200)
            self.assertFormError(response, 'form', None, '创建用户失败')
            mock_create_user.assert_called_once()

为什么之前的方式失败？ #

• 直接patchCustomUserManager.create_user但未关联到模型实际使用的管理器实例，Django的管理器绑定机制会导致patch未生效；
• 创建CustomUser Mock对象的方式，无法覆盖视图中调用的真实模型类的objects管理器实例，因此mock没有作用。

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