如何对与块设备交互的程序进行单元测试

Question

我有一个程序可以与 linux 上的块设备（/dev/sda 等）进行交互和更改。我正在使用各种外部命令（主要是来自 fdisk 和 GNU fdisk 软件包的命令）来控制设备。我已经创建了一个类，作为块设备的大多数基本操作的接口（例如：它的大小是多少？它安装在哪里？等等）

这是一种查询分区大小的方法：

def get_drive_size(device):
    """Returns the maximum size of the drive, in sectors.

    :device the device identifier (/dev/sda and such)"""

    query_proc = subprocess.Popen(["blockdev", "--getsz", device], stdout=subprocess.PIPE) 
    #blockdev returns the number of 512B blocks in a drive
    output, error = query_proc.communicate()
    exit_code = query_proc.returncode
    if exit_code != 0:
        raise Exception("Non-zero exit code", str(error, "utf-8")) #I have custom exceptions, this is slight pseudo-code

    return int(output) #should always be valid

所以这个方法接受一个块设备路径，并返回一个整数。测试将以 root 身份运行，因为整个程序最终都必须以 root 身份运行。

我应该尝试测试这些方法之类的代码吗？如果是这样，怎么做？我可以尝试为每个测试创建和挂载图像文件，但这似乎有很多开销，并且本身可能容易出错。它需要块设备，所以我不能直接对文件系统中的图像文件进行操作。

我可以尝试嘲笑，正如一些答案所暗示的那样，但这感觉不够。似乎我开始测试方法的实现，如果我模拟 Popen 对象，而不是输出。在这种情况下，这是对正确单元测试方法的正确评估吗？

我在这个项目中使用 python3，我还没有选择单元测试框架。如果没有其他原因，我可能会直接使用 Python 自带的默认 unittest 框架。

Answer 1

您应该查看 mock 模块（我认为它现在是 Python 3 中 unittest 模块的一部分）。

它使您无需依赖任何外部资源即可运行测试，同时让您能够控制模拟与代码的交互方式。

我将从Voidspace 中的文档开始

这是一个例子：

import unittest2 as unittest
import mock

class GetDriveSizeTestSuite(unittest.TestCase):

  @mock.patch('path/to/original/file.subprocess.Popen')
  def test_a_scenario_with_mock_subprocess(self, mock_popen):
    mock_popen.return_value.communicate.return_value = ('Expected_value', '')
    mock_popen.return_value.returncode = '0'
    self.assertEqual('expected_value', get_drive_size('some device'))