YfL@dZddlmZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZddlmZddlmZddlmZmZmZddlmZmZmZdd lmZmZdd lm Z m!Z!dd l"m#Z#dd l$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*ddl+m,Z,ddl-m.Z/Gdde,Z0dZ1dZ2GddZ3ee)GddZ4Gdde/Z5Gdde/Z6ee*GddZ7Gdde/Z8Gd d!e!Z9Gd"d#e/Z:d$S)%zh Tests for implementations of L{IHostnameResolver} and their interactions with reactor implementations. ) defaultdict) AF_INETAF_INET6 AF_UNSPEC EAI_NONAME IPPROTO_TCP SOCK_DGRAM SOCK_STREAMgaierror getaddrinfo)Locklocal) implementer) verifyObject) LockWorkerTeamcreateMemoryWorker)ComplexResolverSimplifier GAIResolverSimpleResolverComplexifier) IPv4Address IPv6Address)PluggableResolverMixin ReactorBase)Deferred)DNSLookupError)IHostnameResolverIReactorPluggableNameResolverIResolutionReceiverIResolverSimple) ThreadPool)SynchronousTestCaseceZdZdZdZdS)DeterministicThreadPoolz6 Create a deterministic L{ThreadPool} object. cLd|_d|_d|_g|_||_dS)zE Create a L{DeterministicThreadPool} from a L{Team}. N)minmaxnamethreads_team)selfteams U/var/www/html/env/lib/python3.11/site-packages/twisted/internet/test/test_resolver.py__init__z DeterministicThreadPool.__init__4s+   N)__name__ __module__ __qualname____doc__r/r0r.r$r$/s-r0r$c t\}tttt t fdd|fS)z Create a deterministic threadpool. @return: 2-tuple of L{ThreadPool}, 0-argument C{work} callable; when C{work} is called, do the work. cSNr5workersr.z#deterministicPool..Isvr0cdSr8r5r5r0r.r;z#deterministicPool..Isr0)rr$rrr r)doerr:s @r.deterministicPoolr>?sY&''LFD DFFEGG,,~~~~ M M    r0c\t\}Gfdd}||fS)z Create a deterministic L{IReactorThreads} @return: a 2-tuple consisting of an L{IReactorThreads}-like object and a 0-argument callable that will perform one unit of work invoked via that object's C{callFromThread} method. ceZdZfdZdS)(deterministicReactorThreads..CFTcBfddS)NciSr8r5)afksr.r;zIdeterministicReactorThreads..CFT.callFromThread..[saajajjr0)do)r,rErDrFr:s ```r.callFromThreadz7deterministicReactorThreads..CFT.callFromThreadZs. II(((((( ) ) ) ) )r0N)r1r2r3rHr9sr.CFTrAYs. * * * * * * *r0rI)r)r=rIr:s @r.deterministicReactorThreadsrJOsX&''LFD********** 355$;r0c0eZdZdZdZddZeeedfdZ dS) FakeAddrInfoGetterz/ Test object implementing getaddrinfo. cFg|_tt|_dS)z1 Create a L{FakeAddrInfoGetter}. N)callsrlistresultsr,s r.r/zFakeAddrInfoGetter.__init__es "4(( r0rc|j||||||f|j|}|r|Sttd)a Mock for L{socket.getaddrinfo}. @param host: see L{socket.getaddrinfo} @param port: see L{socket.getaddrinfo} @param family: see L{socket.getaddrinfo} @param socktype: see L{socket.getaddrinfo} @param proto: see L{socket.getaddrinfo} @param flags: see L{socket.getaddrinfo} @return: L{socket.getaddrinfo} z,nodename nor servname provided, or not known)rNappendrPr r)r,hostportfamilysocktypeprotoflagsrPs r.r zFakeAddrInfoGetter.getaddrinfolsR$ 4vxFGGG,t$  WN:'UVV Vr0r0cP|j||||||fdS)a Add a result for a given hostname. When this hostname is resolved, the result will be a L{list} of all results C{addResultForHost} has been called with using that hostname so far. @param host: The hostname to give this result for. This will be the next result from L{FakeAddrInfoGetter.getaddrinfo} when passed this host. @type canonname: native L{str} @param sockaddr: The resulting socket address; should be a 2-tuple for IPv4 or a 4-tuple for IPv6. @param family: An C{AF_*} constant that will be returned from C{getaddrinfo}. @param socktype: A C{SOCK_*} constant that will be returned from C{getaddrinfo}. @param proto: An C{IPPROTO_*} constant that will be returned from C{getaddrinfo}. @param canonname: A canonical name that will be returned from C{getaddrinfo}. @type canonname: native L{str} N)rPrS)r,rTsockaddrrVrWrX canonnames r.addResultForHostz#FakeAddrInfoGetter.addResultForHosts1H T!!68UIx"PQQQQQr0N)rrrr) r1r2r3r4r/r rr rr]r5r0r.rLrL`sm)))WWWW:$R$R$R$R$R$Rr0rLc2eZdZdZdZdZdZdZdZdZ dS) ResultHolderzI A resolution receiver which holds onto the results it received. Fc||_dS)z> Create a L{ResultHolder} with a L{UnitTest}. N) _testCase)r,testCases r.r/zResultHolder.__init__s"r0c0d|_||_g|_dS)zg Hostname resolution began. @param hostResolution: see L{IResolutionReceiver} TN)_started _resolution _addresses)r,hostResolutions r.resolutionBeganzResultHolder.resolutionBegans  )r0c:|j|dS)z^ An address was resolved. @param address: see L{IResolutionReceiver} N)rfrS)r,addresss r.addressResolvedzResultHolder.addressResolveds w'''''r0cd|_dS)z2 Hostname resolution is complete. TN)_endedrQs r.resolutionCompletezResultHolder.resolutionCompletes r0N) r1r2r3r4rdrmr/rhrkrnr5r0r.r_r_sfH F""" (((r0r_ceZdZdZdZdS) HelperTestsz? Tests for error cases of helpers used in this module. ct\|_|_d}|j|||t |tddS)zq L{DeterministicThreadPool} will log any exceptions that its "thread" workers encounter. c ddz S)Nr&rr5r5r0r. divideByZeroz9HelperTests.test_logErrorsInThreads..divideByZeros q5Lr0r&N)r>pool doThreadWork callInThread assertEquallenflushLoggedErrorsZeroDivisionError)r,rss r.test_logErrorsInThreadsz#HelperTests.test_logErrorsInThreadss (9':':$ 4$    |,,,  T334EFFGGKKKKKr0N)r1r2r3r4r{r5r0r.rprps2 L L L L Lr0rpcNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) HostnameResolutionTestsz( Tests for hostname resolution. ct\__t\__t _tjfdjj _ dS)z* Set up a L{GAIResolver}. cjSr8rtrQsr.r;z/HostnameResolutionTests.setUp.. $)r0N) r>rtrurJreactor doReactorWorkrLgetterrr resolverrQs`r.setUpzHostnameResolutionTests.setUpsi(9':':$ 4$+F+H+H( d((** # L++++T[-D   r0ct|}|jdd|j|d}||j|||jd||j d| | ||j d||j tdddgdS) z Resolving an individual hostname that results in one address from getaddrinfo results in a single call each to C{resolutionBegan}, C{addressResolved}, and C{resolutionComplete}. sample.example.com)4.3.2.1rTFTCPrrN)r_rr]rresolveHostNameassertIsrerwrdrmrurrfrr,receiver resolutions r.test_resolveOneHostz+HostnameResolutionTests.test_resolveOneHosts  %% $$%9>JJJ]228=QRR  h*J777 *D111 %000   $/// ,{5)Q/O/O.PQQQQQr0c 0t|}d}d}|jddd||ft|j|d}||j|||j d||j d| | ||j d||j td dd||gd S) a Resolving an individual hostname that results in one address from getaddrinfo results in a single call each to C{resolutionBegan}, C{addressResolved}, and C{resolutionComplete}; C{addressResolved} will receive an L{IPv6Address}. r&r::1r)rVTFrN)r_rr]rrrrrerwrdrmrurrfr)r,rflowInfoscopeIDrs r.test_resolveOneIPv6Hostz/HostnameResolutionTests.test_resolveOneIPv6Host s% %% $$ 5!Xw"? %   ]228=QRR  h*J777 *D111 %000   $///   +eUAx"Q"Q!R     r0ct|}|j|d}||j|||||jd||j d||j gdS)a  Resolving a hostname that results in C{getaddrinfo} raising a L{gaierror} will result in the L{IResolutionReceiver} receiving a call to C{resolutionComplete} with no C{addressResolved} calls in between; no failure is logged. rTN) r_rrrrerurrwrdrmrfrs r. test_gaierrorz%HostnameResolutionTests.test_gaierror!s %%]228=QRR  h*J777   *D111 $/// ,b11111r0cBt|}|j|d|}||j||||jjd\}}}}} } | ||dS)z Verify that the given set of address types results in the given C{AF_} constant being passed to C{getaddrinfo}. @param addrTypes: iterable of L{IAddress} implementers @param expectedAF: an C{AF_*} constant r) addressTypesrN) r_rrrrerurrrNrw) r, addrTypes expectedAFrrrTrUrVrWrXrYs r._resolveOnlyTestz(HostnameResolutionTests._resolveOnlyTest1s %%]22 *3   h*J777  59[5Fq5I2dFHeU ,,,,,r0cH|tgtdS)z When passed an C{addressTypes} parameter containing only L{IPv4Address}, L{GAIResolver} will pass C{AF_INET} to C{getaddrinfo}. N)rrrrQs r.test_resolveOnlyIPv4z,HostnameResolutionTests.test_resolveOnlyIPv4Ds" {mW55555r0cH|tgtdS)z When passed an C{addressTypes} parameter containing only L{IPv6Address}, L{GAIResolver} will pass C{AF_INET6} to C{getaddrinfo}. N)rrrrQs r.test_resolveOnlyIPv6z,HostnameResolutionTests.test_resolveOnlyIPv6Ks" {mX66666r0c|ttgt|dtdS)z When passed an C{addressTypes} parameter containing both L{IPv4Address} and L{IPv6Address} (or the default of C{None}, which carries the same meaning), L{GAIResolver} will pass C{AF_UNSPEC} to C{getaddrinfo}. N)rrrrrQs r.test_resolveBothz(HostnameResolutionTests.test_resolveBothRs; {K8)DDD dI.....r0c&t|}|j|ddt|}|j|dd|||||jjd\}}}}}}|jjd\}}}} }}||t|| tdS)z When passed a C{transportSemantics} paramter, C{'TCP'} (the value present in L{IPv4Address.type} to indicate a stream transport) maps to C{SOCK_STREAM} and C{'UDP'} maps to C{SOCK_DGRAM}. example.comr)transportSemanticsUDPrr&N) r_rrrurrrNrwr r ) r,r receiver2rTrUrV socktypeTrXrY socktypeUs r.#test_transportSemanticsToSocketTypez;HostnameResolutionTests.test_transportSemanticsToSocketType[s  %% %%h RW%XXX &&  %% } &       6:k6G6J3dFIue6:k6G6J3dFIue K000 J/////r0cXt|}d}d}ttfD]L}|jddd||ft ||jddt |M|j|d| | |j \}}}}| |j d| |j d| |j d | |j d d S) z When L{GAIResolver} receives a C{SOCK_DGRAM} result from C{getaddrinfo}, it returns a C{'TCP'} L{IPv4Address} or L{IPv6Address}; if it receives C{SOCK_STREAM} then it returns a C{'UDP'} type of same. r&rrrr)rVrW)z 127.0.0.3rrrN)r_r r rr]rrrrrurrfrwtype) r,rrrrWstream4stream6dgram4dgram6s r.test_socketTypeToAddressTypez4HostnameResolutionTests.test_socketTypeToAddressTypepsH  %%#Z/  H K ( (8W-! )    K ( (/( )     %%h >>>  +3+>(&& u--- u--- e,,, e,,,,,r0N)r1r2r3r4rrrrrrrrrrr5r0r.r}r}s    RRR"   0222 ---&666777///000*-----r0r}c eZdZdZdZddZdS)SillyResolverSimplez6 Trivial implementation of L{IResolverSimple} cg|_dS)zd Create a L{SillyResolverSimple} with a queue of requests it is working on. N) _requestsrQs r.r/zSillyResolverSimple.__init__s r0r5ch|jt|jdS)z Implement L{IResolverSimple.getHostByName}. @param name: see L{IResolverSimple.getHostByName}. @param timeout: see L{IResolverSimple.getHostByName}. @return: see L{IResolverSimple.getHostByName}. )rrSr)r,r)timeouts r. getHostByNamez!SillyResolverSimple.getHostByNames+ hjj)))~b!!r0N)r5)r1r2r3r4r/rr5r0r.rrsA " " " " " "r0rc0eZdZdZdZdZdZdZdZdS)LegacyCompatibilityTestsz Older applications may supply an object to the reactor via C{installResolver} that only provides L{IResolverSimple}. L{SimpleResolverComplexifier} is a wrapper for an L{IResolverSimple}. c.t}t|}t|}||jd||d||jd||jd||jg|jd d||jtdddg||jddS)z L{SimpleResolverComplexifier} translates C{resolveHostName} into L{IResolutionReceiver.addressResolved}. FrTr 192.168.1.1rN) rrr_rwrdrrmrfrcallbackrr,simplecomplexrs r. test_successz%LegacyCompatibilityTests.test_successs %&&,V44%% *E222-888 *D111 %000 ,b111$$]333 ,{5-QR/S/S.TUUU $/////r0c(t}t|}t|}||jd||d||jd||jd||jg|jd td||jd||jgdS)z L{SimpleResolverComplexifier} translates a known error result from L{IResolverSimple.resolveHostName} into an empty result. FrTrnopeN) rrr_rwrdrrmrfrerrbackrrs r. test_failurez%LegacyCompatibilityTests.test_failures %&&,V44%% *E222-888 *D111 %000 ,b111##N6$:$:;;; $/// ,b11111r0ct}t|}t|}||jd||d||jd||jd||jg|jd td|t| td||jd||jgdS)z L{SimpleResolverComplexifier} translates an unknown error result from L{IResolverSimple.resolveHostName} into an empty result and a logged error. FrTrzowr&N) rrr_rwrdrrmrfrrrzrxryrs r. test_errorz#LegacyCompatibilityTests.test_errors+ %&&,V44%% *E222-888 *D111 %000 ,b111##$5e$<$<=== T334EFFGGKKK $/// ,b11111r0ct\__t\__t _tjfdjj _ tj }j dd| d}| d}|jt"|ddS)z L{ComplexResolverSimplifier} translates an L{IHostnameResolver} into an L{IResolverSimple} for applications that still expect the old interfaces to be in place. cjSr8rrQsr.r;z:LegacyCompatibilityTests.test_simplifier..rr0r) 192.168.3.4znx.example.comrN)r>rtrurJrrrLrrr rrr]rrwfailureResultOfrrsuccessResultOf)r,simpleResolversuccessfailures` r.test_simplifierz(LegacyCompatibilityTests.test_simplifiersO (9':':$ 4$+F+H+H( d((** # L++++T[-D   34=AA $$]4IJJJ ..}== ../?@@     --g66;^LLL --g66 FFFFFr0ct}t|}t|}||dd||jd||jd||jg|jd d||jtdddg||jddS) z L{SimpleResolverComplexifier} preserves the C{port} argument passed to C{resolveHostName} in its returned addresses. rrTFrrrN) rrr_rrwrdrmrfrrrrs r.test_portNumberz(LegacyCompatibilityTests.test_portNumbers %&&,V44%%->>> *D111 %000 ,b111$$]333 ,{5-QU/V/V.WXXX $/////r0N) r1r2r3r4rrrrrr5r0r.rrsl 000"222"222&GGG.00000r0rceZdZdZdZdS)JustEnoughReactorzT Just enough subclass implementation to be a valid L{ReactorBase} subclass. cdS)z Do nothing. Nr5rQs r. installWakerzJustEnoughReactor.installWakersr0N)r1r2r3r4rr5r0r.rr s-     r0rc$eZdZdZdZdZdZdS)ReactorInstallationTestsz Tests for installing old and new resolvers onto a L{PluggableResolverMixin} and L{ReactorBase} (from which all of Twisted's reactor implementations derive). ct}tt|tt|jtt |jdS)z L{PluggableResolverMixin} (and its subclasses) implement both L{IReactorPluggableNameResolver} and L{IReactorPluggableResolver}. N)rrrr rr nameResolverr,rs r.test_interfaceCompliancez1ReactorInstallationTests.test_interfaceCompliancesM )**2G<<<_g&6777&(<=====r0ct}t}tt||||jt||jj |dS)zf L{PluggableResolverMixin} will wrap an L{IResolverSimple} in a complexifier. N) rrrr installResolverassertIsInstancerrr_simpleResolver)r,rits r.test_installingOldStyleResolverz8ReactorInstallationTests.test_installingOldStyleResolver(sr )** " "_g&=&=b&A&ABBB g24NOOO g*:B?????r0cvt}||jt||jjt ||jt||jj |||jj |jdS)zD L{ReactorBase} defaults to using a L{GAIResolver}. N) rrrrr _getaddrinfor rr_reactor _nameResolverrs r.test_defaultToGAIResolverz2ReactorInstallationTests.test_defaultToGAIResolver3s$%% g2K@@@ g*7EEE g.0IJJJ g*3W=== g&4g6JKKKKKr0N)r1r2r3r4rrrr5r0r.rrsS >>> @ @ @ L L L L Lr0rN);r4 collectionsrsocketrrrrrr r r r threadingr rzope.interfacerzope.interface.verifyrtwisted._threadsrrrtwisted.internet._resolverrrrtwisted.internet.addressrrtwisted.internet.baserrtwisted.internet.deferrtwisted.internet.errorrtwisted.internet.interfacesrrrr twisted.python.threadpoolr!twisted.trial.unittestr"UnitTestr$r>rJrLr_rpr}rrrrr5r0r.rs $#####                      "!!!!!!!&&&&&&......AAAAAAAAAA >=======EEEEEEEE++++++111111 100000BBBBBB     j       "IRIRIRIRIRIRIRIRX  !!$$$$$$$"!$NLLLLL(LLL(b-b-b-b-b-hb-b-b-J _""""""""4a0a0a0a0a0xa0a0a0H        %L%L%L%L%Lx%L%L%L%L%Lr0