如何在unittest中Mock User类__init__抛出的ValueError异常？

假设你的User类大致实现如下（包含在初始化时根据输入抛出ValueError的逻辑）：

class User:
    def __init__(self, name):
        self.__name = None
        # 当name无效时抛出异常
        if not isinstance(name, str) or len(name.strip()) == 0:
            raise ValueError("Name cannot be empty or non-string")
        self.__name = name

要在测试用例testUserClassInit中验证当输入触发异常的参数时，实例的__name属性是否初始化为None，关键是要让__init__方法执行到属性初始化的步骤，但跳过抛出异常的逻辑。以下是两种可行的Mock方式：

方法1：使用unittest.mock.patch替换__init__行为 #

直接通过patch修改User.__init__的执行逻辑，仅保留__name的初始化，不抛出异常：

import unittest
from unittest.mock import patch
from your_module import User  # 替换为你的User类所在模块

class TestUser(unittest.TestCase):
    def test_user_init_raises_error(self):
        # 正常测试异常抛出的用例
        with self.assertRaises(ValueError):
            User("")

    def testUserClassInit(self):
        # Mock __init__方法，仅初始化__name为None
        with patch.object(User, '__init__', side_effect=lambda self, name: setattr(self, '_User__name', None)):
            # 即使传入无效参数，也不会抛出异常
            user = User("invalid_input")
            # 私有属性需通过Python名称改写后的格式访问
            self.assertEqual(user._User__name, None)

方法2：临时替换__init__方法（无patch依赖） #

如果不想使用mock库，也可以临时替换__init__，测试完成后恢复原方法：

import unittest
from your_module import User

class TestUser(unittest.TestCase):
    def testUserClassInit(self):
        # 保存原__init__方法
        original_init = User.__init__
        
        def mock_init(self, name):
            # 仅执行原__init__中初始化__name的步骤
            self._User__name = None
        
        # 替换__init__
        User.__init__ = mock_init
        
        try:
            user = User("")
            self.assertEqual(user._User__name, None)
        finally:
            # 必须恢复原__init__，避免影响其他测试用例
            User.__init__ = original_init

关键说明 #

• Python的私有属性（以双下划线开头）会被名称改写，所以访问时需要用_类名__属性名的格式，比如_User__name。
• 两种方式的核心都是让__init__在测试时不抛出异常，同时确保__name被初始化为你需要验证的None值。

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