U Tg$@sdZddlZddlmZddlmZmZddlmZm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZdd lmZGd d d eZGdddejZGdddejZdS)z- Test cases for L{twisted.names.srvconnect}. N) verifyObject)deferprotocol)DNSLookupErrorServiceNameUnknownError) IConnector) MemoryReactor)clientdns srvconnect) ResolverBase) DNSNameError)unittestc@s"eZdZdZdddZddZdS) FakeResolvera Resolver that only gives out one given result. Either L{results} or L{failure} must be set and will be used for the return value of L{_lookup} @ivar results: List of L{dns.RRHeader} for the desired result. @type results: C{list} @ivar failure: Failure with an exception from L{twisted.names.error}. @type failure: L{Failure} NcCs||_||_g|_dSN)resultsfailurelookups)selfrrrF/tmp/pip-unpacked-wheel-1_7di8ic/twisted/names/test/test_srvconnect.py__init__$szFakeResolver.__init__cCs@|j||||f|jdk r0t|jggfSt|jSdS)z9 Return the result or failure on lookup. N)rappendrrZsucceedZfailr)rnameclsZqtypetimeoutrrr_lookup)s zFakeResolver._lookup)NN)__name__ __module__ __qualname____doc__rrrrrrrs rc@s eZdZdZddZddZdS) DummyFactoryzL Dummy client factory that stores the reason of connection failure. cCs d|_dSrreasonrrrrr9szDummyFactory.__init__cCs ||_dSrr")r connectorr#rrrclientConnectionFailed<sz#DummyFactory.clientConnectionFailedN)rrrr rr&rrrrr!4sr!c@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd S)!SRVConnectorTestsz/ Tests for L{srvconnect.SRVConnector}. cCsF|tdtt|_t|_t|jdd|j|_ g|_ g|_ dS)N theResolver xmpp-server example.org) patchr rrreactorr!factoryr SRVConnectorr% randIntArgsrandIntResultsr$rrrsetUpEszSRVConnectorTests.setUpcCs|j||f|jdS)a Fake randint. Returns the first element of L{randIntResults} and records the arguments passed to it in L{randIntArgs}. @param min: Lower bound of the random number. @type min: L{int} @param max: Higher bound of the random number. @type max: L{int} @return: Fake random number from L{randIntResults}. @rtype: L{int} r)r/rr0pop)rminmaxrrr_randintOszSRVConnectorTests._randintcCstt|jdS)zF L{srvconnect.SRVConnector} implements L{IConnector}. N)rrr%r$rrrtest_interfacebsz SRVConnectorTests.test_interfacecCsftjdddd}tjdtjtjd|dgtj_|j | |j j | |jjddd dS) zS Test connectTCP gets called with the address from the SRV record. }host.example.org<porttargetttlr*rtyperr=payloadN)r8r7)r Record_SRVRRHeaderSRVINr r(rr%connect assertIsNoner-r# assertEqualr, tcpClientsr2rr@rrrtest_SRVPresenths z!SRVConnectorTests.test_SRVPresentcCsDtdtj_|j||jj| |j j ddddS)zS Test connectTCP gets called with fallback parameters on NXDOMAIN. s example.orgNrAr*r)) r r r(rr%rFrGr-r#rHr,rIr2r$rrrtest_SRVNotPresentws  z$SRVConnectorTests.test_SRVNotPresentcCs@gtj_|j||jj||j j ddddS)zW Test connectTCP gets called with fallback parameters on empty result. NrArL) r r(rr%rFrGr-r#rHr,rIr2r$rrrtest_SRVNoResults z"SRVConnectorTests.test_SRVNoResultcCs\tj|jdd|jdd|_ttj_|j | |jj | |jj ddddS)zY connectTCP gets called with default port if the service is not defined. thisbetternotexistr*f)Z defaultPortNrA)r*rP)r r.r,r-r%rr r(rrFrGr#rHrIr2r$rrr)test_SRVNoResultUnknownServiceDefaultPorts  z;SRVConnectorTests.test_SRVNoResultUnknownServiceDefaultPortcCsBt|jdd|j|_ttj_|j | |jj tdS)zR Connect fails on no result, unknown service and no default port. rOr*N) r r.r,r-r%rr r(rrF assertTruer#checkr$rrr+test_SRVNoResultUnknownServiceNoDefaultPorts  z=SRVConnectorTests.test_SRVNoResultUnknownServiceNoDefaultPortcCsVtjdtjtjdddgtj_|j| |j j | |j jddddS)zU Test connectTCP gets called with fallback parameters on bad result. r*r9Nr>rArL)r rCZCNAMErEr r(rr%rFrGr-r#rHr,rIr2r$rrrtest_SRVBadResults z#SRVConnectorTests.test_SRVBadResultcCshtjdddd}tjdtjtjd|dgtj_|j | |j j |j j t||jjgdS)zH Test that connecting fails when no service is present. i.r9r:r*r>N)r rBrCrDrEr r(rr%rFZassertIsNotNoner-r#ZtraprrHr,rIrJrrrtest_SRVNoServices z#SRVConnectorTests.test_SRVNoServicecCs2gtj_|jtjjdd}|d|dS)zW The lookup name is a native string from service, protocol and domain. rs_xmpp-server._tcp.example.orgN)r r(rr%rFrrH)rrrrrtest_SRVLookupNames z$SRVConnectorTests.test_SRVLookupNamecCs*t|jdd|j|_|d|jjdS)zq L{srvconnect.SRVConnector} automatically encodes unicode domain using C{idna} encoding. z xmpp-clientuéchec.example.orgsxn--chec-9oa.example.orgN)r r.r,r-r%rHdomainr$rrrtest_unicodeDomainsz$SRVConnectorTests.test_unicodeDomaincCstdddd}tdddd}||g|j_g|j_|td|jddg|_|j | |j dd |j | |j d d ddg|_|j | |j d d |j | |j d ddS)a pickServer calculates running sum of weights and calls randint. This exercises the server selection algorithm specified in RFC 2782 by preparing fake L{random.randint} results and checking the values it was called with. rPhost1.example.orghost2.example.orgrandint r)r)rr\rA)rr^N) r rBr%orderedServersserversr+randomr5r0 pickServerrHr/rZrecord1Zrecord2rrrtest_pickServerWeightss       z(SRVConnectorTests.test_pickServerWeightscCsvtdddd}tdddd}||g|j_g|j_|td|jddg|_| d|j | d |j d S) zR Two records with equal priorities compare on weight (ascending). r\rPr]r^r_r`rr]rPr_rPN r rBr%rerfr+rgr5r0rHrhrirrrtest_pickServerSamePrioritiess  z/SRVConnectorTests.test_pickServerSamePrioritiescCsvtdddd}tdddd}||g|j_g|j_|td|jddg|_| d|j | d |j d S) zX Two records with differing priorities compare on priority (ascending). r\rrPr]r^r_r`rkrlNrmrirrrtest_srvDifferentPrioritiess  z-SRVConnectorTests.test_srvDifferentPrioritiesN)rrrr r1r5r6rKrMrNrQrTrUrWrYr[rjrnrorrrrr'@s       !r')r rgZzope.interface.verifyrZtwisted.internetrrZtwisted.internet.errorrrZtwisted.internet.interfacesrZtwisted.internet.testingrZ twisted.namesr r r Ztwisted.names.commonr Ztwisted.names.errorr Z twisted.trialrrZ ClientFactoryr!ZTestCaser'rrrrs