o ¯b¹4ã@sÆdZddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z ddl mZmZddlmZdd lmZdd lmZd d „ZGd d„deƒZee ƒGdd„dƒƒZeƒ e ¡¡dS)z0 Tests for implementations of L{IReactorFDSet}. éN)ÚskipIf)Ú implementer)ÚFileDescriptor)Ú IReactorFDSetÚIReadDescriptor)Ú EINPROGRESSÚ EWOULDBLOCK)ÚReactorBuilder)Úplatform)ÚSkipTestc CsÊt ¡}| d¡| d¡zQt ¡}z7| d¡z | d| ¡df¡WntyB}z|jdtt fvr8‚WYd}~nd}~ww|  ¡\}}Wn t yV|  ¡‚wW|  ¡||fS|  ¡w)N)ú 127.0.0.1réFr r) ÚsocketÚbindÚlistenÚ setblockingÚconnectÚ getsocknameÚOSErrorÚargsrrÚacceptÚ BaseExceptionÚclose)Ú serverSocketÚclientÚeÚserverÚaddr©rúB/usr/lib/python3/dist-packages/twisted/internet/test/test_fdset.pyÚ socketpairs0    ÿ€ÿ þÿ þr c@s”eZdZdZegZdd„Zdd„Zdd„Zdd „Z d d „Z d d „Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zee ¡dƒdd„ƒZdd„ZdS) ÚReactorFDSetTestsBuilderz> Builder defining tests relating to L{IReactorFDSet}. cCs*tƒ\}}| |j¡| |j¡||fS)zL Return the two sockets which make up a new TCP connection. )r Ú addCleanupr)ÚselfrrrrrÚ_connectedPair8s   z'ReactorFDSetTestsBuilder._connectedPaircCs.| ¡}| ¡\}}t|ƒ}|j|_|||fS©N)Ú buildReactorr$rÚfileno)r#ÚreactorrrÚfdrrrÚ _simpleSetupAs   z%ReactorFDSetTestsBuilder._simpleSetupcsD| ¡\‰‰}‡‡fdd„}|ˆ_ˆ ˆ¡| d¡| ˆ¡dS)z© C{reactor.addReader()} accepts an L{IReadDescriptor} provider and calls its C{doRead} method when there may be data available on its C{fileno}. cóˆ ˆ¡ˆ ¡dSr%)Ú removeReaderÚstopr©r)r(rrÚ removeAndStopRó  z>ReactorFDSetTestsBuilder.test_addReader..removeAndStopóxN)r*ÚdoReadÚ addReaderÚsendallÚ runReactor©r#rr/rr.rÚtest_addReaderKs   z'ReactorFDSetTestsBuilder.test_addReadercs`ˆ ¡\}}}‡fdd„}||_| |¡| |¡| d¡| d|jd|j¡ˆ |¡dS)zÈ L{reactor.removeReader()} accepts an L{IReadDescriptor} provider previously passed to C{reactor.addReader()} and causes it to no longer be monitored for input events. cóˆ d¡dS©NzdoRead should not be called©Úfailr©r#rrr;fóz8ReactorFDSetTestsBuilder.test_removeReader..failr1rN)r*r2r3r,r4Ú callLaterr-r5©r#r(r)rr;rr<rÚtest_removeReader^s    z*ReactorFDSetTestsBuilder.test_removeReadercs:| ¡\‰‰}‡‡fdd„}|ˆ_ˆ ˆ¡| ˆ¡dS)z³ C{reactor.addWriter()} accepts an L{IWriteDescriptor} provider and calls its C{doWrite} method when it may be possible to write to its C{fileno}. cr+r%)Ú removeWriterr-rr.rrr/~r0z>ReactorFDSetTestsBuilder.test_addWriter..removeAndStopN)r*ÚdoWriteÚ addWriterr5r6rr.rÚtest_addWritervs  z'ReactorFDSetTestsBuilder.test_addWritercCsª| ¡}t|d|dƒ}t|d|ƒ}t|d|ƒ}| ¡\}}| ||ƒ¡| ||ƒ¡||ƒ| ||ƒ¡| ||ƒ¡||ƒ| ||ƒ¡| ||ƒ¡dS)z= Helper for getReaders and getWriters tests. ÚgetÚsÚaddÚremoveN)r&Úgetattrr$Ú assertNotInÚassertIn)r#Úkindr(rErGrHrrrrrÚ _getFDTestˆs z#ReactorFDSetTestsBuilder._getFDTestcCó| d¡dS)z¡ L{IReactorFDSet.getReaders} reflects the additions and removals made with L{IReactorFDSet.addReader} and L{IReactorFDSet.removeReader}. ÚReaderN©rMr<rrrÚtest_getReadersžóz(ReactorFDSetTestsBuilder.test_getReaderscsVˆ ¡\}}}‡fdd„}||_| |¡| |¡| d|jd|j¡ˆ |¡dS)zÊ L{reactor.removeWriter()} accepts an L{IWriteDescriptor} provider previously passed to C{reactor.addWriter()} and causes it to no longer be monitored for outputability. cr8©NzdoWrite should not be calledr:rr<rrr;­r=z8ReactorFDSetTestsBuilder.test_removeWriter..failrN)r*rBrCrAr>r-r5r?rr<rÚtest_removeWriter¥s   z*ReactorFDSetTestsBuilder.test_removeWritercCrN)z¡ L{IReactorFDSet.getWriters} reflects the additions and removals made with L{IReactorFDSet.addWriter} and L{IReactorFDSet.removeWriter}. ÚWriterNrPr<rrrÚtest_getWriters¼rRz(ReactorFDSetTestsBuilder.test_getWriterscsˆˆ ¡}ˆ ¡\}}}‡fdd„|_‡fdd„|_| d¡| |¡| |¡| ¡}| d|jd|j ¡ˆ  |¡ˆ  ||g¡dS)z¯ C{reactor.removeAll()} removes all registered L{IReadDescriptor} providers and all registered L{IWriteDescriptor} providers and returns them. có ˆ d¡Sr9r:rr<rrÚÍó z9ReactorFDSetTestsBuilder.test_removeAll..crWrSr:rr<rrrXÎrYr1rN) r&r*r2rBr4r3rCÚ removeAllr>r-r5Ú assertEqual)r#r(r)rÚremovedrr<rÚtest_removeAllÃs    z'ReactorFDSetTestsBuilder.test_removeAllcCs8| ¡}t|ƒ}| |j¡| |¡| |jg¡dS)z† A descriptor's C{fileno} method should not be called after the descriptor has been removed from the reactor. N)r&ÚRemovingDescriptorÚcallWhenRunningÚstartr5r[Úcalls)r#r(Ú descriptorrrrÚtest_removedFromReactoràs   z0ReactorFDSetTestsBuilder.test_removedFromReactorcsd| ¡‰| ¡\‰‰G‡‡‡fdd„dtƒ}|ˆƒ}ˆ |¡ˆ d¡| ˆ¡| |jd¡dS)zp If L{FileDescriptor.fileno} returns C{-1}, the descriptor is removed from the reactor. cs:eZdZ” ¡ZdZdd„Z‡‡fdd„Z‡fdd„ZdS) zWReactorFDSetTestsBuilder.test_negativeOneFileDescriptor..DisappearingDescriptorócSs|jSr%)Ú_filenor<rrrr'ùsz^ReactorFDSetTestsBuilder.test_negativeOneFileDescriptor..DisappearingDescriptor.filenocs(d|_|jˆ d¡7_ˆ d¡dS)Néÿÿÿÿr óy)reÚ _receivedÚrecvÚsendr<)rrrrr2üsz^ReactorFDSetTestsBuilder.test_negativeOneFileDescriptor..DisappearingDescriptor.doReadcs ˆ ¡dSr%©r-©r#Úreason©r(rrÚconnectionLosts zfReactorFDSetTestsBuilder.test_negativeOneFileDescriptor..DisappearingDescriptor.connectionLostN)Ú__name__Ú __module__Ú __qualname__r'rerhr2ror©rr(rrrÚDisappearingDescriptorôs rtr1N)r&r$rr3rjr5r[rh)r#rtrbrrsrÚtest_negativeOneFileDescriptorës    z7ReactorFDSetTestsBuilder.test_negativeOneFileDescriptorz*Cannot duplicate socket filenos on Windowscsˆˆ ¡‰ˆjj}|dvrt|›dƒ‚ˆ ¡\}‰G‡‡fdd„dtƒ}ˆ |ƒ¡‡‡fdd„}ˆ d|¡ˆ ˆ¡ˆ  t j ¡dS) a The file descriptor underlying a FileDescriptor may be closed and replaced by another at some point. Bytes which arrive on the new descriptor must not be delivered to the FileDescriptor which was originally registered with the original descriptor of the same number. Practically speaking, this is difficult or impossible to detect. The implementation relies on C{fileno} raising an exception if the original descriptor has gone away. If C{fileno} continues to return the original file descriptor value, the reactor may deliver events from that descriptor. This is a best effort attempt to ease certain debugging situations. Applications should not rely on it intentionally. )Ú EPollReactorÚ KQueueReactorÚ CFReactorÚAsyncioSelectorReactorz$ cannot detect lost file descriptorscs0eZdZdZ‡fdd„Zdd„Z‡fdd„ZdS) z@ReactorFDSetTestsBuilder.test_lostFileDescriptor..VictimaŠ This L{FileDescriptor} will have its socket closed out from under it and another socket will take its place. It will raise a socket.error from C{fileno} after this happens (because socket objects remember whether they have been closed), so as long as the reactor calls the C{fileno} method the problem will be detected. csˆ ¡Sr%)r'r<)rrrr'3ózGReactorFDSetTestsBuilder.test_lostFileDescriptor..Victim.filenocSstdƒ‚)Nz$Victim.doRead should never be called)Ú Exceptionr<rrrr26rzzGReactorFDSetTestsBuilder.test_lostFileDescriptor..Victim.doReadcs ˆ ¡dS)zÊ When the problem is detected, the reactor should disconnect this file descriptor. When that happens, stop the reactor so the test ends. Nrkrlrnrrro9s zOReactorFDSetTestsBuilder.test_lostFileDescriptor..Victim.connectionLostN)rprqrrÚ__doc__r'r2ror)r(rrrÚVictim*s  r}cs:ˆ ¡\}}ˆ ¡}ˆ ¡t | ¡|¡| d¡dS)Nr1)r$r'rÚosÚdup2rj)ÚnewCÚnewSr')r#rrrÚmessItUpHs zBReactorFDSetTestsBuilder.test_lostFileDescriptor..messItUprN) r&Ú __class__rpr r$rr3r>r5ÚflushLoggedErrorsrÚerror)r#Únamerr}r‚r)r(r#rrÚtest_lostFileDescriptor s    z0ReactorFDSetTestsBuilder.test_lostFileDescriptorcCsŠ| ¡}Gdd„dtƒ}| ¡\}}||ƒ}|j|_||ƒ}|j|_| |¡| |¡| |j¡| |¡|  |j ¡|  |j ¡dS)z‡ Any file descriptors added to the reactor have their C{connectionLost} called when C{reactor.stop} is called. c@seZdZdd„Zdd„ZdS)zSReactorFDSetTestsBuilder.test_connectionLostOnShutdown..DoNothingDescriptorcSódSr%rr<rrrr2_ózZReactorFDSetTestsBuilder.test_connectionLostOnShutdown..DoNothingDescriptor.doReadcSrˆr%rr<rrrrBbr‰z[ReactorFDSetTestsBuilder.test_connectionLostOnShutdown..DoNothingDescriptor.doWriteN)rprqrrr2rBrrrrÚDoNothingDescriptor^s rŠN) r&rr$r'r3rCr_r-r5Ú assertTrueÚ disconnected)r#r(rŠrrÚfd1Úfd2rrrÚtest_connectionLostOnShutdownWs      z6ReactorFDSetTestsBuilder.test_connectionLostOnShutdownN)rprqrrr|rÚrequiredInterfacesr$r*r7r@rDrMrQrTrVr]rcrurr Ú isWindowsr‡rrrrrr!1s$     Lr!c@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)r^a˜ A read descriptor which removes itself from the reactor as soon as it gets a chance to do a read and keeps track of when its own C{fileno} method is called. @ivar insideReactor: A flag which is true as long as the reactor has this descriptor as a reader. @ivar calls: A list of the bottom of the call stack for any call to C{fileno} when C{insideReactor} is false. cCs$||_d|_g|_tƒ\|_|_dS©NF)r(Ú insideReactorrar ÚreadÚwrite)r#r(rrrÚ__init__‚szRemovingDescriptor.__init__cCs"d|_|j |¡|j d¡dS)NTóa)r“r(r3r•rjr<rrrr`ˆs zRemovingDescriptor.startcCsdS)NÚfoorr<rrrÚ logPrefixr‰zRemovingDescriptor.logPrefixcCs4|j |¡d|_|j ¡|j ¡|j ¡dSr’)r(r,r“r-r”rr•r<rrrr2s   zRemovingDescriptor.doReadcCs,|js|j tjdddd…¡|j ¡S)Né)Úlimitrf)r“raÚappendÚ tracebackÚ extract_stackr”r'r<rrrr'—s zRemovingDescriptor.filenocCrˆr%rrlrrrroœsz!RemovingDescriptor.connectionLostN) rprqrrr|r–r`r™r2r'rorrrrr^ts  r^)r|r~rrÚunittestrÚzope.interfacerÚtwisted.internet.abstractrÚtwisted.internet.interfacesrrÚtwisted.internet.tcprrÚ#twisted.internet.test.reactormixinsr Útwisted.python.runtimer Útwisted.trial.unittestr r r!r^ÚglobalsÚupdateÚmakeTestCaseClassesrrrrÚs&      E.