Yf$dZddlZddlmZddlmZmZddlmZm Z ddl m Z ddl m Z ddlmZmZmZdd lmZdd lmZdd lmZGd d eZGddejZGddejZdS)z- Test cases for L{twisted.names.srvconnect}. N) verifyObject)deferprotocol)DNSLookupErrorServiceNameUnknownError) IConnector) MemoryReactor)clientdns srvconnect) ResolverBase) DNSNameError)unittestc eZdZdZddZdZdS) 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} Nc0||_||_g|_dSN)resultsfailurelookups)selfrrs T/var/www/html/env/lib/python3.11/site-packages/twisted/names/test/test_srvconnect.py__init__zFakeResolver.__init__$s   c|j||||f|jtj|jggfStj|jS)z9 Return the result or failure on lookup. )rappendrrsucceedfailr)rnameclsqtypetimeouts r_lookupzFakeResolver._lookup)sV T3w7888 < #=$,B!788 8:dl++ +r)NN)__name__ __module__ __qualname____doc__rr#rrrrsA   ,,,,,rrceZdZdZdZdZdS) DummyFactoryzL Dummy client factory that stores the reason of connection failure. cd|_dSrreasonrs rrzDummyFactory.__init__9s  rc||_dSrr,)r connectorr-s rclientConnectionFailedz#DummyFactory.clientConnectionFailed<s  rN)r$r%r&r'rr1r(rrr*r*4s<rr*cleZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS)SRVConnectorTestsz/ Tests for L{srvconnect.SRVConnector}. c |tdtt|_t |_tj|jdd|j|_ g|_ g|_ dS)N theResolver xmpp-server example.org) patchr rr reactorr*factoryr SRVConnectorr0 randIntArgsrandIntResultsr.s rsetUpzSRVConnectorTests.setUpEsk 6=,..999$ #~~ #0 L-    rcn|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<rr=pop)rminmaxs r_randintzSRVConnectorTests._randintOs6 c +++"&&q)))rc:tt|jdS)zF L{srvconnect.SRVConnector} implements L{IConnector}. N)rrr0r.s rtest_interfacez SRVConnectorTests.test_interfacebs Z00000rctjddd}tjdtjtjd|gt j_|j | |j j | |jjddd dS) zS Test connectTCP gets called with the address from the SRV record. }host.example.org<porttargetttlr7rtyper rMpayloadN)rHrG)r Record_SRVRRHeaderSRVINr r5rr0connect assertIsNoner:r- assertEqualr9 tcpClientsr@rrPs rtest_SRVPresentz!SRVConnectorTests.test_SRVPresenths.d3E2NNN L"cf"g   & "     $,-... 04466rr:%>"     $,-... 04466rr:%>"      +112IJJKKKKKrcdtjdtjtjddgtj_|j| |j j | |j jddddS)zU Test connectTCP gets called with fallback parameters on bad result. r7rINrNrQr])r rSCNAMErUr r5rr0rVrWr:r-rXr9rYr@r.s rtest_SRVBadResultz#SRVConnectorTests.test_SRVBadResults L"BPT   & "     $,-...  L # ' ' ) )"1" -/M     rctjddd}tjdtjtjd|gt j_|j | |j j |j j t||jjgdS)zH Test that connecting fails when no service is present. i.rIrJr7rNN)r rRrSrTrUr r5rr0rVassertIsNotNoner:r-traprrXr9rYrZs rtest_SRVNoServicez#SRVConnectorTests.test_SRVNoServices.d4R@@@ L"cf"g   & "     T\0111   000 0"55555rcgtj_|jtjjdd}|d|dS)zW The lookup name is a native string from service, protocol and domain. rs_xmpp-server._tcp.example.orgN)r r5rr0rVrrX)rrs rtest_SRVLookupNamez$SRVConnectorTests.test_SRVLookupNamesU&("    !)"-a0 94@@@@@rctj|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;r9r:r0rXdomainr.s rtest_unicodeDomainz$SRVConnectorTests.test_unicodeDomainsJ $0 L-)A4<   4dn6KLLLLLrctjdddd}tjdddd}||g|j_g|j_|t d|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. rchost1.example.orghost2.example.orgrandint r)r)rryrQ)rr{N) r rRr0orderedServersserversr8randomrCr= pickServerrXr<rrecord1record2s rtest_pickServerWeightsz(SRVConnectorTests.test_pickServerWeightssK.R/BCC.R/BCC)0'(:%!# 69dm444 "1g !!### )!,g666 !!### )!,g666 "1g !!### )!,g666 !!### )!,g66666rctjdddd}tjdddd}||g|j_g|j_|t d|jddg|_| d|j | d |j d S) zR Two records with equal priorities compare on weight (ascending). ryrcrzr{r|r}rrzrcr|rcN r rRr0rrr8rrCr=rXrrs rtest_pickServerSamePrioritiesz/SRVConnectorTests.test_pickServerSamePrioritiess.R/BCC.R/BCC)0'(:%!# 69dm444 !f 4dn6O6O6Q6QRRR 4dn6O6O6Q6QRRRRRrctjdddd}tjdddd}||g|j_g|j_|t d|jddg|_| d|j | d |j d S) zX Two records with differing priorities compare on priority (ascending). ryrrcrzr{r|r}rrNrrs rtest_srvDifferentPrioritiesz-SRVConnectorTests.test_srvDifferentPrioritiess.Q.ABB.Q.ABB)0'(:%!# 69dm444 !f 4dn6O6O6Q6QRRR 4dn6O6O6Q6QRRRRRrN)r$r%r&r'r>rCrEr[r^r`rerirlrqrtrwrrrr(rrr3r3@s!!!***&111 X X X        SSS$ L L L   666 AAAMMM777BSSS SSSSSrr3)r'rzope.interface.verifyrtwisted.internetrrtwisted.internet.errorrrtwisted.internet.interfacesrtwisted.internet.testingr twisted.namesr r r twisted.names.commonr twisted.names.errorr twisted.trialrr ClientFactoryr*TestCaser3r(rrrsv  ......,,,,,,,,JJJJJJJJ2222222222221111111111------,,,,,,"""""",,,,,<,,,:     8)   aSaSaSaSaS)aSaSaSaSaSr