compat.py 716 B

12345678910111213141516171819202122232425262728293031
  1. try:
  2. from StringIO import StringIO
  3. except ImportError:
  4. from io import StringIO
  5. import pytest
  6. StringIO # pyflakes, this is for re-export
  7. if hasattr(pytest, 'hookimpl'):
  8. hookwrapper = pytest.hookimpl(hookwrapper=True)
  9. else:
  10. hookwrapper = pytest.mark.hookwrapper
  11. class SessionWrapper(object):
  12. def __init__(self, session):
  13. self._session = session
  14. if hasattr(session, 'testsfailed'):
  15. self._attr = 'testsfailed'
  16. else:
  17. self._attr = '_testsfailed'
  18. @property
  19. def testsfailed(self):
  20. return getattr(self._session, self._attr)
  21. @testsfailed.setter
  22. def testsfailed(self, value):
  23. setattr(self._session, self._attr, value)