时间戳

首页 > 解决方案 > 如何测试在 tarfile.open 上下文管理器中调用的函数？

python - 如何测试在 tarfile.open 上下文管理器中调用的函数？

问题描述

我编写了一个创建 tarball 的函数。

# tarball.py
import os
import tarfile


def create_tarball():
    with tarfile.open("path/to/tar/file", "w:gz") as tar:
        tar.add(
            "/path/to/included/directory",
            arcname=os.path.basename("/path/to/included/directory"),
        )

tarfile.open我已经在上下文管理器上编写了一个带有断言的通过测试。

from unittest.mock import patch

from tarball import create_tarball


@patch("tarball.tarfile.open")
def test_create_tarball_partial(mock_open):
    create_tarball()

    mock_open.assert_called_with("path/to/tar/file", "w:gz")

如何为tarfile.open上下文管理器内部调用的函数编写测试？

标签： pythonunit-testingpytesttar

解决方案

要为tar.add函数编写测试，请执行以下操作。

  1. 补丁os.path.basename功能
  2. 创建一个tar.add使用MagicMock的模拟
  3. 将上下文管理器中的函数值设置为tar.addMagicMock¹
  4. os.path.basename设置函数的返回值
  5. 断言模拟os.path.basename和tarfile.open函数

例如：

from unittest.mock import patch, MagicMock

from tarball import create_tarball


@patch("tarball.os.path.basename")
@patch("tarball.tarfile.open")
def test_create_tarball_full(mock_open, mock_basename):
    mock_add = MagicMock()
    mock_open.return_value.__enter__.return_value.add = mock_add
    mock_basename.return_value = "/path/to/included/directory"

    create_tarball()

    mock_open.assert_called_with("path/to/tar/file", "w:gz")
    mock_basename.assert_called_with("/path/to/included/directory")
    mock_add.assert_called_with(
        "/path/to/included/directory", arcname="/path/to/included/directory"
    )

¹ Python：模拟上下文管理器

推荐阅读