Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use _setupStdout method in unittest-xml-reporting

Best Python code snippet using unittest-xml-reporting_python

suite.py

Source:suite.py

1"""TestSuite"""2import sys3from . import case4from . import util5__unittest = True6def _call_if_exists(parent, attr):7    func = getattr(parent, attr, lambda: None)8    func()9class BaseTestSuite(object):10    """A simple test suite that doesn't provide class or module shared fixtures.11    """12    def __init__(self, tests=()):13        self._tests = []14        self.addTests(tests)15    def __repr__(self):16        return "<%s tests=%s>" % (util.strclass(self.__class__), list(self))17    def __eq__(self, other):18        if not isinstance(other, self.__class__):19            return NotImplemented20        return list(self) == list(other)21    def __ne__(self, other):22        return not self == other23    # Can't guarantee hash invariant, so flag as unhashable24    __hash__ = None25    def __iter__(self):26        return iter(self._tests)27    def countTestCases(self):28        cases = 029        for test in self:30            cases += test.countTestCases()31        return cases32    def addTest(self, test):33        # sanity checks34        if not hasattr(test, '__call__'):35            raise TypeError("{} is not callable".format(repr(test)))36        if isinstance(test, type) and issubclass(test,37                                                 (case.TestCase, TestSuite)):38            raise TypeError("TestCases and TestSuites must be instantiated "39                            "before passing them to addTest()")40        self._tests.append(test)41    def addTests(self, tests):42        if isinstance(tests, basestring):43            raise TypeError("tests must be an iterable of tests, not a string")44        for test in tests:45            self.addTest(test)46    def run(self, result):47        for test in self:48            if result.shouldStop:49                break50            test(result)51        return result52    def __call__(self, *args, **kwds):53        return self.run(*args, **kwds)54    def debug(self):55        """Run the tests without collecting errors in a TestResult"""56        for test in self:57            test.debug()58class TestSuite(BaseTestSuite):59    """A test suite is a composite test consisting of a number of TestCases.60    For use, create an instance of TestSuite, then add test case instances.61    When all tests have been added, the suite can be passed to a test62    runner, such as TextTestRunner. It will run the individual test cases63    in the order in which they were added, aggregating the results. When64    subclassing, do not forget to call the base class constructor.65    """66    def run(self, result, debug=False):67        topLevel = False68        if getattr(result, '_testRunEntered', False) is False:69            result._testRunEntered = topLevel = True70        for test in self:71            if result.shouldStop:72                break73            if _isnotsuite(test):74                self._tearDownPreviousClass(test, result)75                self._handleModuleFixture(test, result)76                self._handleClassSetUp(test, result)77                result._previousTestClass = test.__class__78                if (getattr(test.__class__, '_classSetupFailed', False) or79                    getattr(result, '_moduleSetUpFailed', False)):80                    continue81            if not debug:82                test(result)83            else:84                test.debug()85        if topLevel:86            self._tearDownPreviousClass(None, result)87            self._handleModuleTearDown(result)88            result._testRunEntered = False89        return result90    def debug(self):91        """Run the tests without collecting errors in a TestResult"""92        debug = _DebugResult()93        self.run(debug, True)94    ################################95    def _handleClassSetUp(self, test, result):96        previousClass = getattr(result, '_previousTestClass', None)97        currentClass = test.__class__98        if currentClass == previousClass:99            return100        if result._moduleSetUpFailed:101            return102        if getattr(currentClass, "__unittest_skip__", False):103            return104        try:105            currentClass._classSetupFailed = False106        except TypeError:107            # test may actually be a function108            # so its class will be a builtin-type109            pass110        setUpClass = getattr(currentClass, 'setUpClass', None)111        if setUpClass is not None:112            _call_if_exists(result, '_setupStdout')113            try:114                setUpClass()115            except Exception as e:116                if isinstance(result, _DebugResult):117                    raise118                currentClass._classSetupFailed = True119                className = util.strclass(currentClass)120                errorName = 'setUpClass (%s)' % className121                self._addClassOrModuleLevelException(result, e, errorName)122            finally:123                _call_if_exists(result, '_restoreStdout')124    def _get_previous_module(self, result):125        previousModule = None126        previousClass = getattr(result, '_previousTestClass', None)127        if previousClass is not None:128            previousModule = previousClass.__module__129        return previousModule130    def _handleModuleFixture(self, test, result):131        previousModule = self._get_previous_module(result)132        currentModule = test.__class__.__module__133        if currentModule == previousModule:134            return135        self._handleModuleTearDown(result)136        result._moduleSetUpFailed = False137        try:138            module = sys.modules[currentModule]139        except KeyError:140            return141        setUpModule = getattr(module, 'setUpModule', None)142        if setUpModule is not None:143            _call_if_exists(result, '_setupStdout')144            try:145                setUpModule()146            except Exception, e:147                if isinstance(result, _DebugResult):148                    raise149                result._moduleSetUpFailed = True150                errorName = 'setUpModule (%s)' % currentModule151                self._addClassOrModuleLevelException(result, e, errorName)152            finally:153                _call_if_exists(result, '_restoreStdout')154    def _addClassOrModuleLevelException(self, result, exception, errorName):155        error = _ErrorHolder(errorName)156        addSkip = getattr(result, 'addSkip', None)157        if addSkip is not None and isinstance(exception, case.SkipTest):158            addSkip(error, str(exception))159        else:160            result.addError(error, sys.exc_info())161    def _handleModuleTearDown(self, result):162        previousModule = self._get_previous_module(result)163        if previousModule is None:164            return165        if result._moduleSetUpFailed:166            return167        try:168            module = sys.modules[previousModule]169        except KeyError:170            return171        tearDownModule = getattr(module, 'tearDownModule', None)172        if tearDownModule is not None:173            _call_if_exists(result, '_setupStdout')174            try:175                tearDownModule()176            except Exception as e:177                if isinstance(result, _DebugResult):178                    raise179                errorName = 'tearDownModule (%s)' % previousModule180                self._addClassOrModuleLevelException(result, e, errorName)181            finally:182                _call_if_exists(result, '_restoreStdout')183    def _tearDownPreviousClass(self, test, result):184        previousClass = getattr(result, '_previousTestClass', None)185        currentClass = test.__class__186        if currentClass == previousClass:187            return188        if getattr(previousClass, '_classSetupFailed', False):189            return190        if getattr(result, '_moduleSetUpFailed', False):191            return192        if getattr(previousClass, "__unittest_skip__", False):193            return194        tearDownClass = getattr(previousClass, 'tearDownClass', None)195        if tearDownClass is not None:196            _call_if_exists(result, '_setupStdout')197            try:198                tearDownClass()199            except Exception, e:200                if isinstance(result, _DebugResult):201                    raise202                className = util.strclass(previousClass)203                errorName = 'tearDownClass (%s)' % className204                self._addClassOrModuleLevelException(result, e, errorName)205            finally:206                _call_if_exists(result, '_restoreStdout')207class _ErrorHolder(object):208    """209    Placeholder for a TestCase inside a result. As far as a TestResult210    is concerned, this looks exactly like a unit test. Used to insert211    arbitrary errors into a test suite run.212    """213    # Inspired by the ErrorHolder from Twisted:214    # http://twistedmatrix.com/trac/browser/trunk/twisted/trial/runner.py215    # attribute used by TestResult._exc_info_to_string216    failureException = None217    def __init__(self, description):218        self.description = description219    def id(self):220        return self.description221    def shortDescription(self):222        return None223    def __repr__(self):224        return "<ErrorHolder description=%r>" % (self.description,)225    def __str__(self):226        return self.id()227    def run(self, result):228        # could call result.addError(...) - but this test-like object229        # shouldn't be run anyway230        pass231    def __call__(self, result):232        return self.run(result)233    def countTestCases(self):234        return 0235def _isnotsuite(test):236    "A crude way to tell apart testcases and suites with duck-typing"237    try:238        iter(test)239    except TypeError:240        return True241    return False242class _DebugResult(object):243    "Used by the TestSuite to hold previous class when running in debug."244    _previousTestClass = None245    _moduleSetUpFailed = False...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.