使用文件输入模块时如何模拟标准输入？

Question

我有一个使用 Python fileinput模块的程序，我正在尝试为main() function 编写单元测试。它们在使用实际文件时可以找到，但会引发OSError: reading from stdin while output is captured当我尝试通过通过标准输入的数据。 使用fileinput时模拟 stdin 输入的正确方法是什么？

示例my_fileinput.py ：

"""
$ echo "42" | python3.8 my_fileinput.py -
answer: 42
"""

import fileinput
import sys

def main():
    for line in fileinput.input(sys.argv[1:]):
        sys.stdout.write(f"answer #{fileinput.lineno()}: {line}")

if __name__ == "__main__":
    main()

示例test_my_fileinput.py ：

"""
$ python3.10 -m pytest test_my_fileinput.py
OSError: reading from stdin while output is captured
"""

import io
from unittest import mock

import my_fileinput

def test_stdin():
    """Test fileinput with stdin."""

    with mock.patch.object(my_fileinput, "raw_input", create=True, return_value="42"):
        with mock.patch("sys.stdout", new=io.StringIO()) as stdout:
            with mock.patch("sys.argv", ["my_fileinput.py", "-"]):
                # Raises OSError: reading from stdin while output is captured
                my_fileinput.main()
                assert stdout.getvalue() == "answer: 42\n"

我尝试了 mocking stdin 的各种方式，结果都相同。 所有这些都会导致相同的 OSError。

Answer 1

更新：不同版本修补sys.stdin而不是inputfile.input

import io

from unittest import mock

import my_fileinput

def test_stdin():
    """Test fileinput with stdin."""

    with mock.patch("sys.stdin", new=io.StringIO("42\n")):
        with mock.patch("sys.stdout", new=io.StringIO()) as stdout:
            with mock.patch("sys.argv", ["my_fileinput.py", "-"]):
                my_fileinput.main()
                assert stdout.getvalue() == "answer: 42\n"

---

警告：原始答案摆脱了OSError ，但使inputfile模块中的其他功能无法使用（请参阅评论）。

原文：将 mock.patch.object 的前两个mock.patch.object更改为fileinput和"input"似乎修复了OSError 。

with mock.patch.object(fileinput, "input", create=True, return_value="42"):

第一个参数是你要打补丁的目标object，也就是fileinput模块。 第二个参数是目标object中要改变的属性，即input 。

import io
import fileinput
from unittest import mock

import my_fileinput

def test_stdin():
    """Test fileinput with stdin."""

    with mock.patch.object(fileinput, "input", create=True, return_value="42"):
        with mock.patch("sys.stdout", new=io.StringIO()) as stdout:
            with mock.patch("sys.argv", ["my_fileinput.py", "-"]):
                my_fileinput.main()
                assert stdout.getvalue() == "answer: 42\n"

Answer 2

您没有必要测试fileinput本身，因为它将由 CPython 自己的测试套件进行测试： Lib/test/test_fileinput.py 。 测试代码的 pytesthonic 方法是这样的，使用固定装置：

import my_fileinput

def test_stdin(mocker, capsys):
    mocker.patch("fileinput.lineno", return_value=1)
    mocker.patch("fileinput.input", return_value=["42\n"])
    my_fileinput.main()
    out, err = capsys.readouterr()
    assert out == "answer #1: 42\n"

capsys fixture 包含在 pytest 中， mocker fixture 由插件pytest-mock提供。