Python项目依赖配置文件时如何合理组织单元测试？

方案1落地实现 #

你可以通过Python模块导入的缓存机制，在测试代码加载业务模块前先替换全局的配置模块指向，全程不需要修改现有业务代码：

1. 先编写你的test_config_file.py，和正式配置文件保持相同的变量名，仅替换为测试相关的路径和参数
2. 在测试文件的最开头先完成配置模块替换，再导入业务模块：

# test_func.py 单元测试文件
import sys
import unittest
# 导入测试配置
import test_config_file
# 把全局的some_config_file指向替换为测试配置
sys.modules["some_config_file"] = test_config_file
# 替换完成后再导入业务模块
from some_module import func

class TestFunc(unittest.TestCase):
    def test_func_normal_logic(self):
        # 此时func内部调用的config已经是测试配置内容
        result = func()
        # 你的断言逻辑

如果需要全局生效所有测试用例，可以把替换逻辑写到测试目录的__init__.py文件中，所有测试运行前会自动执行替换。

方案2优雅优化 #

不需要给每个函数加test参数硬编码判断，用默认参数实现轻量依赖注入即可，仅需修改一次函数定义，原有业务调用完全不用改：

# some_module.py 改造后
import some_config_file as default_config

def func(config = default_config):
    path = config.path_to_some_data
    # 原有逻辑无需改动

测试时直接传入测试配置即可：

# 测试用例中
import test_config_file
from some_module import func

def test_func():
    result = func(config=test_config_file)
    # 断言逻辑

更通用的无侵入解决方案 #

如果不想额外维护测试配置文件，可以直接用单元测试框架自带的mock工具打补丁，全程不需要修改任何业务代码：

# test_func.py
import unittest
from unittest.mock import patch
from some_module import func

class TestFunc(unittest.TestCase):
    # 直接替换func用到的配置项值
    @patch("some_module.config.path_to_some_data", "path/to/test_data")
    def test_func_logic(self):
        result = func()
        # 断言逻辑

如果需要替换多个配置项，也可以批量mock整个配置模块：

@patch("some_module.config")
def test_func_multi_config(self, mock_config):
    mock_config.path_to_some_data = "path/to/test_data"
    mock_config.batch_size = 8
    mock_config.timeout = 10
    result = func()
    # 断言逻辑

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