o i(@sdZddlmZddlmZddlmZddlmZm Z m Z m Z m Z m Z mZmZddlmZmZmZddlmZddlmZGd d d eZGd d d ZGd ddeeZGdddeeZGdddeeZdS)zC Tests for the I{hosts(5)}-based resolver, L{twisted.names.hosts}. ) annotations)Protocol) gatherResults)AAAAINA DomainErrorQueryRecord_A Record_AAAARRHeader)Resolver searchFileForsearchFileForAll)FilePath)SynchronousTestCasec@seZdZdddZdS)_SupportsMktempreturnstrcCsdS)NselfrrcC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/names/test/test_hosts.pymktempsz_SupportsMktemp.mktempN)rr)__name__ __module__ __qualname__rrrrrrrc@seZdZdddZdS) GoodTempPathMixinrrrFilePath[bytes]cCst|dS)Nzutf-8)rrencoderrrrpath!szGoodTempPathMixin.pathN)rrrr)rrrr!rrrrr rrc@8eZdZdZd ddZd ddZd dd Zd d d Zd S)SearchHostsFileTestsz Tests for L{searchFileFor}, a helper which finds the first address for a particular hostname in a I{hosts(5)}-style file. rNonecC*|}|d|dt|jddS)zr If there is an IPv4 address for the hostname passed to L{searchFileFor}, it is returned. 10.2.3.4 foo.example.com z10.2.3.4foo.example.comNr! setContent assertEqualrrhostsrrrtest_findAddress+s z%SearchHostsFileTests.test_findAddresscCs(|}|d|t|jddS)z~ If there is no address information for the hostname passed to L{searchFileFor}, L{None} is returned. r&sbar.example.comN)r!r)Z assertIsNonerr+rrrtest_notFoundAddress4s z)SearchHostsFileTests.test_notFoundAddresscCr%)zS The first address associated with the given hostname is returned. sV::1 foo.example.com 10.1.2.3 foo.example.com fe80::21b:fcff:feee:5a1d foo.example.com ::1r'Nr(r+rrrtest_firstAddress=s z&SearchHostsFileTests.test_firstAddresscCsz|}|d|t|jdd|t|jdd|t|jdd|t|jdd|t|jddd S) z For a host with a canonical name and one or more aliases, L{searchFileFor} can find an address given any of the names. sY127.0.1.1 helmut.example.org helmut # a comment ::1 localhost ip6-localhost ip6-loopback shelmutz 127.0.1.1shelmut.example.orgs ip6-localhostr/s ip6-loopbacks localhostNr(r+rrrtest_searchFileForAliasesIsz.SearchHostsFileTests.test_searchFileForAliasesNrr$)rrr__doc__r-r.r0r1rrrrr#%s    r#c@r")SearchHostsFileForAllTestsz Tests for L{searchFileForAll}, a helper which finds all addresses for a particular hostname in a I{hosts(5)}-style file. rr$cCs,|}|d|gdt|ddS)zt L{searchFileForAll} returns a list of all addresses associated with the name passed to it. sc127.0.0.1 foobar.example.com 127.0.0.2 foobar.example.com ::1 foobar.example.com ) 127.0.0.1z 127.0.0.2r/sfoobar.example.comNr!r)r*rr+rrrtest_allAddressesasz,SearchHostsFileForAllTests.test_allAddressescCs*|}|d|dgt|ddS)zL L{searchFileForAll} searches for names case-insensitively. s!127.0.0.1 foobar.EXAMPLE.com r5sFOOBAR.example.comNr6r+rrrtest_caseInsensitivelyqs z1SearchHostsFileForAllTests.test_caseInsensitivelycCs|gt|ddS)z If there is an error reading the contents of the hosts file, L{searchFileForAll} returns an empty list. s example.comN)r*rr!rrrrtest_readErrorysz)SearchHostsFileForAllTests.test_readErrorcCs,|}|d|ddgt|ddS)zw L{searchFileForAll} ignores any malformed IP addresses associated with the name passed to it. s[127.0.0.1 miser.example.org miser not-an-ip miser not-ascii miser # miser miser ::1 miserr5r/smiserNr6r+rrrtest_malformedIPsz+SearchHostsFileForAllTests.test_malformedIPNr2)rrrr3r7r8r9r:rrrrr4[s    r4c@seZdZdZdddZdddZddd Zdd d Zdd d ZdddZ dddZ dddZ dddZ dddZ dddZdddZdddZdS) HostsTestszJ Tests for the I{hosts(5)}-based L{twisted.names.hosts.Resolver}. rr$cCs,|}|dd|_t|j|j|_dS)Ns  1.1.1.1 EXAMPLE EXAMPLE.EXAMPLETHING ::2 mixed 1.1.1.2 MIXED ::1 ip6thingy 1.1.1.3 multiple 1.1.1.4 multiple ::3 ip6-multiple ::4 ip6-multiple not-an-ip malformed malformed # malformed 1.1.1.5 malformed ::5 malformed ih)r!r)ttlr resolver)rfrrrsetUps zHostsTests.setUpcCst}|d|jdS)z The default hosts file used by L{Resolver} is I{/etc/hosts} if no value is given for the C{file} initializer parameter. s /etc/hostsN)r r*file)rr=rrrtest_defaultPathszHostsTests.test_defaultPathcs,gd}fdd|D}t|dS)z L{hosts.Resolver.getHostByName} returns a L{Deferred} which fires with a string giving the address of the queried name as found in the resolver's hosts file. ))EXAMPLE1.1.1.1)sEXAMPLE.EXAMPLETHINGrC)sMIXED1.1.1.2cs&g|]\}}j|j|qSr)r=Z getHostByNameZ addCallbackr*).0niprrr sz1HostsTests.test_getHostByName..N)successResultOfr)rdataZdsrrrtest_getHostByNames  zHostsTests.test_getHostByNamec C\|jd}||\}}}|tdtt|jtd|jtdtt|jtd|jf|dS)z L{hosts.Resolver.lookupAddress} returns a L{Deferred} which fires with A records from the hosts file. smultiplez1.1.1.3z1.1.1.4N r= lookupAddressrIr*r rrr<r rdZanswers authority additionalrrrtest_lookupAddresss zHostsTests.test_lookupAddressc CrL)z L{hosts.Resolver.lookupIPV6Address} returns a L{Deferred} which fires with AAAA records from the hosts file. s ip6-multiplez::3z::4N r=lookupIPV6AddressrIr*r rrr<r rOrrrtest_lookupIPV6Addresss z!HostsTests.test_lookupIPV6Addressc CsD|jd}||\}}}|tdtt|jtd|jf|dS)z L{hosts.Resolver.lookupAllRecords} returns a L{Deferred} which fires with A records from the hosts file. smixedrDN) r=lookupAllRecordsrIr*r rrr<r rOrrrtest_lookupAllRecordss z HostsTests.test_lookupAllRecordscC||jdtdS)z| L{hosts.Resolver} fails with L{NotImplementedError} for L{IResolver} methods it doesn't implement. rBN)failureResultOfr=ZlookupMailExchangeNotImplementedErrorrrrrtest_notImplementeds zHostsTests.test_notImplementedcCs8|jtd}||\\}}}||jddS)NrBrC)r=queryr rIr*payloadZ dottedQuadrrPZanswerrQrRrrr test_queryszHostsTests.test_querycCrY)z L{hosts.Resolver.lookupAddress} returns a L{Deferred} which fires with L{dns.DomainError} if the name passed in has no addresses in the hosts file. foueoaN)rZr=rNrrrrrtest_lookupAddressNotFoundsz%HostsTests.test_lookupAddressNotFoundcCrY)zj Like L{test_lookupAddressNotFound}, but for L{hosts.Resolver.lookupIPV6Address}. raN)rZr=rUrrrrrtest_lookupIPV6AddressNotFound z)HostsTests.test_lookupIPV6AddressNotFoundcCrY)zi Like L{test_lookupAddressNotFound}, but for L{hosts.Resolver.lookupAllRecords}. raN)rZr=rWrrrrrtest_lookupAllRecordsNotFoundrdz(HostsTests.test_lookupAllRecordsNotFoundc CD|jd}||\\}}}|tdtt|jtd|j|dS)z L{hosts.Resolver.lookupAddress} returns a L{Deferred} which fires with the valid addresses from the hosts file, ignoring any entries that aren't valid IP addresses. malformedz1.1.1.5NrMr_rrrtest_lookupMalformeds zHostsTests.test_lookupMalformedc Crf)zk Like L{test_lookupAddressMalformed}, but for L{hosts.Resolver.lookupIPV6Address}. rgz::5NrTr_rrrtest_lookupIPV6Malformed's z#HostsTests.test_lookupIPV6MalformedNr2)rrrr3r?rArKrSrVrXr\r`rbrcrerhrirrrrr;s           r;N)r3 __future__rZtyping_extensionsrZtwisted.internet.deferrZtwisted.names.dnsrrrrr r r r Ztwisted.names.hostsr rrZtwisted.python.filepathrZtwisted.trial.unittestrrrr#r4r;rrrrs   (  69