52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
import py
|
|
|
|
class IocapturePlugin:
|
|
""" capture sys.stdout/sys.stderr / fd1/fd2. """
|
|
def pytest_funcarg__stdcapture(self, request):
|
|
capture = Capture(py.io.StdCapture)
|
|
request.addfinalizer(capture.finalize)
|
|
return capture
|
|
|
|
def pytest_funcarg__stdcapturefd(self, request):
|
|
capture = Capture(py.io.StdCaptureFD)
|
|
request.addfinalizer(capture.finalize)
|
|
return capture
|
|
|
|
class Capture:
|
|
def __init__(self, captureclass):
|
|
self._captureclass = captureclass
|
|
self._capture = self._captureclass()
|
|
|
|
def finalize(self):
|
|
self._capture.reset()
|
|
|
|
def reset(self):
|
|
res = self._capture.reset()
|
|
self._capture = self._captureclass()
|
|
return res
|
|
|
|
def test_generic(plugintester):
|
|
plugintester.hookcheck(IocapturePlugin)
|
|
|
|
class TestCapture:
|
|
def test_std_functional(self, testdir):
|
|
testdir.plugins.append(IocapturePlugin())
|
|
evrec = testdir.inline_runsource("""
|
|
def test_hello(stdcapture):
|
|
print 42
|
|
out, err = stdcapture.reset()
|
|
assert out.startswith("42")
|
|
""")
|
|
evrec.assertoutcome(passed=1)
|
|
|
|
def test_stdfd_functional(self, testdir):
|
|
testdir.plugins.append(IocapturePlugin())
|
|
evrec = testdir.inline_runsource("""
|
|
def test_hello(stdcapturefd):
|
|
import os
|
|
os.write(1, "42")
|
|
out, err = stdcapturefd.reset()
|
|
assert out.startswith("42")
|
|
""")
|
|
evrec.assertoutcome(passed=1)
|