模拟位于 __init__.py 中的方法

Question

我想模拟init .py 中的一个方法，但实际上它不起作用。

有一个示例来演示该问题以及我如何尝试编写单元测试：

被测代码：src.main.myfile：

from src.main.utils import a_plus_b

def method_under_test():
    a_plus_b()

a_plus_b位于src.main.utils模块中的__init__.py中：

def a_plus_b():
    print("a + b")

单元测试：

import src.main.utils
import unittest
from mock import patch
from src.main.myfile import method_under_test

class my_Test(unittest.TestCase):
    def a_plus_b_side_effect():
       print("a_plus_b_side_effect")

    @patch.object(utils, 'a_plus_b')
    def test(self, mock_a_plus_b):
        mock_a_plus_b.side_effect = self.a_plus_b_side_effect
        method_under_test()

单元测试打印“a + b”，而不是副作用。谁能帮我解决我做错了什么？

Answer 1

您需要修补的名称不是src.main.utils.a_plus_b，而是src.main.myfile.a_plus_b，因为这就是method_under_test使用的名称。

@patch('src.main.myfile.a_plus_b')
def test(self, mock_a_plus_b):
    mock_a_plus_b.side_effect = self.a_plus_b_side_effect
    method_under_test()