python：如何模拟 os.statvfs 作为 returns 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 数据类型中 return 值，它会给出错误：

TypeError: Required argument 'sequence' (pos 1) not found

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

str() is not having 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)

python：如何模拟 os.statvfs 作为 returns posix.statvfs_result 数据类型

python: How to mock os.statvfs as it returns posix.statvfs_result datatype

python

python-mock