Pytest如何mock Django管理命令类handle方法中的path变量？

方案1：直接patch os.path.join（适合命令中只有这一处os.path.join调用的场景） #

直接在测试用例中patch你命令模块内的os.path.join方法，固定返回临时CSV文件路径即可：

@pytest.mark.django_db
def test_function(mocker, create_test_csv_file):
    # 按实际路径替换你命令模块的导入路径
    mocker.patch("app.my_app.my_command.os.path.join", return_value=create_test_csv_file)
    call_command("my_command")
    # 后续补充你的业务逻辑断言

方案2：带参数判断的patch（适合命令中存在多处os.path.join调用的场景） #

如果你的handle方法里还有其他os.path.join调用，不想影响其他逻辑，可以通过side_effect做参数匹配，只把生成data.csv路径的调用替换成临时文件路径：

import os

@pytest.mark.django_db
def test_function(mocker, create_test_csv_file):
    def mock_os_join(*args):
        # 只拦截最后一个参数为data.csv的join调用
        if args[-1] == "data.csv":
            return create_test_csv_file
        # 其他调用走原生os.path.join逻辑
        return os.path.join(*args)
    
    mocker.patch("app.my_app.my_command.os.path.join", side_effect=mock_os_join)
    call_command("my_command")
    # 后续补充你的业务逻辑断言

注意：patch的目标路径必须是你实际调用os.path.join的模块路径，也就是你写Command类的my_command.py的导入路径，遵循在哪里使用就在哪里patch的mock原则。

更推荐的优化写法（可选） #

如果可以修改原有命令代码，更建议把文件路径改成命令的可选参数，不需要mock也能直接测试：

修改后的命令代码 #

class Command(BaseCommand):
    # 添加可选参数
    def add_arguments(self, parser):
        parser.add_argument("--csv-path", default="data.csv", help="CSV文件路径")
    
    def handle(self,  *args, **options):
        csv_path = options["csv_path"]
        path = (os.path.join(os.path.abspath(os.path.dirname(__name__)), csv_path))
        # other logic with file

对应的测试用例 #

@pytest.mark.django_db
def test_function(create_test_csv_file):
    # 直接通过命令参数传入临时文件路径
    call_command("my_command", csv_path=create_test_csv_file)
    # 后续补充你的业务逻辑断言

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