编程语言
首页 > 编程语言> > python – 使用fixture时没有被pytest capsys捕获的stdout

python – 使用fixture时没有被pytest capsys捕获的stdout

作者:互联网

我正在使用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.如果我从sys import stdout和args.out = stdout直接进入测试函数,事情按预期工作.但这会使事情变得更加混乱,因为我必须为每个测试重新声明这些语句.难道我做错了什么?我可以使用带固定装置的帽子吗?

解决方法:

在测试运行之前调用Fixture.在您的示例中,shared_args fixture在其他函数可以向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

标签:python,fixtures,pytest
来源: https://codeday.me/bug/20190710/1428446.html