o i9@s~dZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZmZmZdd lmZdd lmZdd lmZmZmZdd lmZddlmZddlmZm Z zddl!Z"Wn e#yodZ!Ynwe"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@seZdZdZddZdeeefdejdej ddfd d Z d d Z d dZ ddZ defddZdeeefdejdej ddfddZddZdS) FakeReactorzl A fake reactor implementation which just supports enough reactor APIs for L{ThreadedResolver}. cs>t_jj_t_jfdd_t_dS)NcsjSN) _threadpoolselfreC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/internet/test/test_base.py1sz&FakeReactor.__init__..) r_clockZ callLaterrrstartZ getThreadPoolr _threadCallsrrrr__init__+s    zFakeReactor.__init__callableargskwargsreturnNcOs|j|||fdSr)rputrr!r"r#rrrcallFromThread5szFakeReactor.callFromThreadcCs"|j\}}}||i|dSr)rget)rfr"r#rrr_runThreadCalls:szFakeReactor._runThreadCallscCs|jdSr)rstoprrrr_stop>szFakeReactor._stopcCdSrrrrrrgetDelayedCallsAzFakeReactor.getDelayedCallscCr-rrrrrrsecondsEr/zFakeReactor.secondscOr-rrr&rrr callInThreadIszFakeReactor.callInThreadcCr-rr)rsizerrrsuggestThreadPoolSizeOr/z!FakeReactor.suggestThreadPoolSize)__name__ __module__ __qualname____doc__r rrobjectr"r#r'r*r,r.floatr0r1r3rrrrr$s4    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)appendnameipZlookedUprrfakeGetHostByNameis z=ThreadedResolverTests.test_success..fakeGetHostByName gethostbynameN)r addCleanupr,patchsocketr getHostByName addCallbackr=r* assertEqualradvancecalls)rr?timeoutreactorZ resolvedTorBresolverdrr@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<cSstd)NzENOBUFS (this is a funny joke))OSErrorr>rrrrBz=ThreadedResolverTests.test_failure..fakeGetHostByNamerC some.namerDN)rrEr,rFrGr rH assertFailurer rIr=r*rJlenrrKrLrrMrNrBZ failedWithrOrPrrr 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}. csr)r(r>resultrrrBrSz=ThreadedResolverTests.test_timeout..fakeGetHostByNamerCrTrDzThe I/O was errorfulN)rrEr,rrFrGr rHrUr rIr=rrKrJrVr%IOErrorrWrrZr 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?ZtimeoutsrLrrrHs zOThreadedResolverTests.test_resolverGivenStr..FakeResolver.getHostByNameN)r)r4r5r6rHrr^rr FakeResolversr_c@seZdZddZdS)zFThreadedResolverTests.test_resolverGivenStr..JustEnoughReactorcSr-rrrrrr installWakerzSThreadedResolverTests.test_resolverGivenStr..JustEnoughReactor.installWakerN)r4r5r6r`rrrrJustEnoughReactors rbzexample.exampleuvääntynyt.exampleu рф.examplezxn----7sbb4ac0ad0be6cf.xn--p1airrDzxn--vntynyt-5waa.examplezxn--p1ai.exampleN)rrr ZinstallResolverrr Z nameResolverZresolveHostNamerJrVlistmaptypestr)rr_rbZfakerNZrecrr^rtest_resolverGivenStrs(  z+ThreadedResolverTests.test_resolverGivenStrN)r4r5r6r7rQrXr]rkrrrrr;Ts # !r;cCdS)z8 Function used by L{DelayedCallTests.test_str}. Nrrrrrnothingrmc@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)callrrrnoopraz0DelayedCallMixin._getDelayedCallAt..noopcSr-rrrrrrrrnz4DelayedCallMixin._getDelayedCallAt..rN)r)rtimerrrrr_getDelayedCallAtsz"DelayedCallMixin._getDelayedCallAtcCs|d|_|d|_dS)zb Create two L{DelayedCall} instanced scheduled to run at different times. rrDN)rtzeroonerrrrsetUps 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. reArcNcSrlNg?rrrrrrrnz+DelayedCallMixin.test_str..z?)rrmrJrjidrdcrrrtest_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{ NcSrl)Ng?rrrrrrrnz,DelayedCallMixin.test_repr..)rrmrJrjreprr~rrr 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}. Nrurv assertTrue assertFalserrurvrrrtest_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_ge5rzDelayedCallMixin.test_gecCs:||j|jk||j|jk||j|jkdS)zD A L{DelayedCall} instance is only equal to itself. N)rrurvrrrrrtest_eqAzDelayedCallMixin.test_eqcCs:||j|jk||j|jk||j|jkdS)zM A L{DelayedCall} instance is not equal to any other object. N)rrurvrrrrrtest_neKrzDelayedCallMixin.test_neN)r4r5r6r7rtrwrrrrrrrrrrrrros      roc@(eZdZdZddZddZddZdS) DelayedCallNoDebugTestsrpcC|tddt|dS)z! Turn debug off. debugFNrFrrorwrrrrrw[zDelayedCallNoDebugTests.setUpc C>tdtdddidddd}dt|}|t||dS) rxryrzr{rcNcSrlr|rrrrrrhrnz2DelayedCallNoDebugTests.test_str..zA)rrmformatr}rJrj)rrexpectedrrrrbs  z DelayedCallNoDebugTests.test_strc C6tddddittdd}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. recSr-rrrrrrrvrnz.rcSrlNrdrrrrrrvrnrDtraceback at creationN)rrmrZ assertNotInrr~rrrtest_switchToDebugoz*DelayedCallNoDebugTests.test_switchToDebugN)r4r5r6r7rwrrrrrrrVs  rc@r) DelayedCallDebugTestsrpcCr)z Turn debug on. rTNrrrrrrwrzDelayedCallDebugTests.setUpc Cr) rxryrzr{rcNcSrlr|rrrrrrrnz0DelayedCallDebugTests.test_str..z\.rcSrlrrrrrrrrnrrN)rrmrZassertInrr~rrrtest_switchFromDebugrz*DelayedCallDebugTests.test_switchFromDebugN)r4r5r6r7rwrrrrrrr{s  rc@seZdZdZddZdS)TestSpySignalCapturingReactorza Subclass of ReactorBase to capture signals delivered to the reactor for inspection. cCrl)z* Required method, unused. Nrrrrrr`rnz*TestSpySignalCapturingReactor.installWakerN)r4r5r6r7r`rrrrrs rzsignal module not availablec@r:) 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 _exitSignalrrNrrrtest_exitSignalDefaultsToNonesz4ReactorBaseSignalTests.test_exitSignalDefaultsToNonecC(t}|tjd|tj|jdS)zn ReactorBase's SIGINT handler saves the value of SIGINT to the _exitSignal attribute. N)rZsigIntsignalSIGINT assertEqualsrrrrrtest_captureSIGINTz)ReactorBaseSignalTests.test_captureSIGINTcCr)zp ReactorBase's SIGTERM handler saves the value of SIGTERM to the _exitSignal attribute. N)rZsigTermrSIGTERMrrrrrrtest_captureSIGTERMrz*ReactorBaseSignalTests.test_captureSIGTERMcCs:ttds tdt}|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)r4r5r6r7rrrrrrrrrs  r)-r7rGqueuertypingrZunittestrZzope.interfacerZtyping_extensionsrZtwisted.internet._resolverrZtwisted.internet.baserr r Ztwisted.internet.deferr Ztwisted.internet.errorr Ztwisted.internet.interfacesr rrZtwisted.internet.taskrZtwisted.python.threadpoolrZtwisted.trial.unittestrrr_signal ImportErrorrrr;rmrorrrrrrrrsB            / r%&