U 42i<@sdZddlmZddlZddlZddlZddlZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lm Z dd lmZdd lmZddlZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddl m!Z!ddl m"Z"ddl m#Z#ddl$m%Z%ddl&Z&ej'dddkr\ddl(m)Z)errddl*Z*ddl+Z,ee e e-e-ej.fe dfZ/dddd d!d"d#Z0Gd$d%d%e!Z1Gd&d'd'e"Z2ed(d)d*d+d,d-d.d/Z3d0a4ed(d1d*d2d3d4d5Z5d6d7d8d9Z6dS):z4Discover and run std-library "unittest" style tests.) annotationsN)Any)Callable) Generator)Iterable)Tuple)Type) TYPE_CHECKING)Union)is_async_function)hookimpl)FixtureRequest) Collector)Item)exit)fail)skip)xfail)Class)Function)Module)CallInfo) )ExceptionGroup)NNNzModule | ClassstrobjectzUnitTestCase | None) collectornameobjreturncCsVz tjd}t||jsWdSWntk r6YdSXt|rFdStj|||dS)Nunittest)rr ) sysmodules issubclassTestCase Exceptioninspect isabstract UnitTestCase from_parent)rrr utr-4/tmp/pip-unpacked-wheel-7n2p7kht/_pytest/unittest.pypytest_pycollect_makeitem3s    r/c@sFeZdZdZddZddddZdd d d d Zdd d d dZdS)r*TcCs |dS)NrunTestr selfr-r-r. newinstanceLszUnitTestCase.newinstancezIterable[Item | Collector]r!c csddlm}|j}t|dds"dSt|}|sJ||||||jj | |j |}d}| |jD]2}t|j|}t|ddsqvtj||dVd}qv|st|jdd}|dk rtjdd} | dks|| jjkrtj|ddVdS) Nr) TestLoaderZ__test__TF)rr0twisted.trial.unittest)r"r6r getattr _is_skipped'_register_unittest_setup_method_fixture&_register_unittest_setup_class_fixtureZ_register_setup_class_fixturesession_fixturemanagerZparsefactoriesr4nodeidZgetTestCaseNamesTestCaseFunctionr+r#r$getr&r0) r3r6clsskippedloaderZfoundsomethingrxruntestr,r-r-r.collectTs0      zUnitTestCase.collecttypeNone)rAr!cstddtdddkr,dkr,dStdddddfd d d d d fdd }|jjjdj||jddddS)zSRegister an auto-use fixture to invoke setUpClass and tearDownClass (#517).Z setUpClassNZ tearDownClassZdoClassCleanupscSsdSNr-r-r-r-r.zzEUnitTestCase._register_unittest_setup_class_fixture..rHr5csFtdd}|sdSdd|D}t|dkr8|dn td|dS)NZtearDown_exceptionscSsg|]\}}}|qSr-r-).0_excr-r-r. szlUnitTestCase._register_unittest_setup_class_fixture..process_teardown_exceptions..rzUnittest class cleanup errors)r8lenr)Z exc_infos exceptions)rAr-r.process_teardown_exceptions|s   zXUnitTestCase._register_unittest_setup_class_fixture..process_teardown_exceptionsr Generator[None]requestr!c3s|j}t|r$|j}tjj|dddk rZz Wn"tk rXYnXdVzdk rpW5XdSNT)Z_use_item_location)rAr9__unittest_skip_why__pytestrr')rVrAreason)cleanuprSsetupteardownr-r.unittest_setup_class_fixtures"  zYUnitTestCase._register_unittest_setup_class_fixture..unittest_setup_class_fixtureZ_unittest_setUpClass_fixture_classTrfuncr>ZscopeZautouser8r<r=Z_register_fixture __qualname__r>)r3rAr^r-)r[rArSr\r]r.r;ss   z3UnitTestCase._register_unittest_setup_class_fixturecsht|ddt|dddkr,dkr,dSdddfdd }|jjjd |j||jd d d dS) zWRegister an auto-use fixture to invoke setup_method and teardown_method (#517).Z setup_methodNZteardown_methodr rTrUc3sV|j}t|r$|j}tjj|dddk r8||jdVdk rR||jdSrW)instancer9rXrYrr'function)rVr3rZr\r]r-r.unittest_setup_method_fixtures z[UnitTestCase._register_unittest_setup_method_fixture..unittest_setup_method_fixtureZ_unittest_setup_method_fixture_reTr`rb)r3rArgr-rfr.r:s    z4UnitTestCase._register_unittest_setup_method_fixtureN)__name__ __module__rc nofuncargsr4rFr;r:r-r-r-r.r*Gs 8r*cs.eZdZUdZdZded<ddZeddZd d fd d Z d d fd d Z dd dddZ dd dddZ ddd dddZ ddd dddZddd dddZd6dddd d!d"d#Zd7dd$d dd%d&Zdd dd'd(Zdd dd)d*Zdd+d d,d-d.Zd d d/d0Zd1d2d3fd4d5 ZZS)8r?TNz7list[_pytest._code.ExceptionInfo[BaseException]] | None_excinfocCst|jtst|j|jSrI) isinstanceparentr*AssertionErrorr rr2r-r-r. _getinstanceszTestCaseFunction._getinstancecCs|jSrI)rdr2r-r-r. _testcaseszTestCaseFunction._testcaserHr5csd|_tdSrI)_explicit_tearDownsuperr\r2 __class__r-r.r\szTestCaseFunction.setupcs0|jdk r|d|_d|_|`tdSrI)rq_objZ _instancerrr]r2rsr-r.r]s  zTestCaseFunction.teardownzunittest.TestCase)testcaser!cCsdSrIr-r3rvr-r-r. startTestszTestCaseFunction.startTest_SysExcInfoType) rawexcinfor!c Cst|d|}z"tjjt|}|j}|j}Wntk rzrz,tj |}| ddt d |ddWn@t j tfk rYn&tk rt d|ddYnXWn6tk rYn"t j k rtjj}YnXYnX|jdg|dS) N _rawexcinforzCNOTE: Incompatible Exception Representation, displaying natively: FZpytracez6ERROR: Unknown Incompatible Exception representation: rk)r8_pytest_codeZ ExceptionInfo BaseExceptionZ from_exc_infovalue traceback TypeErrorformat_exceptioninsertrjoinr'KeyboardInterruptZ from_current__dict__ setdefaultappend)r3rzexcinforMvaluesr-r-r. _addexcinfos8    zTestCaseFunction._addexcinfo)rvrzr!cCsFz"t|dtjr t|djWntk r6YnX||dS)NrP)rlrr'msgrrr3rvrzr-r-r.addError s zTestCaseFunction.addErrorcCs||dSrI)rrr-r-r. addFailureszTestCaseFunction.addFailurer)rvrZr!cCs>ztjj|ddWn$tjk r8|tYnXdSrW)rYrr'rr#exc_info)r3rvrZr-r-r.addSkipszTestCaseFunction.addSkipr|)rvrzrZr!cCs:ztt|Wn$tjk r4|tYnXdSrI)rrr'rr#r)r3rvrzrZr-r-r.addExpectedFailure sz#TestCaseFunction.addExpectedFailurez"twisted.trial.unittest.Todo | NonecCsRd}|r|d|j7}zt|ddWn$tjk rL|tYnXdS)NzUnexpected successz: Fr})rZrr'rr#r)r3rvrZrr-r-r.addUnexpectedSuccess+sz%TestCaseFunction.addUnexpectedSuccesscCsdSrIr-rwr-r-r. addSuccess9szTestCaseFunction.addSuccesscCsdSrIr-rwr-r-r.stopTest<szTestCaseFunction.stopTestfloat)rvelapsedr!cCsdSrIr-)r3rvrr-r-r. addDuration?szTestCaseFunction.addDurationc Csddlm}|j}|dk st||t|jr<||dn|t|jtsLtt |jp`t |jj}|j dr|s|j |_ t|dddt||j|jz||dW5t||jXdS)Nr)&maybe_wrap_pytest_function_for_tracing)resultZusepdbtearDowncWsdSrIr-)argsr-r-r.rJYrKz*TestCaseFunction.runtest..)Z_pytest.debuggingrrdrnr r rlrmr*r9configZ getoptionrrqsetattrrdelattr)r3rrvrBr-r-r.rEBs    zTestCaseFunction.runtestz*_pytest._code.ExceptionInfo[BaseException]z_pytest._code.Traceback)rr!cs&t|}|dd}|s"|}|S)NcSs|jjd S)NZ __unittest)frame f_globalsr@)rDr-r-r.rJhrKz4TestCaseFunction._traceback_filter..)rr_traceback_filterfilter)r3rrZ ntracebackrsr-r.rcs z"TestCaseFunction._traceback_filter)r|)N)rhrircrjrk__annotations__ropropertyrpr\r]rxrrrrrrrrrrEr __classcell__r-r-rsr.r?s*   #  !r?T)ZtryfirstrzCallInfo[None]rH)itemcallr!cst|tr<|jr<|jd|_z|`Wntk r:YnXtj d}|r|jrt|jj |j r|jt d fdd|j}|j|_dS)Nrr"csttjSrI)rYrrrr-rr-r.rJrKz+pytest_runtest_makereport..)rlr?rkpoprrAttributeErrorr#r$r@rZSkipTestrZ from_callwhen)rrr"Zcall2r-rr.pytest_runtest_makereportos   rF)wrapperzGenerator[None, object, object])rr!c#st|tr~dtjkr~tjd}|jjtsPddlm}ddl m }|t|dad fdd }||j_z dV}W5|j_XndV}|S) Nr7ztwisted.python.failurer) IReporter)classImplementsTcsn|dkrt|_n|dkr$t|}|||f|_z|||||dWn"tk rh||||YnXdS)N) captureVars)r#rr{rGr)r3 exc_valueexc_typeexc_tbrZFailure__init__r-r.excstores   z)pytest_runtest_protocol..excstore)NNNN) rlr?r#r$ZFailure__init__classImplements_has_runZtwisted.trial.itrialrZzope.interfacer)rr,rrrresr-rr.pytest_runtest_protocols&      rboolr5cCstt|ddS)zDReturn True if the given object has been marked with @unittest.skip.Z__unittest_skip__F)rr8r1r-r-r.r9sr9)7__doc__ __future__rr(r#rtypestypingrrrrrrr r Z _pytest._coder~Z_pytest.compatr Z_pytest.configr Z_pytest.fixturesr Z _pytest.nodesrrZ_pytest.outcomesrrrrZ_pytest.pythonrrrZ_pytest.runnerrrY version_infoZexceptiongrouprr"Ztwisted.trial.unittestZtwistedr TracebackTyperyr/r*r?rrrr9r-r-r-r.sb                       &&