如何在 Python 中模拟实例方法

Question

我正在尝试测试一个类并模拟它的一个方法，但我似乎无法用我的模拟行为替换其中一个实例方法的行为。 我的代码是这样组织的：

--src ----module ------__init__.py ------my_class.py --tst ----__init__.py ----test_my_class.py

my_class.py包含以下内容

class MyClass:

    def __init__(self):
        pass

    def do_something(self):
        return 'Real Output'

我的测试文件test_my_class.py包含以下内容。

from unittest.mock import patch
from src.module.my_class import MyClass

def test_my_class():
    my_class = MyClass()
    assert my_class.do_something() == 'Real Output'

@patch('src.module.my_class.MyClass')
def test_mock_my_class(mock_my_class):
    mock_my_class.return_value.do_something.return_value = 'Mocked Output'
    my_class = MyClass()
    assert my_class.do_something() == 'Mocked Output'

第一个测试工作得很好（到目前为止不涉及嘲笑）。 然而，第二个测试给了我以下断言错误。 我希望do_something()方法被模拟并返回“模拟输出”，并且断言语句评估为真。 我在哪里错了？

AssertionError: assert <bound method MyClass.do_something of <src.module.my_class.MyClass object at 0x1057133c8>> == 'Mocked Output' E + where <bound method MyClass.do_something of <src.module.my_class.MyClass object at 0x1057133c8>> = <src.module.my_class.MyClass object at 0x1057133c8>.do_something

PS。 我咨询了以下资源但没有成功：

• Python模拟类实例变量
• 在 Python 中模拟实例变量的方法
• Python mock：包装实例方法
• 如何为 python 单元测试提供模拟类方法？
• https://medium.com/python-pandemonium/python-mocking-you-are-a-tricksy-beast-6c4a1f8d19b2

最后一个链接起初看起来特别有用，因为我几乎逐字逐句地遵循其中一个示例，但它仍然不起作用。

@mock.patch("simple.SimpleClass")
def mock_simple_class(mock_class):
    mock_class.return_value.explode.return_value = "BOO!"
    inst = simple.SimpleClass()
    result = inst.explode()
    print(result)

Answer 1

解决这个问题的更合适的方法是使用 patch.object

@patch.object(MyClass, "do_something")
def test_mock_my_class(m):
    m.side_effect = ['Mocked Output']
    my_class = MyClass()
    assert my_class.do_something() == 'Mocked Output'

Answer 2

就个人而言， patch.object让我感到困惑。 读完这篇文章后，我决定只使用patch 。

问题中patch的问题在于，打补丁的MyClass不是test函数中使用的MyClass 。 有关应在何处应用补丁的更深入说明，请参阅本指南。

补丁的路径应该是tst.test_my_class.MyClass ，如下面的代码所示。

from unittest.mock import patch
import unittest
from src.module.my_class import MyClass


class Test(unittest.TestCase):
    def test_my_class(self):
        my_class = MyClass()
        self.assertTrue(my_class.do_something() == 'Real Output')

    @patch('test.test_my_class.MyClass')
    def test_mock_my_class(self, mock_my_class):
        mock_my_class.return_value.do_something.return_value = 'Mocked Output'
        my_class = MyClass()
        self.assertTrue(my_class.do_something() == 'Mocked Output')

运行此命令以在终端中执行它。

python3 -m unittest discover