o >h(@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)Nselfrr}/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/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) 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-localhostr0s ip6-loopbacks localhostNr(r+rrrtest_searchFileForAliasesIsz.SearchHostsFileTests.test_searchFileForAliasesNrr$)rrr__doc__r-r/r1r2rrrrr#%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.2r0sfoobar.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 r6sFOOBAR.example.comNr7r+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 miserr6r0smiserNr7r+rrrtest_malformedIPsz+SearchHostsFileForAllTests.test_malformedIPNr3)rrrr4r8r9r:r;rrrrr5[s    r5c@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.EXAMPLETHINGrD)sMIXED1.1.1.2cs&g|]\}}j|j|qSr)r> 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 r> lookupAddressrLr*r rrr=r rdanswers authority additionalrrrtest_lookupAddresss zHostsTests.test_lookupAddressc CrP)z L{hosts.Resolver.lookupIPV6Address} returns a L{Deferred} which fires with AAAA records from the hosts file. s ip6-multiplez::3z::4N r>lookupIPV6AddressrLr*r rrr=r rSrrrtest_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. smixedrEN) r>lookupAllRecordsrLr*r rrr=r rSrrrtest_lookupAllRecordss z HostsTests.test_lookupAllRecordscC||jdtdS)z| L{hosts.Resolver} fails with L{NotImplementedError} for L{IResolver} methods it doesn't implement. rCN)failureResultOfr>lookupMailExchangeNotImplementedErrorrrrrtest_notImplementeds zHostsTests.test_notImplementedcCs8|jtd}||\\}}}||jddS)NrCrD)r>queryr rLr*payload dottedQuadrrTanswerrVrWrrr test_queryszHostsTests.test_querycCr^)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)r_r>rRrrrrrtest_lookupAddressNotFoundsz%HostsTests.test_lookupAddressNotFoundcCr^)zj Like L{test_lookupAddressNotFound}, but for L{hosts.Resolver.lookupIPV6Address}. riN)r_r>rZrrrrrtest_lookupIPV6AddressNotFound z)HostsTests.test_lookupIPV6AddressNotFoundcCr^)zi Like L{test_lookupAddressNotFound}, but for L{hosts.Resolver.lookupAllRecords}. riN)r_r>r\rrrrrtest_lookupAllRecordsNotFoundrlz(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.5NrQrfrrrtest_lookupMalformeds zHostsTests.test_lookupMalformedc Crn)zk Like L{test_lookupAddressMalformed}, but for L{hosts.Resolver.lookupIPV6Address}. roz::5NrYrfrrrtest_lookupIPV6Malformed's z#HostsTests.test_lookupIPV6MalformedNr3)rrrr4r@rBrOrXr[r]rbrhrjrkrmrprqrrrrr<s           r<N)r4 __future__rtyping_extensionsrtwisted.internet.deferrtwisted.names.dnsrrrrr r r r twisted.names.hostsr rrtwisted.python.filepathrtwisted.trial.unittestrrrr#r5r<rrrrs   (  69