从同一软件包中修补一个函数不是修补

Question

我正在使用内置的Unitest库。我正在尝试测试一个函数，在此示例a中，它调用函数b。这两个功能都位于软件包的最高级别，不是基于类的功能。我已经阅读了有关修补何处的Unitest文档，并发现您必须使用该补丁来针对导入其他函数的函数（a）（b），并且不需函数本身（b），但是所有答案和文档似乎仅基于参考类功能。

Heirarchy：顶级

mypkg
|- __init__.py
|- a.py
|- b.py
|- test
  |- __init__.py
  |- test_a.py

__ INT __

from .a import a
from .b import b
__all__ = ['a','b']

有

import .b import b

def a():
  ...
  functionB({"USERNAME": username, "OTHER": other})

以下

def b(**kwargs):
 // do stuff with kwargs like calling APIs etc
 // This function doesn't return anything

。

import unittest
from unittest.mock import patch
from ..a import a

fake_return_value = "heythere"

class TestFunctionA(unittest.TestCase):
 @patch('mypkg.a.b', return_value=fake_return_value)
 # @patch('mympkg.b') # Doesn't work, does not override the original function
 def test_mytest(self, mocked_func):
   a()
   mocked_func.assert_called_once_with(stuff)

我 原始功能被调用，而不是模拟功能。

我在这里做错了什么？我不认为单位测试很难，我来自单元测试React应用程序，所以这有点令人沮丧，但我假设用户错误。到目前为止，我所做的所有文档和谷歌搜索似乎都应该起作用，或者至少上面评论了补丁变体。这是因为我的软件包结构并试图在同一软件包中模拟功能还是？

Answer 1

让我们看一下您的顶级__ init __. py，其中包含在导入时间上执行的代码。

语句来自。 em>函数 a包含在模块a中。因此，导入mypkg.a的所有调用（例如，测试模块中的补丁）将返回 function a而不是模块。这就是为什么您获得错误attributeError：函数A在您的最后一个答案中的评论中没有属性'b'。

要修补模块a的函数用法b，您的测试代码需要能够导入模块a。有几种方法可以做到这一点。

• 从.a __ init __. py 中删除.a导入。
  • 您仍然可以从mypkg导入或导入mypkg.a a a a li>
  • 您仍然可以通过mypkg..a导入或导入mypkg.aa a a a
• 重命名函数 a对其他东西。

Answer 2

注意：此答案地址为a 上一个修订版问题的问题

已注释的patch确实是正确的一个要使用的一种，因为您正在尝试修补对b的引用。 b是类还是一个函数，这并不重要，因为像JavaScript一样，两个都是python中的一流对象，并通过unittest.mock.patch < /代码>以相同的方式。

我怀疑您更复杂的系统中的问题与示例中的以下错误之一有关：

• 补丁应从UnitTest.mock导入import fatch 将其导入为。
• 模拟的return_value应在您的测试中，而不是在装饰器中分配。
• 东西不在测试类的范围内。

工作，等效的示例

这是您描述的系统的工作示例：

• 两个模块， a 和b，每个模块都带有顶级功能
• 模块a从b调用功能。
• 您想在b中修补该函数以隔离测试模块a。

文件层次结构：

mypkg
|- __init__.py    // empty
|- a.py
|- b.py
|- test
  |- __init__.py  // empty
  |- test_a.py

b.py

def b():
  return "World!"

a.py

from .b import b

def a():
  return "Hello " + b()

test_a.py

import unittest
from unittest.mock import patch
from ..a import a

class TestFunctionA(unittest.TestCase):
    def test_a_no_mock(self):
        message = a()
        self.assertEqual(message, "Hello World!")

    @patch('mypkg.a.b')
    def test_mytest(self, mocked_func):
        mocked_func.return_value = "Stack Overflow!"
        message = a()
        self.assertEqual(message, "Hello Stack Overflow!")
        mocked_func.assert_called_once()

从文件夹中运行包含 的测试mypkg

$ ls
mypkg
$ python -m unittest discover
..
----------------------------------------------------------------------
Ran 2 tests in 0.001s

OK