pytest_iocapture plugin
=======================

convenient capturing of writes to stdout/stderror streams

and file descriptors. 

Example Usage
----------------------

You can use the `capsys funcarg`_ to capture writes 
to stdout and stderr streams by using it in a test 
likes this:

.. sourcecode:: python

    def test_myoutput(capsys):
        print "hello" 
        print >>sys.stderr, "world"
        out, err = capsys.reset()
        assert out == "hello\n"
        assert err == "world\n"
        print "next"
        out, err = capsys.reset()
        assert out == "next\n" 

The ``reset()`` call returns a tuple and will restart 
capturing so that you can successively check for output. 
After the test function finishes the original streams
will be restored.
.. _`capsys funcarg`:


the 'capsys' test function argument
-----------------------------------

captures writes to sys.stdout/sys.stderr and makes 
them available successively via a ``capsys.reset()`` method 
which returns a ``(out, err)`` tuple of captured strings. 
.. _`capfd funcarg`:


the 'capfd' test function argument
----------------------------------

captures writes to file descriptors 1 and 2 and makes 
them available successively via a ``capsys.reset()`` method 
which returns a ``(out, err)`` tuple of captured strings. 

Getting and improving this plugin
---------------------------------


Do you find the above documentation or the plugin itself lacking,
not fit for what you need?  Here is a **30 seconds guide**
to get you started on improving the plugin:

1. Download `pytest_iocapture.py`_ plugin source code 
2. put it somewhere as ``pytest_iocapture.py`` into your import path 
3. a subsequent test run will now use your local version! 

Further information: extend_ documentation, other plugins_ or contact_.  

For your convenience here is also an inlined version of ``pytest_iocapture.py``:

.. sourcecode:: python

    """
    convenient capturing of writes to stdout/stderror streams 
    and file descriptors. 
    
    Example Usage
    ----------------------
    
    You can use the `capsys funcarg`_ to capture writes 
    to stdout and stderr streams by using it in a test 
    likes this:
    
    .. sourcecode:: python
    
        def test_myoutput(capsys):
            print "hello" 
            print >>sys.stderr, "world"
            out, err = capsys.reset()
            assert out == "hello\\n"
            assert err == "world\\n"
            print "next"
            out, err = capsys.reset()
            assert out == "next\\n" 
    
    The ``reset()`` call returns a tuple and will restart 
    capturing so that you can successively check for output. 
    After the test function finishes the original streams
    will be restored. 
    """
    
    import py
    
    def pytest_funcarg__capsys(request):
        """captures writes to sys.stdout/sys.stderr and makes 
        them available successively via a ``capsys.reset()`` method 
        which returns a ``(out, err)`` tuple of captured strings. 
        """ 
        capture = Capture(py.io.StdCapture)
        request.addfinalizer(capture.finalize)
        return capture 
    
    def pytest_funcarg__capfd(request):
        """captures writes to file descriptors 1 and 2 and makes 
        them available successively via a ``capsys.reset()`` method 
        which returns a ``(out, err)`` tuple of captured strings. 
        """ 
        capture = Capture(py.io.StdCaptureFD)
        request.addfinalizer(capture.finalize)
        return capture 
    
    def pytest_pyfunc_call(pyfuncitem):
        if hasattr(pyfuncitem, 'funcargs'):
            for funcarg, value in pyfuncitem.funcargs.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)
    
        def test_funcall_yielded_no_funcargs(self, testdir):        
            reprec = testdir.inline_runsource("""
                def test_hello():
                    yield lambda: None
            """)
            reprec.assertoutcome(passed=1)

.. _`pytest_iocapture.py`: http://bitbucket.org/hpk42/py-trunk/raw/3d511a58acc10477ff4f766df4a9224cfc3546af/py/test/plugin/pytest_iocapture.py
.. _`extend`: ../extend.html
.. _`plugins`: index.html
.. _`contact`: ../../contact.html
.. _`checkout the py.test development version`: ../../download.html#checkout