Python中如何mock方法里用到的本地不存在的指定文件路径

方案1：直接mock read_location_file 方法（测试get_matches_mr时推荐） #

如果测试目标是get_matches_mr的逻辑，不需要校验文件读取逻辑，直接mock该依赖方法的返回值即可，完全绕过本地文件不存在的问题：

from unittest.mock import patch

def test_get_matches_mr():
    # 自定义你需要的测试路径列表作为模拟返回值
    mock_locations = ["/src/service1", "/src/service2", "/config/common"]
    # patch路径替换为你代码中实际导入read_location_file的模块路径
    with patch("your_module.read_location_file", return_value=mock_locations):
        # 按需mock MergeRequest的get_matches方法返回测试用数据
        with patch("your_module.MergeRequest.get_matches", return_value={"matches": ["/src/service1"]}):
            # 实例化你的业务类并调用方法
            your_obj = YourClass()
            result = your_obj.get_matches_mr()
            # 此处添加你的断言逻辑
            assert result == {"matches": ["/src/service1"]}

方案2：mock文件系统接口（需要测试read_location_file逻辑时使用） #

如果你要校验read_location_file的文件校验、yaml解析逻辑，可以mock操作系统的文件相关接口，模拟文件存在且内容符合预期：

from unittest.mock import patch, mock_open

def test_read_location_file():
    # 模拟yaml文件的内容
    mock_yaml = """
paths:
  - /src/service1
  - /src/service2
"""
    # 模拟os.path.isfile返回True，代表文件存在
    with patch("os.path.isfile", return_value=True):
        # 模拟open方法返回上面定义的yaml内容
        with patch("builtins.open", mock_open(read_data=mock_yaml)):
            result = read_location_file()
            assert result == ["/src/service1", "/src/service2"]

如果不想手动mock多个接口，也可以安装第三方库pyfakefs直接模拟整个文件系统，在模拟环境中直接创建.test/location.yml文件即可，不需要修改业务代码。

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