不能在 Python 单元测试中猴子补丁模块变量

Question

我正在尝试测试模块：

包/模块.py

DATA_PATH = os.path.join(os.path.dirname(__file__), "data")
class SomeClass:
    def __init__(self):
        self.filename = os.path.join(DATA_PATH, "ABC.txt")

在tests/module_test.py中我正在尝试做

from package import module
@patch("package.module.DATA_PATH", "data_path_here") # doesn't work
class TestSomeClass(unittest.TestCase):
    def setUp(self):
        module.DATA_PATH = "data_path_here" # doesn't work
        self.obj= SomeClass()

    @patch("package.module.DATA_PATH", "data_path_here") # doesn't work either
    def test_constructor(self):
        self.assertEqual(r"data_path_here\ABC.txt", self.obj.filename)

但是 DATA_PATH 仍然没有被模拟出来。我想我尝试了所有可能的选项来模拟它，但它仍然返回原始路径而不是“data_path_here”

我究竟做错了什么？

编辑：它不是在 Python 单元测试框架中修改全局变量的副本， 因为该解决方案不起作用

Answer 1

您不需要修补，因为您使用的是来自另一个模块的全局变量：

#module.py

DATA_PATH = 1

def getData():
    return DATA_PATH


#tests.py
from package import module

print(module.DATA_PATH, module.getData())
module.DATA_PATH = 2
print(module.DATA_PATH, module.getData())

输出：

1 1
2 2

Answer 2

对我来说，使用模拟/补丁是一个痛苦的练习。另一方面，通过设置（和恢复）测试的全局来完成是微不足道的：

import mock_module

class TestSomeClass(unittest2.TestCase):
    def setUp(self):
        self._original = mock_module.DATA_PATH
        mock_module.DATA_PATH = 'data_path_here'

    def tearDown(self):
        mock_module.DATA_PATH = self._original

    def test_constructor(self):
        obj = mock_module.SomeClass()
        self.assertEqual(r"data_path_here\ABC.txt", obj.filename)

请注意，对于我的操作系统路径连接，分隔符是 \ 但您的用法可能会有所不同。

在 0.005 秒内运行 1 次测试

好的