o b%@sdZddlmZddlmZmZddlmZddlm Z dZ zddl m Z ddl mZWn ey7d Z Ynwdd l mZdd lmZGd d d eZGddde ZGddde ZGdddeZGddde ZGddde ZGddde ZdS)z> Tests for L{twisted.internet.posixbase} and supporting code. )Deferred)PosixReactorBase_Waker) ServerFactory)TestCaseN)unix) ClientProtoz)Platform does not support AF_UNIX socketsreactor)Portc@s4eZdZddZddZddZddZd d Zd S) TrivialReactorcCsi|_i|_t|dSN)_readers_writersr__init__selfrF/usr/lib/python3/dist-packages/twisted/internet/test/test_posixbase.pyrszTrivialReactor.__init__cCd|j|<dSNTrrreaderrrr addReaderzTrivialReactor.addReadercC |j|=dSr rrrrr removeReader" zTrivialReactor.removeReadercCrrrrwriterrrr addWriter%rzTrivialReactor.addWritercCrr rr rrr removeWriter(rzTrivialReactor.removeWriterN)__name__ __module__ __qualname__rrrr"r#rrrrr s  r c@s0eZdZdZddZddZddZdd Zd S) PosixReactorBaseTestsz( Tests for L{PosixReactorBase}. cCs2||jt||j|j||j|jdSr )assertIsInstancewakerrassertIn_internalReadersrrr rrr _checkWaker1sz!PosixReactorBaseTests._checkWakercCst}||dS)z When L{PosixReactorBase} is instantiated, it creates a waker and adds it to its internal readers set. N)r r-r,rrrtest_wakerIsInternalReader6sz0PosixReactorBaseTests.test_wakerIsInternalReadercCs\t}t}|j|||||j|j||| ||j| ||jdS)z Any L{IReadDescriptors} in L{PosixReactorBase._internalReaders} are left alone by L{PosixReactorBase._removeAll}. N) r objectr+addr _removeAllrrr-r*)rr extrarrr"test_removeAllSkipsInternalReaders>s   z8PosixReactorBaseTests.test_removeAllSkipsInternalReaderscCsjt}t}t}||||||j|j}|t|||h| ||j| ||jdS)z L{PosixReactorBase._removeAll} returns a list of removed L{IReadDescriptor} and L{IWriteDescriptor} objects. N) r r/rr"r1rr assertEqualset assertNotIn)rr rr!removedrrr'test_removeAllReturnsRemovedDescriptorsLs  z=PosixReactorBaseTests.test_removeAllReturnsRemovedDescriptorsN)r$r%r&__doc__r-r.r3r8rrrrr',s  r'c@s&eZdZdZeees dZddZdS) TCPPortTestsz1 Tests for L{twisted.internet.tcp.Port}. zNon-posixbase reactorcCs,tdt}d|_dd|_||tS)z L{Port.stopListening} returns a L{Deferred} which errbacks if L{Port.connectionLost} raises an exception. i90TcSsddS)Nrr)reasonrrrkz8TCPPortTests.test_connectionLostFailed..)r r connectedconnectionLost assertFailure stopListeningZeroDivisionError)rportrrrtest_connectionLostFailedds  z&TCPPortTests.test_connectionLostFailedN) r$r%r&r9 isinstancer rskiprErrrrr:\s   r:c@s8eZdZdZddZddZddZdd Zd d Zd S) TimeoutReportReactorz A reactor which is just barely runnable and which cannot monitor any readers or writers, and which fires a L{Deferred} with the timeout passed to its C{doIteration} method as soon as that method is invoked. cCst|t|_d|_dS)Nd)rrriterationTimeoutnowrrrrrvs  zTimeoutReportReactor.__init__cCsdS)z Ignore the reader. This is necessary because the waker will be added. However, we won't actually monitor it for any events. Nrrrrrr{zTimeoutReportReactor.addReadercCsgS)z There are no readers or writers, so there is nothing to remove. This will be called when the reactor stops, though, so it must be implemented. rrrrr removeAllszTimeoutReportReactor.removeAllcCs|jS)zx Override the real clock with a deterministic one that can be easily controlled in a unit test. )rKrrrrsecondsszTimeoutReportReactor.secondscCs&|j}|durd|_||dSdSr )rJcallback)rtimeoutdrrr doIterations z TimeoutReportReactor.doIterationN) r$r%r&r9rrrMrNrRrrrrrHos rHc@sPeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS)IterationTimeoutTestsz Tests for the timeout argument L{PosixReactorBase.run} calls L{PosixReactorBase.doIteration} with in the presence of various delayed calls. cs6g}j|jjfdd|dS)NcsSr )stop)ignoredr rrr=r>z>IterationTimeoutTests._checkIterationTimeout..r)rJ addCallbackappendrunrr rPrr r_checkIterationTimeouts z,IterationTimeoutTests._checkIterationTimeoutcCst}||}||dS)zl If there are no delayed calls, C{doIteration} is called with a timeout of L{None}. N)rHrZ assertIsNonerYrrr test_noCallss z"IterationTimeoutTests.test_noCallscCs0t}|ddd||}||ddS)z If there is a delayed call, C{doIteration} is called with a timeout which is the difference between the current time and the time at which that call is to run. rIcSdSr rrrrrr=rLz8IterationTimeoutTests.test_delayedCall..NrH callLaterrZr4rYrrrtest_delayedCalls z&IterationTimeoutTests.test_delayedCallcCs>t}|ddd|jd7_||}||ddS)z If a delayed call is scheduled and then some time passes, the timeout passed to C{doIteration} is reduced by the amount of time which passed. rIcSr]r rrrrrr=rLz7IterationTimeoutTests.test_timePasses..KN)rHr_rKrZr4rYrrrtest_timePassess  z%IterationTimeoutTests.test_timePassescCsPt}|ddd|ddd|ddd||}||ddS) z If there are several delayed calls, C{doIteration} is called with a timeout which is the difference between the current time and the time at which the earlier of the two calls is to run. 2cSr]r rrrrrr=rLzAIterationTimeoutTests.test_multipleDelayedCalls.. cSr]r rrrrrr=rLrIcSr]r rrrrrr=rLNr^rYrrrtest_multipleDelayedCallss  z/IterationTimeoutTests.test_multipleDelayedCallscCsHt}|ddd}|jd7_|d||}||ddS)z If a delayed call is reset, the timeout passed to C{doIteration} is based on the interval between the time when reset is called and the new delay of the call. rdcSr]r rrrrrr=rLz=IterationTimeoutTests.test_resetDelayedCall..raN)rHr_rKresetrZr4rr callrPrrrtest_resetDelayedCalls   z+IterationTimeoutTests.test_resetDelayedCallcCsHt}|ddd}|jd7_|d||}||ddS)z If a delayed call is re-delayed, the timeout passed to C{doIteration} is based on the remaining time before the call would have been made and the additional amount of time passed to the delay method. rdcSr]r rrrrrr=rLz=IterationTimeoutTests.test_delayDelayedCall..re<N)rHr_rKdelayrZr4rirrrtest_delayDelayedCalls   z+IterationTimeoutTests.test_delayDelayedCallcCs6t}|ddd}|||}||dS)zp If the only delayed call is canceled, L{None} is the timeout passed to C{doIteration}. rdcSr]r rrrrrr=rLz>IterationTimeoutTests.test_cancelDelayedCall..N)rHr_cancelrZr[rirrrtest_cancelDelayedCalls  z,IterationTimeoutTests.test_cancelDelayedCallN) r$r%r&r9rZr\r`rcrfrkrorqrrrrrSs     rSc@s,eZdZdZedur eZddZddZdS)ConnectedDatagramPortTestsz/ Test connected datagram UNIX sockets. Ncs.fdd}tdt}||_|ddS)z L{ConnectedDatagramPort} does not call the deprecated C{loseConnection} in L{ConnectedDatagramPort.connectionFailed}. csddS)z Dummy C{loseConnection} method. C{loseConnection} is deprecated and should not get called. z7loseConnection is deprecated and should not get called.N)failrrrrloseConnectionsz`ConnectedDatagramPortTests.test_connectionFailedDoesntCallLoseConnection..loseConnectionNgoodbye)rConnectedDatagramPortrrtconnectionFailed)rrtrDrrr-test_connectionFailedDoesntCallLoseConnections zHConnectedDatagramPortTests.test_connectionFailedDoesntCallLoseConnectioncs@d_fdd}tdt}||_|djdS)z L{ConnectedDatagramPort} calls L{ConnectedDatagramPort.stopListening} instead of the deprecated C{loseConnection} in L{ConnectedDatagramPort.connectionFailed}. Fcs d_dS)z8 Dummy C{stopListening} method. TN)calledrrrrrBs zYConnectedDatagramPortTests.test_connectionFailedCallsStopListening..stopListeningNru)ryrrvrrBrw assertTrue)rrBrDrrr'test_connectionFailedCallsStopListenings   zBConnectedDatagramPortTests.test_connectionFailedCallsStopListening)r$r%r&r9 skipSocketsrGrxr{rrrrrrs  rrc@seZdZddZdS) WakerTestscCs0tdd}|}|d|t|ddS)Nr r)r flushWarningsr@r4len)rr)warningsrrr test_noWakerConstructionWarnings's  z+WakerTests.test_noWakerConstructionWarningsN)r$r%r&rrrrrr}&s r})r9twisted.internet.deferrtwisted.internet.posixbaserrtwisted.internet.protocolrtwisted.trial.unittestrr|twisted.internetrtwisted.test.test_unixr ImportErrorr twisted.internet.tcpr r r'r:rHrSrrr}rrrrs*       0(b-