模拟 pathlib.Path.exists

Question

我的 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

如您所见，我的副作用是没有得到任何基于路径的 args/kwargs。

Answer 1

您的示例按预期工作，因为 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())

模拟 pathlib.Path.exists

Mock pathlib.Path.exists

python

unit-testing

mocking

pathlib