File: //usr/lib/python3/dist-packages/twisted/names/test/__pycache__/test_hosts.cpython-38.pyc
U
W[c! � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
mZmZmZmZmZ ddlmZmZmZ G dd � d e�ZG d
d � d ee�ZG d d
� d
ee�ZG dd� dee�ZdS )zC
Tests for the I{hosts(5)}-based resolver, L{twisted.names.hosts}.
� )�division�absolute_import)�TestCase)�FilePath��
gatherResults)�A�AAAA�IN� DomainError�RRHeader�Query�Record_A� Record_AAAA)�Resolver�
searchFileFor�searchFileForAllc @ s e Zd Zdd� ZdS )�GoodTempPathMixinc C s t | �� �d��S )Nzutf-8)r Zmktemp�encode��self� r �?/usr/lib/python3/dist-packages/twisted/names/test/test_hosts.py�path s zGoodTempPathMixin.pathN)�__name__�
__module__� __qualname__r r r r r r s r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � 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.
c C s* | � � }|�d� | �dt|j d�� dS )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.4� foo.example.comN�r �
setContent� assertEqualr �r Zhostsr r r �test_findAddress s �
�z%SearchHostsFileTests.test_findAddressc C s( | � � }|�d� | �t|j d�� dS )z~
If there is no address information for the hostname passed to
L{searchFileFor}, L{None} is returned.
r s bar.example.comN)r r! Z assertIsNoner r# r r r �test_notFoundAddress* s
�z)SearchHostsFileTests.test_notFoundAddressc C s* | � � }|�d� | �dt|j d�� dS )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# r r r �test_firstAddress5 s
�z&SearchHostsFileTests.test_firstAddressc C sz | � � }|�d� | �t|j d�d� | �t|j d�d� | �t|j d�d� | �t|j d�d� | �t|j d�d� d 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.
sY 127.0.1.1 helmut.example.org helmut
# a comment
::1 localhost ip6-localhost ip6-loopback
s helmutz 127.0.1.1s helmut.example.orgs
ip6-localhostr& s ip6-loopbacks localhostNr r# r r r �test_searchFileForAliasesA s �
�z.SearchHostsFileTests.test_searchFileForAliasesN)r r r �__doc__r$ r% r' r( r r r r r s
r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �SearchHostsFileForAllTestsz�
Tests for L{searchFileForAll}, a helper which finds all addresses for a
particular hostname in a I{hosts(5)}-style file.
c C s. | � � }|�d� | �dddgt|d�� dS )zt
L{searchFileForAll} returns a list of all addresses associated with the
name passed to it.
sc 127.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& s foobar.example.comN�r r! r" r r# r r r �test_allAddressesY s ��z,SearchHostsFileForAllTests.test_allAddressesc C s* | � � }|�d� | �dgt|d�� dS )zL
L{searchFileForAll} searches for names case-insensitively.
s! 127.0.0.1 foobar.EXAMPLE.com
r+ s FOOBAR.example.comNr, r# r r r �test_caseInsensitivelyh s
�z1SearchHostsFileForAllTests.test_caseInsensitivelyc C s | � g t| �� d�� dS )z�
If there is an error reading the contents of the hosts file,
L{searchFileForAll} returns an empty list.
s example.comN)r" r r r r r r �test_readErrorr s �z)SearchHostsFileForAllTests.test_readErrorN)r r r r) r- r. r/ r r r r r* T s
r* c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
HostsTestszJ
Tests for the I{hosts(5)}-based L{twisted.names.hosts.Resolver}.
c C s, | � � }|�d� d| _t|j | j�| _d S )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
ih )r r! �ttlr �resolver)r �fr r r �setUp� s
zHostsTests.setUpc C s t � }| �d|j� dS )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)r r2 r r r �test_defaultPath� s zHostsTests.test_defaultPathc s$ dddg}� fdd�|D �}t |�S )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.
)� EXAMPLE�1.1.1.1)s EXAMPLE.EXAMPLETHINGr8 )s MIXED�1.1.1.2c s& g | ] \}}� j �|��� j|��qS r )r2 Z
getHostByName� addCallbackr" )�.0�nZipr r r �
<listcomp>� s �z1HostsTests.test_getHostByName.<locals>.<listcomp>r )r �dataZdsr r r �test_getHostByName� s �
�z HostsTests.test_getHostByNamec s& � j �d�}� fdd�}|�|� |S )z
L{hosts.Resolver.lookupAddress} returns a L{Deferred} which fires with A
records from the hosts file.
� multiplec sJ | \}}}� � tdtt� jtd� j��tdtt� jtd� j��f|� d S )Nr@ z1.1.1.3z1.1.1.4�r" r r r
r1 r �ZresultsZanswersZ authorityZ
additionalr r r �resolved� s
�
���z/HostsTests.test_lookupAddress.<locals>.resolved)r2 �
lookupAddressr: �r �drC r r r �test_lookupAddress� s
z HostsTests.test_lookupAddressc s& � j �d�}� fdd�}|�|� |S )z�
L{hosts.Resolver.lookupIPV6Address} returns a L{Deferred} which fires
with AAAA records from the hosts file.
� ip6-multiplec sJ | \}}}� � tdtt� jtd� j��tdtt� jtd� j��f|� d S )NrH z::3z::4)r" r r r
r1 r rB r r r rC � s
�
���z3HostsTests.test_lookupIPV6Address.<locals>.resolved)r2 �lookupIPV6Addressr: rE r r r �test_lookupIPV6Address� s
z!HostsTests.test_lookupIPV6Addressc s& � j �d�}� fdd�}|�|� |S )z�
L{hosts.Resolver.lookupAllRecords} returns a L{Deferred} which fires
with A records from the hosts file.
� mixedc
s2 | \}}}� � tdtt� jtd� j��f|� d S )NrK r9 rA rB r r r rC � s
��z2HostsTests.test_lookupAllRecords.<locals>.resolved)r2 �lookupAllRecordsr: rE r r r �test_lookupAllRecords� s
z HostsTests.test_lookupAllRecordsc C s | � | j�d�t�S )Nr7 )�
assertFailurer2 ZlookupMailExchange�NotImplementedErrorr r r r �test_notImplemented� s �z HostsTests.test_notImplementedc s& � j �td��}|�� fdd�� |S )Nr7 c s � � | d d j�� d�S )Nr r8 )r" ZpayloadZ
dottedQuad)�xr r r �<lambda>� s �z'HostsTests.test_query.<locals>.<lambda>)r2 Zqueryr
r: )r rF r r r �
test_query� s zHostsTests.test_queryc C s | � | j�d�t�S )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.
� foueoa)rN r2 rD r r r r r �test_lookupAddressNotFound� s �z%HostsTests.test_lookupAddressNotFoundc C s | � | j�d�t�S )zj
Like L{test_lookupAddressNotFound}, but for
L{hosts.Resolver.lookupIPV6Address}.
rT )rN r2 rI r r r r r � test_lookupIPV6AddressNotFound� s �z)HostsTests.test_lookupIPV6AddressNotFoundc C s | � | j�d�t�S )zi
Like L{test_lookupAddressNotFound}, but for
L{hosts.Resolver.lookupAllRecords}.
rT )rN r2 rL r r r r r � test_lookupAllRecordsNotFound� s �z(HostsTests.test_lookupAllRecordsNotFoundN)r r r r) r4 r6 r? rG rJ rM rP rS rU rV rW r r r r r0 | s
r0 N) r) Z
__future__r r Ztwisted.trial.unittestr Ztwisted.python.filepathr Ztwisted.internet.deferr Ztwisted.names.dnsr r r
r r r
r r Ztwisted.names.hostsr r r �objectr r r* r0 r r r r �<module> s (;(