o bL@sdZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZmZmZmZGd d d eZGdddeZGdddZGdddZGdddeZGdddeZGdddeZGdddeZGdddeZ dS)z! Tests for L{twisted.trial.util} N)StringIO) implementer) DelayedCall)IProcessTransport)filepath)Failure)util)SynchronousTestCase)DirtyReactorAggregateError_JanitoracquireAttributeexcInfoOrFailureToExcInfoc@0eZdZdZddZddZddZdd Zd S) MktempTestszm Tests for L{TestCase.mktemp}, a helper function for creating temporary file or directory names. cCs8|}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>/usr/lib/python3/dist-packages/twisted/trial/test/test_util.pyr$s zMktempTests.test_namecCs|}|||dS)zF Repeated calls to C{mktemp} return different values. N)rassertNotEqual)rrrrr test_unique/szMktempTests.test_uniquecCs<|}tj|}|tj||tj|dS)zS The directory part of the path name returned by C{mktemp} exists. N)rrrr assertTrueexists assertFalse)rrrrrr test_created6s zMktempTests.test_createdcCs(tj|}||tdS)zZ The path returned by C{mktemp} is beneath the current working directory. N)rrabspathrr startswithgetcwd)rrrrr test_location?szMktempTests.test_locationN)__name__ __module__ __qualname____doc__rrr#r'rrrrrs   rc@(eZdZdZddZddZddZdS) DirtyReactorAggregateErrorTestsz6 Tests for the L{DirtyReactorAggregateError}. cCs tddg}|t|ddS)z5 Delayed calls are formatted nicely. FoobarzhReactor was unclean. DelayedCalls: (set twisted.internet.base.DelayedCall.debug = True to debug) Foo barNr rstrrerrorrrrtest_formatDelayedCallLs z6DirtyReactorAggregateErrorTests.test_formatDelayedCallcCs"tgddg}|t|ddS)z3 Selectables are formatted nicely. z selectable 1z selectable 2z;Reactor was unclean. Selectables: selectable 1 selectable 2Nr0r2rrrtest_formatSelectablesZs 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 Sel2Nr0r2rrr%test_formatDelayedCallsAndSelectableshs zEDirtyReactorAggregateErrorTests.test_formatDelayedCallsAndSelectablesN)r(r)r*r+r4r5r:rrrrr-Gs  r-c@s4eZdZdZd ddZd ddZddZd d ZdS) 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}. NcCs$||_g|_d|_|s g}||_dS)z @param delayedCalls: See L{StubReactor.delayedCalls}. @param selectables: See L{StubReactor.selectables}. rN) delayedCalls iterationsremoveAllCalled selectables)rr<r?rrr__init__s  zStubReactor.__init__cC|j|dS)z/ Increment C{self.iterations}. N)r=append)rtimeoutrrriteratezStubReactor.iteratecCs|jS)z. Return C{self.delayedCalls}. )r<rrrrgetDelayedCallsszStubReactor.getDelayedCallscCs|jd7_|jS)zS Increment C{self.removeAllCalled} and return C{self.selectables}. )r>r?rFrrr removeAllszStubReactor.removeAllN)r(r)r*r+r@rDrGrIrrrrr;zs   r;c@ eZdZdZddZddZdS)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}. cC g|_dSrJ)errorsrFrrrr@ zStubErrorReporter.__init__cCs|j||fdS)z6 Record parameters in C{self.errors}. N)rNrB)rtestr3rrraddErrorszStubErrorReporter.addErrorN)r(r)r*r+r@rQrrrrrLs rLc@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS) JanitorTestsz Tests for L{_Janitor}! cCs4tg}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=rrTjanrrrtest_cleanPendingSpinsReactorsz*JanitorTests.test_cleanPendingSpinsReactorcCsVdd}g}td|di|jdd}t|g}tdd|d}||||gdS) zW During pending-call cleanup, the janitor cancels pending timed calls. cSdS)NLulzrrrrrfuncz8JanitorTests.test_cleanPendingCancelsCalls..func,rcSdSrJrxrrrz.NrS)rrBr;r rUr)rr[ cancelled delayedCallrTrWrrrtest_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. r]cSr^rJrrrrrrarbzIJanitorTests.test_cleanPendingReturnsDelayedCallStrings..rcSr^rJrr_rrrrarbcSrYNrrrrrrrarbsecondsNrS)rr1r;r rUr)rrddelayedCallStringrTrWstringsrrr*test_cleanPendingReturnsDelayedCallStringss z7JanitorTests.test_cleanPendingReturnsDelayedCallStringscCs0tg}tdd|d}|||jddS)zM The Janitor will remove selectables during reactor cleanup. NrSrH)r;r _cleanReactorrr>rVrrr#test_cleanReactorRemovesSelectablessz0JanitorTests.test_cleanReactorRemovesSelectablescCsRttGddd}|}tg|g}tdd|d}|||jdgdS)zI The Janitor will kill processes during reactor cleanup. c@rK)zJJanitorTests.test_cleanReactorKillsProcesses..StubProcessTransportz A stub L{IProcessTransport} provider which records signals. @ivar signals: The signals passed to L{signalProcess}. cSrMrJ)signalsrFrrrr@rOzSJanitorTests.test_cleanReactorKillsProcesses..StubProcessTransport.__init__cSrA)zF Append C{signal} to C{self.signals}. N)rnrB)rsignalrrr signalProcessrEzXJanitorTests.test_cleanReactorKillsProcesses..StubProcessTransport.signalProcessN)r(r)r*r+r@rprrrrStubProcessTransports rqNrSKILL)rrr;r rlrrn)rrqptrTrWrrrtest_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@seZdZdZdefddZdS)zJJanitorTests.test_cleanReactorReturnsSelectableStrings..Selectablezl A stub Selectable which only has an interesting string representation. returncSrY)N (SELECTABLE!)rrFrrr__repr__r\zSJanitorTests.test_cleanReactorReturnsSelectableStrings..Selectable.__repr__N)r(r)r*r+r1rwrrrr SelectablesrxNrSrv)r;r rrl)rrxrTrWrrr)test_cleanReactorReturnsSelectableStrings s 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. rSN)r;objectrLr r postCaseCleanuprrNrrTrPreporterrWrrrtest_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. r]cSr^rJrrrrrra3rbz=JanitorTests.test_postCaseCleanupWithErrors..rcSr^rJrr_rrrra3rbcSrYrfrrrrrra3rbrgrSrHrN) rr1r;rzrLr r"r{rlenrNvaluer<rrdrirTrPr}rWrrrtest_postCaseCleanupWithErrors,s  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. rSN)r;rzrLr postClassCleanuprrNr|rrrtest_postClassCleanupNoErrors>s 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. r]cSr^rJrrrrrraPrbzIJanitorTests.test_postClassCleanupWithPendingCallErrors..rcSr^rJrr_rrrraPrbcSrYrfrrrrrraPrbrgrSrHrN) rr1r;rzrLr rrrrNrr<rrrr*test_postClassCleanupWithPendingCallErrorsJs  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 HERErSrHrN) r;rzrLr rrrrNrr?repr)r selectablerTrPr}rWrrr)test_postClassCleanupWithSelectableErrors[s $z6JanitorTests.test_postClassCleanupWithSelectableErrorsN)r(r)r*r+rXrerkrmrtryr~rrrrrrrrrRs    rRc@r,) RemoveSafelyTestsz* Tests for L{util._removeSafely}. cCs8|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) rencodermkdirrFilePath assertRaisesr_NoTrialMarker _removeSafely)r directorydirPathrrrtest_removeSafelyNoTrialMarkeros  z0RemoveSafelyTests.test_removeSafelyNoTrialMarkercCspdd}t}|td||d}t|t|}| d ||_ t ||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. cSst) Raise an C{OSError} to emulate the branch of L{util._removeSafely} in which path removal fails. OSErrorrrrr dummyRemoveszORemoveSafelyTests.test_removeSafelyRemoveFailsMoveSucceeds..dummyRemovestdoutr _trial_markercould not remove FilePathN)rpatchsysrrrrrrchildtouchremoverrassertIngetvalue)rroutrrrrr(test_removeSafelyRemoveFailsMoveSucceedszs   z:RemoveSafelyTests.test_removeSafelyRemoveFailsMoveSucceedscCsdd}dd}t}|td||d}t|t|}| d ||_ ||_ | ttj|}|t|d|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. If that attempt fails, the L{OSError} is re-raised. cStd)rzpath removal failedrrrrrrzLRemoveSafelyTests.test_removeSafelyRemoveFailsMoveFails..dummyRemovecSr)z Raise an C{OSError} to emulate the branch of L{util._removeSafely} in which path movement fails. path movement failedr)rrrr dummyMoveTorzLRemoveSafelyTests.test_removeSafelyRemoveFailsMoveFails..dummyMoveTorrrrrN)rrrrrrrrrrrrmoveTorrrrrr1rr)rrrrrrr3rrr%test_removeSafelyRemoveFailsMoveFailss  z7RemoveSafelyTests.test_removeSafelyRemoveFailsMoveFailsN)r(r)r*r+rrrrrrrrjs  rc@rK) ExcInfoTestsz1 Tests for L{excInfoOrFailureToExcInfo}. cCs$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 )rinforrr 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. rHrN) BaseExceptionrrtypertbr )rfrrr test_failures     zExcInfoTests.test_failureN)r(r)r*r+rrrrrrrs rc@r) AcquireAttributeTestsz( Tests for L{acquireAttribute}. cCs*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. rNrzrr r rrrrrtest_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. rNrrrrrtest_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)rAttributeErrorr rzrFrrrtest_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)rzr r )rdefaultrrrtest_notFoundDefaultsz*AcquireAttributeTests.test_notFoundDefaultN)r(r)r*r+rrrrrrrrrs   rc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ 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. cCs$g}d}t|d}|||dS)zB If things is empty, an empty string is returned. andNr _listToPhraserrsampleexpectedresultrrr test_emptys zListToPhraseTests.test_emptycCs&dg}d}t|d}|||dS)z; With a single item, the item is returned. OnerNrrrrr test_oneWords zListToPhraseTests.test_oneWordcCs(ddg}d}t|d}|||dS)zA Two words are separated by the final delimiter. rTwoz One and TworNrrrrr 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 ThreerNrrrrrtest_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; ) delimiterNrrrrrtest_fourWords+sz ListToPhraseTests.test_fourWordscCr)zS If something in things is not a string, it is converted into one. )rHthreez1, 2, and threerNrrrrrtest_notString4rz 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 TypeErrorrrrr1rrr3rrrtest_stringTypeError=sz&ListToPhraseTests.test_stringTypeErrorcCs2tgd}|ttj|d}|t|ddS)zB If things is an iterator, a TypeError is raised. )rHrrrN)iterrrrrrr1rrrrtest_iteratorTypeErrorEs z(ListToPhraseTests.test_iteratorTypeErrorcCs.dd}|ttj|d}|t|ddS)zB If things is a generator, a TypeError is raised. csstdEdHdS)Nr)rangerrrrrRsz9ListToPhraseTests.test_generatorTypeError..samplerrNrrrrrtest_generatorTypeErrorMsz)ListToPhraseTests.test_generatorTypeErrorN) r(r)r*r+rrrrrrrrrrrrrrs       r)!r+rriorzope.interfacertwisted.internet.basertwisted.internet.interfacesrtwisted.pythonrtwisted.python.failurer twisted.trialrtwisted.trial.unittestr twisted.trial.utilr r r r rr-r;rLrRrrrrrrrrs,        )3+3U'