U  W[ON@sdZddlmZmZeZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZddlmZmZddlmZddlmZddlmZmZmZmZdd lmZ dd l!m"Z"dd l#m$Z$m%Z%m&Z&dd l'm(Z(m)Z)dd l*m+Z+m,Z,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3Gddde"e4Z5ddZ6ddZ7Gddde4Z8eeGddde4Z9Gddde Z:Gddde Z;eeGdd d e4ZGd%d&d&e e4Z?d'S)(zh Tests for implementations of L{IHostnameResolver} and their interactions with reactor implementations. )divisionabsolute_import) defaultdict) getaddrinfogaierror EAI_NONAMEAF_INETAF_INET6 AF_UNSPEC SOCK_STREAM SOCK_DGRAM IPPROTO_TCP)localLock) implementer) verifyObject)IResolutionReceiverIResolverSimpleIReactorPluggableNameResolverIHostnameResolver)SynchronousTestCase) ThreadPool)createMemoryWorkerTeam LockWorker) IPv4Address IPv6Address) GAIResolverSimpleResolverComplexifierComplexResolverSimplifier)Deferred)DNSLookupError) ReactorBasec@seZdZdZddZdS)DeterministicThreadPoolz6 Create a deterministic L{ThreadPool} object. cCs"d|_d|_d|_g|_||_dS)zE Create a L{DeterministicThreadPool} from a L{Team}. N)minmaxnameZthreadsZ_team)selfZteamr)E/usr/lib/python3/dist-packages/twisted/internet/test/test_resolver.py__init__2s z DeterministicThreadPool.__init__N)__name__ __module__ __qualname____doc__r+r)r)r)r*r#.sr#cs4t\}tttttfdddd|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. csSNr)r)Zworkerr)r*Hz#deterministicPool..cSsdSr0r)r)r)r)r*r2Hr3)rr#rrrr)doerr)r1r*deterministicPool>s  r5cs(t\}Gfdddt}||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. cseZdZfddZdS)z(deterministicReactorThreads..CFTcsfdddS)Ncs Sr0r)r)afkr)r*r2Yr3zIdeterministicReactorThreads..CFT.callFromThread..)Zdo)r(r8r7r9r1r6r*callFromThreadXsz7deterministicReactorThreads..CFT.callFromThreadN)r,r-r.r:r)r1r)r*CFTWsr;)robject)r4r;r)r1r*deterministicReactorThreadsNs r=c@s4eZdZdZddZd ddZeeedfdd Z d S) FakeAddrInfoGetterz/ Test object implementing getaddrinfo. cCsg|_tt|_dS)z1 Create a L{FakeAddrInfoGetter}. N)callsrlistresultsr(r)r)r*r+cszFakeAddrInfoGetter.__init__rcCs8|j||||||f|j|}|r*|SttddS)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 knownN)r?appendrArr)r(hostportfamilysocktypeprotoflagsrAr)r)r*rks zFakeAddrInfoGetter.getaddrinfor3cCs|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)rArC)r(rDZsockaddrrFrGrHZ canonnamer)r)r*addResultForHosts  z#FakeAddrInfoGetter.addResultForHostN)rrrr) r,r-r.r/r+rrr r rJr)r)r)r*r>^s r>c@s8eZdZdZdZdZddZddZddZd d Z d S) ResultHolderzI A resolution receiver which holds onto the results it received. FcCs ||_dS)z> Create a L{ResultHolder} with a L{UnitTest}. N)Z _testCase)r(ZtestCaser)r)r*r+szResultHolder.__init__cCsd|_||_g|_dS)zg Hostname resolution began. @param hostResolution: see L{IResolutionReceiver} TN)_started _resolution _addresses)r(ZhostResolutionr)r)r*resolutionBeganszResultHolder.resolutionBegancCs|j|dS)z^ An address was resolved. @param address: see L{IResolutionReceiver} N)rNrC)r(Zaddressr)r)r*addressResolvedszResultHolder.addressResolvedcCs d|_dS)z2 Hostname resolution is complete. TN)_endedrBr)r)r*resolutionCompleteszResultHolder.resolutionCompleteN) r,r-r.r/rLrQr+rOrPrRr)r)r)r*rKs  rKc@seZdZdZddZdS) HelperTestsz? Tests for error cases of helpers used in this module. cCsDt\|_|_dd}|j|||t|tddS)zq L{DeterministicThreadPool} will log any exceptions that its "thread" workers encounter. cSsddS)Nr$rr)r)r)r)r* divideByZerosz9HelperTests.test_logErrorsInThreads..divideByZeror$N)r5pool doThreadWorkZ callInThread assertEquallenflushLoggedErrorsZeroDivisionError)r(rTr)r)r*test_logErrorsInThreadss  z#HelperTests.test_logErrorsInThreadsN)r,r-r.r/r[r)r)r)r*rSsrSc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)HostnameResolutionTestsz( Tests for hostname resolution. csDt\__t\__t_tjfddjj _ dS)z* Set up a L{GAIResolver}. csjSr0rUr)rBr)r*r2r3z/HostnameResolutionTests.setUp..N) r5rUrVr=reactor doReactorWorkr>getterrrresolverrBr)rBr*setUps zHostnameResolutionTests.setUpcCst|}|jdd|j|d}||j|||jd||j d| | ||j d||j t dddgdS) 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.1rTFTCPrdrN)rKr`rJraresolveHostNameassertIsrMrWrLrQrVr_rNrr(receiver resolutionr)r)r*test_resolveOneHosts z+HostnameResolutionTests.test_resolveOneHostc Cst|}d}d}|jjddd||ftd|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$rc::1r)rFTFreN)rKr`rJr rarfrgrMrWrLrQrVr_rNr)r(riflowInfoscopeIDrjr)r)r*test_resolveOneIPv6Host s& z/HostnameResolutionTests.test_resolveOneIPv6HostcCsbt|}|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. rcTN) rKrarfrgrMrVr_rWrLrQrNrhr)r)r* test_gaierror%sz%HostnameResolutionTests.test_gaierrorc Cs`t|}|jj|d|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 rc)Z addressTypesrN) rKrarfrgrMrVr_r`r?rW) r(Z addrTypesZ expectedAFrirjrDrErFrGrHrIr)r)r*_resolveOnlyTest7s z(HostnameResolutionTests._resolveOnlyTestcCs|tgtdS)z When passed an C{addressTypes} parameter containing only L{IPv4Address}, L{GAIResolver} will pass C{AF_INET} to C{getaddrinfo}. N)rrrrrBr)r)r*test_resolveOnlyIPv4Ksz,HostnameResolutionTests.test_resolveOnlyIPv4cCs|tgtdS)z When passed an C{addressTypes} parameter containing only L{IPv6Address}, L{GAIResolver} will pass C{AF_INET6} to C{getaddrinfo}. N)rrrr rBr)r)r*test_resolveOnlyIPv6Ssz,HostnameResolutionTests.test_resolveOnlyIPv6cCs |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)rrrrr rBr)r)r*test_resolveBoth[sz(HostnameResolutionTests.test_resolveBothc Cst|}|jj|dddt|}|jj|ddd|||||jjd\}}}}}}|jjd\}}}} }}||t|| t dS)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.comre)ZtransportSemanticsUDPrr$N) rKrarfrVr_r`r?rWr r ) r(riZ receiver2rDrErFZ socktypeTrHrIZ socktypeUr)r)r*#test_transportSemanticsToSocketTypees    z;HostnameResolutionTests.test_transportSemanticsToSocketTypec Cst|}d}d}ttfD]4}|jjddd||ft|d|jjddt|dq|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$rlrvrmr)rFrG)z 127.0.0.3rrerwN)rKr r r`rJr rrarfrVr_rNrWtype) r(rirnrorGZstream4Zstream6Zdgram4Zdgram6r)r)r*test_socketTypeToAddressType{s0  z4HostnameResolutionTests.test_socketTypeToAddressTypeN)r,r-r.r/rbrkrprqrrrsrtrurxrzr)r)r)r*r\s  r\c@s"eZdZdZddZdddZdS) SillyResolverSimplez6 Trivial implementation of L{IResolverSimple} cCs g|_dS)zd Create a L{SillyResolverSimple} with a queue of requests it is working on. N) _requestsrBr)r)r*r+szSillyResolverSimple.__init__r)cCs|jt|jdS)z Implement L{IResolverSimple.getHostByName}. @param name: see L{IResolverSimple.getHostByName}. @param timeout: see L{IResolverSimple.getHostByName}. @return: see L{IResolverSimple.getHostByName}. )r|rCr )r(r'Ztimeoutr)r)r* getHostByNames z!SillyResolverSimple.getHostByNameN)r))r,r-r.r/r+r~r)r)r)r*r{sr{c@s8eZdZdZddZddZddZdd Zd d Zd S) 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}. cCst}t|}t|}||jd||d||jd||jd||jg|jd d||jt dddg||jddS)z L{SimpleResolverComplexifier} translates C{resolveHostName} into L{IResolutionReceiver.addressResolved}. FrvTr 192.168.1.1reN) r{rrKrWrLrfrQrNr|callbackrr(Zsimplecomplexrir)r)r* test_successs  z%LegacyCompatibilityTests.test_successcCst}t|}t|}||jd||d||jd||jd||jg|jd t d||jd||jgdS)z L{SimpleResolverComplexifier} translates a known error result from L{IResolverSimple.resolveHostName} into an empty result. FrvTrZnopeN) r{rrKrWrLrfrQrNr|errbackr!rr)r)r* test_failures z%LegacyCompatibilityTests.test_failurecCst}t|}t|}||jd||d||jd||jd||jg|jd t d|t | t d||jd||jgdS)z L{SimpleResolverComplexifier} translates an unknown error result from L{IResolverSimple.resolveHostName} into an empty result and a logged error. FrvTrZzowr$N) r{rrKrWrLrfrQrNr|rrZrXrYrr)r)r* test_errors z#LegacyCompatibilityTests.test_errorcst\__t\__t_tjfddjj _ t j }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. csjSr0r]r)rBr)r*r2r3z:LegacyCompatibilityTests.test_simplifier..rv) 192.168.3.4znx.example.comrN)r5rUrVr=r^r_r>r`rrrarrJr~rWZfailureResultOfryr!ZsuccessResultOf)r(ZsimpleResolverZsuccessZfailurer)rBr*test_simplifiers    z(LegacyCompatibilityTests.test_simplifiercCst}t|}t|}||dd||jd||jd||jg|jd d||jt dddg||jddS) z L{SimpleResolverComplexifier} preserves the C{port} argument passed to C{resolveHostName} in its returned addresses. rvrTFrrreN) r{rrKrfrWrLrQrNr|rrrr)r)r*test_portNumber s z(LegacyCompatibilityTests.test_portNumberN) r,r-r.r/rrrrrr)r)r)r*rs rc@seZdZdZddZdS)JustEnoughReactorzT Just enough subclass implementation to be a valid L{ReactorBase} subclass. cCsdS)z Do nothing. Nr)rBr)r)r* installWaker!szJustEnoughReactor.installWakerN)r,r-r.r/rr)r)r)r*rsrc@s(eZdZdZddZddZddZdS) ReactorInstallationTestsz Tests for installing old and new resolvers onto a L{ReactorBase} (from which all of Twisted's reactor implementations derive). cCs,t}tt|tt|jtt|jdS)z L{ReactorBase} (and its subclasses) provide both L{IReactorPluggableNameResolver} and L{IReactorPluggableResolver}. N)rrrrrar nameResolverr(r^r)r)r*test_interfaceCompliance.s  z1ReactorInstallationTests.test_interfaceCompliancecCsXt}||jt||jjt||jt||jj |||jj |jdS)zD L{ReactorBase} defaults to using a L{GAIResolver}. N) rassertIsInstancerrrgZ _getaddrinforrarZ_reactorZ _nameResolverrr)r)r*test_defaultToGAIResolver9s z2ReactorInstallationTests.test_defaultToGAIResolvercCs>t}t}tt||||jt||jj |dS)zS L{ReactorBase} will wrap an L{IResolverSimple} in a complexifier. N) rr{rrZinstallResolverrrrrgZ_simpleResolver)r(r^itr)r)r*test_installingOldStyleResolverEs z8ReactorInstallationTests.test_installingOldStyleResolverN)r,r-r.r/rrrr)r)r)r*r(s  rN)@r/Z __future__rrryZ __metaclass__ collectionsrZsocketrrrrr r r r r Z threadingrrZzope.interfacerZzope.interface.verifyrZtwisted.internet.interfacesrrrrZtwisted.trial.unittestrZUnitTestZtwisted.python.threadpoolrZtwisted._threadsrrrZtwisted.internet.addressrrZtwisted.internet._resolverrrrZtwisted.internet.deferr Ztwisted.internet.errorr!Ztwisted.internet.baser"r<r#r5r=r>rKrSr\r{rrrr)r)r)r*s< ,       L*1i