似乎无法在另一个文件中修补 Class & 方法

Question

我一直在用这样的小模型撞墙：

这是树：

src
├── __init__.py
├── file_a.py
├── file_b.py


test
├── test_a.py

在file_a中：

class qaz(object):
    def __init__(self):
        print("\n\nin qaz")

    def exec_bar(self):
        bar_inst = bar()
        bar_inst.execute("a", "b")

在file_b中：

class bar(object):
    def __init__(self, a, b):
        print("\n\nin bar")

    def execute(self, c, d):
        print("\n\nin bar -> execute")

所以，我想模拟 bar 以便我可以毫无问题地测试 a。

在test_a中：

from unittest.mock import patch, MagicMock
from src.file_a import qaz
from src.file_b import bar

class BarTester(unittest.TestCase):

    @patch('src.file_b.bar')
    def test_bar(self, mock_bar):
        bar_inst = MagicMock()
        bar_inst.execute.return_value = None
        mock_bar.return_value = bar_inst

        q = qaz()
        q.exec_bar()

每次都失败，如下所示：

TypeError: __init__() missing 2 required positional arguments: 'a' and 'b'

这意味着模拟不工作。我似乎无法弄清楚我错了什么。

Answer 1

in file_b，您期望在“init”
中传递 2 个参数 def __init__(self, a, b):

但是在 file_a 中，当您为 class 栏创建对象时，您没有传递任何参数
bar_inst = bar()

这就是您看到错误的原因
TypeError: __init__() missing 2 required positional arguments: 'a' and 'b'

你做了两件事：

从 def __init__(self, a, b):
在

新解决方案：

from mock import patch
import unittest
from src.file_a import qaz
from src.file_b import bar

class BarTester(unittest.TestCase):

    @patch.object(bar, '__init__', return_value=None)
    def test_bar(self, *mock_stdout):
        q = qaz()
        q.exec_bar()

Answer 2

Sai Ram 解决了它，但想 post 代码以备将来参考。测试最终是这样的；

从unittest.mock导入补丁，MagicMock 从 src.file_a 导入 qaz 来自 src.file_b 导入栏

class BarTester(unittest.TestCase):

    @patch.object(bar, '__init__', return_value=None)
    @patch.object(bar, 'execute', return_value=None)
    def test_bar(self, *mock_bar):
        q = qaz()
        q.exec_bar()

似乎无法在另一个文件中修补 Class & 方法

Cannot Seem to Patch Class & Method In Another File

python

mocking

python-3.x

python-unittest

magicmock