使用fixture时,stdout没有被pytest capsys捕获

时间:2017-02-28 23:07:07

标签: python pytest fixtures

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

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 stdoutargs.out = stdout直接移动到测试函数中,事情就会按预期工作。但这会使事情变得更加混乱,因为我必须为每个测试重新声明这些语句。难道我做错了什么?我可以使用带有固定装置的帽子吗?

1 个答案:

答案 0 :(得分:0)

在运行测试之前调用固定。在您的示例中,shared_args固定装置正在读取stdout,其他函数才能将任何内容写入stdout。

解决问题的一种方法是使灯具返回一个可以执行您想要的功能的函数。您可以根据用例确定灯具的范围。

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