如何使用 unittest.mock.MagicMock 在 Python 中对类进行模拟测试

2023-04-13 00:00:00 如何使用模拟中对

unittest.mock.MagicMock 是 Python 中的一个模拟框架，它可以模拟类、函数等，在测试中可以用来代替真实的对象，以避免测试中对真实对象的影响。

下面我们来演示如何使用该框架来模拟一个类：

from unittest.mock import MagicMock
import unittest

# 定义一个待测试的类
class Myclass:
    def get_name(self):
        return "pidancode.com"
    def get_hello(self):
        return "Hello, pidancode.com!"

class TestClass(unittest.TestCase):
    def test_myclass_get_name(self):
        # 使用 MagicMock 模拟 Myclass 类
        myclass_mock = MagicMock(spec=Myclass)
        # 模拟 get_name 方法，让其返回皮蛋编程
        myclass_mock.get_name.return_value = "皮蛋编程"
        # 使用模拟类进行测试
        self.assertEqual(myclass_mock.get_name(),"皮蛋编程")

    def test_myclass_get_hello(self):
        # 使用 MagicMock 模拟 Myclass 类
        myclass_mock = MagicMock(spec=Myclass)
        # 模拟 get_hello 方法
        myclass_mock.get_hello.return_value = "Hello, pidancode.com!"
        # 使用模拟类进行测试
        self.assertEqual(myclass_mock.get_hello(),"Hello, pidancode.com!")

if __name__ == '__main__':
    unittest.main()

在上面的代码中，我们定义了一个 Myclass 类，其中包含 get_name 和 get_hello 两个方法，分别用于获取名称和打招呼。

在测试 TestClass 中，我们使用 MagicMock 对 Myclass 进行模拟，然后分别测试 get_name 和 get_hello 方法。

通过模拟框架，我们可以很方便地模拟类和方法，从而更加灵活和高效地进行单元测试。

相关文章