模拟路径lib.Path.exists

问题描述

我的 Python 函数创建 pathlib.Path 对象。我想模拟一下，所以当代码调用 pathlib.Path("/tmp/a").exists() 时它会得到 True，而当代码调用 pathlib.Path("/tmp/b").exists() 时它会得到 False。

我试过了：

import pathlib
from unittest.mock import patch

def my_side_effect(*args,**kwargs):
    print (f" args = {args},kwargs={kwargs}")
    return True

with patch.object(pathlib.Path,'exists') as mock_exists:
    mock_exists.side_effect = my_side_effect
    print(pathlib.Path("a").exists())

输出：

 args = (),kwargs={}
True

如您所见，我的副作用是没有根据路径获得任何参数/kwargs。

解决方法

您的示例按预期工作，因为 pathlib.Path.exists 不接受任何参数。您感兴趣的路径保存在 Path 对象中。
您可以不使用 side_effect，而是使用自己的函数修补 pathlib.Path.exists 来满足您的需要：

import pathlib
from unittest import TestCase
from unittest.mock import patch

def my_exists(self):
    # self is the Path instance,str(Path) returns the path string
    return str(self) == "a"

class MyTest(TestCase):
    def test_exists(self):
        with patch.object(pathlib.Path,'exists',my_exists):
            self.assertTrue(pathlib.Path("a").exists())
            self.assertFalse(pathlib.Path("b").exists())