U TgV@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@sHeZdZdZddddZddddZdddd Zddd d Zd S) MktempTestszm Tests for L{TestCase.mktemp}, a helper function for creating temporary file or directory names. NonereturncCs:|}tj|tjdd}||dddgdS)z The path name returned by C{mktemp} is directly beneath a directory which identifies the test method which created the name. Nztwisted.trial.test.test_utilr test_name)mktempospathdirnamesplitsep assertEqual)selfnamedirsr#@/tmp/pip-unpacked-wheel-1_7di8ic/twisted/trial/test/test_util.pyr)s zMktempTests.test_namecCs|}|||dS)zF Repeated calls to C{mktemp} return different values. N)rZassertNotEqual)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_locationN)__name__ __module__ __qualname____doc__rr%r)r-r#r#r#r$r#s   rc@s:eZdZdZddddZddddZdddd Zd S) DirtyReactorAggregateErrorTestsz6 Tests for the L{DirtyReactorAggregateError}. rrcCs tddg}|t|ddS)z5 Delayed calls are formatted nicely. ZFoobarzhReactor 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 2Nr4r6r#r#r$test_formatSelectables_s z6DirtyReactorAggregateErrorTests.test_formatSelectablescCs&tddgddg}|t|ddS)zR Both delayed calls and selectables can appear in the same error. ZbleckZBoozoZSel1ZSel2zReactor was unclean. DelayedCalls: (set twisted.internet.base.DelayedCall.debug = True to debug) bleck Boozo Selectables: Sel1 Sel2Nr4r6r#r#r$%test_formatDelayedCallsAndSelectablesms zEDirtyReactorAggregateErrorTests.test_formatDelayedCallsAndSelectablesN)r.r/r0r1r8r9r:r#r#r#r$r2Lsr2c@sReZdZdZdddddddZdd dd d d Zdd ddZdd ddZdS) 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}. Nzlist[DelayedCall]zlist[object] | Noner) delayedCalls selectablesrcCs$||_g|_d|_|sg}||_dS)z @param delayedCalls: See L{StubReactor.delayedCalls}. @param selectables: See L{StubReactor.selectables}. rN)r< iterationsremoveAllCalledr=)r r<r=r#r#r$__init__s zStubReactor.__init__z float | None)timeoutrcCs|j|dS)z/ Increment C{self.iterations}. N)r>append)r rAr#r#r$iterateszStubReactor.iteratercCs|jS)z. Return C{self.delayedCalls}. )r<r r#r#r$getDelayedCallsszStubReactor.getDelayedCallsz list[object]cCs|jd7_|jS)zS Increment C{self.removeAllCalled} and return C{self.selectables}. )r?r=rDr#r#r$ removeAllszStubReactor.removeAll)N)N)r.r/r0r1r@rCrErGr#r#r#r$r;s  r;c@s0eZdZdZddddZddddd 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}. rrcCs g|_dSN)errorsrDr#r#r$r@szStubErrorReporter.__init__objectr )testr7rcCs|j||fdS)z6 Record parameters in C{self.errors}. N)rJrB)r rLr7r#r#r$addErrorszStubErrorReporter.addErrorN)r.r/r0r1r@rMr#r#r#r$rHsrHc@seZdZdZddddZddddZdddd Zddd d Zddd d ZddddZ ddddZ ddddZ ddddZ ddddZ ddddZdS) 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)r;r _cleanPendingrr>r rPjanr#r#r$test_cleanPendingSpinsReactorsz*JanitorTests.test_cleanPendingSpinsReactorcCs\dddd}g}td|di|jdd}t|g}td d |d }||||gd S) zW During pending-call cleanup, the janitor cancels pending timed calls. r5rcSsdS)NZLulzr#r#r#r#r$funcsz8JanitorTests.test_cleanPendingCancelsCalls..func,r#cSsdSrIr#xr#r#r$z.NrO)rrBr;rrQr)r rUZ cancelled delayedCallrPrSr#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. rVcSsdSrIr#r#r#r#r$rYrZzIJanitorTests.test_cleanPendingReturnsDelayedCallStrings..r#cSsdSrIr#rWr#r#r$rYrZcSsdSNrr#r#r#r#r$rYrZsecondsNrO)rr5r;rrQr)r r[delayedCallStringrPrSstringsr#r#r$*test_cleanPendingReturnsDelayedCallStringss z7JanitorTests.test_cleanPendingReturnsDelayedCallStringscCs0tg}tdd|d}|||jddS)zM The Janitor will remove selectables during reactor cleanup. NrOrF)r;r _cleanReactorrr?rRr#r#r$#test_cleanReactorRemovesSelectablessz0JanitorTests.test_cleanReactorRemovesSelectablescCsRttGddd}|}tg|g}tdd|d}|||jdgdS)zI The Janitor will kill processes during reactor cleanup. c@s.eZdZdZddddZddddd Zd S) zJJanitorTests.test_cleanReactorKillsProcesses..StubProcessTransportz A stub L{IProcessTransport} provider which records signals. @ivar signals: The signals passed to L{signalProcess}. rrcSs g|_dSrI)signalsrDr#r#r$r@szSJanitorTests.test_cleanReactorKillsProcesses..StubProcessTransport.__init__z str | int)signalrcSs|j|dS)zF Append C{signal} to C{self.signals}. N)rerB)r rfr#r#r$ signalProcessszXJanitorTests.test_cleanReactorKillsProcesses..StubProcessTransport.signalProcessN)r.r/r0r1r@rgr#r#r#r$StubProcessTransportsrhNrOZKILL)rr r;rrcrre)r rhptrPrSr#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@seZdZdZddddZdS)zJJanitorTests.test_cleanReactorReturnsSelectableStrings..Selectablezl A stub Selectable which only has an interesting string representation. r5rcSsdS)N (SELECTABLE!)r#rDr#r#r$__repr__ szSJanitorTests.test_cleanReactorReturnsSelectableStrings..Selectable.__repr__N)r.r/r0r1rlr#r#r#r$ SelectablesrmNrOrk)r;rrrc)r rmrPrSr#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. rON)r;rKrHrr&postCaseCleanuprrJr rPrLreporterrSr#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. rVcSsdSrIr#r#r#r#r$rY:rZz=JanitorTests.test_postCaseCleanupWithErrors..r#cSsdSrIr#rWr#r#r$rY:rZcSsdSr]r#r#r#r#r$rY:rZr^rOrFrN) rr5r;rKrHrr(rorlenrJvaluer<r r[r`rPrLrqrSr#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. rON)r;rKrHrpostClassCleanuprrJrpr#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. rVcSsdSrIr#r#r#r#r$rYWrZzIJanitorTests.test_postClassCleanupWithPendingCallErrors..r#cSsdSrIr#rWr#r#r$rYWrZcSsdSr]r#r#r#r#r$rYWrZr^rOrFrN) rr5r;rKrHrrwrrsrJrtr<rur#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 HERErOrFrN) r;rKrHrrwrrsrJrtr=repr)r Z selectablerPrLrqrSr#r#r$)test_postClassCleanupWithSelectableErrorsbs z6JanitorTests.test_postClassCleanupWithSelectableErrorsN)r.r/r0r1rTr\rbrdrjrnrrrvrxryr{r#r#r#r$rNs    rNc@s:eZdZdZddddZddddZdddd Zd S) 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 Z_NoTrialMarker _removeSafely)r directorydirPathr#r#r$test_removeSafelyNoTrialMarkervs  z0RemoveSafelyTests.test_removeSafelyNoTrialMarkercCsvdddd}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. rrcSs tdS) Raise an C{OSError} to emulate the branch of L{util._removeSafely} in which path removal fails. NOSErrorr#r#r#r$ dummyRemoveszORemoveSafelyTests.test_removeSafelyRemoveFailsMoveSucceeds..dummyRemovestdoutr} _trial_markercould not remove FilePathN)rpatchsysrr~rrr rchildtouchremover rassertIngetvalue)r routrrr#r#r$(test_removeSafelyRemoveFailsMoveSucceedss   z:RemoveSafelyTests.test_removeSafelyRemoveFailsMoveSucceedscCsdddd}dddddd 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. rrcSs tddS)rzpath removal failedNrr#r#r#r$rszLRemoveSafelyTests.test_removeSafelyRemoveFailsMoveFails..dummyRemoveTrKbool) destination followLinksrcSs tddS)z Raise an C{OSError} to emulate the branch of L{util._removeSafely} in which path movement fails. path movement failedNr)rrr#r#r$ dummyMoveToszLRemoveSafelyTests.test_removeSafelyRemoveFailsMoveFails..dummyMoveTorr}rrrN)T)rrrrr~rrr rrrrZmoveTorrr rrr5rr)r rrrrrr7r#r#r$%test_removeSafelyRemoveFailsMoveFailss  z7RemoveSafelyTests.test_removeSafelyRemoveFailsMoveFailsN)r.r/r0r1rrrr#r#r#r$r|qs r|c@s,eZdZdZddddZddddZdS) ExcInfoTestsz1 Tests for L{excInfoOrFailureToExcInfo}. rrcCs$ttddf}||t|kdS)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_excInfocCsHz ddWntk r&t}YnX||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. rFrN) BaseExceptionr rtyperttbr)r fr#r#r$ test_failures   zExcInfoTests.test_failureN)r.r/r0r1rrr#r#r#r$rsrc@sHeZdZdZddddZddddZdddd Zddd d Zd S) AcquireAttributeTestsz( Tests for L{acquireAttribute}. rrcCs*t|_}||t|tgdkdS)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. rtNrKrtr&rr rtr#r#r$test_foundOnEarlierObjects z/AcquireAttributeTests.test_foundOnEarlierObjectcCs*t|_}||tt|gdkdS)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. rtNrrr#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)rAttributeErrorrrKrDr#r#r$test_notFoundExceptionsz,AcquireAttributeTests.test_notFoundExceptioncCs$t}||ttgd|kdS)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)rKr&r)r defaultr#r#r$test_notFoundDefaultsz*AcquireAttributeTests.test_notFoundDefaultN)r.r/r0r1rrrrr#r#r#r$rs  rc@seZdZdZddddZddddZdddd Zddd d Zddd d ZddddZ ddddZ ddddZ ddddZ 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 s zListToPhraseTests.test_twoWordscCs*dddg}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)s  z!ListToPhraseTests.test_threeWordscCs0ddddg}d}tj|ddd}|||d S) zW If a delimiter is specified, it is used instead of the default comma. rrrZFourzOne; Two; Three; or Fourorz; ) delimiterNrrr#r#r$test_fourWords2s z ListToPhraseTests.test_fourWordscCs*dddg}d}t|d}|||dS)zS If something in things is not a string, it is converted into one. rFZthreez1, 2, and threerNrrr#r#r$test_notString;s  z 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 rrr5r rr7r#r#r$test_stringTypeErrorDsz&ListToPhraseTests.test_stringTypeErrorcCs4tdddg}|ttj|d}|t|ddS)zB If things is an iterator, a TypeError is raised. rFrrrN)iterrrr rrr5rr#r#r$test_iteratorTypeErrorLsz(ListToPhraseTests.test_iteratorTypeErrorcCs4dddd}|ttj|d}|t|ddS)zB If things is a generator, a TypeError is raised. zGenerator[int, None, None]rcsstdEdHdS)Nr)ranger#r#r#r$rYsz9ListToPhraseTests.test_generatorTypeError..samplerrNrrr#r#r$test_generatorTypeErrorTsz)ListToPhraseTests.test_generatorTypeErrorN) r.r/r0r1rrrrrrrrrr#r#r#r$rs      rc@s,eZdZdZddddZddddZdS) OpenTestLogTestsz# Tests for C{openTestLog}. rrc Cst}|tjtj|ttjdd}t|}t|}| |W5QRXt |j d}| }W5QRXt |dt|dS)zR The log file is opened in text mode and uses UTF-8 for encoding. )CasciiuHere comes the ☉rbr}N)locale getlocaleZ addCleanup setlocaleLC_ALLr rrrwriteopenrreadrr~r)r Z currentLocaletextprZwrittenr#r#r$ test_utf8es zOpenTestLogTests.test_utf8c Csd}d}dtjdtj}t|}t|}||W5QRXt|}||W5QRXt| 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.r}N) rlinesepr rrrrrZ getContentdecoder)r Z existingTextZnewTextrrrr#r#r$ test_append{s   zOpenTestLogTests.test_appendN)r.r/r0r1rrr#r#r#r$r`sr)+r1 __future__rrrriortypingrZzope.interfacerZhamcrestrrZtwisted.internet.baserZtwisted.internet.interfacesr Ztwisted.pythonr Ztwisted.python.failurer Z twisted.trialr Ztwisted.trial.unittestr Ztwisted.trial.utilrrrrrrr2r;rHrNr|rrrrr#r#r#r$s4           )3-3U'Y