U Acj\@sdZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZddlmZddlmZmZmZdd lmZmZmZmZmZdd lmZdd lmZmZm Z m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddZ/Gdddej0Z1ej"dddddddddZ2ej"dddd d!d"d#d$dZ3ej"d%d&d'd(dd)dd*Z4e1de2fe2e!d+e5d,e6d-e6d.d/e7d0d1ej8d2d3d4e9d5e:d6e;d7ee=d?d@dAedCej?dDdEdFej@ddGdHejAdIdJdKeBdLejCdMeDdd0dNdOdPeEdQgeFdRdSdTdUge!dVe!dWge!dXe!dYge9dge!dZgd[d\ZGe1de3fd]eHd^e3gid\ZIe1d_e4fd_e4ej!dZd`daej5dbdcdaej5dddedaejFdRdSdTdUdfdagid\ZJGdgdhdhe.jKZLGdidjdje.jKZMGdkdldle.jKZNGdmdndne.jKZOGdodpdpe.jKZPGdqdrdre.jKZQGdsdtdte.jKZRGdudvdve.jKZSGdwdxdxe.jKZTdyZUGdzd{d{e.jKZVd|ZWGd}d~d~e.jKZXdS)z Test cases for twisted.names. N)partialreduce)BytesIO)pack)defererrorreactor)succeed)MemoryReactorClockStringTransportwaitUntilAllDisconnected) authorityclientcommondnsserver)Resolver)SOAMessageQueryRecord_A Record_SOARRHeader) DomainError)SecondaryAuthoritySecondaryAuthorityService) nativeString)FilePath)unittestcCsdd|dDS)NcSsg|] }|jqSpayload.0rrrA/tmp/pip-unpacked-wheel-nrvc_22y/twisted/names/test/test_names.py "szjustPayload..rrresultsrrr% justPayload!sr)c@seZdZddZdS)NoFileAuthoritycCstj||||_|_dSN)rZ ResolverBase__init__soarecords)selfr-r.rrr%r,&s zNoFileAuthority.__init__N)__name__ __module__ __qualname__r,rrrr%r*%sr*test-domain.comzroot.test-domain.comdiH?&mnamernameserialrefreshminimumexpireretryttls93.84.28.in-addr.arpaxi1i~Ϊ my-domain.comspostmaster.test-domain.comi90i?B)r:r;r<r=r>r?r@s 127.0.0.1s 39.28.189.39#v=spf1 mx/30 mx:example.org/30 -allv=spf1 +mx a:colo.example.com/28 -all not valid host.test-domain.comLinuxA Fast One, Dontcha know)oscpuscanonical.name.comsmailbox.test-domain.comsmail.group.someplaceA First piece of Texta SecoNd pieces ABCD::4321 s 0:0069::0ssome.network.tlds$0:5634:1294:AFCB:56AC:48EF:34C3:01FFs tra.la.la.net+Some more text, haha! Yes. Still here?smail.redirect.or.whatevers r mail boxs e mail box)rmailbxemailbxsafsdb.test-domain.com)subtypehostnameswhatever.i.dunnossome.more.text)mboxtxts 12.54.78.12usip+E2U!!^.*$!sip:information@domain.tld!s'AF43:5634:1294:AFCB:56AC:48EF:34C3:01FF?骪ssome.other.place.fool 123.242.1.5 0.255.0.255s255.255.255.254s0.0.0.0s1.2.3.4)r3shttp.tcp.test-domain.comhost.test-domain.comshost-two.test-domain.comscname.test-domain.comsanothertest-domain.comr-r.s123.93.84.28.in-addr.arpastest.host-reverse.lookup.coms my-domain.com1SrAs ns1.domains2Ms ns2.domain3H4Dc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9Zd:d;Z d.checkResults) addCallback)r/Zqueryingrrrrr% namesTests zServerDNSTests.namesTestcCs ||jdtjdddgS)z"Test simple DNS 'A' record queriestest-domain.comror6rirrz lookupAddressrrr/rrr%test_addressRecord1s z"ServerDNSTests.test_addressRecord1cCs,||jdtjdddtjdddgS)z1Test DNS 'A' record queries with multiple answersrLz 123.242.1.5r6riz 0.255.0.255rrrrr%test_addressRecord2s    z"ServerDNSTests.test_addressRecord2cCs,||jdtjdddtjdddgS)z+Test DNS 'A' record queries with edge caseszhost-two.test-domain.comz255.255.255.254r6ri0.0.0.0rrrrr%test_addressRecord3s    z"ServerDNSTests.test_addressRecord3cCs||jdtgS)zTest DNS 'SOA' record queriesr)rrzZlookupAuthority soa_recordrrrr%test_authoritys zServerDNSTests.test_authorityc Cs:||jdtjddddtjdddtjdddgS)z The DNS client can issue an MX query and receive a response including an MX record as well as any A record hints. r3rKrfr6rirdre)rrzlookupMailExchanger Record_MXrrrrr%test_mailExchangeRecords   z&ServerDNSTests.test_mailExchangeRecordcCs ||jdtjdddgS)zTest DNS 'NS' record queriesrz 39.28.189.39r6ri)rrzlookupNameserversr Record_NSrrrr%test_nameservers zServerDNSTests.test_nameservercCs"||jdtjddddgS)zTest DNS 'HINFO' record queriesrrMrNr6)rOrPrA)rrzZlookupHostInfor Record_HINFOrrrr% test_HINFOs zServerDNSTests.test_HINFOcCs ||jdtjdddgS)zTest DNS 'PTR' record queriesz123.93.84.28.in-addr.arpaztest.host-reverse.lookup.comrCri)rrz lookupPointerr Record_PTRrrrr%test_PTR s zServerDNSTests.test_PTRcCs ||jdtjdddgS)zTest DNS 'CNAME' record queriesrzcanonical.name.comr6ri)rrzlookupCanonicalNamer Record_CNAMErrrr% test_CNAME's zServerDNSTests.test_CNAMEcCs ||jdtjdddgS)zTest DNS 'MB' record queriesrzmailbox.test-domain.comr6ri)rrzZ lookupMailBoxr Record_MBrrrr%test_MB.s zServerDNSTests.test_MBcCs ||jdtjdddgS)zTest DNS 'MG' record queriesrzmail.group.someplacer6ri)rrzZlookupMailGroupr Record_MGrrrr%test_MG5s zServerDNSTests.test_MGcCs ||jdtjdddgS)zTest DNS 'MR' record queriesrzmail.redirect.or.whateverr6ri)rrzZlookupMailRenamer Record_MRrrrr%test_MR<s zServerDNSTests.test_MRcCs"||jdtjddddgS)zTest DNS 'MINFO' record queriesrz r mail boxz e mail boxr6)rWrXrA)rrzZlookupMailboxInfor Record_MINFOrrrr% test_MINFOCs zServerDNSTests.test_MINFOc Cs&||jdtjddddddgS)zTest DNS 'SRV' record querieszhttp.tcp.test-domain.comrarbrcsome.other.place.foolr6ri)rrzZ lookupServicer Record_SRVrrrr%test_SRVNs zServerDNSTests.test_SRVcCs"||jdtjddddgS)zTest DNS 'AFSDB' record queriesrr8zafsdb.test-domain.comr6)rYrZrA)rrzZlookupAFSDatabaser Record_AFSDBrrrr% test_AFSDBUs zServerDNSTests.test_AFSDBcCs"||jdtjddddgS)zTest DNS 'RP' record queriesrzwhatever.i.dunnozsome.more.textr6)r[r\rA)rrzZlookupResponsibilityr Record_RPrrrr%test_RP`s zServerDNSTests.test_RPcCs.||jdtjddddtjdddgS)zTest DNS 'TXT' record queriesrrQrRr6rirV)rrzZ lookupTextr Record_TXTrrrr%test_TXTks zServerDNSTests.test_TXTc Cs.||jdtjdddtjddddgS)zH L{DNSServerFactory} can serve I{SPF} resource records. rrHr6rirIrJ)rrzZlookupSenderPolicyr Record_SPFrrrr%test_spfys  zServerDNSTests.test_spfc Cs&||jdtjdtjdddgS)zTest DNS 'WKS' record queriesrz 12.54.78.12r]r6ri)rrzZlookupWellKnownServicesr Record_WKSsocket IPPROTO_TCPrrrr%test_WKSs zServerDNSTests.test_WKScCs^tt}tj|_||jd|tjdddtj dddtj dddtj d d d d d dgS)NrF1.2.3.4rhriz ns1.domainZ2Mz ns2.domainrjrarbrcrrk) copymy_soar?rArrzlookupAllRecordsrrrr)r/Z result_soarrr%test_someRecordsWithTTLss     z'ServerDNSTests.test_someRecordsWithTTLscCs ||jdtjdddgS)z%Test DNS 'AAAA' record queries (IPv6)rz'AF43:5634:1294:AFCB:56AC:48EF:34C3:01FFr6ri)rrzlookupIPV6Addressr Record_AAAArrrr% test_AAAAs zServerDNSTests.test_AAAAc CsD||jdtjdddddtjddd ddtjd d d ddgS) z#Test DNS 'A6' record queries (IPv6)rrz ABCD::4321r6rirTz 0:0069::0zsome.network.tldrUz$0:5634:1294:AFCB:56AC:48EF:34C3:01FFz tra.la.la.net)rrzZlookupAddress6r Record_A6rrrr%test_A6s zServerDNSTests.test_A6cCsZtj}ddttjtjD}|D]}|jdkr&||_q&| |j d dd|S)z9 Test DNS 'AXFR' queries (Zone transfer) cSsg|]}t|qSr)rr"rrr%r&sz4ServerDNSTests.test_zoneTransfer..NrcSs|dddfS)Nrr)r$rrr%rSz2ServerDNSTests.test_zoneTransfer..) rr?roperatoraddrqr.valuesrArrz lookupZoner)r/Z default_ttlr(r$rrr%test_zoneTransfers  z ServerDNSTests.test_zoneTransfercCs tdgd}||dtjS)zu A failed AXFR TCP connection errbacks the L{Deferred} returned from L{Resolver.lookupZone}. )znameserver.invalid5rpzimpossible.invalid)r assertFailurerrZDNSLookupError)r/rzrrr% test_zoneTransferConnectionFailss  z/ServerDNSTests.test_zoneTransferConnectionFailscCs ||jdtjdddgS)z6Tests that unrelated zones don't mess with each other.zanothertest-domain.comrr6rirrrrr%test_similarZonesDontInterferes z-ServerDNSTests.test_similarZonesDontInterferec Cs(||jdtjdddddddgS) z2 Test DNS 'NAPTR' record queries. rr4rKr^r_r`r6ri)rrzZlookupNamingAuthorityPointerr Record_NAPTRrrrr% test_NAPTRs zServerDNSTests.test_NAPTRN)!r0r1r2__doc__r|rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr%rls<        rlc@seZdZddZdS) HelperTestscCs@|}t|}tdD] }t|}|||k|}qdS)N)mktempr Z getSerialrange assertTrue)r/faibrrr%test_serialGenerators    z HelperTests.test_serialGeneratorN)r0r1r2rrrrr%rsrc@s4eZdZddZddZddZddZd d Zd S) AXFRTestscCsd|_t|_|j|jtd|j|_t j dt j t j ddt j ddddddd d d d |_|jt j dt jt j d dt jdd dd t j dt jt j d dt jddd dd t j dt jt j d dt jdd dd |jg|_dS)N fooby.comiQFzhooj.fooby.comr4i,iiiX)r:r;r<r=r@r?r>rAnametypeclsrAauthr!ns.twistedmatrix.comrrArKz mail.mv3d.com) preferenceZexchangerAz 64.123.27.105)addressrA)r(rZDeferreddr _gotResultsrZAXFRController controllerrrrINrr-NSrZMXrArr.rrrr%r|s`   zAXFRTests.setUpc Cstjdddddddddd S)Nr8r idanswerZopCodeZrecDesZrecAvrZrCodetruncZmaxSize)rrrrrr% _makeMessage2szAXFRTests._makeMessagecCsH|}tdtjtjg|_|j|_|j |d| |j |jdS)Nr) rrrAXFRrqueriesr.answersrmessageReceivedrr()r/mrrr%test_bindAndTNamesStyle@s z!AXFRTests.test_bindAndTNamesStylecCs ||_dSr+r')r/resultrrr%rHszAXFRTests._gotResultscCsR|jdd}|r>|}g|_|dg|_|j|dq||j|jdSNr) r.rrpoprrrrr()r/r.rrrr% test_DJBStyleKszAXFRTests.test_DJBStyleN)r0r1r2r|rrrrrrrr%rs 5rc@seZdZddZddZdS)ResolvConfHandlingTestscCs2|}tj|d}||jdg|jdS)NZresolvror)rrrr dynServers _parseCallcancelr/Z resolvConfr$rrr% test_missingWs z$ResolvConfHandlingTests.test_missingcCs@|}t|dtj|d}||jdg|jdS)Nwrr) ropencloserrrrrrrrrr% test_empty]s  z"ResolvConfHandlingTests.test_emptyN)r0r1r2rr rrrr%rVsrc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)AuthorityTestszs Tests for the basic response record selection code in L{FileAuthority} (independent of its fileness). cCs4t}d|jd}|||}||jtdS)aJ L{FileAuthority} lookup methods errback with L{DomainError} if the requested C{name} shares a common suffix with its zone but is not actually a descendant of its zone, in terms of its sequence of DNS name labels. eg www.the-example.com has nothing to do with the zone example.com. nonexistent.prefix-rN)rqr-failureResultOfrassertIsInstancevaluer)r/ testDomaintestDomainNamerrrr%'test_domainErrorForNameWithCommonSuffixksz6AuthorityTests.test_domainErrorForNameWithCommonSuffixc Csttjjtftjjttdgid}||tjj\}}}||g||tj tjjtj tj tddg||gdS)a! If a L{FileAuthority} has a zone which includes an I{NS} record for a particular name and that authority is asked for another record for the same name which does not exist, the I{NS} record is not included in the authority section of the response. rrgTrAr!rN) r*rr:rrrsuccessResultOfrrrTYPEr?)r/r r additionalrrr%test_recordMissingxs0     z!AuthorityTests.test_recordMissingcCsJt}d|jd}ttjd}||t||d}||j t dS)z Requesting a record of unknown type where no records exist for the name in question results in L{DomainError}. r rr8rrN) rqr-maxr typeToMethodr queryrr rr)r/rr unknownTyperrrr%test_unknownTypeNXDOMAINsz'AuthorityTests.test_unknownTypeNXDOMAINcCs<ttjd}|ttd|d\}}}||gdS)z Requesting a record of unknown type where other records exist for the name in question results in an empty answer set. r8rFrN)rrrrrsrrr)r/rrr rrrr%test_unknownTypeMissings  z&AuthorityTests.test_unknownTypeMissingc Csdtjj}td}ttjjtf||gid}t|||}||\}}}||g||tj |tj tj |ddg||gdS)z Create an authority and make a request against it. Then verify that the result is a referral, including no records in the answers or additional sections, but with an I{NS} record in the authority section. sexample.rrgFrN) rr:rrrr*getattrrrrrr?)r/methodZ subdomainZ nameserverr rrrrrr% _referralTests0     zAuthorityTests._referralTestcCs|ddS)a  When an I{NS} record is found for a child zone, it is included in the authority section of the response. It is marked as non-authoritative if the authority is not also authoritative for the child zone (RFC 2181, section 6.1). rNr rrrr% test_referralszAuthorityTests.test_referralcCs|ddS)zT A referral is also generated for a request of type C{ALL_RECORDS}. rNr!rrrr%test_allRecordsReferralsz&AuthorityTests.test_allRecordsReferralN) r0r1r2rrrrrr r"r#rrrr%r es "  ! r c@seZdZdZedZedZddZ ddZ dd Z d d Z d d Z ddZddZddZddZddZddZddZddZddZd d!Zd"S)#AdditionalProcessingTestszz Tests for L{FileAuthority}'s additional processing for those record types which require it (MX, CNAME, etc). s10.0.0.1sf080::1cCs8t|jj|f|jj||g||id}t||tjjS)a Perform a DNS lookup against a L{FileAuthority} configured with records as defined by C{makeRecord} and C{addresses}. @param method: The name of the lookup method to use; for example, C{"lookupNameservers"}. @type method: L{str} @param soa: A L{Record_SOA} for the zone for which the L{FileAuthority} is authoritative. @param makeRecord: A one-argument callable which accepts a name and returns an L{IRecord} provider. L{FileAuthority} is constructed with this record. The L{FileAuthority} is queried for a record of the resulting type with the given name. @param target: The extra name which the record returned by C{makeRecord} will be pointed at; this is the name which might require extra processing by the server so that all the available, useful information is returned. For example, this is the target of a CNAME record or the mail exchange host pointed to by an MX record. @type target: L{bytes} @param addresses: A L{list} of records giving addresses of C{target}. @return: A L{Deferred} that fires with the result of the resolver method give by C{method}. rg)r*r:rrr)r/rr- makeRecordtarget addressesr rrr%_lookupSomeRecordss z,AdditionalProcessingTests._lookupSomeRecordscCs(dd}|t||dt||ddS)a  Assert that the L{RRHeader} instances given by C{expected} and C{computed} carry all the same information but without requiring the records appear in the same order. @param expected: A L{list} of L{RRHeader} instances giving the expected records. @param computed: A L{list} of L{RRHeader} instances giving the records computed by the scenario under test. @raise self.failureException: If the two collections of records disagree. cSs|jSr+)r)rrrrr%r$rSz>AdditionalProcessingTests.assertRecordsMatch..)keyN)rsorted)r/expectedZcomputedr*rrr%assertRecordsMatchsz,AdditionalProcessingTests.assertRecordsMatchcsLdtjj||t||}||\}}}|fdd|D|dS)aT Verify that certain address records are included in the I{additional} section of a response generated by L{FileAuthority}. @param method: See L{_lookupSomeRecords} @param makeRecord: See L{_lookupSomeRecords} @param addresses: A L{list} of L{IRecord} providers which the I{additional} section of the response is required to match (ignoring order). @raise self.failureException: If the I{additional} section of the response consists of different records than those given by C{addresses}. smail.c s$g|]}tj|jtj|ddqSTrrrrrr?r#rr&rr%r&=sz=AdditionalProcessingTests._additionalTest..N)rr:rr(rr-)r/rr%r'rrr rrr1r%_additionalTest's   z)AdditionalProcessingTests._additionalTestcCs|dttjd|dS)z Verify that a response to an MX query has certain records in the I{additional} section. @param addresses: See C{_additionalTest} rrKN)r2rrrr/r'rrr%_additionalMXTestJs  z+AdditionalProcessingTests._additionalMXTestcCs||jgdS)z If the name of the MX response has A records, they are included in the additional section of the response. N)r4_Arrrr%test_mailExchangeAdditionalAUsz6AdditionalProcessingTests.test_mailExchangeAdditionalAcCs||jgdS)z If the name of the MX response has AAAA records, they are included in the additional section of the response. N)r4_AAAArrrr%test_mailExchangeAdditionalAAAA\sz9AdditionalProcessingTests.test_mailExchangeAdditionalAAAAcCs||j|jgdS)z If the name of the MX response has both A and AAAA records, they are all included in the additional section of the response. N)r4r5r7rrrr%test_mailExchangeAdditionalBothcsz9AdditionalProcessingTests.test_mailExchangeAdditionalBothcCs|dtj|dS)z Verify that a response to an NS query has certain records in the I{additional} section. @param addresses: See C{_additionalTest} rN)r2rrr3rrr%_additionalNSTestjsz+AdditionalProcessingTests._additionalNSTestcCs||jgdS)z If the name of the NS response has A records, they are included in the additional section of the response. N)r:r5rrrr%test_nameserverAdditionalAssz4AdditionalProcessingTests.test_nameserverAdditionalAcCs||jgdS)z If the name of the NS response has AAAA records, they are included in the additional section of the response. N)r:r7rrrr%test_nameserverAdditionalAAAAzsz7AdditionalProcessingTests.test_nameserverAdditionalAAAAcCs||j|jgdS)z If the name of the NS response has both A and AAAA records, they are all included in the additional section of the response. N)r:r5r7rrrr%test_nameserverAdditionalBothsz7AdditionalProcessingTests.test_nameserverAdditionalBothcsvdtjj|dttj|}||\}}}tjtjjtjtj tdd}| fdd|D|g|dS)z Verify that a response to a CNAME query has certain records in the I{answer} section. @param addresses: See C{_additionalTest} swww.rTrc s$g|]}tj|jtj|ddqSr.r/r0r1rr%r&sz>AdditionalProcessingTests._answerCNAMETest..N) rr:rr(rrrrZCNAMEr?r-)r/r'rrr raliasrr1r%_answerCNAMETests0    z*AdditionalProcessingTests._answerCNAMETestcCs||jgdS)z If the name of the CNAME response has A records, they are included in the answer section of the response. N)r?r5rrrr%test_canonicalNameAnswerAsz3AdditionalProcessingTests.test_canonicalNameAnswerAcCs||jgdS)z If the name of the CNAME response has AAAA records, they are included in the answer section of the response. N)r?r7rrrr%test_canonicalNameAnswerAAAAsz6AdditionalProcessingTests.test_canonicalNameAnswerAAAAcCs||j|jgdS)z If the name of the CNAME response has both A and AAAA records, they are all included in the answer section of the response. N)r?r5r7rrrr%test_canonicalNameAnswerBothsz6AdditionalProcessingTests.test_canonicalNameAnswerBothN)r0r1r2rrrr5rr7r(r-r2r4r6r8r9r:r;r<r=r?r@rArBrrrr%r$s$  &#  #r$c@seZdZddZdS)NoInitialResponseTestsc sfdd}dd}tdgd}||_g|tjd|tjdtjtjd d tjd d d d g}tj dddddddddd }||_ || | dtjS)z If a request returns a L{dns.NS} response, but we can't connect to the given server, the request fails with the error returned at connection. cstdSr)r r)r/argsmessagesrr%rsz3NoInitialResponseTests.test_noAnswer..queryc_sttdS)NzCouldn't connect)rZfailrgaierror)r/rDkwargsrrr% queryProtocolsz;NoInitialResponseTests.test_noAnswer..queryProtocol)rrrprz fooba.comrFrrrrr8rrr)rZ_querypatchrrurrrrrrappendrZ getHostByNamerrG)r/rrIrzr.rrrEr% test_noAnswers:      z$NoInitialResponseTests.test_noAnswerN)r0r1r2rLrrrr%rCsrCc@s(eZdZdZddZddZddZdS) SecondaryAuthorityServiceTestsz Tests for L{SecondaryAuthorityService}, a service which keeps one or more authorities up to date by doing zone transfers from a master. cCsd}t|ddg}||j|||jd||jdj|||jdjd||jdjd||jdj|||jdjd||jdjddS) a L{SecondaryAuthorityService} can be constructed with a C{str} giving a master server address and several domains, causing the creation of a secondary authority for each domain and that master server address and the default DNS port. 192.168.1.2 example.comz example.orgrrr8s example.orgN)rrprimary_portdomainsdomainr/rPservicerrr%test_constructAuthorityFromHostsz>SecondaryAuthorityServiceTests.test_constructAuthorityFromHostcCsd}d}t||fddg}||j|||jd||jdj|||jdj|||jdjd||jdj|||jdj|||jdjddS) a\ L{SecondaryAuthorityService.fromServerAddressAndDomains} constructs a new L{SecondaryAuthorityService} from a C{str} giving a master server address and DNS port and several domains, causing the creation of a secondary authority for each domain and that master server address and the given DNS port. 192.168.1.3iz example.net example.edur example.netr8N)rZfromServerAddressAndDomainsrrPrQrRrS)r/rPrwrUrrr%&test_constructAuthorityFromHostAndPort szESecondaryAuthorityServiceTests.test_constructAuthorityFromHostAndPortcCsxd}t|ddg}||j|||jdj|||jdjd||jdj|||jdjddS)aQ L{SecondaryAuthorityService.fromServerAddressAndDomains} constructs a new L{SecondaryAuthorityService} from a C{bytes} giving a master server address and several domains, causing the creation of a secondary authority for each domain and that master server address and the given DNS port. rWrYz example.edurr8rXN)rencoderrPrRrSrTrrr% test_constructAuthorityFromBytes#sz?SecondaryAuthorityServiceTests.test_constructAuthorityFromBytesN)r0r1r2rrVrZr\rrrr%rMsrMc@s0eZdZdZddZddZddZdd Zd S) SecondaryAuthorityTestsz L{twisted.names.secondary.SecondaryAuthority} correctly constructs objects with a specified IP address and optionally specified DNS port. cCs8tdd}||jd||jd||jddS)zp When constructed using L{SecondaryAuthority.__init__}, the default port of 53 is used. 192.168.1.1 inside.comr inside.comN)rrrPrQrSr/ secondaryrrr%test_defaultPort?s z(SecondaryAuthorityTests.test_defaultPortcCs:tdd}||jd||jd||jddS)z~ When constructed using L{SecondaryAuthority.fromServerAddressAndDomain}, the specified port is used. )r^r_r^rdr`N)rfromServerAddressAndDomainrrPrQrSrarrr%test_explicitPortIsz)SecondaryAuthorityTests.test_explicitPortc Cstdd}t|_}||jd\}}}}}||d||d|||f}t } | | t } | t | dd|tdtjtjg| jdS)z An attempt is made to transfer the zone for the domain the L{SecondaryAuthority} was constructed with from the server address it was constructed with when L{SecondaryAuthority.transfer} is called. rNr5 example.comrrNr5rmN)rrer _reactortransfer tcpClientsrr buildProtocolr makeConnectionrdecoderrrrrrr) r/rbrhostrwrttimeout bindAddressprotor~msgrrr% test_transferUs    z%SecondaryAuthorityTests.test_transferc Cs.tdd}t|_}||jd\}}}}}|||f}t} | | t ddd} | t | ddtdd d d d d ddd} tddd} t | jddd} | jtdt| dtd| dtdt| dg| }|tdt||||d}|td| ddgggf|dS)z L{SecondaryAuthority.lookupAddress} returns a L{Deferred} that fires with the I{A} records the authority has cached from the primary. rgrOrr8)rrrmNsns1.example.comzadmin.example.comi@iii i%i.r9s 192.168.1.2ri)rrr)rr!r z!HrhT)r!r)rrer rirjrkrrlr rmrrnrrrrrrextendrrZtoStrZ dataReceivedrlenrrr)r/rbrrorwrtrprqrrr~rr-rrdatarrrr%test_lookupAddressssJ       z*SecondaryAuthorityTests.test_lookupAddressN)r0r1r2rrcrfrtrxrrrr%r]9s   r]sY$ORIGIN example.com. $TTL 1w example.com. IN SOA dns.example.com ( 2013120201 ; serial number of this zone file 1d ; slave refresh 2h ; slave retry time in case of a problem 4w ; slave expiration time 1h ; maximum caching time in case of failed lookups ) ; A comment. @ IN AAAA 2001:db8:10::1 example.com. IN A 10.0.0.1 no-in.example.com. A 10.0.0.2 ; technically wrong but used to work not-fqdn IN MX 10 mx.example.com www IN CNAME example.comc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)BindAuthorityTestsz= Tests for L{twisted.names.authority.BindAuthority}. cCs(t|d}||t|jS)z Create a new L{twisted.names.authority.BindAuthority} from C{s}. @param s: A string with BIND zone data. @type s: bytes @return: a new bind authority @rtype: L{twisted.names.authority.BindAuthority} ascii)rrr[Z setContentr Z BindAuthoritypath)r/sfprrr%loadBindStrings  z!BindAuthorityTests.loadBindStringcCs|t|_dSr+)r~sampleBindZonerrrrr%r|szBindAuthorityTests.setUpcCs8|jjD]&}|jj|D]}|d|jkqq dS)zG Loads the default $TTL and applies it to all records. : N)rr.keysrrA)r/domZrecrrr%test_ttlszBindAuthorityTests.test_ttlcCs$|d|jj|d|jjdS)z, Loads the default $ORIGIN. s example.com.not-fqdn.example.comN)rroriginZassertInr.rrrr%test_originFromFilesz&BindAuthorityTests.test_originFromFilecCsBdD]8\}}||j|\\}\\|t|d|jqdS)' A records are loaded. )rOz10.0.0.1)no-in.example.comz10.0.0.2rNrrrrrrr!r/ripr)rrr% test_aRecordss z BindAuthorityTests.test_aRecordscCs4||jd\\}\\|tdd|jdS)* AAAA records are loaded. rO2001:db8:10::1rNrrrrrrr!r/r)rrr%test_aaaaRecordss z#BindAuthorityTests.test_aaaaRecordscCs8||jd\\}\\|tjdddd|jdS)( MX records are loaded. rrKzmx.example.comr)rrrANrrrrrrr!rrrr%test_mxRecordss z!BindAuthorityTests.test_mxRecordscCs<||jd\}\\|d}|tjddd|jdS)+ CNAME records are loaded. www.example.comrrhrrNrrrrrrr!r/rr)rrr%test_cnameRecordss z$BindAuthorityTests.test_cnameRecordsc Cs8|t}|dW5QRX|d|jjddS)zO loadBindString raises NotImplementedError on invalid records. sexample.com. IN LOL 192.168.0.1zRecord type 'LOL' not supportedrN) assertRaisesNotImplementedErrorr~r exceptionrD)r/errr%test_invalidRecordClass%s z*BindAuthorityTests.test_invalidRecordClassc CsNdD]D}|t}||dW5QRX|t|d|jjdqdS)zC $INCLUDE and $GENERATE raise NotImplementedError. )s$INCLUDEs $GENERATEs doesNotMatters directive not implementedrN)rrr~rrrrD)r/ directiverrrr%test_invalidDirectives-s   z)BindAuthorityTests.test_invalidDirectivesN)r0r1r2rr~r|rrrrrrrrrrrr%rys ryazone = [ SOA( # For whom we are the authority 'example.com', # This nameserver's name mname = "dns.example.com", # Mailbox of individual who handles this rname = "root.example.com", # Unique serial identifying this SOA data serial = 86400, # Time interval before zone should be refreshed refresh = "2H", # Interval before failed refresh should be retried retry = "1H", # Upper limit on time interval before expiry expire = "1H", # Minimum TTL minimum = "3H" ), AAAA('example.com', '2001:db8:10::1'), A('example.com', '10.0.0.1'), NS('example.com', 'dns.example.com'), A('no-in.example.com', '10.0.0.2'), PTR('2.0.0.10.in-addr.arpa', 'no-in.example.com'), CNAME('www.example.com', 'example.com'), CNAME('ftp.example.com', 'example.com'), MX('not-fqdn.example.com', 10, 'mail.example.com'), ] c@sPeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ dS)PySourceAuthorityTestszA Tests for L{twisted.names.authority.PySourceAuthority}. c Cs:t|}t|jd}||W5QRXt|jS)a Create a new L{twisted.names.authority.PySourceAuthority} from C{s}. @param s: A string with BIND zone data in a Python source file. @type s: L{str} @return: a new bind authority @rtype: L{twisted.names.authority.PySourceAuthority} r)rrrr{writer ZPySourceAuthority)r/r|r}rrrr%loadPySourceStringjs z)PySourceAuthorityTests.loadPySourceStringcCs|t|_dSr+)rsamplePySourcerrrrr%r|zszPySourceAuthorityTests.setUpcCs@dD]6\}}||j|\\}\\|t||jqdS)rrNrrrrr%r}s  z$PySourceAuthorityTests.test_aRecordscCs2||jd\\}\\|td|jdS)rrOrNrrrrr%rs z'PySourceAuthorityTests.test_aaaaRecordscCs6||jd\\}\\|tjddd|jdS)rrrKzmail.example.com)rrNrrrrr%rs z%PySourceAuthorityTests.test_mxRecordscCs:||jd\}\\|d}|tjdd|jdS)rrrrhrNrrrrr%rs z(PySourceAuthorityTests.test_cnameRecordscCs:||jd\}\\|d}|tjdd|jdS)z) PTR records are loaded. s2.0.0.10.in-addr.arparrrN)rrrrrrr!rrrr%rs zPySourceAuthorityTests.test_PTRcCsd}|t|j|dS)z1 Input file has no zone variable z nothing = []N)r ValueErrorr)r/Z badPySourcerrr%test_badInputNoZonesz*PySourceAuthorityTests.test_badInputNoZoneN) r0r1r2rrr|rrrrrrrrrr%res r)Yrrrr functoolsrriorstructrZtwisted.internetrrrZtwisted.internet.deferr Ztwisted.internet.testingr r r Z twisted.namesr rrrrZtwisted.names.clientrZtwisted.names.dnsrrrrrrZtwisted.names.errorrZtwisted.names.secondaryrrZtwisted.python.compatrZtwisted.python.filepathrZ twisted.trialrr)Z FileAuthorityr*rZ reverse_soarrrrrrrrrrrrrrrrrrrrqrrrrsZTestCaserlrrrr r$rCrMr]rryrrrrrr%s                         3    \ Z~_0Hm+