用虚拟数据覆盖所有类方法

使用mock库可以很方便地覆盖所有类方法的虚拟数据。以下是一个示例代码：

from unittest import TestCase
from unittest.mock import patch, MagicMock

class MyClass:
    def method1(self):
        return "original method1"

    def method2(self):
        return "original method2"

class MyTest(TestCase):
    def test_mock_methods(self):
        # 创建一个mock对象来替代MyClass的实例
        my_obj = MagicMock(spec=MyClass)

        # 设置mock对象的方法返回值
        my_obj.method1.return_value = "mocked method1"
        my_obj.method2.return_value = "mocked method2"

        # 使用patch装饰器将my_obj应用到测试方法中
        with patch("__main__.MyClass", return_value=my_obj):
            # 实例化一个MyClass对象
            obj = MyClass()

            # 调用方法1，将返回mocked method1
            self.assertEqual(obj.method1(), "mocked method1")

            # 调用方法2，将返回mocked method2
            self.assertEqual(obj.method2(), "mocked method2")

在上面的示例中，我们使用了MagicMock类来创建一个可以模拟类方法行为的对象。然后，我们使用return_value属性来设置每个方法的返回值。最后，我们使用patch装饰器将mock对象应用到测试方法中，并在测试方法中实例化一个原始类对象并调用其方法。这样，我们就可以使用虚拟数据来覆盖所有类方法。