python:如何模拟os.statvfs，因为它返回posix.statvfs_result数据类型

Question

我正在尝试编写一个单元测试用例 下面是原始函数的部分代码：

def disk_space(something, otherthing):
"""
Comments
"""
dir_map = something['dir_map']
for dir, expected_free_gb in dir_map.items():
try:
result = os.statvfs(dir)
block_size = result.f_frsize
avail_blocks = result.f_bavail
giga = 1024 * 1024 * 1024
free_size = avail_blocks * block_size / giga
if free_size >= float(expected_free_gb):
print "No need to clear space"
else:
print "Clear some space"

单元测试功能：

@patch('os.statvfs', return_value=posix.statvfs_result(f_frsize=4096, f_bavail=2599495))
def test_disk_space(self, os_statvfs):
something
something

现在在这里，如果我以posix.statvfs_result数据类型返回值，它会给出错误：

类型错误： 找不到必需的参数"序列"(位置 1(

如果我只是给return_value=(f_frsize=4096, f_bavail=2599495)它说：

str(( 没有f_frsize。

Answer 1

使用以下函数

@patch('os.statvfs', return_value=os.statvfs_result((4096, 4096, 1909350, 1491513, 1394521, 971520, 883302, 883302, 0, 255)))
def test_disk_space(self, os_statvfs):
something
something

这些值对应于以下字段，因此请相应地进行更改。另请注意，值应位于双大括号内。

posix.statvfs_result(f_bsize=4096, f_frsize=4096, f_blocks=1909350L, f_bfree=1491513L,
f_bavail=1394521L, f_files=971520L, f_ffree=883302L, f_fvail=883302L, f_flag=0,
f_namemax=255)