模拟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())