o ¯bæDã@s–dZddlZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z Gdd „d ƒZGd d „d e ƒZGd d „d eƒZGdd„de ƒZdS)z? Tests for Trial's interaction with the Python warning system. éN)ÚStringIO)Ú TestResult)ÚFilePath)Ú_collectWarningsÚ_setWarningRegistryToNone)ÚSynchronousTestCasec@s eZdZdZGdd„deƒZdS)ÚMaskzQ Hide a test case definition from trial's automatic discovery mechanism. c@s(eZdZdZdZeZdd„Zdd„ZdS)zMask.MockTestsz A test case which is used by L{FlushWarningsTests} to verify behavior which cannot be verified by code inside a single test method. úsome warning textcCst |j|j¡dS)z@ Generate a warning and don't flush it. N)ÚwarningsÚwarnÚmessageÚcategory©Úself©rúA/usr/lib/python3/dist-packages/twisted/trial/test/test_warning.pyÚtest_unflushed!szMask.MockTests.test_unflushedcCs(t |j|j¡| t| ¡ƒd¡dS)z: Generate a warning and flush it. éN)r r r r Ú assertEqualÚlenÚ flushWarningsrrrrÚ test_flushed'szMask.MockTests.test_flushedN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r Ú UserWarningr rrrrrrÚ MockTestss  rN)rrrrrrrrrrrsrc@s˜eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$S)%ÚFlushWarningsTestsze Tests for C{flushWarnings}, an API for examining the warnings emitted so far in a test. cCs&| ¡D] \}}| |||¡qdS)z Assert that all the keys present in C{subset} are also present in C{set} and that the corresponding values are equal. N)Úitemsr)rÚsetÚsubsetÚkÚvrrrÚassertDictSubset5sÿz#FlushWarningsTests.assertDictSubsetcCs8| t|ƒt|ƒ¡t||ƒD] \}}| ||¡qdS)zµ For each pair of corresponding elements in C{sets} and C{subsets}, assert that the element from C{subsets} is a subset of the element from C{sets}. N)rrÚzipr$)rÚsetsÚsubsetsÚaÚbrrrÚassertDictSubsets=sÿz$FlushWarningsTests.assertDictSubsetscCs| | ¡g¡dS)zg If no warnings are emitted by a test, C{flushWarnings} returns an empty list. N)rrrrrrÚ test_noneGszFlushWarningsTests.test_nonecCsPd}t}tj||dd}t}tj||d| | ¡||dœ||dœg¡dS)z| If several warnings are emitted by a test, C{flushWarnings} returns a list containing all of them. zfirst warning message©r r zsecond warning message©r r N)rr r ÚRuntimeWarningr*r)rÚ firstMessageÚ firstCategoryÚ secondMessageÚsecondCategoryrrrÚ test_severalNsþþzFlushWarningsTests.test_severalcCsDd}t}tdƒD] }tj||dq| | ¡||dœgd¡dS)zŠ The same warning triggered twice from the same place is included twice in the list returned by C{flushWarnings}. ú the messageér,r-N)r.Úranger r r*r)rr r ÚirrrÚ test_repeatedcs ÿz FlushWarningsTests.test_repeatedcCsBd}t}tj||d| | ¡||dœg¡| | ¡g¡dS)z‰ After a particular warning event has been returned by C{flushWarnings}, it is not returned by subsequent calls. r4r,r-N)r.r r r*rr)rr r rrrÚ test_clearedqsÿzFlushWarningsTests.test_clearedcCsªtƒ}t d¡}| |¡| tjjg¡}| |ddd¡| |ddt¡t |ƒjj }|j }|j d}| |dd|¡| |dd|¡| t |ƒd ¡d S) zx Any warnings emitted by a test which are not flushed are emitted to the Python warning system. rrr r r éÚfilenameÚlinenorN)rrrÚrunrrrÚassertIdenticalrÚtypeÚ__code__Ú co_filenameÚco_firstlinenor)rÚresultÚcaseÚ warningsShownÚwherer;r<rrrr~s    z!FlushWarningsTests.test_unflushedcCsJtƒ}t d¡}tƒ}| td|¡}| |¡| ¡| |  ¡d¡dS)zx Any warnings emitted by a test which are flushed are not emitted to the Python warning system. rÚstdoutÚN) rrrrÚpatchÚsysr=ÚrestorerÚgetvalue)rrCrDÚoutputÚmonkeyrrrr•s  zFlushWarningsTests.test_flushedc Cs¸Gdd„dtƒ}tƒ}t d¡}||_tjdd…}z9t d¡| |¡|  t |j ƒd¡|  |j dd|¡|  |j dd ¡d d ¡¡W|tjdd…<dS|tjdd…<w) z¾ If a warnings filter has been installed which turns warnings into exceptions, tests have an error added to the reporter for them for each unflushed warning. c@ó eZdZdS)zIFlushWarningsTests.test_warningsConfiguredAsErrors..CustomWarningN©rrrrrrrÚ CustomWarning©órQrNÚerrorrréÿÿÿÿz CustomWarning: some warning text)ÚWarningrrrr r ÚfiltersÚ simplefilterr=rrÚerrorsr>Ú assertTrueÚ splitlinesÚendswith©rrQrCrDÚoriginalWarningsrrrÚtest_warningsConfiguredAsErrors¢s"    ÿû"z2FlushWarningsTests.test_warningsConfiguredAsErrorsc Cs|Gdd„dtƒ}tƒ}t d¡}||_tjdd…}zt d¡| |¡|  |j g¡W|tjdd…<dS|tjdd…<w)zÊ If a warnings filter has been installed which turns warnings into exceptions, tests which emit those warnings but flush them do not have an error added to the reporter. c@rO)zPFlushWarningsTests.test_flushedWarningsConfiguredAsErrors..CustomWarningNrPrrrrrQÇrRrQrNrS) rUrrrr r rVrWr=rrXr\rrrÚ&test_flushedWarningsConfiguredAsErrorsÀs   "z9FlushWarningsTests.test_flushedWarningsConfiguredAsErrorscCs@t d¡| t| ¡ƒd¡t d¡| t| ¡ƒd¡dS)zƒ Any warnings emitted after a call to C{flushWarnings} can be flushed by another call to C{flushWarnings}. z first messagerzsecond messageN)r r rrrrrrrÚtest_multipleFlushesÖs  z'FlushWarningsTests.test_multipleFlushescsxd‰t‰‡‡fdd„}d‰t‰‡‡fdd„}|ƒ|ƒ| |j|gdˆˆdœg¡| |j|gdˆˆdœg¡d S) zå The list returned by C{flushWarnings} includes only those warnings which refer to the source of the function passed as the value for C{offendingFunction}, if a value is passed for that parameter. zfirst warning textcótjˆˆdddS©Nr)Ú stacklevel©r r r)r0r/rrÚoneéóz>FlushWarningsTests.test_filterOnOffendingFunction..onez some textcrarbrdr)r2r1rrÚtwoïrfz>FlushWarningsTests.test_filterOnOffendingFunction..two©ÚoffendingFunctionsr-N)rr.r*r)rrergr)r0r/r2r1rÚtest_filterOnOffendingFunctionàs   þ  þz1FlushWarningsTests.test_filterOnOffendingFunctioncCs,dd„}|ƒ| t|j|gdƒd¡dS)z Verify that warnings emitted at the very edges of a function are still determined to be emitted from that function. cSs"t d¡t d¡t d¡dS)Nzfirst line warningzinternal line warningzlast line warningrdrrrrÚwarners  z:FlushWarningsTests.test_functionBoundaries..warnerrhéN©rrr)rrkrrrÚtest_functionBoundariesþsz*FlushWarningsTests.test_functionBoundariescCs4t d¡| t|jdg¡| t| ¡ƒd¡dS)zÏ If an object which is neither a function nor a method is included in the C{offendingFunctions} list, C{flushWarnings} raises L{ValueError}. Such a call flushes no warnings. úoh noNr)r r Ú assertRaisesÚ ValueErrorrrrrrrrÚtest_invalidFilter s z%FlushWarningsTests.test_invalidFiltercCsÐt| ¡ d¡ƒ d¡}| ¡| d¡ d¡| d¡ d¡| ¡j d¡}t j  d|¡|  t jj |¡ddl m}|  t jjd ¡|  t jj|j¡| d¡  ¡| ¡| t| |jg¡ƒd ¡d S) zx Warnings emitted by a function the source code of which is not available can still be flushed. úutf-8ótwisted_private_helperó __init__.pyósmissingsourcefile.pyó7 import warnings def foo(): warnings.warn("oh no") r)ÚmissingsourcefileÚtwisted_private_helperrN)rÚmktempÚencodeÚchildÚmakedirsÚ setContentÚparentÚpathÚdecoderJÚinsertÚ addCleanupÚremoveryrxÚmodulesÚpoprÚfoorrr)rÚpackageÚ pathEntryrxrrrÚtest_missingSources"ÿ ÿ z%FlushWarningsTests.test_missingSourcecCst| ¡ d¡ƒ d¡}| ¡| d¡ d¡| d¡ d¡| ¡j d¡}t j  d|¡|  t jj |¡ddl m}t jd =t j|j=zdd lm}Wn tyWYnw|ƒ| | d ¡¡ddlm}|  t jjd ¡|  t jj|j¡| ¡| t| |jg¡ƒd ¡dS)a  Warnings emitted by a function defined in a file which has been renamed since it was initially compiled can still be flushed. This is testing the code which specifically supports working around the unfortunate behavior of CPython to write a .py source file name into the .pyc files it generates and then trust that it is correct in various places. If source files are renamed, .pyc files may not be regenerated, but they will contain incorrect filenames. rsrtrurvs module.pyrwr)Úmodulery)Úinvalidate_cachesstwisted_renamed_helperÚtwisted_renamed_helperrN)rrzr{r|r}r~rr€rrJr‚rƒr„ryr‹r…rÚ importlibrŒÚ ImportErrorÚmoveToÚsiblingrr†r‡rrr)rrˆr‰r‹rŒrrrÚtest_renamedSource3s4 ÿ ÿ   ÿ z%FlushWarningsTests.test_renamedSourcecCs,ddd„}|ƒ| t| |g¡ƒd¡dS)a¨ In Python 3.6 the dis.findlinestarts documented behaviour was changed such that the reported lines might not be sorted ascending. In Python 3.10 PEP 626 introduced byte-code change such that the last line of a function wasn't always associated with the last byte-code. In the past flushWarning was not detecting that such a function was associated with any warnings. rcSs |r|r t d¡dS dSdS)Nrord)r(r)rrrr‡ys üzCFlushWarningsTests.test_offendingFunctions_deep_branch..fooN)rrrm)rr‡rrrÚ#test_offendingFunctions_deep_branchos z6FlushWarningsTests.test_offendingFunctions_deep_branchN)rrrrr$r*r+r3r8r9rrr^r_r`rjrnrrrŠr’r“rrrrr/s&      .frr–r:r—é)rUrr˜rr Ústrr)rrœrr™rÚtest_callsObservers  z'CollectWarningsTests.test_callsObservercCs:tƒ}| td|¡tdd„tjdƒ| | ¡d¡dS)z” Any warnings emitted by a call to a function passed to L{_collectWarnings} are not actually emitted to the warning system. rGcSódS©Nr©ÚxrrrÚ°óz6CollectWarningsTests.test_suppresses..ÚtextrHN)rrIrJrr r rrL)rrMrrrÚtest_suppresses©sz$CollectWarningsTests.test_suppressescsXg‰tƒ‰‡‡fdd„}tdd„|ddddd }| ˆd ddd œfg¡| |ˆ¡d S) z€ L{_collectWarnings} returns the result of calling the callable passed to it with the parameters given. csˆ ||f¡ˆSr¡)r˜)ÚargsÚkwargs©Ú argumentsÚvaluerrrœ»sz2CollectWarningsTests.test_callsFunction..fcSr r¡rr¢rrrr¤¿r¥z9CollectWarningsTests.test_callsFunction..rr(r5Úd)r)Úc)rr(N)Úobjectrrr>)rrœrCrrªrÚtest_callsFunction³s z'CollectWarningsTests.test_callsFunctioncCsdbdd„}t d¡|ƒg}t|j|ƒ| t|ƒd¡| |djd¡| t| ¡ƒd¡dS)zì Subsequent emissions of a warning from a particular source site can be collected by L{_collectWarnings}. In particular, the per-module emitted-warning cache should be bypassed (I{__warningregistry__}). cSst d¡dS)Nr‡rdrrrrrœÑóz>CollectWarningsTests.test_duplicateWarningCollected..fÚdefaultrrr‡N) Ú__warningregistry__r rWrr˜rrr r)rrœršrrrÚtest_duplicateWarningCollectedÃs   z3CollectWarningsTests.test_duplicateWarningCollectedcCs,tƒ}|tj|<| tjj|¡| ¡dS)z± L{_collectWarnings}'s behavior is not altered by the presence of an object which cannot have attributes set on it as a value in C{sys.modules}. N)r¯rJr…rƒr†r´©rÚkeyrrrÚtest_immutableObjectÜs  z)CollectWarningsTests.test_immutableObjectcsni‰G‡fdd„dƒ}tƒ}tƒ}||ƒˆ|<tƒ}tƒ}||ƒˆ|<tˆƒ| ||||htˆ ¡ƒ¡dS)aÙ If the dictionary passed to L{_setWarningRegistryToNone} changes size partway through the process, C{_setWarningRegistryToNone} continues to set C{__warningregistry__} to L{None} on the rest of the values anyway. This might be caused by C{sys.modules} containing something that's not really a module and imports things on setattr. py.test does this, as does L{twisted.python.deprecate.deprecatedModuleAttribute}. cs eZdZdd„Z‡fdd„ZdS)zKCollectWarningsTests.test_setWarningRegistryChangeWhileIterating..AcSs||jd<dS)NÚ_key)Ú__dict__rµrrrÚ__init__õr±zTCollectWarningsTests.test_setWarningRegistryChangeWhileIterating..A.__init__csdˆ|j<dSr¡)r¸)rr¬Úitem©r­rrÚ __setattr__ør±zWCollectWarningsTests.test_setWarningRegistryChangeWhileIterating..A.__setattr__N)rrrrºr½rr¼rrÚAôsr¾N)r¯rrr Úkeys)rr¾Úkey1Úkey2Úkey3Úkey4rr¼rÚ+test_setWarningRegistryChangeWhileIteratingçs    z@CollectWarningsTests.test_setWarningRegistryChangeWhileIteratingN) rrrrrŸr§r°r´r·rÄrrrrr•‹s  r•)rrJr ÚiorÚunittestrÚtwisted.python.filepathrÚtwisted.trial._synctestrrÚtwisted.trial.unittestrrrrUr”r•rrrrÚs    Z