o b&@sdZddlmZddlmZmZmZmZmZm Z m Z m Z ddl m Z mZmZddlmZddlmZGdddZGd d d eeZGd d d eeZGd ddeeZdS)zC Tests for the I{hosts(5)}-based resolver, L{twisted.names.hosts}. ) gatherResults)AAAAINA DomainErrorQueryRecord_A Record_AAAARRHeader)Resolver searchFileForsearchFileForAll)FilePath)SynchronousTestCasec@seZdZddZdS)GoodTempPathMixincCst|dS)Nzutf-8)rmktempencodeselfr?/usr/lib/python3/dist-packages/twisted/names/test/test_hosts.pypathszGoodTempPathMixin.pathN)__name__ __module__ __qualname__rrrrrrs rc@0eZdZdZddZddZddZdd 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. cC*|}|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 assertEqualr rhostsrrrtest_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. rsbar.example.comN)rr! assertIsNoner r#rrrtest_notFoundAddress-s 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 ::1rNr r#rrrtest_firstAddress6s 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_searchFileForAliasesBsz.SearchHostsFileTests.test_searchFileForAliasesN)rrr__doc__r%r'r)r*rrrrrs   rc@r) SearchHostsFileForAllTestsz Tests for L{searchFileForAll}, a helper which finds all addresses for a particular hostname in a I{hosts(5)}-style file. 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.comNrr!r"r r#rrrtest_allAddressesZsz,SearchHostsFileForAllTests.test_allAddressescCs*|}|d|dgt|ddS)zL L{searchFileForAll} searches for names case-insensitively. s!127.0.0.1 foobar.EXAMPLE.com r-sFOOBAR.example.comNr.r#rrrtest_caseInsensitivelyjs 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"r rrrrrtest_readErrorrsz)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 miserr-r(smiserNr.r#rrrtest_malformedIPysz+SearchHostsFileForAllTests.test_malformedIPN)rrrr+r/r0r1r2rrrrr,Ts  r,c@sxeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZdS) HostsTestszJ Tests for the I{hosts(5)}-based L{twisted.names.hosts.Resolver}. 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)rr!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)rr5rrrtest_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.EXAMPLETHINGr;)sMIXED1.1.1.2cs&g|]\}}j|j|qSr)r5 getHostByName addCallbackr").0niprrr sz1HostsTests.test_getHostByName..N)successResultOfr)rdatadsrrrtest_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 r5 lookupAddressrCr"r rrr4rrdanswers authority additionalrrrtest_lookupAddresss zHostsTests.test_lookupAddressc CrG)z L{hosts.Resolver.lookupIPV6Address} returns a L{Deferred} which fires with AAAA records from the hosts file. s ip6-multiplez::3z::4N r5lookupIPV6AddressrCr"r rrr4r 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) r5lookupAllRecordsrCr"r rrr4rrJrrrtest_lookupAllRecordss z HostsTests.test_lookupAllRecordscC||jdtdS)z| L{hosts.Resolver} fails with L{NotImplementedError} for L{IResolver} methods it doesn't implement. r:N)failureResultOfr5lookupMailExchangeNotImplementedErrorrrrrtest_notImplementeds zHostsTests.test_notImplementedcCs8|jtd}||\\}}}||jddS)Nr:r;)r5queryrrCr"payload dottedQuadrrKanswerrMrNrrr test_queryszHostsTests.test_querycCrU)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)rVr5rIrrrrrtest_lookupAddressNotFoundsz%HostsTests.test_lookupAddressNotFoundcCrU)zj Like L{test_lookupAddressNotFound}, but for L{hosts.Resolver.lookupIPV6Address}. r`N)rVr5rQrrrrrtest_lookupIPV6AddressNotFoundz)HostsTests.test_lookupIPV6AddressNotFoundcCrU)zi Like L{test_lookupAddressNotFound}, but for L{hosts.Resolver.lookupAllRecords}. r`N)rVr5rSrrrrrtest_lookupAllRecordsNotFound rcz(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.5NrHr]rrrtest_lookupMalformeds zHostsTests.test_lookupMalformedc Cre)zk Like L{test_lookupAddressMalformed}, but for L{hosts.Resolver.lookupIPV6Address}. rfz::5NrPr]rrrtest_lookupIPV6Malformed s z#HostsTests.test_lookupIPV6MalformedN)rrrr+r7r9rFrOrRrTrYr_rarbrdrgrhrrrrr3s   r3N)r+twisted.internet.deferrtwisted.names.dnsrrrrrrr r twisted.names.hostsr r r twisted.python.filepathrtwisted.trial.unittestrrrr,r3rrrrs (  69