U  W['@sdZddlmZmZddlZddlmZddlmZddl m Z m Z ddl m Z mZddlmZdd lmZmZmZdd lmZdd lmZdd lmZdd lmZGdddeZGddde jZGdddej Z!dS)z- Test cases for L{twisted.names.srvconnect}. )absolute_importdivisionN) verifyObject)unittest)deferprotocol)DNSLookupErrorServiceNameUnknownError) IConnector)clientdns srvconnect) ResolverBase) DNSNameError) nativeString) MemoryReactorc@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__'szFakeResolver.__init__cCs@|j||||f|jdk r0t|jggfSt|jSdS)z9 Return the result or failure on lookup. N)rappendrrZsucceedZfailr)rnameclsZqtypeZtimeoutrrr_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|_dSrreasonrrrrr<szDummyFactory.__init__cCs ||_dSrr$)r connectorr%rrrclientConnectionFailed?sz#DummyFactory.clientConnectionFailedN)rr r!r"rr(rrrrr#8sr#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&rrrsetUpIs 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)r1rr2pop)rminmaxrrr_randintSszSRVConnectorTests._randintcCstt|jdS)zF L{srvconnect.SRVConnector} implements L{IConnector}. N)rr r'r&rrrtest_interfacegsz 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<Zporttargetttlr,rtyperr>payloadN)r:r9)r Record_SRVRRHeaderSRVINr r*rr'connect assertIsNoner/r% assertEqualr. tcpClientsr4rrArrrtest_SRVPresentns  z!SRVConnectorTests.test_SRVPresentcCsDtdtj_|j||jj| |j j ddddS)zS Test connectTCP gets called with fallback parameters on NXDOMAIN. r,NrBr,r+) rr r*rr'rGrHr/r%rIr.rJr4r&rrrtest_SRVNotPresent~s  z$SRVConnectorTests.test_SRVNotPresentcCs@gtj_|j||jj||j j ddddS)zW Test connectTCP gets called with fallback parameters on empty result. NrBrM) r r*rr'rGrHr/r%rIr.rJr4r&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 defaultPortNrB)r,rQ)r r0r.r/r'r r r*rrGrHr%rIrJr4r&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. rPr,N) r r0r.r/r'r r r*rrGZ assertTruer%Zcheckr&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;Nr?rBrM)r rDZCNAMErFr r*rr'rGrHr/r%rIr.rJr4r&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,r?N)r rCrDrErFr r*rr'rGZassertIsNotNoner/r%ZtraprrIr.rJrKrrrtest_SRVNoServices  z#SRVConnectorTests.test_SRVNoServicecCs6gtj_|jtjjdd}|td|dS)zW The lookup name is a native string from service, protocol and domain. rz_xmpp-server._tcp.example.orgN)r r*rr'rGrrIr)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.orgzxn--chec-9oa.example.orgN)r r0r.r/r'rIZdomainr&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. rQhost1.example.orghost2.example.orgrandint r)r)rrZrB)rr\N) r rCr'orderedServersserversr-randomr7r2 pickServerrIr1rZrecord1Zrecord2rrrtest_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). rZrQr[r\r]r^rr[rQr]rQN r rCr'rcrdr-rer7r2rIrfrgrrrtest_pickServerSamePriorities s  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). rZrrQr[r\r]r^rirjNrkrgrrrtest_srvDifferentPrioritiess  z-SRVConnectorTests.test_srvDifferentPrioritiesN)rr r!r"r3r7r8rLrNrOrRrSrTrVrXrYrhrlrmrrrrr)Ds      "r))"r"Z __future__rrreZzope.interface.verifyrZ twisted.trialrZtwisted.internetrrZtwisted.internet.errorrr Ztwisted.internet.interfacesr Z twisted.namesr r r Ztwisted.names.commonrZtwisted.names.errorrZtwisted.python.compatrZtwisted.test.proto_helpersrrZ ClientFactoryr#ZTestCaser)rrrrs