o bt5@sndZddlZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZmZmZdd lmZdd lmZdd lmZmZmZdd lmZdd lmZddlmZmZzddl Z!Wn e"ykdZ Ynwe!Z e eeGdddZ#GdddeZ$ddZ%GdddZ&Gddde&eZ'Gddde&eZ(GdddeZ)ee dGdddeZ*dS) z% Tests for L{twisted.internet.base}. N)Queue)AnyCallable)skipIf) implementer) FirstOneWins) DelayedCall ReactorBaseThreadedResolver)Deferred)DNSLookupError)IReactorThreads IReactorTimeIResolverSimple)Clock) ThreadPool)SkipTestTestCasec@sleZdZdZddZdedeffddZdd Zd d Z d d Z ddZ dedeffddZ ddZ dS) 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-sz&FakeReactor.__init__..) r_clock callLaterrrstart getThreadPoolr _threadCallsrrrr__init__'s    zFakeReactor.__init__callable.cOs|j|||fdSr)r putrr"argskwargsrrrcallFromThread1szFakeReactor.callFromThreadcCs"|j\}}}||i|dSr)r get)rfr%r&rrr_runThreadCalls4szFakeReactor._runThreadCallscCs|jdSr)rstoprrrr_stop8szFakeReactor._stopcCdSrrrrrrgetDelayedCalls;zFakeReactor.getDelayedCallscCr-rrrrrrseconds?r/zFakeReactor.secondscOr-rrr$rrr callInThreadCr/zFakeReactor.callInThreadcCr-rr)rsizerrrsuggestThreadPoolSizeGr/z!FakeReactor.suggestThreadPoolSizeN)__name__ __module__ __qualname____doc__r!rrr'r*r,r.r0r1r3rrrrr s  rc@0eZdZdZddZddZddZdd 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)appendnameiplookedUprrfakeGetHostByNameas z=ThreadedResolverTests.test_success..fakeGetHostByName gethostbynameN)r addCleanupr,patchsocketr getHostByName addCallbackr;r* assertEqualradvancecalls)rr=timeoutreactor resolvedTorAresolverdrr>r test_successQs"  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:cSstd)NzENOBUFS (this is a funny joke))OSErrorr<rrrrA~z=ThreadedResolverTests.test_failure..fakeGetHostByNamerB some.namerCN)rrDr,rErFr rG assertFailurer rHr;r*rIlenrrJrKrrLrMrA failedWithrOrPrrr test_failurets   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}. csr)r(r<resultrrrArSz=ThreadedResolverTests.test_timeout..fakeGetHostByNamerBrTrCzThe I/O was errorfulN)rrDr,rrErFr rGrUr rHr;rrJrIrVr#IOErrorrWrr[r test_timeouts       z"ThreadedResolverTests.test_timeoutcsgttGfddd}Gdddt}|}|}||tt}|j|d|j|d|j|d|j|d|j|d|t d |t t t t gd |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 )rr=timeoutsrKrrrGs zOThreadedResolverTests.test_resolverGivenStr..FakeResolver.getHostByNameN)r)r4r5r6rGrr`rr FakeResolversrac@seZdZddZdS)zFThreadedResolverTests.test_resolverGivenStr..JustEnoughReactorcSr-rrrrrr installWakerzSThreadedResolverTests.test_resolverGivenStr..JustEnoughReactor.installWakerN)r4r5r6rbrrrrJustEnoughReactors rdzexample.exampleuvääntynyt.exampleu рф.examplezxn----7sbb4ac0ad0be6cf.xn--p1airrCzxn--vntynyt-5waa.examplezxn--p1ai.exampleN)rrr installResolverrr nameResolverresolveHostNamerIrVlistmaptypestr)rrardfakerMrecrr`rtest_resolverGivenStrs(  z+ThreadedResolverTests.test_resolverGivenStrN)r4r5r6r7rQrYr^rrrrrrr9Ls # !r9cCdS)z8 Function used by L{DelayedCallTests.test_str}. Nrrrrrnothingrtc@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. cSr-rr)callrrrnooprcz0DelayedCallMixin._getDelayedCallAt..noopcSr-rrrrrrrruz4DelayedCallMixin._getDelayedCallAt..rN)r)rtimeryrrr_getDelayedCallAtsz"DelayedCallMixin._getDelayedCallAtcCs|d|_|d|_dS)zb Create two L{DelayedCall} instanced scheduled to run at different times. rrCN)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. rgAreNcSrsNg?rrrrrrruz+DelayedCallMixin.test_str..z?)rrtrIroidrdcrrrtest_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}. )r NcSrs)Ng?rrrrrrruz,DelayedCallMixin.test_repr..)rrtrIroreprrrrr 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}. Nr|r} assertTrue assertFalserr|r}rrrtest_lt  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}. Nrrrrrtest_le 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}. Nrrrrrtest_gt"rzDelayedCallMixin.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}. Nrrrrrtest_ge-rzDelayedCallMixin.test_gecCs:||j|jk||j|jk||j|jkdS)zD A L{DelayedCall} instance is only equal to itself. N)rr|r}rrrrrtest_eq9zDelayedCallMixin.test_eqcCs:||j|jk||j|jk||j|jkdS)zM A L{DelayedCall} instance is not equal to any other object. N)rr|r}rrrrrtest_neCrzDelayedCallMixin.test_neN)r4r5r6r7r{r~rrrrrrrrrrrrrvs      rvc@ eZdZdZddZddZdS)DelayedCallNoDebugTestsrwcC|tddt|dS)z! Turn debug off. debugFNrErrvr~rrrrr~SzDelayedCallNoDebugTests.setUpc C>tdtdddidddd}dt|}|t||dS) rrrrreNcSrsrrrrrrr`ruz2DelayedCallNoDebugTests.test_str..zA)rrtformatrrIro)rrexpectedrrrrZs  z DelayedCallNoDebugTests.test_strNr4r5r6r7r~rrrrrrN rc@r)DelayedCallDebugTestsrwcCr)z Turn debug on. rTNrrrrrr~mrzDelayedCallDebugTests.setUpc Cr) rrrrreNcSrsrrrrrrrzruz0DelayedCallDebugTests.test_str..z\s>          + r