python - 使用 fixture 时 pytest capsys 未捕获标准输出

Question

我正在使用 pytest fixture 来模拟用于测试脚本的命令行参数。这样每个测试函数共享的参数只需要在一个地方声明。我也在尝试使用 pytest 的 capsys 来捕获脚本打印的输出。考虑以下愚蠢的例子。

from __future__ import print_function
import pytest
import othermod
from sys import stdout


@pytest.fixture
def shared_args():
    args = type('', (), {})()
    args.out = stdout
    args.prefix = 'dude:'
    return args


def otherfunction(message, prefix, stream):
    print(prefix, message, file=stream)


def test_dudesweet(shared_args, capsys):
    otherfunction('sweet', shared_args.prefix, shared_args.out)
    out, err = capsys.readouterr()
    assert out == 'dude: sweet\n'

在这里，capsys 没有正确捕获 sys.stderr。如果我将 from sys import stdout 和 args.out = stdout 直接移动到测试函数中，一切都会按预期进行。但这让事情变得更加困惑，因为我必须为每个测试重新声明这些语句。难道我做错了什么？我可以将 capsys 与固定装置一起使用吗？

Answer 1

fixture 在测试运行之前被调用。在您的示例中，shared_args fixture 在其他函数可以将任何内容写入 stdout 之前读取 stdout。

解决您的问题的一种方法是让您的 fixture 返回一个可以执行您希望它执行的操作的函数。您可以根据您的用例确定 fixture 的范围。

from __future__ import print_function
import pytest
from sys import stdout
import os


@pytest.fixture(scope='function')
def shared_args():
    def args_func():
        args = type('', (), {})()
        args.out = stdout
        args.prefix = 'dude:'
        return args
    return args_func


def otherfunction(message, prefix, stream):
    print(prefix, message, file=stream)


def test_dudesweet(shared_args, capsys):
    prefix, out = shared_args().prefix, shared_args().out
    otherfunction('sweet', prefix, out)
    out, err = capsys.readouterr()
    assert out == 'dude: sweet\n'

您没有正确使用 capsys.readouterr()。在此处查看 capsys 的正确用法:https://stackoverflow.com/a/26618230/2312300