U lj(@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@seZdZddddZdS)_SupportsMktempstrreturncCsdS)NselfrrA/tmp/pip-unpacked-wheel-aghyhbke/twisted/names/test/test_hosts.pymktempsz_SupportsMktemp.mktempN)__name__ __module__ __qualname__rrrrrrsrc@seZdZdddddZdS)GoodTempPathMixinrzFilePath[bytes])rrcCst|dS)Nzutf-8)rrencoderrrrpath!szGoodTempPathMixin.pathN)rrrr rrrrr src@sHeZdZdZddddZddddZdddd Zddd 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. NonercCs*|}|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_notFoundAddresscCs*|}|d|dt|jddS)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_searchFileForAliasesN)rrr__doc__r*r+r-r.rrrrr!%s    r!c@sHeZdZdZddddZddddZdddd Zddd d Zd S) SearchHostsFileForAllTestsz Tests for L{searchFileForAll}, a helper which finds all addresses for a particular hostname in a I{hosts(5)}-style file. r"rcCs.|}|d|dddgt|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 r1sFOOBAR.example.comNr2r(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 miserr1r,smiserNr2r(rrrtest_malformedIPsz+SearchHostsFileForAllTests.test_malformedIPN)rrrr/r3r4r5r6rrrrr0[s r0c@seZdZdZddddZddddZdddd Zddd d Zddd d ZddddZ ddddZ ddddZ ddddZ ddddZ ddddZddddZddddZdS) HostsTestszJ Tests for the I{hosts(5)}-based L{twisted.names.hosts.Resolver}. r"rcCs,|}|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)rr9rrrtest_defaultPathszHostsTests.test_defaultPathcs.dddg}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.EXAMPLETHINGr?)sMIXED1.1.1.2cs&g|]\}}j|j|qSr)r9Z getHostByNameZ addCallbackr').0niprrr sz1HostsTests.test_getHostByName..N)successResultOfr)rdataZdsrrrtest_getHostByNames zHostsTests.test_getHostByNamec Cs\|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 r9 lookupAddressrEr'r rrr8r rdZanswers authority additionalrrrtest_lookupAddresss zHostsTests.test_lookupAddressc Cs\|jd}||\}}}|tdtt|jtd|jtdtt|jtd|jf|dS)z L{hosts.Resolver.lookupIPV6Address} returns a L{Deferred} which fires with AAAA records from the hosts file. s ip6-multiplez::3z::4N r9lookupIPV6AddressrEr'r rrr8r rJrrrtest_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. smixedr@N) r9lookupAllRecordsrEr'r rrr8r rJrrrtest_lookupAllRecordss  z HostsTests.test_lookupAllRecordscCs||jdtdS)z| L{hosts.Resolver} fails with L{NotImplementedError} for L{IResolver} methods it doesn't implement. r>N)failureResultOfr9ZlookupMailExchangeNotImplementedErrorrrrrtest_notImplementeds zHostsTests.test_notImplementedcCs8|jtd}||\\}}}||jddS)Nr>r?)r9queryr rEr'payloadZ dottedQuadrrKZanswerrLrMrrr test_queryszHostsTests.test_querycCs||jdtdS)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)rTr9rIrrrrrtest_lookupAddressNotFoundsz%HostsTests.test_lookupAddressNotFoundcCs||jdtdS)zj Like L{test_lookupAddressNotFound}, but for L{hosts.Resolver.lookupIPV6Address}. r[N)rTr9rPrrrrrtest_lookupIPV6AddressNotFound sz)HostsTests.test_lookupIPV6AddressNotFoundcCs||jdtdS)zi Like L{test_lookupAddressNotFound}, but for L{hosts.Resolver.lookupAllRecords}. r[N)rTr9rRrrrrrtest_lookupAllRecordsNotFoundsz(HostsTests.test_lookupAllRecordsNotFoundc CsD|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.5NrHrYrrrtest_lookupMalformeds  zHostsTests.test_lookupMalformedc CsD|jd}||\\}}}|tdtt|jtd|j|dS)zk Like L{test_lookupAddressMalformed}, but for L{hosts.Resolver.lookupIPV6Address}. r_z::5NrOrYrrrtest_lookupIPV6Malformed's  z#HostsTests.test_lookupIPV6MalformedN)rrrr/r;r=rGrNrQrSrVrZr\r]r^r`rarrrrr7s   r7N)r/ __future__rZtyping_extensionsrZtwisted.internet.deferrZtwisted.names.dnsrrrrr r r r Ztwisted.names.hostsr rrZtwisted.python.filepathrZtwisted.trial.unittestrrrr!r0r7rrrrs   (   69