floraachy
/
pytest2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merged in nicoddemus/pytest/pastebin-xdist (pull request #239) 

Using pytest-xdist and --paste=all results in error
This commit is contained in:
holger krekel 2015-01-19 22:34:43 +01:00
parent 8859936301 7e15fb7f2d
commit 394367e1d2
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
_pytest/pastebin.py
View File

@ -14,13 +14,17 @@ def pytest_addoption(parser):
@pytest.mark.trylast @pytest.mark.trylast
def pytest_configure(config): def pytest_configure(config):
if config.option.pastebin == "all": if config.option.pastebin == "all":
config._pastebinfile = tempfile.TemporaryFile('w+')
tr = config.pluginmanager.getplugin('terminalreporter') tr = config.pluginmanager.getplugin('terminalreporter')
oldwrite = tr._tw.write # if no terminal reporter plugin is present, nothing we can do here;
def tee_write(s, **kwargs): # this can happen when this function executes in a slave node
oldwrite(s, **kwargs) # when using pytest-xdist, for example
config._pastebinfile.write(str(s)) if tr is not None:
tr._tw.write = tee_write config._pastebinfile = tempfile.TemporaryFile('w+')
oldwrite = tr._tw.write
def tee_write(s, **kwargs):
oldwrite(s, **kwargs)
config._pastebinfile.write(str(s))
tr._tw.write = tee_write
def pytest_unconfigure(config): def pytest_unconfigure(config):
if hasattr(config, '_pastebinfile'): if hasattr(config, '_pastebinfile'):