U  W[X@sdZddlmZmZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZmZdd lmZdd lmZmZmZmZmZdd lmZmZmZm Z m!Z!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/m0Z0m1Z1ddZ2Gdddej3Z4ej!dddddddddZ5ej!ddd d!d"d#d$d%dZ6ej!d&d'd(d)dd*dd+Z7e4de5fe5e d,e8d-e9d.e9d/d0e:d1d2ej;d3d4d5ed8e?d9d:e@dd;d<e@d=d>d?e@d@dAdBe?dCeAdDejBdEdFdGejCddHdIejDdJdKdLeEdMejFdNeGdd1dOdPdQeHdRgeIdSdTdUdVge dWe dXge dYe dZge$szjustPayload..rr!resultsr!r!r' justPayload#sr+c@seZdZddZdS)NoFileAuthoritycCstj||||_|_dSN)rZ ResolverBase__init__soarecords)selfr/r0r!r!r'r.'s zNoFileAuthority.__init__N)__name__ __module__ __qualname__r.r!r!r!r'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@rArBs 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)r5shttp.tcp.test-domain.comhost.test-domain.comshost-two.test-domain.comscname.test-domain.comsanothertest-domain.comr/r0s123.93.84.28.in-addr.arpastest.host-reverse.lookup.coms my-domain.com1SrCs 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)r1Zqueryingrrr!rr' namesTests zServerDNSTests.namesTestcCs ||jdtjdddgS)z"Test simple DNS 'A' record queriestest-domain.comrqr8rkrr{ lookupAddressrrr1r!r!r'test_addressRecord1s z"ServerDNSTests.test_addressRecord1cCs,||jdtjdddtjdddgS)z1Test DNS 'A' record queries with multiple answersrNz 123.242.1.5r8rkz 0.255.0.255rrr!r!r'test_addressRecord2s    z"ServerDNSTests.test_addressRecord2cCs,||jdtjdddtjdddgS)z+Test DNS 'A' record queries with edge caseszhost-two.test-domain.comz255.255.255.254r8rk0.0.0.0rrr!r!r'test_addressRecord3s z"ServerDNSTests.test_addressRecord3cCs||jdtgS)zTest DNS 'SOA' record queriesr)rr{ZlookupAuthority soa_recordrr!r!r'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. r5rMrhr8rkrfrg)rr{lookupMailExchanger Record_MXrrr!r!r'test_mailExchangeRecords   z&ServerDNSTests.test_mailExchangeRecordcCs ||jdtjdddgS)zTest DNS 'NS' record queriesrz 39.28.189.39r8rk)rr{lookupNameserversr Record_NSrr!r!r'test_nameservers zServerDNSTests.test_nameservercCs"||jdtjddddgS)zTest DNS 'HINFO' record queriesrrOrPr8)rQrRrC)rr{ZlookupHostInfor Record_HINFOrr!r!r' test_HINFOs  zServerDNSTests.test_HINFOcCs ||jdtjdddgS)zTest DNS 'PTR' record queriesz123.93.84.28.in-addr.arpaztest.host-reverse.lookup.comrErk)rr{Z lookupPointerr Record_PTRrr!r!r'test_PTR"s zServerDNSTests.test_PTRcCs ||jdtjdddgS)zTest DNS 'CNAME' record queriesrzcanonical.name.comr8rk)rr{lookupCanonicalNamer Record_CNAMErr!r!r' test_CNAME*s zServerDNSTests.test_CNAMEcCs ||jdtjdddgS)zTest DNS 'MB' record queriesrzmailbox.test-domain.comr8rk)rr{Z lookupMailBoxr Record_MBrr!r!r'test_MB1s zServerDNSTests.test_MBcCs ||jdtjdddgS)zTest DNS 'MG' record queriesrzmail.group.someplacer8rk)rr{ZlookupMailGroupr Record_MGrr!r!r'test_MG9s zServerDNSTests.test_MGcCs ||jdtjdddgS)zTest DNS 'MR' record queriesrzmail.redirect.or.whateverr8rk)rr{ZlookupMailRenamer Record_MRrr!r!r'test_MRAs zServerDNSTests.test_MRcCs"||jdtjddddgS)zTest DNS 'MINFO' record queriesrz r mail boxz e mail boxr8)rYrZrC)rr{ZlookupMailboxInfor Record_MINFOrr!r!r' test_MINFOIs zServerDNSTests.test_MINFOc Cs&||jdtjddddddgS)zTest DNS 'SRV' record querieszhttp.tcp.test-domain.comrcrdresome.other.place.foolr8rk)rr{Z lookupServicer Record_SRVrr!r!r'test_SRVQs zServerDNSTests.test_SRVcCs"||jdtjddddgS)zTest DNS 'AFSDB' record queriesrr:zafsdb.test-domain.comr8)r[r\rC)rr{ZlookupAFSDatabaser Record_AFSDBrr!r!r' test_AFSDBXs zServerDNSTests.test_AFSDBcCs"||jdtjddddgS)zTest DNS 'RP' record queriesrzwhatever.i.dunnozsome.more.textr8)r]r^rC)rr{ZlookupResponsibilityr Record_RPrr!r!r'test_RP`s zServerDNSTests.test_RPcCs.||jdtjddddtjdddgS)zTest DNS 'TXT' record queriesrrSrTr8rkrX)rr{Z lookupTextr Record_TXTrr!r!r'test_TXThs zServerDNSTests.test_TXTc Cs.||jdtjdddtjddddgS)zH L{DNSServerFactory} can serve I{SPF} resource records. rrJr8rkrKrL)rr{ZlookupSenderPolicyr Record_SPFrr!r!r'test_spfss zServerDNSTests.test_spfc Cs&||jdtjdtjdddgS)zTest DNS 'WKS' record queriesrz 12.54.78.12r_r8rk)rr{ZlookupWellKnownServicesr Record_WKSsocket IPPROTO_TCPrr!r!r'test_WKSs  zServerDNSTests.test_WKScCs^tt}tj|_||jd|tjdddtj dddtj dddtj d d d d d dgS)NrH1.2.3.4rjrkz ns1.domainZ2Mz ns2.domainrlrcrdrerrm) copymy_soarArCrr{lookupAllRecordsrrrr)r1Z result_soar!r!r'test_someRecordsWithTTLss     z'ServerDNSTests.test_someRecordsWithTTLscCs ||jdtjdddgS)z%Test DNS 'AAAA' record queries (IPv6)rz'AF43:5634:1294:AFCB:56AC:48EF:34C3:01FFr8rk)rr{lookupIPV6Addressr Record_AAAArr!r!r' test_AAAAs zServerDNSTests.test_AAAAc CsD||jdtjdddddtjddd ddtjd d d ddgS) z#Test DNS 'A6' record queries (IPv6)rrz ABCD::4321r8rkrVz 0:0069::0zsome.network.tldrWz$0:5634:1294:AFCB:56AC:48EF:34C3:01FFz tra.la.la.net)rr{ZlookupAddress6r Record_A6rr!r!r'test_A6s zServerDNSTests.test_A6cCsZtj}ddttjtjD}|D]}|jdkr&||_q&| |j d dd|S)z9 Test DNS 'AXFR' queries (Zone transfer) cSsg|]}t|qSr!)rr$r!r!r'r(sz4ServerDNSTests.test_zoneTransfer..NrcSs|dddfS)Nrr!)r&r!r!r'rUz2ServerDNSTests.test_zoneTransfer..) rrAroperatoraddrsr0valuesrCrr{ lookupZoner)r1Z default_ttlr*r&r!r!r'test_zoneTransfers z ServerDNSTests.test_zoneTransfercCs tdgd}||dtjS)zu A failed AXFR TCP connection errbacks the L{Deferred} returned from L{Resolver.lookupZone}. )znameserver.invalid5rrzimpossible.invalid)r assertFailurerr ZDNSLookupError)r1r{r!r!r' test_zoneTransferConnectionFailss  z/ServerDNSTests.test_zoneTransferConnectionFailscCs ||jdtjdddgS)z6Tests that unrelated zones don't mess with each other.zanothertest-domain.comrr8rkrrr!r!r'test_similarZonesDontInterferes z-ServerDNSTests.test_similarZonesDontInterferec Cs(||jdtjdddddddgS) z2 Test DNS 'NAPTR' record queries. rr6rMr`rarbr8rk)rr{ZlookupNamingAuthorityPointerr Record_NAPTRrr!r!r' test_NAPTRs  zServerDNSTests.test_NAPTRN)!r2r3r4__doc__r}rrrrrrrrrrrrrrrrrrrrrrrrrrrrr!r!r!r'rns<        rnc@seZdZddZdS) HelperTestscCs@|}t|}tdD] }t|}|||k|}qdS)N)mktemprZ getSerialrange assertTrue)r1faibr!r!r'test_serialGenerators    z HelperTests.test_serialGeneratorN)r2r3r4rr!r!r!r'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.comr6i,iiiX)r<r=r>r?rBrAr@rCnametypeclsrCauthr#ns.twistedmatrix.comrrCrMz mail.mv3d.com) preferenceZexchangerCz 64.123.27.105)addressrC)r*r ZDeferreddr _gotResultsr ZAXFRController controllerrrrINrr/NSrZMXrArr0rr!r!r'r}s6    zAXFRTests.setUpc Cstjdddddddddd S)Nr:r idanswerZopCodeZrecDesZrecAvrZrCodeZtruncZmaxSize)rrrr!r!r' _makeMessageszAXFRTests._makeMessagecCsH|}tdtjtjg|_|j|_|j |d| |j |jdS)Nr) rrrAXFRrqueriesr0answersrmessageReceivedrr*)r1mr!r!r'test_bindAndTNamesStyles z!AXFRTests.test_bindAndTNamesStylecCs ||_dSr-r))r1resultr!r!r'rszAXFRTests._gotResultscCsR|jdd}|r>|}g|_|dg|_|j|dq||j|jdSNr) r0rrpoprrrrr*)r1r0rr!r!r' test_DJBStyle szAXFRTests.test_DJBStyleN)r2r3r4r}rrrrr!r!r!r'rs rc@seZdZddZddZdS)ResolvConfHandlingTestscCs2|}tj|d}||jdg|jdS)NZresolvrqr)rr rr dynServers _parseCallcancelr1Z resolvConfr&r!r!r' test_missings z$ResolvConfHandlingTests.test_missingcCs@|}t|dtj|d}||jdg|jdS)Nwrr) ropencloser rrrrrrr!r!r' test_emptys  z"ResolvConfHandlingTests.test_emptyN)r2r3r4rrr!r!r!r'rsrc@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)rsr/failureResultOfrassertIsInstancevaluer)r1 testDomaintestDomainNamerr!r!r''test_domainErrorForNameWithCommonSuffix,sz6AuthorityTests.test_domainErrorForNameWithCommonSuffixc Cstttjtfttjttdgid}|ttj}g}||j|d\}}}| |g| |tj ttjtj 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. rrirTrCr#rN) r,strrr<rrrrappendrrTYPErA)r1rrrr additionalr!r!r'test_recordMissing:s2   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}. rrr:rrN) rsr/maxr typeToMethodrqueryrr r r)r1r r  unknownTyperr!r!r'test_unknownTypeNXDOMAINVsz'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. r:rHrN)rrrsuccessResultOfrurrr)r1rrrrr!r!r'test_unknownTypeMissingcs  z&AuthorityTests.test_unknownTypeMissingc Csdttj}td}tttjtf||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. zexample.rriFrN) rrr<rrr,getattrrrrrrA)r1methodZ subdomainZ nameserverrrrrr!r!r' _referralTestos.   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). rNrrr!r!r' test_referralszAuthorityTests.test_referralcCs|ddS)zT A referral is also generated for a request of type C{ALL_RECORDS}. rNrrr!r!r'test_allRecordsReferralsz&AuthorityTests.test_allRecordsReferralN) r2r3r4rr rrrrr r!r!r!r!r'r&s   rc@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}. ri)r,r<rrr)r1rr/ makeRecordtarget addressesrr!r!r'_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)rrr!r!r'rrUz>AdditionalProcessingTests.assertRecordsMatch..)keyN)rsorted)r1ZexpectedZcomputedr(r!r!r'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|ddqSTrrrrrrAr%rr$r!r'r(sz=AdditionalProcessingTests._additionalTest..N)rr<rr&rr*)r1rr#r%rrrrr!r.r'_additionalTests  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} rrMN)r/rrrr1r%r!r!r'_additionalMXTests  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)r1_Arr!r!r'test_mailExchangeAdditionalAsz6AdditionalProcessingTests.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)r1_AAAArr!r!r'test_mailExchangeAdditionalAAAAsz9AdditionalProcessingTests.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)r1r2r4rr!r!r'test_mailExchangeAdditionalBothsz9AdditionalProcessingTests.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)r/rrr0r!r!r'_additionalNSTests z+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)r7r2rr!r!r'test_nameserverAdditionalA*sz4AdditionalProcessingTests.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)r7r4rr!r!r'test_nameserverAdditionalAAAA2sz7AdditionalProcessingTests.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)r7r2r4rr!r!r'test_nameserverAdditionalBoth:sz7AdditionalProcessingTests.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,r-r.r!r'r(Ssz>AdditionalProcessingTests._answerCNAMETest..N) rr<rr&rrrrZCNAMErAr*)r1r%rrrraliasr!r.r'_answerCNAMETestBs0  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<r2rr!r!r'test_canonicalNameAnswerAZsz3AdditionalProcessingTests.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<r4rr!r!r'test_canonicalNameAnswerAAAAbsz6AdditionalProcessingTests.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<r2r4rr!r!r'test_canonicalNameAnswerBothjsz6AdditionalProcessingTests.test_canonicalNameAnswerBothN)r2r3r4rrrr2rr4r&r*r/r1r3r5r6r7r8r9r:r<r=r>r?r!r!r!r'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)r1argsZmessagesr!r'r{sz3NoInitialResponseTests.test_noAnswer..queryc_sttdS)NzCouldn't connect)r Zfailrgaierror)r1rAkwargsr!r!r' queryProtocolsz;NoInitialResponseTests.test_noAnswer..queryProtocol)rrrrrz fooba.comrFrrrrr:rrr)rZ_queryZpatchrrwrrrrrrrrZ getHostByNamerrC)r1rrEr{r0rr!rBr' test_noAnswerus0   z$NoInitialResponseTests.test_noAnswerN)r2r3r4rFr!r!r!r'r@ssr@c@s eZdZdZddZddZdS)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.orgrrr:N)rrprimary_portdomainsdomain)r1rJservicer!r!r'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. z 192.168.1.3iz example.netz example.edurr:N)rZfromServerAddressAndDomainsrrJrKrLrM)r1rJrxrNr!r!r'&test_constructAuthorityFromHostAndPortszESecondaryAuthorityServiceTests.test_constructAuthorityFromHostAndPortN)r2r3r4rrOrPr!r!r!r'rGsrGc@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.comrN)rrrJrKrMr1 secondaryr!r!r'test_defaultPorts z(SecondaryAuthorityTests.test_defaultPortcCs:tdd}||jd||jd||jddS)z~ When constructed using L{SecondaryAuthority.fromServerAddressAndDomain}, the specified port is used. )rRrSrRrWN)rfromServerAddressAndDomainrrJrKrMrTr!r!r'test_explicitPortsz)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. rHr7rIrrHr7roN)rrXr_reactortransfer tcpClientsrr buildProtocolrmakeConnectionrdecoderr rrrrr) r1rUr hostrxrvtimeout bindAddressprotor~msgr!r!r' test_transfers$    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. rZ example.comrr:)rrroNsns1.example.comzadmin.example.comi@iii i%i.r;s 192.168.1.2rk)rrr)rr#r"z!HrIT)r#r)rrXrr[r\r]rr^rr_rr`rr rrrrextendrrZtoStrZ dataReceivedrlenrrr)r1rUr rarxrvrbrcrdr~rr/rrdatarr!r!r'test_lookupAddress sN       z*SecondaryAuthorityTests.test_lookupAddressN)r2r3r4rrVrYrfrkr!r!r!r'rQs   rQsY$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)rrencodeZ setContentrZ BindAuthoritypath)r1sfpr!r!r'loadBindStringQs  z!BindAuthorityTests.loadBindStringcCs|t|_dSr-)rrsampleBindZonerrr!r!r'r}aszBindAuthorityTests.setUpcCs8|jjD]&}|jj|D]}|d|jkqq dS)zG Loads the default $TTL and applies it to all records. : N)rr0keysrrC)r1domZrecr!r!r'test_ttles zBindAuthorityTests.test_ttlcCs$|d|jj|d|jjdS)z, Loads the default $ORIGIN. s example.com.not-fqdn.example.comN)rroriginZassertInr0rr!r!r'test_originFromFilepsz&BindAuthorityTests.test_originFromFilecCsBdD]8\}}||j|dd}|t|d|jqdS)z' A records are loaded. ))rgz10.0.0.1)sno-in.example.comz10.0.0.2rrtN)rrrrrrr#)r1rvZipr'r!r!r' test_aRecords|s  z BindAuthorityTests.test_aRecordscCs4||jddd}|tdd|jdS)z* AAAA records are loaded. rgrz2001:db8:10::1rtN)rrrrrrr#r1r'r!r!r'test_aaaaRecordss z#BindAuthorityTests.test_aaaaRecordscCs8||jddd}|tjdddd|jdS)z( MX records are loaded. rxrrMzmx.example.comrt)rrrCN)rrrrrrr#r|r!r!r'test_mxRecordss z!BindAuthorityTests.test_mxRecordscCs6||jddd}|tjddd|jdS)z+ CNAME records are loaded. swww.example.comrrIrtrN)rrrrrrr#r|r!r!r'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) assertRaisesNotImplementedErrorrrr exceptionrA)r1er!r!r'test_invalidRecordClasss  z*BindAuthorityTests.test_invalidRecordClassc CsNdD]D}|t}||dW5QRX|t|d|jjdqdS)zC $INCLUDE and $GENERATE raise NotImplementedError. )s$INCLUDEs $GENERATEs doesNotMatters directive not implementedrN)rrrrrrrrA)r1Z directiverr!r!r'test_invalidDirectivess   z)BindAuthorityTests.test_invalidDirectivesN)r2r3r4rrrr}rwrzr{r}r~rrrr!r!r!r'rlMs   rl)ZrZ __future__rrrrrior functoolsrrZstructrZ twisted.trialrZtwisted.internetr r r Ztwisted.internet.deferr Z twisted.namesr rrrrZtwisted.names.dnsrrrrrrZtwisted.names.errorrZtwisted.names.clientrZtwisted.names.secondaryrrZtwisted.python.compatrZtwisted.python.filepathrZtwisted.test.proto_helpersrrr r+Z FileAuthorityr,rZ reverse_soarrrrrrrrrrrrrrrrrrrrsrrtruZTestCasernrrrrr"r@rGrQrsrlr!r!r!r's                        5     9 9s[%5l