U q[94@srdZddlZzddlmZWn ek r<ddlmZYnXddlmZddlmZddl m Z m Z m Z ddl mZddlmZdd lmZdd lmZmZmZdd lmZdd lmZmZee e Gd ddeZGdddeZddZGdddeZ Gddde eZ!Gddde eZ"GdddeZ#GdddeZ$z ddl%Z%Wnek rlde$_&YnXdS)z% Tests for L{twisted.internet.base}. N)Queue) implementer) ThreadPool) IReactorTimeIReactorThreadsIResolverSimple)DNSLookupError) FirstOneWins)Deferred)ThreadedResolver DelayedCall ReactorBase)Clock)TestCaseSkipTestc@s0eZdZdZddZddZddZdd Zd S) FakeReactorzl A fake reactor implementation which just supports enough reactor APIs for L{ThreadedResolver}. cs>t_jj_t_jfdd_t_dS)NcsjSN) _threadpoolselfrA/usr/lib/python3/dist-packages/twisted/internet/test/test_base.py(z&FakeReactor.__init__..) r_clockZ callLaterrrstartZ getThreadPoolr _threadCallsrrrr__init__"s   zFakeReactor.__init__cOs|j|||fdSr)rputrfargskwargsrrrcallFromThread-szFakeReactor.callFromThreadcCs|j\}}}|||dSr)rgetrrrr_runThreadCalls1szFakeReactor._runThreadCallscCs|jdSr)rstoprrrr_stop6szFakeReactor._stopN)__name__ __module__ __qualname____doc__rr#r%r'rrrrrs  rc@s0eZdZdZddZddZddZdd Zd S) ThreadedResolverTestsz( Tests for L{ThreadedResolver}. csdd}d}t}||jgg}fdd}|td|t|}|||f}||j| | |g| |g|j |d| |j j gdS) z L{ThreadedResolver.getHostByName} returns a L{Deferred} which fires with the value returned by the call to L{socket.gethostbyname} in the threadpool of the reactor passed to L{ThreadedResolver.__init__}. z 10.0.0.17zfoo.bar.example.comcs|Sr)appendnameZipZlookedUprrfakeGetHostByNameNs z=ThreadedResolverTests.test_success..fakeGetHostByName gethostbynameN)r addCleanupr'patchsocketr getHostByName addCallbackr.r% assertEqualradvancecalls)rr0timeoutreactorZ resolvedTor2resolverdrr1r test_success?s"  z"ThreadedResolverTests.test_successcCsd}t}||jdd}|td|g}t|}|d|f}||t| |j | | t |d|j|d| |jjgdS)z L{ThreadedResolver.getHostByName} returns a L{Deferred} which fires a L{Failure} if the call to L{socket.gethostbyname} raises an exception. r-cSs tddS)NzENOBUFS (this is a funny joke))IOErrorr/rrrr2lsz=ThreadedResolverTests.test_failure..fakeGetHostByNamer3 some.namer4N)rr5r'r6r7r r8 assertFailurerr9r.r%r:lenrr;r<rr=r>r2Z failedWithr?r@rrr test_failurebs   z"ThreadedResolverTests.test_failurecsd}t}||jtfdd}|td|g}t|}|d|f}||t | |j |j |d||g|j d|t|dtddS) z If L{socket.gethostbyname} does not complete before the specified timeout elapsed, the L{Deferred} returned by L{ThreadedResolver.getHostByName} fails with L{DNSLookupError}. cs dSr)r$r/resultrrr2sz=ThreadedResolverTests.test_timeout..fakeGetHostByNamer3rCr4zThe I/O was errorfulN)rr5r'rr6r7r r8rDrr9r.rr;r:rErrBrFrrIr test_timeouts       z"ThreadedResolverTests.test_timeoutcsgttGfdddt}Gdddt}|}|}||tt}|j|d|j|d|j|d|j|d|j|d| t d | t t t tgd | dd | dd | d d | dd| dddS)zm L{ThreadedResolver.getHostByName} is passed L{str}, encoded using IDNA if required. cseZdZdfdd ZdS)zAThreadedResolverTests.test_resolverGivenStr..FakeResolverrcs|tSr)r.r )rr0Ztimeoutsr<rrr8s zOThreadedResolverTests.test_resolverGivenStr..FakeResolver.getHostByNameN)r)r(r)r*r8rrLrr FakeResolversrMc@seZdZddZdS)zFThreadedResolverTests.test_resolverGivenStr..JustEnoughReactorcSsdSrrrrrr installWakerszSThreadedResolverTests.test_resolverGivenStr..JustEnoughReactor.installWakerN)r(r)r*rNrrrrJustEnoughReactorsrOzexample.exampleuvääntynyt.exampleu рф.examplezxn----7sbb4ac0ad0be6cf.xn--p1airr4zxn--vntynyt-5waa.examplezxn--p1ai.exampleN)rrobjectr ZinstallResolverr r Z nameResolverZresolveHostNamer:rElistmaptypestr)rrMrOZfaker>ZrecrrLrtest_resolverGivenStrsF  z+ThreadedResolverTests.test_resolverGivenStrN)r(r)r*r+rArGrKrYrrrrr,;s #!r,cCsdS)z8 Function used by L{DelayedCallTests.test_str}. NrrrrrnothingsrZc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)DelayedCallMixin L{DelayedCall} cCs dd}t|dddi||dS)z Get a L{DelayedCall} instance at a given C{time}. @param time: The absolute time at which the returned L{DelayedCall} will be scheduled. cSsdSrr)callrrrnoopsz0DelayedCallMixin._getDelayedCallAt..noopcSsdSrrrrrrrrz4DelayedCallMixin._getDelayedCallAt..rN)r )rtimer^rrr_getDelayedCallAtsz"DelayedCallMixin._getDelayedCallAtcCs|d|_|d|_dS)zb Create two L{DelayedCall} instanced scheduled to run at different times. rr4N)r`zeroonerrrrsetUps zDelayedCallMixin.setUpc Cs:tdtdddidddd}|t|dt|fdS) The string representation of a L{DelayedCall} instance, as returned by L{str}, includes the unsigned id of the instance, as well as its state, the function to be called, and the function arguments. rRArPNcSsdSNg?rrrrrrrz+DelayedCallMixin.test_str..z?)r rZr:rXidrdcrrrtest_strszDelayedCallMixin.test_strc Cs4tdtdddidddd}|t|t|dS)z The string representation of a L{DelayedCall} instance, as returned by {repr}, is identical to that returned by L{str}. )rg NcSsdS)Ng?rrrrrrrz,DelayedCallMixin.test_repr..)r rZr:rXreprrjrrr test_reprszDelayedCallMixin.test_reprcCsJ|j|j}}|||k|||k|||k|||kdS)z For two instances of L{DelayedCall} C{a} and C{b}, C{a < b} is true if and only if C{a} is scheduled to run before C{b}. Nrarb assertTrue assertFalserrarbrrrtest_lts zDelayedCallMixin.test_ltcCsJ|j|j}}|||k|||k|||k|||kdS)z For two instances of L{DelayedCall} C{a} and C{b}, C{a <= b} is true if and only if C{a} is scheduled to run before C{b} or at the same time as C{b}. Nrrrurrrtest_les zDelayedCallMixin.test_lecCsJ|j|j}}|||k|||k|||k|||kdS)z For two instances of L{DelayedCall} C{a} and C{b}, C{a > b} is true if and only if C{a} is scheduled to run after C{b}. Nrrrurrrtest_gts zDelayedCallMixin.test_gtcCsJ|j|j}}|||k|||k|||k|||kdS)z For two instances of L{DelayedCall} C{a} and C{b}, C{a > b} is true if and only if C{a} is scheduled to run after C{b} or at the same time as C{b}. Nrrrurrrtest_ge's zDelayedCallMixin.test_gecCs:||j|jk||j|jk||j|jkdS)zD A L{DelayedCall} instance is only equal to itself. N)rtrarbrsrrrrtest_eq4szDelayedCallMixin.test_eqcCs:||j|jk||j|jk||j|jkdS)zM A L{DelayedCall} instance is not equal to any other object. N)rsrarbrtrrrrtest_ne?szDelayedCallMixin.test_neN)r(r)r*r+r`rcrlrqrvrwrxryrzr{rrrrr[s        r[c@s eZdZdZddZddZdS)DelayedCallNoDebugTestsr\cCs|tddt|dS)z! Turn debug off. debugFNr6r r[rcrrrrrcOszDelayedCallNoDebugTests.setUpc Cs>tdtdddidddd}dt|}|t||dS) rdrerfrgrPNcSsdSrhrrrrrr]rz2DelayedCallNoDebugTests.test_str..zA)r rZformatrir:rX)rrkZexpectedrrrrlWs z DelayedCallNoDebugTests.test_strNr(r)r*r+rcrlrrrrr|Ksr|c@s eZdZdZddZddZdS)DelayedCallDebugTestsr\cCs|tddt|dS)z Turn debug on. r}TNr~rrrrrciszDelayedCallDebugTests.setUpc Cs>tdtdddidddd}dt|}|t||dS) rdrerfrgrPNcSsdSrhrrrrrrwrz0DelayedCallDebugTests.test_str..z\rrrtest_exitSignalDefaultsToNonesz4ReactorBaseSignalTests.test_exitSignalDefaultsToNonecCs(t}|tjd|tj|jdS)zn ReactorBase's SIGINT handler saves the value of SIGINT to the _exitSignal attribute. N)rZsigIntsignalSIGINT assertEqualsrrrrrtest_captureSIGINTsz)ReactorBaseSignalTests.test_captureSIGINTcCs(t}|tjd|tj|jdS)zp ReactorBase's SIGTERM handler saves the value of SIGTERM to the _exitSignal attribute. N)rZsigTermrSIGTERMrrrrrrtest_captureSIGTERMsz*ReactorBaseSignalTests.test_captureSIGTERMcCs:ttdstdt}|tjd|tj|jdS)zr ReactorBase's SIGBREAK handler saves the value of SIGBREAK to the _exitSignal attribute. SIGBREAKz$signal module does not have SIGBREAKN)hasattrrrrZsigBreakrrrrrrrtest_captureSIGBREAKs  z+ReactorBaseSignalTests.test_captureSIGBREAKN)r(r)r*r+rrrrrrrrrs   rzsignal module not available)'r+r7r ImportErrorZqueueZzope.interfacerZtwisted.python.threadpoolrZtwisted.internet.interfacesrrrZtwisted.internet.errorrZtwisted.internet._resolverr Ztwisted.internet.deferr Ztwisted.internet.baser r r Ztwisted.internet.taskrZtwisted.trial.unittestrrrTrr,rZr[r|rrrrskiprrrrs8      y0