o i\@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.0rrrcC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/names/test/test_names.py "zjustPayload..rrresultsrrr% justPayload!sr*c@eZdZddZdS)NoFileAuthoritycCstj||||_|_dSN)rZ ResolverBase__init__soarecords)selfr/r0rrr%r.&s zNoFileAuthority.__init__N)__name__ __module__ __qualname__r.rrrr%r,% r,test-domain.comzroot.test-domain.comdiH?&mnamernameserialrefreshminimumexpireretryttls93.84.28.in-addr.arpaxi1i~Ϊ my-domain.comspostmaster.test-domain.comi90i?B)r=r>r?r@rArBrCs 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)r6shttp.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.com1SrDs 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.checkResultsN) addCallback)r1Zqueryingrrrrr% namesTests zServerDNSTests.namesTestcC ||jdtjdddgS)z"Test simple DNS 'A' record queriestest-domain.comrrr9rlNrr} lookupAddressrrr1rrr%test_addressRecord1 z"ServerDNSTests.test_addressRecord1cC,||jdtjdddtjdddgS)z1Test DNS 'A' record queries with multiple answersrOz 123.242.1.5r9rlz 0.255.0.255Nrrrrr%test_addressRecord2    z"ServerDNSTests.test_addressRecord2cCr)z+Test DNS 'A' record queries with edge caseszhost-two.test-domain.comz255.255.255.254r9rl0.0.0.0Nrrrrr%test_addressRecord3rz"ServerDNSTests.test_addressRecord3cCs||jdtgS)zTest DNS 'SOA' record queriesrN)rr}ZlookupAuthority soa_recordrrrr%test_authorityszServerDNSTests.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. r6rNrir9rlrgrhN)rr}lookupMailExchanger Record_MXrrrrr%test_mailExchangeRecords   z&ServerDNSTests.test_mailExchangeRecordcCr)zTest DNS 'NS' record queriesrz 39.28.189.39r9rlN)rr}lookupNameserversr Record_NSrrrr%test_nameserverrzServerDNSTests.test_nameservercC"||jdtjddddgS)zTest DNS 'HINFO' record queriesrrPrQr9)rRrSrDN)rr}ZlookupHostInfor Record_HINFOrrrr% test_HINFO zServerDNSTests.test_HINFOcCr)zTest DNS 'PTR' record queriesz123.93.84.28.in-addr.arpaztest.host-reverse.lookup.comrFrlN)rr} lookupPointerr Record_PTRrrrr%test_PTR rzServerDNSTests.test_PTRcCr)zTest DNS 'CNAME' record queriesrzcanonical.name.comr9rlN)rr}lookupCanonicalNamer Record_CNAMErrrr% test_CNAME'rzServerDNSTests.test_CNAMEcCr)zTest DNS 'MB' record queriesrzmailbox.test-domain.comr9rlN)rr}Z lookupMailBoxr Record_MBrrrr%test_MB.rzServerDNSTests.test_MBcCr)zTest DNS 'MG' record queriesrzmail.group.someplacer9rlN)rr}ZlookupMailGroupr Record_MGrrrr%test_MG5rzServerDNSTests.test_MGcCr)zTest DNS 'MR' record queriesrzmail.redirect.or.whateverr9rlN)rr}ZlookupMailRenamer Record_MRrrrr%test_MR<rzServerDNSTests.test_MRcCr)zTest DNS 'MINFO' record queriesrz r mail boxz e mail boxr9)rZr[rDN)rr}ZlookupMailboxInfor Record_MINFOrrrr% test_MINFOCrzServerDNSTests.test_MINFOc Cs&||jdtjddddddgS) zTest DNS 'SRV' record querieszhttp.tcp.test-domain.comrdrerfsome.other.place.foolr9rlN)rr}Z lookupServicer Record_SRVrrrr%test_SRVNs zServerDNSTests.test_SRVcCr)zTest DNS 'AFSDB' record queriesrr;zafsdb.test-domain.comr9)r\r]rDN)rr}ZlookupAFSDatabaser Record_AFSDBrrrr% test_AFSDBUrzServerDNSTests.test_AFSDBcCr)zTest DNS 'RP' record queriesrzwhatever.i.dunnozsome.more.textr9)r^r_rDN)rr}ZlookupResponsibilityr Record_RPrrrr%test_RP`rzServerDNSTests.test_RPcCs.||jdtjddddtjdddgS)zTest DNS 'TXT' record queriesrrTrUr9rlrYN)rr}Z lookupTextr Record_TXTrrrr%test_TXTks zServerDNSTests.test_TXTc Cs.||jdtjdddtjddddgS)zH L{DNSServerFactory} can serve I{SPF} resource records. rrKr9rlrLrMN)rr}ZlookupSenderPolicyr Record_SPFrrrr%test_spfys  zServerDNSTests.test_spfc Cs&||jdtjdtjdddgS)zTest DNS 'WKS' record queriesrz 12.54.78.12r`r9rlN)rr}ZlookupWellKnownServicesr Record_WKSsocket IPPROTO_TCPrrrr%test_WKSs zServerDNSTests.test_WKScCs^tt}tj|_||jd|tjdddtj dddtj dddtj d d d d d dgS)NrI1.2.3.4rkrlz ns1.domainZ2Mz ns2.domainrmrdrerfrrn) copymy_soarBrDrr}lookupAllRecordsrrrr)r1Z result_soarrr%test_someRecordsWithTTLss     z'ServerDNSTests.test_someRecordsWithTTLscCr)z%Test DNS 'AAAA' record queries (IPv6)rz'AF43:5634:1294:AFCB:56AC:48EF:34C3:01FFr9rlN)rr}lookupIPV6Addressr Record_AAAArrrr% test_AAAArzServerDNSTests.test_AAAAc CsD||jdtjdddddtjddd ddtjd d d ddgS)z#Test DNS 'A6' record queries (IPv6)rrz ABCD::4321r9rlrWz 0:0069::0zsome.network.tldrXz$0:5634:1294:AFCB:56AC:48EF:34C3:01FFz tra.la.la.netN)rr}ZlookupAddress6r Record_A6rrrr%test_A6s zServerDNSTests.test_A6cCsZtj}ddttjtjD}|D] }|jdur||_q| |j d dd|S)z9 Test DNS 'AXFR' queries (Zone transfer) cSsg|]}t|qSr)rr"rrr%r&s z4ServerDNSTests.test_zoneTransfer..NrcSs|dddfS)Nrr)r$rrr%r'z2ServerDNSTests.test_zoneTransfer..) rrBroperatoraddrtr0valuesrDrr} lookupZoner)r1Z 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.invalid5rszimpossible.invalidN)r assertFailurerrZDNSLookupError)r1r}rrr% test_zoneTransferConnectionFailss  z/ServerDNSTests.test_zoneTransferConnectionFailscCr)z6Tests that unrelated zones don't mess with each other.zanothertest-domain.comrr9rlNrrrrr%test_similarZonesDontInterfererz-ServerDNSTests.test_similarZonesDontInterferec Cs(||jdtjdddddddgS) z2 Test DNS 'NAPTR' record queries. rr7rNrarbrcr9rlN)rr}ZlookupNamingAuthorityPointerr Record_NAPTRrrrr% test_NAPTRs zServerDNSTests.test_NAPTRN)!r2r3r4__doc__rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr%ros>        roc@r+) HelperTestscCs@|}t|}tdD]}t|}|||k|}q dS)N)mktempr Z getSerialrange assertTrue)r1faibrrr%test_serialGenerators   z HelperTests.test_serialGeneratorN)r2r3r4rrrrr%rr5rc@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.comr7i,iiiX)r=r>r?r@rCrBrArDnametypeclsrDauthr!ns.twistedmatrix.comrrDrNz mail.mv3d.com) preferenceZexchangerDz 64.123.27.105)addressrD)r)rZDeferreddr _gotResultsrZAXFRController controllerrrrINrr/NSrZMXrArr0rrrr%rs`    zAXFRTests.setUpc Cstjdddddddddd S)Nr;r idanswerZopCodeZrecDesZrecAvrZrCodetruncZmaxSize)rrrrrr% _makeMessage2szAXFRTests._makeMessagecCsH|}tdtjtjg|_|j|_|j |d| |j |jdS)Nr) rrrAXFRrqueriesr0answersrmessageReceivedrr))r1mrrr%test_bindAndTNamesStyle@s z!AXFRTests.test_bindAndTNamesStylecCs ||_dSr-r()r1resultrrr%rHs zAXFRTests._gotResultscCsT|jdd}|r |}g|_|dg|_|j|d|s ||j|jdSNr) r0rrpoprrrrr))r1r0rrrr% test_DJBStyleKszAXFRTests.test_DJBStyleN)r2r3r4rrrrrrrrr%rs 5 rc@seZdZddZddZdS)ResolvConfHandlingTestscCs2|}tj|d}||jdg|jdS)NZresolvrrr)rrrr dynServers _parseCallcancelr1Z resolvConfr$rrr% test_missingWs z$ResolvConfHandlingTests.test_missingcCs@|}t|dtj|d}||jdg|jdS)Nwrr ) ropencloserrrr r r r rrr% test_empty]s  z"ResolvConfHandlingTests.test_emptyN)r2r3r4rrrrrr%rVs rc@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)rtr/failureResultOfrassertIsInstancevaluer)r1 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. rrjTrDr!rN) r,rr=rrrsuccessResultOfrrrTYPErB)r1r 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}. rrr;rrN) rtr/maxr typeToMethodrqueryrrrr)r1rr 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. r;rIr N)r!rr"rrvr#rr)r1r$rr 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.rrjFrN) rr=rrrr,getattrrrrrrB)r1methodZ subdomainZ nameserverr rrrrrr% _referralTests0     zAuthorityTests._referralTestcC|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_referralcCr*)zT A referral is also generated for a request of type C{ALL_RECORDS}. rNr+rrrr%test_allRecordsReferralsz&AuthorityTests.test_allRecordsReferralN) r2r3r4rrrr%r&r)r,r-rrrr%res "  ! 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}. rjN)r,r=rr'r)r1r(r/ 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$sz>AdditionalProcessingTests.assertRecordsMatch..)keyN)rsorted)r1expectedZcomputedr4rrr%assertRecordsMatchs z,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 $g|]}tj|jtj|ddqSTrrrrrrBr#rr0rr%r&=z=AdditionalProcessingTests._additionalTest..N)rr=rr2rr7)r1r(r/r1rrr rrr<r%_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} rrNN)r>rrrr1r1rrr%_additionalMXTestJsz+AdditionalProcessingTests._additionalMXTestcC||jgdS)z If the name of the MX response has A records, they are included in the additional section of the response. N)r@_Arrrr%test_mailExchangeAdditionalAUz6AdditionalProcessingTests.test_mailExchangeAdditionalAcCrA)z If the name of the MX response has AAAA records, they are included in the additional section of the response. N)r@_AAAArrrr%test_mailExchangeAdditionalAAAA\rDz9AdditionalProcessingTests.test_mailExchangeAdditionalAAAAcC||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)r@rBrErrrr%test_mailExchangeAdditionalBothcz9AdditionalProcessingTests.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>rrr?rrr%_additionalNSTestjsz+AdditionalProcessingTests._additionalNSTestcCrA)z If the name of the NS response has A records, they are included in the additional section of the response. N)rJrBrrrr%test_nameserverAdditionalAsrDz4AdditionalProcessingTests.test_nameserverAdditionalAcCrA)z If the name of the NS response has AAAA records, they are included in the additional section of the response. N)rJrErrrr%test_nameserverAdditionalAAAAzrDz7AdditionalProcessingTests.test_nameserverAdditionalAAAAcCrG)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)rJrBrErrrr%test_nameserverAdditionalBothrIz7AdditionalProcessingTests.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 r8r9r:r;r<rr%r&r=z>AdditionalProcessingTests._answerCNAMETest..N) rr=rr2rrrrZCNAMErBr7)r1r1rrr raliasrr<r%_answerCNAMETests(     z*AdditionalProcessingTests._answerCNAMETestcCrA)z If the name of the CNAME response has A records, they are included in the answer section of the response. N)rOrBrrrr%test_canonicalNameAnswerArDz3AdditionalProcessingTests.test_canonicalNameAnswerAcCrA)z If the name of the CNAME response has AAAA records, they are included in the answer section of the response. N)rOrErrrr%test_canonicalNameAnswerAAAArDz6AdditionalProcessingTests.test_canonicalNameAnswerAAAAcCrG)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)rOrBrErrrr%test_canonicalNameAnswerBothrIz6AdditionalProcessingTests.test_canonicalNameAnswerBothN)r2r3r4rrrrBrrEr2r7r>r@rCrFrHrJrKrLrMrOrPrQrRrrrr%r.s&  &#  # r.c@r+)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)r1argsmessagesrr%r#sz3NoInitialResponseTests.test_noAnswer..queryc_sttdS)NzCouldn't connect)rZfailrgaierror)r1rTkwargsrrr% queryProtocolz;NoInitialResponseTests.test_noAnswer..queryProtocol)rrrsr#z fooba.comrFrrrrr;rrrN)rZ_querypatchrrxrrrrrrappendrZ getHostByNamerrW)r1r#rYr}r0rrrUr% test_noAnswers:     z$NoInitialResponseTests.test_noAnswerN)r2r3r4r]rrrr%rSr5rSc@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.orgrrr;s example.orgN)rrprimary_portdomainsdomainr1raservicerrr%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.netr;N)rZfromServerAddressAndDomainsrrarbrcrd)r1rarzrfrrr%&test_constructAuthorityFromHostAndPort s zESecondaryAuthorityServiceTests.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. rhrjz example.edurr;riN)rencoderrarcrdrerrr% test_constructAuthorityFromBytes#sz?SecondaryAuthorityServiceTests.test_constructAuthorityFromBytesN)r2r3r4rrgrkrmrrrr%r^s  r^c@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)rrrarbrdr1 secondaryrrr%test_defaultPort?s z(SecondaryAuthorityTests.test_defaultPortcCs:tdd}||jd||jd||jddS)z~ When constructed using L{SecondaryAuthority.fromServerAddressAndDomain}, the specified port is used. )rorprorurqN)rfromServerAddressAndDomainrrarbrdrrrrr%test_explicitPortIs z)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. r_r8 example.comrr_r8rpN)rrvr _reactortransfer tcpClientsrr buildProtocolr makeConnectionrdecoderrrrrrr) r1rsrhostrzrwtimeout bindAddressprotormsgrrr% 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. rxr`rr;)rrrpNsns1.example.comzadmin.example.comi@iii i%i.r<s 192.168.1.2rl)rrr)rr!r z!HryT)r!r)rrvr rzr{r|rr}r r~rrrrrrrrextendrrZtoStrZ dataReceivedrlenrrr)r1rsrrrzrwrrrrr#r/rrdatarrrr%test_lookupAddressssF       z*SecondaryAuthorityTests.test_lookupAddressN)r2r3r4rrtrwrrrrrr%rn9s   rnsY$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} asciiN)rrrlZ setContentr Z BindAuthoritypath)r1sfprrr%loadBindStrings  z!BindAuthorityTests.loadBindStringcC|t|_dSr-)rsampleBindZonerrrrr%rrZzBindAuthorityTests.setUpcCs8|jjD]}|jj|D] }|d|jkqqdS)zG Loads the default $TTL and applies it to all records. : N)rr0keysrrD)r1domZrecrrr%test_ttls zBindAuthorityTests.test_ttlcCs$|d|jj|d|jjdS)z, Loads the default $ORIGIN. s example.com.not-fqdn.example.comN)rroriginZassertInr0rrrr%test_originFromFilesz&BindAuthorityTests.test_originFromFilecCsBdD]\}}||j|\\}\\|t|d|jqdS)' A records are loaded. )r`z10.0.0.1)no-in.example.comz10.0.0.2rNrrrrrrr!r1ripr3rrr% test_aRecordss z BindAuthorityTests.test_aRecordscCs4||jd\\}\\|tdd|jdS)* AAAA records are loaded. r`2001:db8:10::1rNrrrrrrr!r1r3rrr%test_aaaaRecordss  z#BindAuthorityTests.test_aaaaRecordscCs8||jd\\}\\|tjdddd|jdS)( MX records are loaded. rrNzmx.example.comr)rrrDNrrrrrrr!rrrr%test_mxRecordss  z!BindAuthorityTests.test_mxRecordscCs<||jd\}\\|d}|tjddd|jdS)+ CNAME records are loaded. www.example.comrryrrNrrrrrrr!r1rr3rrr%test_cnameRecordss  z$BindAuthorityTests.test_cnameRecordscCsL|t }|dWdn1swY|d|jjddS)zO loadBindString raises NotImplementedError on invalid records. sexample.com. IN LOL 192.168.0.1NzRecord type 'LOL' not supportedr) assertRaisesNotImplementedErrorrr exceptionrT)r1errr%test_invalidRecordClass%s  z*BindAuthorityTests.test_invalidRecordClassc CsbdD],}|t}||dWdn1swY|t|d|jjdqdS)zC $INCLUDE and $GENERATE raise NotImplementedError. )s$INCLUDEs $GENERATEs doesNotMatterNs directive not implementedr)rrrrrrrT)r1 directiverrrr%test_invalidDirectives-s   z)BindAuthorityTests.test_invalidDirectivesN)r2r3r4rrrrrrrrrrrrrrr%rs  razone = [ 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}. cCsNt|}t|jd }||Wdn1swYt|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} rN)rrrrwriter ZPySourceAuthority)r1rrrrrr%loadPySourceStringjs   z)PySourceAuthorityTests.loadPySourceStringcCrr-)rsamplePySourcerrrrr%rzrZzPySourceAuthorityTests.setUpcCs@dD]\}}||j|\\}\\|t||jqdS)rrNrrrrr%r}s z$PySourceAuthorityTests.test_aRecordscCs2||jd\\}\\|td|jdS)rr`rNrrrrr%rs  z'PySourceAuthorityTests.test_aaaaRecordscCs6||jd\\}\\|tjddd|jdS)rrrNzmail.example.com)rrNrrrrr%rs  z%PySourceAuthorityTests.test_mxRecordscC:||jd\}\\|d}|tjdd|jdS)rrrryrNrrrrr%r  z(PySourceAuthorityTests.test_cnameRecordscCr)z) PTR records are loaded. s2.0.0.10.in-addr.arparrrN)rrrrrrr!rrrr%rrzPySourceAuthorityTests.test_PTRcCsd}|t|j|dS)z1 Input file has no zone variable z nothing = []N)r ValueErrorr)r1Z badPySourcerrr%test_badInputNoZonesz*PySourceAuthorityTests.test_badInputNoZoneN) r2r3r4rrrrrrrrrrrrr%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_soarrrrrrrrrrrrrrrrrrrrtrrurvZTestCaserorrrrr.rSr^rnrrrrrrrr%s                          3    \ Z~_0Hm+