o b$@sdZddlZddlmZddlmZmZddlmZm Z ddl m Z ddl m Z mZmZddlmZdd lmZdd lmZdd lmZGd d d eZGdddejZGdddejZdS)z- Test cases for L{twisted.names.srvconnect}. N) verifyObject)deferprotocol)DNSLookupErrorServiceNameUnknownError) IConnector)clientdns srvconnect) ResolverBase) DNSNameError) MemoryReactor)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)selfrrrD/usr/lib/python3/dist-packages/twisted/names/test/test_srvconnect.py__init__$s zFakeResolver.__init__cCs<|j||||f|jdurt|jggfSt|jS)z9 Return the result or failure on lookup. N)rappendrrsucceedfailr)rnameclsqtypetimeoutrrr_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|_dSrreasonrrrrr9 zDummyFactory.__init__cCs ||_dSrr%)r connectorr&rrrclientConnectionFailed<r(z#DummyFactory.clientConnectionFailedN)r r!r"r#rr*rrrrr$4s r$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) patchrrr reactorr$factoryr SRVConnectorr) randIntArgsrandIntResultsr'rrrsetUpEs  zSRVConnectorTests.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)r3rr4pop)rminmaxrrr_randintOs zSRVConnectorTests._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.rtyperrApayloadN)r<r;)r Record_SRVRRHeaderSRVINrr,rr)connect assertIsNoner1r& assertEqualr0 tcpClientsr6rrDrrrtest_SRVPresenths  z!SRVConnectorTests.test_SRVPresentcCsDtdtj_|j||jj| |j j ddddS)zS Test connectTCP gets called with fallback parameters on NXDOMAIN. s example.orgNrEr.r-) r rr,rr)rJrKr1r&rLr0rMr6r'rrrtest_SRVNotPresentws  z$SRVConnectorTests.test_SRVNotPresentcCs@gtj_|j||jj||j j ddddS)zW Test connectTCP gets called with fallback parameters on empty result. NrErP) rr,rr)rJrKr1r&rLr0rMr6r'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) defaultPortNrE)r.rT)r r2r0r1r)rrr,rrJrKr&rLrMr6r'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. rSr.N) r r2r0r1r)rrr,rrJ 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.r=NrBrErP)r rGCNAMErIrr,rr)rJrKr1r&rLr0rMr6r'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.r=r>r.rBN)r rFrGrHrIrr,rr)rJassertIsNotNoner1r&traprrLr0rMrNrrrtest_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)rr,rr)rJrrL)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 r2r0r1r)rLdomainr'rrrtest_unicodeDomains z$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. rThost1.example.orghost2.example.orgrandint r)r)rrdrE)rrfN) r rFr)orderedServersserversr/randomr9r4 pickServerrLr3rrecord1record2rrrtest_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). rdrTrerfrgrhrrerTrgrTN r rFr)rmrnr/ror9r4rLrprqrrrtest_pickServerSamePriorities  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). rdrrTrerfrgrhrurvNrwrqrrrtest_srvDifferentPrioritiesryz-SRVConnectorTests.test_srvDifferentPrioritiesN)r r!r"r#r5r9r:rOrQrRrVrYr[r_rarcrtrxrzrrrrr+@s"      ! r+)r#rozope.interface.verifyrtwisted.internetrrtwisted.internet.errorrrtwisted.internet.interfacesr twisted.namesrr r twisted.names.commonr twisted.names.errorr twisted.test.proto_helpersr twisted.trialrr ClientFactoryr$TestCaser+rrrrs