U P|j9@sdZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZmZmZdd lmZdd lmZdd lmZmZmZdd lmZddlmZddlmZm Z z ddl!Z"Wne#k rdZ!YnXe"Z!e dZ$e eeGdddZ%Gddde Z&ddZ'GdddZ(Gddde(e Z)Gddde(e Z*GdddeZ+ee! dGd d!d!e Z,dS)"z% Tests for L{twisted.internet.base}. N)Queue)Callable)skipIf) implementer) ParamSpec) FirstOneWins) DelayedCall ReactorBaseThreadedResolver)Deferred)DNSLookupError)IReactorThreads IReactorTimeIResolverSimple)Clock) ThreadPool)SkipTestTestCase_Pc@seZdZdZddZeeefejej ddddZ dd Z d d Z d d Z edddZeeefejej ddddZddZdS) FakeReactorzl A fake reactor implementation which just supports enough reactor APIs for L{ThreadedResolver}. cs>t_jj_t_jfdd_t_dS)NcsjSN) _threadpoolselfrC/tmp/pip-unpacked-wheel-audpp6lv/twisted/internet/test/test_base.py1z&FakeReactor.__init__..) r_clockZ callLaterrrstartZ getThreadPoolr _threadCallsrrrr__init__+s   zFakeReactor.__init__N)callableargskwargsreturncOs|j|||fdSr)r putrr"r#r$rrrcallFromThread5szFakeReactor.callFromThreadcCs|j\}}}|||dSr)r get)rfr#r$rrr_runThreadCalls:szFakeReactor._runThreadCallscCs|jdSr)rstoprrrr_stop>szFakeReactor._stopcCsdSrrrrrrgetDelayedCallsAszFakeReactor.getDelayedCalls)r%cCsdSrrrrrrsecondsEszFakeReactor.secondscOsdSrrr'rrr callInThreadIszFakeReactor.callInThreadcCsdSrr)rsizerrrsuggestThreadPoolSizeOsz!FakeReactor.suggestThreadPoolSize)__name__ __module__ __qualname____doc__r!rrobjectr#r$r(r+r-r.floatr/r0r2rrrrr$s"    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)appendnameipZlookedUprrfakeGetHostByNameis z=ThreadedResolverTests.test_success..fakeGetHostByName gethostbynameN)r addCleanupr-patchsocketr getHostByName addCallbackr;r+ assertEqualradvancecalls)rr=timeoutreactorZ resolvedTor@resolverdrr>r test_successYs"  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))OSErrorr<rrrr@sz=ThreadedResolverTests.test_failure..fakeGetHostByNamerA some.namerBN)rrCr-rDrEr rF assertFailurer rGr;r+rHlenrrIrJrrKrLr@Z failedWithrMrNrrr test_failure|s   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<resultrrr@sz=ThreadedResolverTests.test_timeout..fakeGetHostByNamerArQrBzThe I/O was errorfulN)rrCr-rrDrEr rFrRr rGr;rrIrHrSr&IOErrorrTrrWr 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=ZtimeoutsrJrrrFs zOThreadedResolverTests.test_resolverGivenStr..FakeResolver.getHostByNameN)r)r3r4r5rFrr[rr FakeResolversr\c@seZdZddZdS)zFThreadedResolverTests.test_resolverGivenStr..JustEnoughReactorcSsdSrrrrrr installWakerszSThreadedResolverTests.test_resolverGivenStr..JustEnoughReactor.installWakerN)r3r4r5r]rrrrJustEnoughReactorsr^zexample.exampleuvääntynyt.exampleu рф.examplezxn----7sbb4ac0ad0be6cf.xn--p1airrBzxn--vntynyt-5waa.examplezxn--p1ai.exampleN)rrr ZinstallResolverrr Z nameResolverZresolveHostNamerHrSlistmaptypestr)rr\r^ZfakerLZrecrr[rtest_resolverGivenStrs(  z+ThreadedResolverTests.test_resolverGivenStrN)r3r4r5r6rOrUrZrgrrrrr9Ts #!r9cCsdS)z8 Function used by L{DelayedCallTests.test_str}. Nrrrrrnothingsrhc@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)rtimerlrrr_getDelayedCallAtsz"DelayedCallMixin._getDelayedCallAtcCs|d|_|d|_dS)zb Create two L{DelayedCall} instanced scheduled to run at different times. rrBN)rnzeroonerrrrsetUps 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. raAr_NcSsdSNg?rrrrrrrz+DelayedCallMixin.test_str..z?)rrhrHrfidrdcrrrtest_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}. )ru NcSsdS)Ng?rrrrrrrz,DelayedCallMixin.test_repr..)rrhrHrfreprrxrrr test_repr szDelayedCallMixin.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}. Nrorp assertTrue assertFalserrorprrrtest_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}. Nrrrrrtest_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}. Nrrrrrtest_gt*s 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}. Nrrrrrtest_ge5s zDelayedCallMixin.test_gecCs:||j|jk||j|jk||j|jkdS)zD A L{DelayedCall} instance is only equal to itself. N)rrorprrrrrtest_eqAszDelayedCallMixin.test_eqcCs:||j|jk||j|jk||j|jkdS)zM A L{DelayedCall} instance is not equal to any other object. N)rrorprrrrrtest_neKszDelayedCallMixin.test_neN)r3r4r5r6rnrqrzrrrrrrrrrrrris       ric@s(eZdZdZddZddZddZdS) DelayedCallNoDebugTestsrjcCs|tddt|dS)z! Turn debug off. debugFNrDrrirqrrrrrq[szDelayedCallNoDebugTests.setUpc Cs>tdtdddidddd}dt|}|t||dS) rrrsrtrur_NcSsdSrvrrrrrrhrz2DelayedCallNoDebugTests.test_str..zA)rrhformatrwrHrf)rryexpectedrrrrzbs z DelayedCallNoDebugTests.test_strc Cs6tddddittdd}d|_|dt|dS) z If L{DelayedCall.debug} changes from C{0} to C{1} between L{DelayeCall.__init__} and L{DelayedCall.__repr__} then L{DelayedCall.__repr__} returns a string that does not include the creator stack. racSsdSrrrrrrrvrz.rcSsdSNr`rrrrrrvrrBtraceback at creationN)rrhrZ assertNotInr~rxrrrtest_switchToDebugosz*DelayedCallNoDebugTests.test_switchToDebugN)r3r4r5r6rqrzrrrrrrVs rc@s(eZdZdZddZddZddZdS) DelayedCallDebugTestsrjcCs|tddt|dS)z Turn debug on. rTNrrrrrrqszDelayedCallDebugTests.setUpc Cs>tdtdddidddd}dt|}|t||dS) rrrsrtrur_NcSsdSrvrrrrrrrz0DelayedCallDebugTests.test_str..z\.rcSsdSrrrrrrrrrrN)rrhrZassertInr~rxrrrtest_switchFromDebugsz*DelayedCallDebugTests.test_switchFromDebugN)r3r4r5r6rqrzrrrrrr{src@seZdZdZddZdS)TestSpySignalCapturingReactorza Subclass of ReactorBase to capture signals delivered to the reactor for inspection. cCsdS)z* Required method, unused. Nrrrrrr]sz*TestSpySignalCapturingReactor.installWakerN)r3r4r5r6r]rrrrrsrzsignal module not availablec@s0eZdZdZddZddZddZdd Zd S) ReactorBaseSignalTestszE Tests to exercise ReactorBase's signal exit reporting path. cCst}|d|jdS)zI The default value of the _exitSignal attribute is None. N)rZassertIs _exitSignalrrLrrrtest_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)r3r4r5r6rrrrrrrrrs   r)-r6rEqueuertypingrZunittestrZzope.interfacerZtyping_extensionsrZtwisted.internet._resolverrZtwisted.internet.baserr r Ztwisted.internet.deferr Ztwisted.internet.errorr Ztwisted.internet.interfacesr rrZtwisted.internet.taskrZtwisted.python.threadpoolrZtwisted.trial.unittestrrr_signal ImportErrorrrr9rhrirrrrrrrrs>            / r%&