如何使用pathlib.Path对象模拟文件大小？

Question

在我的函数中，我以前使用os.path.getsize函数来获取文件大小。 因此，在我的测试套装中，我使用以下方法来测试函数的结果值：

# init test values
test_value = 7.5
size_name = ("B", "KB", "MB", "GB", "TB", "PB", "EB", "ZB", "YB")
        
# mock 0 B file 
with patch('os.path.getsize', return_value=0):
    txt = su.get_file_size('random')
    self.assertEqual(txt, '0B')
        
# mock every pow of 1024 to YB
for i in range(9):
    with patch('os.path.getsize', return_value=test_value*(1024**i)):
        txt = su.get_file_size('random')
        self.assertEqual(txt, f'7.5 {size_name[i]}')

现在我在内部使用pathlib.Path对象作为文件。所以我现在要模拟Path(filename).stat().st_size。有人能告诉我应该在with语句中使用什么而不是patch吗

Answer 1

如果您在测试模块中使用from pathlib import Path导入Path，您可以使用类似的方法（有关更多信息，请查看where to patch）：

   with patch('yourmodule.su.Path.stat') as stat:
        stat.return_value.st_size = 0
        txt = get_file_size('random')

您不能直接模拟整个表达式，因为方法调用（stat()）必须使用return_value进行模拟，但它应该足够简单