模拟由静态方法调用的模块函数不起作用

Question

好的，我有一系列 Python3 代码，如下所示：

foo.py

from bar import ClassB

class ClassA:

    @staticmethod
    def get_all(fn):
        something = ClassB(fn)
        return something.lines()

bar.py

def baz(f):
    with open(f, 'rt') as fin:
        ln = fin.readlines()
    return ln

class ClassB:

    def __init__(self, f):
        self._baz = baz(f)

    def lines():
        return self._baz

现在我这样写单元测试：

test_foo.py

from unittest import TestCase, mock
from foo import ClassA

class Test_A(TestCase):

    @mock.patch("bar.baz")
    def setUp(self, mock_baz):
        mock_baz.return_value = ['a', 'b']

    def test_1(self):
        self.assertEqual(ClassA.get_all('whatever'), ['a', 'b']

问题：

运行测试导致 FileNotFoundError，这意味着 mock.patch("bar.baz") 不起作用。

如何解决此问题以便 baz() 函数得到正确模拟？

Answer 1

您的补丁似乎不适用于您的测试 - 您正在模拟的函数仅在 setUp 本身中被模拟。您可以尝试以下两种解决方案之一：

在设置中手动创建补丁：

class Test_A(TestCase):

    def setUp(self):
        mock_baz = mock.Mock(return_value=['a', 'b'])
        self.baz_patch = mock.patch("bar.baz", mock_baz)
        self.baz_patch.start()

    def tearDown(self):
        self.baz_patch.stop()
    ...

或者，仅在您的测试中模拟它：

class Test_A(TestCase):

    @mock.patch("bar.baz")
    def test_1(self, mock_baz):
        mock_baz.return_value = ['a', 'b']
        self.assertEqual(ClassA.get_all('whatever'), ['a', 'b']

模拟由静态方法调用的模块函数不起作用

Mocking a module function called by a static method does not work

python

python-3.x

python-unittest

python-unittest.mock

foo.py

bar.py

test_foo.py

问题：