o i$@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)selfrrrhC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-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)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|_dSrreasonrrrrr9 zDummyFactory.__init__cCs ||_dSrr")r connectorr#rrrclientConnectionFailed<r%z#DummyFactory.clientConnectionFailedN)rrrr 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) patchr rrreactorr!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} rN)r0rr1pop)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+rtyperr>payloadN)r9r8)r Record_SRVRRHeaderSRVINr r)rr&connect assertIsNoner.r# assertEqualr- tcpClientsr3rrArrrtest_SRVPresenths  z!SRVConnectorTests.test_SRVPresentcCsDtdtj_|j||jj| |j j ddddS)zS Test connectTCP gets called with fallback parameters on NXDOMAIN. s example.orgNrBr+r*) r r r)rr&rGrHr.r#rIr-rJr3r$rrrtest_SRVNotPresentws  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-rJr3r$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 r/r-r.r&rr r)rrGrHr#rIrJr3r$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 r/r-r.r&rr r)rrG 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:Nr?rBrM)r rDZCNAMErFr r)rr&rGrHr.r#rIr-rJr3r$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_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&rGrrI)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&rIdomainr$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. rQhost1.example.orghost2.example.orgrandint r)r)rr]rB)rr_N) r rCr&orderedServersserversr,randomr6r1 pickServerrIr0rZrecord1Zrecord2rrrtest_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]rQr^r_r`rarr^rQr`rQN r rCr&rfrgr,rhr6r1rIrirjrrrtest_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). r]rrQr^r_r`rarlrmNrnrjrrrtest_srvDifferentPrioritiesrpz-SRVConnectorTests.test_srvDifferentPrioritiesN)rrrr r2r6r7rLrNrOrRrUrVrXrZr\rkrorqrrrrr(@s"      ! r()r rhZzope.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