Question

我编写了一个创建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上下文管理器内部调用的函数编写测试？

Answer 1

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

修补os.path.basename函数
使用MagicMock创建tar.add的模拟物
将上下文管理器中的tar.add函数的值设置为MagicMock¹
设置os.path.basename函数的返回值
在模拟的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: Mocking a context manager

测试在tarfile.open上下文管理器内部调用的函数？

1 个答案: