65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
"""
|
|
'capsys' and 'capfd' funcargs for capturing stdout/stderror either
|
|
by intercepting sys.stdout/stderr or File Descriptors 1/2.
|
|
|
|
Calling the reset() method of the capture funcargs gives
|
|
a out/err tuple of strings representing the captured streams.
|
|
You can call reset() multiple times each time getting
|
|
the chunk of output that was captured between the invocations.
|
|
|
|
"""
|
|
import py
|
|
|
|
def pytest_funcarg__capsys(request):
|
|
""" capture writes to sys.stdout/sys.stderr. """
|
|
capture = Capture(py.io.StdCapture)
|
|
request.addfinalizer(capture.finalize)
|
|
return capture
|
|
|
|
def pytest_funcarg__capfd(request):
|
|
""" capture writes to filedescriptors 1 and 2"""
|
|
capture = Capture(py.io.StdCaptureFD)
|
|
request.addfinalizer(capture.finalize)
|
|
return capture
|
|
|
|
def pytest_pyfunc_call(pyfuncitem, args, kwargs):
|
|
for funcarg, value in kwargs.items():
|
|
if funcarg == "capsys" or funcarg == "capfd":
|
|
value.reset()
|
|
|
|
class Capture:
|
|
_capture = None
|
|
def __init__(self, captureclass):
|
|
self._captureclass = captureclass
|
|
|
|
def finalize(self):
|
|
if self._capture:
|
|
self._capture.reset()
|
|
|
|
def reset(self):
|
|
res = None
|
|
if self._capture:
|
|
res = self._capture.reset()
|
|
self._capture = self._captureclass()
|
|
return res
|
|
|
|
class TestCapture:
|
|
def test_std_functional(self, testdir):
|
|
reprec = testdir.inline_runsource("""
|
|
def test_hello(capsys):
|
|
print 42
|
|
out, err = capsys.reset()
|
|
assert out.startswith("42")
|
|
""")
|
|
reprec.assertoutcome(passed=1)
|
|
|
|
def test_stdfd_functional(self, testdir):
|
|
reprec = testdir.inline_runsource("""
|
|
def test_hello(capfd):
|
|
import os
|
|
os.write(1, "42")
|
|
out, err = capfd.reset()
|
|
assert out.startswith("42")
|
|
""")
|
|
reprec.assertoutcome(passed=1)
|