U ~HjL@sdZddlmZddlmZmZmZmZmZm 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/Gddde,Z0ddZ1ddZ2GdddZ3ee)GdddZ4Gddde/Z5Gddde/Z6ee*GdddZ7Gd d!d!e/Z8Gd"d#d#e!Z9Gd$d%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)SynchronousTestCasec@seZdZdZddZdS)DeterministicThreadPoolz6 Create a deterministic L{ThreadPool} object. cCs"d|_d|_d|_g|_||_dS)zE Create a L{DeterministicThreadPool} from a L{Team}. N)minmaxnamethreadsZ_team)selfZteamr)G/tmp/pip-unpacked-wheel-ewod1tgb/twisted/internet/test/test_resolver.py__init__4s 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*Iz#deterministicPool..cSsdSr0r)r)r)r)r*r2Ir3)rr"rrr r )doerr)r1r*deterministicPool?s   r5cs&t\}Gfddd}||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*r2[r3zIdeterministicReactorThreads..CFT.callFromThread..)Zdo)r(r8r7r9r1r6r*callFromThreadZsz7deterministicReactorThreads..CFT.callFromThreadN)r,r-r.r:r)r1r)r*CFTYsr;)r)r4r;r)r1r*deterministicReactorThreadsOs 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+eszFakeAddrInfoGetter.__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>appendr@r r)r(hostportfamilysocktypeprotoflagsr@r)r)r*r ls  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)r@rB)r(rCZsockaddrrErFrG canonnamer)r)r*addResultForHosts$z#FakeAddrInfoGetter.addResultForHostN)rrrr) r,r-r.r/r+r rr rrJr)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)rNrB)r(addressr)r)r*addressResolvedszResultHolder.addressResolvedcCs d|_dS)z2 Hostname resolution is complete. TN)_endedrAr)r)r*resolutionCompleteszResultHolder.resolutionCompleteN) r,r-r.r/rLrRr+rOrQrSr)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(rUr)r)r*test_logErrorsInThreadss  z#HelperTests.test_logErrorsInThreadsN)r,r-r.r/r\r)r)r)r*rTsrTc@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}. csjSr0rVr)rAr)r*r2r3z/HostnameResolutionTests.setUp..N) r5rVrWr<reactor doReactorWorkr=getterrr resolverrAr)rAr*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.1rTFTCPrerN)rKrarJrbresolveHostNameassertIsrMrXrLrRrWr`rNrr(receiver resolutionr)r)r*test_resolveOneHostsz+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#rd::1r)rETFrfN)rKrarJrrbrgrhrMrXrLrRrWr`rNr)r(rjflowInfoscopeIDrkr)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. rdTN) rKrbrgrhrMrWr`rXrLrRrNrir)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 rd)Z addressTypesrN) rKrbrgrhrMrWr`rar>rX) r(Z addrTypesZ expectedAFrjrkrCrDrErFrGrHr)r)r*_resolveOnlyTest1s 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)rsrrrAr)r)r*test_resolveOnlyIPv4Dsz,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)rsrrrAr)r)r*test_resolveOnlyIPv6Ksz,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)rsrrrrAr)r)r*test_resolveBothRsz(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.comrf)ZtransportSemanticsUDPrr#N) rKrbrgrWr`rar>rXr r) r(rjZ receiver2rCrDrEZ socktypeTrGrHZ socktypeUr)r)r*#test_transportSemanticsToSocketType[s  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#rmrwrnr)rErF)z 127.0.0.3rrfrxN)rKr rrarJrrrbrgrWr`rNrXtype) r(rjrorprFZstream4Zstream6Zdgram4Zdgram6r)r)r*test_socketTypeToAddressTypeps0  z4HostnameResolutionTests.test_socketTypeToAddressTypeN)r,r-r.r/rcrlrqrrrsrtrurvryr{r)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) _requestsrAr)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}rBr)r(r&timeoutr)r)r* getHostByNames z!SillyResolverSimple.getHostByNameN)r))r,r-r.r/r+rr)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}. FrwTr 192.168.1.1rfN) r|rrKrXrLrgrRrNr}callbackrr(simplecomplexrjr)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. FrwTrZnopeN) r|rrKrXrLrgrRrNr}errbackrrr)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. FrwTrZzowr#N) r|rrKrXrLrgrRrNr}rr[rYrZrr)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)rAr)r*r2r3z:LegacyCompatibilityTests.test_simplifier..rw) 192.168.3.4znx.example.comrN)r5rVrWr<r_r`r=rarr rbrrJrrXZfailureResultOfrzrZsuccessResultOf)r(ZsimpleResolversuccessZfailurer)rAr*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. rwrTFrrrfN) r|rrKrgrXrLrRrNr}rrrr)r)r*test_portNumbersz(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)rAr)r)r* installWakerszJustEnoughReactor.installWakerN)r,r-r.r/rr)r)r)r*r src@s(eZdZdZddZddZddZdS) ReactorInstallationTestsz Tests for installing old and new resolvers onto a L{PluggableResolverMixin} and L{ReactorBase} (from which all of Twisted's reactor implementations derive). cCs,t}tt|tt|jtt|jdS)z L{PluggableResolverMixin} (and its subclasses) implement both L{IReactorPluggableNameResolver} and L{IReactorPluggableResolver}. N)rrrrrbr nameResolverr(r_r)r)r*test_interfaceCompliances  z1ReactorInstallationTests.test_interfaceCompliancecCs>t}t}tt||||jt||jj |dS)zf L{PluggableResolverMixin} will wrap an L{IResolverSimple} in a complexifier. N) rr|rrZinstallResolverassertIsInstancerrrhZ_simpleResolver)r(r_itr)r)r*test_installingOldStyleResolver(s z8ReactorInstallationTests.test_installingOldStyleResolvercCsXt}||jt||jjt||jt||jj |||jj |jdS)zD L{ReactorBase} defaults to using a L{GAIResolver}. N) rrrrrhZ _getaddrinfor rbrZ_reactorZ _nameResolverrr)r)r*test_defaultToGAIResolver3s z2ReactorInstallationTests.test_defaultToGAIResolverN)r,r-r.r/rrrr)r)r)r*rs  rN);r/ collectionsrsocketrrrrrrr r r threadingr r Zzope.interfacerZzope.interface.verifyrZtwisted._threadsrrrZtwisted.internet._resolverrrrZtwisted.internet.addressrrZtwisted.internet.baserrZtwisted.internet.deferrZtwisted.internet.errorrZtwisted.internet.interfacesrrrrZtwisted.python.threadpoolr Ztwisted.trial.unittestr!ZUnitTestr"r5r<r=rKrTr]r|rrrr)r)r)r*s8 ,       L'&d