使用Mock测试Python包时遇__init__.py模块结构问题求助

看起来你遇到的是Mock路径匹配错误的典型问题——这也是很多开发者在多目录包结构里写测试时容易踩的坑。我来一步步帮你理清问题并解决：

核心问题：Mock的目标路径不对 #

unittest.mock.patch的关键原则是：你需要patch的是被测试代码实际引用的那个对象实例，而不是测试文件里的导入路径，也不是全局的模块路径。

你的原代码里用了@mock.patch('module.os')，但这个路径大概率和被测试模块里实际使用的os对象的路径不匹配，导致Mock没有生效或者抛出异常。

具体修正步骤 #

1. 先明确你的项目结构（修正排版后的清晰版本） #

假设你的结构是这样的（如果不是，调整对应路径即可）：

你的项目根目录/
├── pkg/
│   ├── __init__.py       # 标记pkg为Python包
│   ├── module.py         # 被测试的业务模块
│   └── tests/
│       ├── __init__.py   # 标记tests为子包
│       └── test_basic.py # 你的测试文件

2. 确认被测试模块的导入方式 #

假设你的module.py里是这样使用os的：

# pkg/module.py
import os

def get_filesize(file_path):
    if os.path.isfile(file_path):
        return os.stat(file_path).st_size
    return 0

3. 修正测试代码的Mock路径 #

在test_basic.py里，你需要patch的是pkg.module.os——也就是被测试模块module.py中实际引用的os对象的完整路径：

# pkg/tests/test_basic.py
import unittest
from unittest import mock
# 用绝对导入或者相对导入引入被测试函数
from pkg.module import get_filesize
# 或者用相对导入：from ..module import get_filesize

class TestFileSize(unittest.TestCase):
    # 这里的路径必须和被测试模块中os的引用路径一致
    @mock.patch('pkg.module.os')
    def test_filesize(self, mock_os):
        # 定义模拟的stat返回对象
        class MockFileInfo:
            st_size = 1000
        
        # 设置Mock的返回值
        mock_os.path.isfile.return_value = True
        mock_os.stat.return_value = MockFileInfo()
        
        # 执行测试并断言结果
        result = get_filesize("test_file.txt")
        self.assertEqual(result, 1000)
        
        # 可选：验证Mock方法是否被正确调用
        mock_os.path.isfile.assert_called_once_with("test_file.txt")
        mock_os.stat.assert_called_once_with("test_file.txt")

if __name__ == '__main__':
    unittest.main()

4. 正确运行测试 #

从你的项目根目录执行测试命令，确保Python能正确识别pkg包：

# 使用unittest运行
python -m unittest pkg.tests.test_basic

# 或者用pytest（如果安装了）
pytest pkg/tests/test_basic.py -v

额外调试技巧 #

如果还是有问题，可以在被测试模块里加一行代码，打印出os对象的实际路径：

# pkg/module.py
import os
print(os.__module__)  # 会输出os，但结合模块导入链，你需要patch的是`pkg.module.os`

另外还要注意：

• 不要让你的模块名和标准库模块重名（比如不要叫os.py）
• 确保项目根目录在Python的PYTHONPATH中，或者用-m参数运行测试来自动识别包结构

内容的提问来源于stack exchange，提问作者tyleax