test_ok1/py/test/plugin/pytest_tmpdir.py

"""
example:

    pytest_plugins = "pytest_tmpdir" 

    def test_plugin(tmpdir):
        tmpdir.join("hello").write("hello")

"""
import py

class TmpdirPlugin:
    """ provide temporary directories to test functions and methods. 
    """ 

    def pytest_configure(self, config):
        # XXX make ensuretemp live on config 
        self.basetmp = py.test.ensuretemp("tmpdir")

    def pytest_pyfuncarg_tmpdir(self, pyfuncitem):
        name = pyfuncitem.name
        for i in range(10000):
            try:
                tmpdir = self.basetmp.mkdir(name + (i and str(i) or ''))
            except py.error.EEXIST:
                continue
            break
        return tmpdir

# ===============================================================================
#
# plugin tests 
#
# ===============================================================================
#
def test_generic(plugintester):
    plugintester.apicheck(TmpdirPlugin)

def test_pyfuncarg(testdir):
    item = testdir.getitem("def test_func(tmpdir): pass")
    plugin = TmpdirPlugin()
    plugin.pytest_configure(item._config)
    p = plugin.pytest_pyfuncarg_tmpdir(item)
    assert p.check()
    bn = p.basename.strip("0123456789")
    assert bn.endswith("test_func")