o >hV@s`dZddlmZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZmZmZmZm Z GdddeZ!GdddeZ"GdddZ#GdddZ$GdddeZ%GdddeZ&GdddeZ'GdddeZ(Gdd d eZ)Gd!d"d"eZ*dS)#z! Tests for L{twisted.trial.util} ) annotationsN)StringIO) Generator) implementer) assert_thatequal_to) DelayedCall)IProcessTransport)filepath)Failure)util)SynchronousTestCase)DirtyReactorAggregateError_JanitoracquireAttributeexcInfoOrFailureToExcInfo openTestLogc@8eZdZdZd ddZd ddZd dd Zd d d Zd S) MktempTestszm Tests for L{TestCase.mktemp}, a helper function for creating temporary file or directory names. returnNonecCs8|}tj|tjdd}||gddS)z The path name returned by C{mktemp} is directly beneath a directory which identifies the test method which created the name. N)ztwisted.trial.test.test_utilr test_name)mktempospathdirnamesplitsep assertEqual)selfnamedirsr#|/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/trial/test/test_util.pyr)s zMktempTests.test_namecCs|}|||dS)zF Repeated calls to C{mktemp} return different values. N)rassertNotEqual)r r!r#r#r$ test_unique4szMktempTests.test_uniquecCs<|}tj|}|tj||tj|dS)zS The directory part of the path name returned by C{mktemp} exists. N)rrrr assertTrueexists assertFalse)r r!rr#r#r$ test_created;s zMktempTests.test_createdcCs(tj|}||tdS)zZ The path returned by C{mktemp} is beneath the current working directory. N)rrabspathrr' startswithgetcwd)r rr#r#r$ test_locationDszMktempTests.test_locationNrr)__name__ __module__ __qualname____doc__rr&r*r.r#r#r#r$r#s    rc@.eZdZdZd ddZd ddZd dd Zd S) DirtyReactorAggregateErrorTestsz6 Tests for the L{DirtyReactorAggregateError}. rrcCs tddg}|t|ddS)z5 Delayed calls are formatted nicely. FoobarzhReactor was unclean. DelayedCalls: (set twisted.internet.base.DelayedCall.debug = True to debug) Foo barNrrstrr errorr#r#r$test_formatDelayedCallQs z6DirtyReactorAggregateErrorTests.test_formatDelayedCallcCs"tgddg}|t|ddS)z3 Selectables are formatted nicely. z selectable 1z selectable 2z;Reactor was unclean. Selectables: selectable 1 selectable 2Nr8r:r#r#r$test_formatSelectables_s z6DirtyReactorAggregateErrorTests.test_formatSelectablescCs&tddgddg}|t|ddS)zR Both delayed calls and selectables can appear in the same error. bleckBoozoSel1Sel2zReactor was unclean. DelayedCalls: (set twisted.internet.base.DelayedCall.debug = True to debug) bleck Boozo Selectables: Sel1 Sel2Nr8r:r#r#r$%test_formatDelayedCallsAndSelectablesms zEDirtyReactorAggregateErrorTests.test_formatDelayedCallsAndSelectablesNr/)r0r1r2r3r<r=rBr#r#r#r$r5Ls   r5c@s>eZdZdZ ddd d Zddd dZdddZdddZdS) StubReactora~ A reactor stub which contains enough functionality to be used with the L{_Janitor}. @ivar iterations: A list of the arguments passed to L{iterate}. @ivar removeAllCalled: Number of times that L{removeAll} was called. @ivar selectables: The value that will be returned from L{removeAll}. @ivar delayedCalls: The value to return from L{getDelayedCalls}. N delayedCallslist[DelayedCall] selectableslist[object] | NonerrcCs$||_g|_d|_|s g}||_dS)z @param delayedCalls: See L{StubReactor.delayedCalls}. @param selectables: See L{StubReactor.selectables}. rN)rD iterationsremoveAllCalledrF)r rDrFr#r#r$__init__s  zStubReactor.__init__timeout float | NonecC|j|dS)z/ Increment C{self.iterations}. N)rHappend)r rKr#r#r$iteratezStubReactor.iteratecCs|jS)z. Return C{self.delayedCalls}. )rDr r#r#r$getDelayedCallsszStubReactor.getDelayedCalls list[object]cCs|jd7_|jS)zS Increment C{self.removeAllCalled} and return C{self.selectables}. )rIrFrQr#r#r$ removeAllszStubReactor.removeAllN)rDrErFrGrr)rKrLrr)rrE)rrS)r0r1r2r3rJrOrRrUr#r#r#r$rCs  rCc@s$eZdZdZd ddZdd d Zd S)StubErrorReporterz A subset of L{twisted.trial.itrial.IReporter} which records L{addError} calls. @ivar errors: List of two-tuples of (test, error) which were passed to L{addError}. rrcC g|_dSrV)errorsrQr#r#r$rJ zStubErrorReporter.__init__testobjectr;r cCs|j||fdS)z6 Record parameters in C{self.errors}. N)rYrN)r r[r;r#r#r$addErrorszStubErrorReporter.addErrorNr/)r[r\r;r rr)r0r1r2r3rJr]r#r#r#r$rWs rWc@s~eZdZdZdddZdddZddd Zdd d Zdd d ZdddZ dddZ dddZ dddZ dddZ dddZdS) JanitorTestsz Tests for L{_Janitor}! rrcCs4tg}tdd|d}|||jddgdS)a During pending-call cleanup, the reactor will be spun twice with an instant timeout. This is not a requirement, it is only a test for current behavior. Hopefully Trial will eventually not do this kind of reactor stuff. Nreactorr)rCr _cleanPendingrrHr r`janr#r#r$test_cleanPendingSpinsReactorsz*JanitorTests.test_cleanPendingSpinsReactorcCsXd dd}g}td|di|jdd}t|g}td d |d }||||gd S) zW During pending-call cleanup, the janitor cancels pending timed calls. rr9cSdS)NLulzr#r#r#r#r$funcz8JanitorTests.test_cleanPendingCancelsCalls..func,r#cSdSrVr#xr#r#r$z.Nr_rr9)rrNrCrrar)r rg cancelled delayedCallr`rcr#r#r$test_cleanPendingCancelsCallss  z*JanitorTests.test_cleanPendingCancelsCallsc Cs`tddddiddddddd}t|}t|g}tdd|d }|}|||gdS) aK The Janitor produces string representations of delayed calls from the delayed call cleanup method. It gets the string representations *before* cancelling the calls; this is important because cancelling the call removes critical debugging information from the string representation. ricSrjrVr#r#r#r#r$rmrnzIJanitorTests.test_cleanPendingReturnsDelayedCallStrings..r#cSrjrVr#rkr#r#r$rmrncSreNrr#r#r#r#r$rmrnsecondsNr_)rr9rCrrar)r rqdelayedCallStringr`rcstringsr#r#r$*test_cleanPendingReturnsDelayedCallStringss z7JanitorTests.test_cleanPendingReturnsDelayedCallStringscCs0tg}tdd|d}|||jddS)zM The Janitor will remove selectables during reactor cleanup. Nr_rT)rCr _cleanReactorrrIrbr#r#r$#test_cleanReactorRemovesSelectablessz0JanitorTests.test_cleanReactorRemovesSelectablescCsRttGddd}|}tg|g}tdd|d}|||jdgdS)zI The Janitor will kill processes during reactor cleanup. c@s$eZdZdZd ddZd dd Zd S) zJJanitorTests.test_cleanReactorKillsProcesses..StubProcessTransportz A stub L{IProcessTransport} provider which records signals. @ivar signals: The signals passed to L{signalProcess}. rrcSrXrV)signalsrQr#r#r$rJrZzSJanitorTests.test_cleanReactorKillsProcesses..StubProcessTransport.__init__signal str | intcSrM)zF Append C{signal} to C{self.signals}. N)r{rN)r r|r#r#r$ signalProcessrPzXJanitorTests.test_cleanReactorKillsProcesses..StubProcessTransport.signalProcessNr/)r|r}rr)r0r1r2r3rJr~r#r#r#r$StubProcessTransports rNr_KILL)rr rCrryrr{)r rptr`rcr#r#r$test_cleanReactorKillsProcessess z,JanitorTests.test_cleanReactorKillsProcessescCs@Gddd}tg|g}tdd|d}||dgdS)z The Janitor returns string representations of the selectables that it cleaned up from the reactor cleanup method. c@seZdZdZdddZdS)zJJanitorTests.test_cleanReactorReturnsSelectableStrings..Selectablezl A stub Selectable which only has an interesting string representation. rr9cSre)N (SELECTABLE!)r#rQr#r#r$__repr__ rhzSJanitorTests.test_cleanReactorReturnsSelectableStrings..Selectable.__repr__Nro)r0r1r2r3rr#r#r#r$ SelectablesrNr_r)rCrrry)r rr`rcr#r#r$)test_cleanReactorReturnsSelectableStringss z6JanitorTests.test_cleanReactorReturnsSelectableStringscCsBtg}t}t}t|||d}||||jgdS)z The post-case cleanup method will return True and not call C{addError} on the result if there are no pending calls. r_N)rCr\rWrr'postCaseCleanuprrYr r`r[reporterrcr#r#r$test_postCaseCleanupNoErrors's z)JanitorTests.test_postCaseCleanupNoErrorsc Cstddddiddddddd}t|}t|gg}t}t}t|||d}|||t |j d ||j d d j j |gd S) z The post-case cleanup method will return False and call C{addError} on the result with a L{DirtyReactorAggregateError} Failure if there are pending calls. ricSrjrVr#r#r#r#r$rm:rnz=JanitorTests.test_postCaseCleanupWithErrors..r#cSrjrVr#rkr#r#r$rm:rncSrersr#r#r#r#r$rm:rnrtr_rTrN) rr9rCr\rWrr)rrlenrYvaluerDr rqrvr`r[rrcr#r#r$test_postCaseCleanupWithErrors3s  z+JanitorTests.test_postCaseCleanupWithErrorscCs<tg}t}t}t|||d}|||jgdS)z The post-class cleanup method will not call C{addError} on the result if there are no pending calls or selectables. r_N)rCr\rWrpostClassCleanuprrYrr#r#r$test_postClassCleanupNoErrorsEs z*JanitorTests.test_postClassCleanupNoErrorsc Cstddddiddddddd}t|}t|gg}t}t}t|||d}||t|j d ||j d d j j |gd S) z The post-class cleanup method call C{addError} on the result with a L{DirtyReactorAggregateError} Failure if there are pending calls. ricSrjrVr#r#r#r#r$rmWrnzIJanitorTests.test_postClassCleanupWithPendingCallErrors..r#cSrjrVr#rkr#r#r$rmWrncSrersr#r#r#r#r$rmWrnrtr_rTrN) rr9rCr\rWrrrrrYrrDrr#r#r$*test_postClassCleanupWithPendingCallErrorsQs  z7JanitorTests.test_postClassCleanupWithPendingCallErrorscCshd}tg|g}t}t}t|||d}||t|jd||jddjj t |gdS)z The post-class cleanup method call C{addError} on the result with a L{DirtyReactorAggregateError} Failure if there are selectables. zSELECTABLE HEREr_rTrN) rCr\rWrrrrrYrrFrepr)r selectabler`r[rrcr#r#r$)test_postClassCleanupWithSelectableErrorsbs $z6JanitorTests.test_postClassCleanupWithSelectableErrorsNr/)r0r1r2r3rdrrrxrzrrrrrrrr#r#r#r$r^s       r^c@r4) RemoveSafelyTestsz* Tests for L{util._removeSafely}. rrcCs8|d}t|t|}|tjtj |dS)z If a path doesn't contain a node named C{"_trial_marker"}, that path is not removed by L{util._removeSafely} and a L{util._NoTrialMarker} exception is raised instead. utf-8N) rencodermkdirr FilePath assertRaisesr _NoTrialMarker _removeSafely)r directorydirPathr#r#r$test_removeSafelyNoTrialMarkervs  z0RemoveSafelyTests.test_removeSafelyNoTrialMarkercCsrd dd}t}|td||d}t|t|}| d ||_ t ||d|d S) z If an L{OSError} is raised while removing a path in L{util._removeSafely}, an attempt is made to move the path to a new name. rrcSst) Raise an C{OSError} to emulate the branch of L{util._removeSafely} in which path removal fails. OSErrorr#r#r#r$ dummyRemoveszORemoveSafelyTests.test_removeSafelyRemoveFailsMoveSucceeds..dummyRemovestdoutr _trial_markercould not remove FilePathNr/)rpatchsysrrrrr rchildtouchremover rassertIngetvalue)r routrrr#r#r$(test_removeSafelyRemoveFailsMoveSucceedss    z:RemoveSafelyTests.test_removeSafelyRemoveFailsMoveSucceedscCsddd}ddd d }t}|td ||d }t|t|}| d ||_ ||_ | ttj|}|t|d|d|dS)z If an L{OSError} is raised while removing a path in L{util._removeSafely}, an attempt is made to move the path to a new name. If that attempt fails, the L{OSError} is re-raised. rrcStd)rzpath removal failedrr#r#r#r$rzLRemoveSafelyTests.test_removeSafelyRemoveFailsMoveFails..dummyRemoveT destinationr\ followLinksboolcSr)z Raise an C{OSError} to emulate the branch of L{util._removeSafely} in which path movement fails. path movement failedr)rrr#r#r$ dummyMoveTorzLRemoveSafelyTests.test_removeSafelyRemoveFailsMoveFails..dummyMoveTorrrrrNr/)T)rr\rrrr)rrrrrrrr rrrrmoveTorrr rrr9rr)r rrrrrr;r#r#r$%test_removeSafelyRemoveFailsMoveFailss    z7RemoveSafelyTests.test_removeSafelyRemoveFailsMoveFailsNr/)r0r1r2r3rrrr#r#r#r$rqs   rc@$eZdZdZd ddZd ddZdS) ExcInfoTestsz1 Tests for L{excInfoOrFailureToExcInfo}. rrcCs$ttddf}||t|udS)z L{excInfoOrFailureToExcInfo} returns exactly what it is passed, if it is passed a tuple like the one returned by L{sys.exc_info}. fooN) ValueErrorr'r)r infor#r#r$ test_excInfoszExcInfoTests.test_excInfocCsFzddWn tyt}Ynw||j|j|jft|dS)z When called with a L{Failure} instance, L{excInfoOrFailureToExcInfo} returns a tuple like the one returned by L{sys.exc_info}, with the elements taken from the type, value, and traceback of the failure. rTrN) BaseExceptionr rtypertbr)r fr#r#r$ test_failures     zExcInfoTests.test_failureNr/)r0r1r2r3rrr#r#r#r$rs rc@r)AcquireAttributeTestsz( Tests for L{acquireAttribute}. rrcCs*t|_}||t|tgdudS)z The value returned by L{acquireAttribute} is the value of the requested attribute on the first object in the list passed in which has that attribute. rNr\rr'rr rr#r#r$test_foundOnEarlierObjects z/AcquireAttributeTests.test_foundOnEarlierObjectcCs*t|_}||tt|gdudS)z The same as L{test_foundOnEarlierObject}, but for the case where the 2nd element in the object list has the attribute and the first does not. rNrrr#r#r$test_foundOnLaterObjects z-AcquireAttributeTests.test_foundOnLaterObjectcCs|tttgddS)z If none of the objects passed in the list to L{acquireAttribute} have the requested attribute, L{AttributeError} is raised. rN)rAttributeErrorrr\rQr#r#r$test_notFoundExceptionsz,AcquireAttributeTests.test_notFoundExceptioncCs$t}||ttgd|udS)z If none of the objects passed in the list to L{acquireAttribute} have the requested attribute and a default value is given, the default value is returned. rN)r\r'r)r defaultr#r#r$test_notFoundDefaultsz*AcquireAttributeTests.test_notFoundDefaultNr/)r0r1r2r3rrrrr#r#r#r$rs   rc@sjeZdZdZdddZdddZddd Zdd d Zdd d ZdddZ dddZ dddZ dddZ dS)ListToPhraseTestsz Input is transformed into a string representation of the list, with each item separated by delimiter (defaulting to a comma) and the final two being separated by a final delimiter. rrcCs$g}d}t|d}|||dS)zB If things is empty, an empty string is returned. andNr _listToPhraserr sampleexpectedresultr#r#r$ test_emptys zListToPhraseTests.test_emptycCs&dg}d}t|d}|||dS)z; With a single item, the item is returned. OnerNrrr#r#r$ test_oneWords zListToPhraseTests.test_oneWordcCs(ddg}d}t|d}|||dS)zA Two words are separated by the final delimiter. rTwoz One and TworNrrr#r#r$ test_twoWords  zListToPhraseTests.test_twoWordscC(gd}d}t|d}|||dS)zY With more than two words, the first two are separated by the delimiter. )rrThreezOne, Two, and ThreerNrrr#r#r$test_threeWords)rz!ListToPhraseTests.test_threeWordscCs,gd}d}tj|ddd}|||dS)zW If a delimiter is specified, it is used instead of the default comma. )rrrFourzOne; Two; Three; or Fourorz; ) delimiterNrrr#r#r$test_fourWords2sz ListToPhraseTests.test_fourWordscCr)zS If something in things is not a string, it is converted into one. )rTthreez1, 2, and threerNrrr#r#r$test_notString;rz ListToPhraseTests.test_notStringcCs*d}|ttj|d}|t|ddS)z? If things is a string, a TypeError is raised. zOne, two, threer Things must be a list or a tupleNr TypeErrorr rrr9r rr;r#r#r$test_stringTypeErrorDsz&ListToPhraseTests.test_stringTypeErrorcCs2tgd}|ttj|d}|t|ddS)zB If things is an iterator, a TypeError is raised. )rTrrrN)iterrrr rrr9rr#r#r$test_iteratorTypeErrorLs z(ListToPhraseTests.test_iteratorTypeErrorcCs0ddd}|ttj|d}|t|ddS) zB If things is a generator, a TypeError is raised. rGenerator[int, None, None]csstdEdHdS)Nr)ranger#r#r#r$rYsz9ListToPhraseTests.test_generatorTypeError..samplerrN)rrrrr#r#r$test_generatorTypeErrorTs z)ListToPhraseTests.test_generatorTypeErrorNr/) r0r1r2r3rrrrrrrrrr#r#r#r$rs  rc@r) OpenTestLogTestsz# Tests for C{openTestLog}. rrcCst}|tjtj|ttjdd}t|}t| }| |Wdn1s1wYt |j d }| }Wdn1sKwYt |dt|dS)zR The log file is opened in text mode and uses UTF-8 for encoding. )CasciiuHere comes the ☉Nrbr)locale getlocale addCleanup setlocaleLC_ALLr rrrwriteopenrreadrrr)r currentLocaletextprwrittenr#r#r$ test_utf8es   zOpenTestLogTests.test_utf8cCsd}d}dtjdtj}t|}t| }||Wdn1s)wYt| }||Wdn1sBwYt| dt |dS)z The log file is opened in append mode so if runner configuration specifies an existing log file its contents are not wiped out. zHello, world. zGoodbye, world. z Hello, world.z Goodbye, world.Nr) rlinesepr rrrrr getContentdecoder)r existingTextnewTextrrrr#r#r$ test_append{s     zOpenTestLogTests.test_appendNr/)r0r1r2r3rrr#r#r#r$r`s r)+r3 __future__rrrriortypingrzope.interfacerhamcrestrrtwisted.internet.basertwisted.internet.interfacesr twisted.pythonr twisted.python.failurer twisted.trialr twisted.trial.unittestr twisted.trial.utilrrrrrrr5rCrWr^rrrrrr#r#r#r$s6           )3-3U'Y