← Back
Editing: test_client.cpython-38.pyc
U W[?� � @ s� d Z ddlmZmZ ddlmZmZ ddlmZ ddl m Z ddl m Z ddl mZ ddlmZmZ dd lmZ dd lmZ dd lmZ dd lmZmZmZm Z m Z dd l m Z ddl m!Z! ddl"m#Z# ddl$m%Z% ddl&m'Z' ddl(m)Z) e �*� r�dZ+ndZ+G dd� de!�Z,G dd� de-�Z.G dd� de-�Z/G dd � d e)j0�Z1G d d � d e)j0e#�Z2G d d � d e)j0�Z3G d!d"� d"e)j0�Z4G d#d$� d$e)j0�Z5G d%d&� d&e-�Z6G d'd(� d(e)j0�Z7G d)d*� d*e)j0�Z8dS )+z) Test cases for L{twisted.names.client}. � )�division�absolute_import)� verifyClass� verifyObject)�failure)�FilePath)�platform)�defer)�CannotListenError�ConnectionRefusedError)� IResolver)�AlternateReactor)�Clock)�error�client�dns�hosts�cache)�DNSQueryTimeoutError)� ResolverBase)�GoodTempPathMixin)� test_util)� proto_helpers)�unittestz:These tests need more work before they'll work on Windows.Nc @ s e Zd Zdd� ZdS )� FakeResolverc C sZ |dkr*t j|t j|dt jddd�d�}nt j|||dd�}|g}g }g }t�|||f�S )z� The getHostByNameTest does a different type of query that requires it return an A record from an ALL_RECORDS lookup, so we accommodate that here. � getHostByNameTest�< � 127.0.0.1)�address�ttl)�name�type�clsr Zpayload)r r! r" r )r ZRRHeader�AZRecord_Ar �succeed) �selfr r" �qtype�timeoutZrr�results� authority� additional� r+ �@/usr/lib/python3/dist-packages/twisted/names/test/test_client.py�_lookup* s �zFakeResolver._lookupN)�__name__� __module__� __qualname__r- r+ r+ r+ r, r ( s r c @ s e Zd ZdZdZdd� ZdS )�StubPortz� A partial implementation of L{IListeningPort} which only keeps track of whether it has been stopped. @ivar disconnected: A C{bool} which is C{False} until C{stopListening} is called, C{True} afterwards. Fc C s d| _ d S )NT)� disconnected�r% r+ r+ r, � stopListeningG s zStubPort.stopListeningN)r. r/ r0 �__doc__r2 r4 r+ r+ r+ r, r1 = s r1 c @ s" e Zd ZdZdd� Zddd�ZdS ) �StubDNSDatagramProtocolz� L{dns.DNSDatagramProtocol}-alike. @ivar queries: A C{list} of tuples giving the arguments passed to C{query} along with the L{defer.Deferred} which was returned from the call. c C s g | _ t� | _d S �N��queriesr1 � transportr3 r+ r+ r, �__init__T s z StubDNSDatagramProtocol.__init__� Nc C s" t �� }| j�|||||f� |S )zv Record the given arguments and return a Deferred which will not be called back by this code. )r �Deferredr9 �append)r% r r9 r' �id�resultr+ r+ r, �queryY s z StubDNSDatagramProtocol.query)r< N)r. r/ r0 r5 r; rA r+ r+ r+ r, r6 L s r6 c @ s( e Zd ZdZereZdd� Zdd� ZdS )�GetResolverTestsz* Tests for L{client.getResolver}. c C s2 t t� �� t�� }W 5 Q R X | �tt|�� dS )zQ L{client.getResolver} returns an object providing L{IResolver}. N)r r r � getResolver� assertTruer r �r% �resolverr+ r+ r, �test_interfacek s z GetResolverTests.test_interfacec C s6 t t� �� t�� }t�� }W 5 Q R X | �||� dS )zn Multiple calls to L{client.getResolver} return the same L{IResolver} implementation. N)r r r rC �assertIs)r% �a�br+ r+ r, �test_idempotentt s z GetResolverTests.test_idempotentN)r. r/ r0 r5 � windowsSkip�skiprG rK r+ r+ r+ r, rB d s rB c @ sh e Zd ZdZereZdd� Zdd� Zdd� Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zdd� ZdS )�CreateResolverTestsz- Tests for L{client.createResolver}. c C s6 dd� |j D �}| �dt|�� | �|d j|� d S )Nc S s g | ]}t |tj�r|�qS r+ )� isinstancer �Resolver��.0�rr+ r+ r, � <listcomp>� s z2CreateResolverTests._hostsTest.<locals>.<listcomp>� r )� resolvers� assertEqual�len�file�r% rF �filename�resr+ r+ r, � _hostsTest� s z CreateResolverTests._hostsTestc C s. t t� �� t�� }W 5 Q R X | �|d� dS )z� L{client.createResolver} returns a L{resolve.ResolverChain} including a L{hosts.Resolver} using I{/etc/hosts} if no alternate hosts file is specified. s /etc/hostsN�r r r �createResolverr] rE r+ r+ r, �test_defaultHosts� s z%CreateResolverTests.test_defaultHostsc C s2 t t� �� tjdd�}W 5 Q R X | �|d� dS )z� The I{hosts} parameter to L{client.createResolver} overrides the hosts file used by the L{hosts.Resolver} in the L{resolve.ResolverChain} it returns. � /foo/bar)r Nr^ rE r+ r+ r, �test_overrideHosts� s z&CreateResolverTests.test_overrideHostsc C s6 dd� |j D �}| �dt|�� | �|d j|� dS )z| Verify that C{resolver} has a L{client.Resolver} with a configuration filename set to C{filename}. c S s g | ]}t |tj�r|�qS r+ �rO r rP rQ r+ r+ r, rT � s z7CreateResolverTests._resolvConfTest.<locals>.<listcomp>rU r N)rV rW rX �resolvrZ r+ r+ r, �_resolvConfTest� s z#CreateResolverTests._resolvConfTestc C sX t � }t|�� t�� }W 5 Q R X dd� |jD �}| �dt|�� | �||d j� dS )z� The L{client.Resolver} included in the L{resolve.ResolverChain} returned by L{client.createResolver} uses the global reactor. c S s g | ]}t |tj�r|�qS r+ rc rQ r+ r+ r, rT � s z4CreateResolverTests.test_reactor.<locals>.<listcomp>rU r N) r r r r_ rV rW rX rH �_reactor)r% �reactorrF r\ r+ r+ r, � test_reactor� s z CreateResolverTests.test_reactorc C s. t t� �� t�� }W 5 Q R X | �|d� dS )z� L{client.createResolver} returns a L{resolve.ResolverChain} including a L{client.Resolver} using I{/etc/resolv.conf} if no alternate resolver configuration file is specified. s /etc/resolv.confN�r r r r_ re rE r+ r+ r, �test_defaultResolvConf� s z*CreateResolverTests.test_defaultResolvConfc C s2 t t� �� tjdd�}W 5 Q R X | �|d� dS )z� The I{resolvconf} parameter to L{client.createResolver} overrides the resolver configuration file used by the L{client.Resolver} in the L{resolve.ResolverChain} it returns. ra �� resolvconfNri rE r+ r+ r, �test_overrideResolvConf� s z+CreateResolverTests.test_overrideResolvConfc C s� | � � }|�d� tt� �� tj|j d�}W 5 Q R X dd� |jD �}| �dt|�� | �g |d j � | �dg|d j � dS ) z� If no servers are given, addresses are taken from the file given by the I{resolvconf} parameter to L{client.createResolver}. � nameserver 127.1.2.3 rk c S s g | ]}t |tj�r|�qS r+ rc rQ r+ r+ r, rT � s z;CreateResolverTests.test_defaultServers.<locals>.<listcomp>rU r �z 127.1.2.3�5 N� �pathZ setContentr r r r_ rV rW rX �servers� dynServers�r% rl rF r\ r+ r+ r, �test_defaultServers� s z'CreateResolverTests.test_defaultServersc C s� | � � }|�d� tt� �� tjdg|j d�}W 5 Q R X dd� |jD �}| �dt|�� | �dg|d j � | �dg|d j � d S ) z� Servers passed to L{client.createResolver} are used in addition to any found in the file given by the I{resolvconf} parameter. rn )z 127.3.2.1rp )rs rl c S s g | ]}t |tj�r|�qS r+ rc rQ r+ r+ r, rT � s z<CreateResolverTests.test_overrideServers.<locals>.<listcomp>rU r ro Nrq ru r+ r+ r, �test_overrideServers� s �z(CreateResolverTests.test_overrideServersc C sB t t� �� t�� }W 5 Q R X dd� |jD �}| �dt|�� dS )zy L{client.createResolver} returns a L{resolve.ResolverChain} including a L{cache.CacheResolver}. c S s g | ]}t |tj�r|�qS r+ )rO r Z CacheResolverrQ r+ r+ r, rT � s z2CreateResolverTests.test_cache.<locals>.<listcomp>rU N)r r r r_ rV rW rX )r% rF r\ r+ r+ r, � test_cache� s z CreateResolverTests.test_cacheN)r. r/ r0 r5 rL rM r] r` rb re rh rj rm rv rw rx r+ r+ r+ r, rN � s rN c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6S )7� ResolverTestsz' Tests for L{client.Resolver}. c C s t tt� dS )z] L{client} provides L{IResolver} through a series of free functions. N)r r r r3 r+ r+ r, � test_clientProvidesIResolver s z*ResolverTests.test_clientProvidesIResolverc C s t ttj� dS )z; L{client.Resolver} provides L{IResolver}. N)r r r rP r3 r+ r+ r, �$test_clientResolverProvidesIResolver s z2ResolverTests.test_clientResolverProvidesIResolverc C s | � ttj� dS )z� L{client.Resolver} raises L{ValueError} if constructed with neither servers nor a nameserver configuration file. N)Z assertRaises� ValueErrorr rP r3 r+ r+ r, �test_noServers s z ResolverTests.test_noServersc C s( t j| �� t� d�}| �dg|j� dS )z� A missing nameserver configuration file results in no server information being loaded from it (ie, not an exception) and a default server being provided. �rd rg �r rp N)r rP �mktempr rW rt rE r+ r+ r, �test_missingConfiguration s z'ResolverTests.test_missingConfigurationc sR t | �� �}|jdd�� G � fdd�dtj�}|ddgdt� d� | �� j� d S ) z� As part of its constructor, C{StubResolver} opens C{/etc/resolv.conf}; then, explicitly closes it and does not count on the GC to do so for it. zw+)�modec s e Zd Z� fdd�ZdS )z9ResolverTests.test_closesResolvConf.<locals>.StubResolverc s � S r7 r+ )r% r �Z resolvConfr+ r, � _openFile+ s zCResolverTests.test_closesResolvConf.<locals>.StubResolver._openFileN)r. r/ r0 r� r+ r� r+ r, � StubResolver* s r� � example.comrp z/etc/resolv.conf)rs rd rg N)r r� �openr rP r rD �closed)r% Zhandler� r+ r� r, �test_closesResolvConf" s �z#ResolverTests.test_closesResolvConfc C s, t jdgd�}|�dg� | �d|j� dS )z� L{client.Resolver.parseConfig} treats a I{domain} line without an argument as indicating a domain of C{b""}. r �rs s domain � N)r rP � parseConfigrW ZdomainrE r+ r+ r, �test_domainEmptyArgument2 s z&ResolverTests.test_domainEmptyArgumentc C s, t jdgd�}|�dg� | �g |j� dS )z� L{client.Resolver.parseConfig} treats a I{search} line without an argument as indicating an empty search suffix. r r� s search N)r rP r� rW �searchrE r+ r+ r, �test_searchEmptyArgument< s z&ResolverTests.test_searchEmptyArgumentc sd t � � ddg}ddg}tj|d�}||_� fdd�|_t� }|�d�}|�| j|� | �t � j �d � | � � j d d |d � � j d d � t d �� | �t � j �d � | � � j d d |d � � j d d � t d �� | �t � j �d � | � � j d d |d � � j d d � t d �� | �t � j �d� | � � j d d |d � � j d d �|� |S )a L{client.Resolver.queryUDP} should issue queries to its L{dns.DNSDatagramProtocol} with server addresses taken from its own C{servers} and C{dynServers} lists, proceeding through them in order as L{DNSQueryTimeoutError}s occur. �z::1rp )z::2rp )z::3rp )z::4rp r� c s � S r7 r+ )� interface��protocolr+ r, �<lambda>S r� z=ResolverTests.test_datagramQueryServerOrder.<locals>.<lambda>NrU r ���� � � )r6 r rP rt �_connectedProtocol�objectZqueryUDP� addCallbackrW rX r9 rH �errbackr �callback)r% rs rt rF ZexpectedResult� queryResultr+ r� r, � test_datagramQueryServerOrderF s, z+ResolverTests.test_datagramQueryServerOrderc s� t � �tjdgd�}�fdd�|_�j}t�dtjtj�}|� |�}�� t |�d� |� |�}�� t |�d� t � � t� � }|j�� � |�� d �|� t�||g�}� �fdd �}|�|� |S ) z� L{client.Resolver.query} only issues one request at a time per query. Subsequent requests made before responses to prior ones are received are queued and given the same response as is given to the first one. �r� rp r� c s � S r7 r+ r+ r� r+ r, r� q r� z<ResolverTests.test_singleConcurrentRequest.<locals>.<lambda>� foo.example.comrU r� c s4 | \}}�� |� gg g f� �� |� gg g f� d S r7 �rW )Z responsesZ firstResponseZsecondResponse)�answerr% r+ r, � cbFinished� s z>ResolverTests.test_singleConcurrentRequest.<locals>.cbFinished)r6 r rP r� r9 r �Queryr# �INrA rW rX r� �Message�answersr> �popr� r � gatherResultsr� ) r% rF r9 rA � firstResult� secondResult�response�dr� r+ )r� r� r% r, � test_singleConcurrentRequesti s" z*ResolverTests.test_singleConcurrentRequestc s� t � � tjdgd�}� fdd�|_� j}t�dtj�}|�|� | � t |�d� t�dtj�}|�|� | � t |�d� t�dtj �}|�|� | � t |�d � d S ) zh L{client.Resolver.query} issues a request for each different concurrent query. r� r� c s � S r7 r+ r+ r� r+ r, r� � r� z?ResolverTests.test_multipleConcurrentRequests.<locals>.<lambda>r� rU � bar.example.comr� r� N) r6 r rP r� r9 r r� r# rA rW rX �A6)r% rF r9 Z firstQueryZ secondQueryZ thirdQueryr+ r� r, � test_multipleConcurrentRequests� s z-ResolverTests.test_multipleConcurrentRequestsc s� t � � tjdgd�}� fdd�|_� j}t�dtj�}|�|� | � t |�d� |� � d � t� � � |�|� | � t |�d� dS ) z� After a response is received to a query issued with L{client.Resolver.query}, another query with the same parameters results in a new network request. r� r� c s � S r7 r+ r+ r� r+ r, r� � r� z?ResolverTests.test_multipleSequentialRequests.<locals>.<lambda>r� rU r� N)r6 r rP r� r9 r r� r# rA rW rX r� r� r� )r% rF r9 rA r+ r� r, � test_multipleSequentialRequests� s z-ResolverTests.test_multipleSequentialRequestsc s� t � � tjdgd�}� fdd�|_� j}t�dtj�}|�|�}|�|�}G dd� dt �}|� � d � t � |� �� t�| �||�| �||�g�S ) z� If the result of a request is an error response, the Deferreds for all concurrently issued requests associated with that result fire with the L{Failure}. r� r� c s � S r7 r+ r+ r� r+ r, r� � r� z>ResolverTests.test_multipleConcurrentFailure.<locals>.<lambda>r� c @ s e Zd ZdS )zGResolverTests.test_multipleConcurrentFailure.<locals>.ExpectedExceptionN�r. r/ r0 r+ r+ r+ r, �ExpectedException� s r� r� )r6 r rP r� r9 r r� r# rA � Exceptionr� r� r �Failurer r� � assertFailure)r% rF r9 rA r� r� r� r+ r� r, � test_multipleConcurrentFailure� s �z,ResolverTests.test_multipleConcurrentFailurec C st t jdgd�}|�� }|�� }| �|j� | �|j� | �|j�� j|j�� j� t� t� |jj �t� |jj �g�S )z� L{client.Resolver._connectedProtocol} returns a new L{DNSDatagramProtocol} connected to a new address with a cryptographically secure random port number. r� r� ) r rP r� ZassertIsNotNoner: ZassertNotEqual�getHost�portr r� Z maybeDeferredr4 )r% rF Z firstProtoZ secondProtor+ r+ r, �test_connectedProtocol� s � �z$ResolverTests.test_connectedProtocolc C s4 t �� }tj| �� |d�}|�� }| �|j|� dS )z� If a reactor instance is supplied to L{client.Resolver} L{client.Resolver._connectedProtocol} should pass that reactor to L{twisted.names.dns.DNSDatagramProtocol}. r~ N)r � MemoryReactorr rP r� r� rH rf )r% rg rF �protor+ r+ r, �)test_resolverUsesOnlyParameterizedReactor� s z7ResolverTests.test_resolverUsesOnlyParameterizedReactorc sd t jdgd�}g � G � fdd�dt�}||_|�t�d�� |�t�d�� | �tt � ��d� dS ) z� L{client.Resolver._connectedProtocol} is called once each time a UDP request needs to be issued and the resulting protocol instance is used for that request. r� r� c s" e Zd Zdd� Zd� fdd� ZdS )z:ResolverTests.test_differentProtocol.<locals>.FakeProtocolc S s t � | _d S r7 �r1 r: r3 r+ r+ r, r; s zCResolverTests.test_differentProtocol.<locals>.FakeProtocol.__init__r< Nc s � � | � t�t�� �S r7 )r> r r$ r r� �r% r rA r' r? �� protocolsr+ r, rA s z@ResolverTests.test_differentProtocol.<locals>.FakeProtocol.query)r< N�r. r/ r0 r; rA r+ r� r+ r, � FakeProtocol s r� r� r� r� N) r rP r� r� rA r r� rW rX �set�r% rF r� r+ r� r, �test_differentProtocol� s z$ResolverTests.test_differentProtocolc C sR t �� }tjdg|d�}|�t�d�� |j�� \\}}|� � j }| � d|� dS )z< If the resolver is ipv6, open a ipv6 port. r� �rs rg r� z::N) r r� r rP rA r r� ZudpPorts�itemsr� �hostrW )r% ZfakerF r� r: r� r+ r+ r, �test_ipv6Resolver s � z ResolverTests.test_ipv6Resolverc sN g � G � fdd�dt �}tjdgd�}|� |_|�� | �tt� ��d� dS )z� If a port number is initially selected which cannot be bound, the L{CannotListenError} is handled and another port number is attempted. c s e Zd Z� fdd�ZdS )z6ResolverTests.test_disallowedPort.<locals>.FakeReactorc s& � � |� t� �dkr"td |d ��d S )NrU )r> rX r )r% r� �args�kwargs�Zportsr+ r, � listenUDP) s z@ResolverTests.test_disallowedPort.<locals>.FakeReactor.listenUDPN)r. r/ r0 r� r+ r� r+ r, � FakeReactor( s r� r� r� r� N)r� r rP rf r� rW rX r� )r% r� rF r+ r� r, �test_disallowedPort! s z!ResolverTests.test_disallowedPortc sx t jdgd�}g � t�t�td���t�t� � �g�G � �fdd�dt �}||_ |� t� d�� | �tt� ��d� dS )z| When a query issued by L{client.Resolver.query} times out, the retry uses a new protocol instance. r� r� Nc s$ e Zd Zdd� Zd� �fdd� ZdS )zFResolverTests.test_differentProtocolAfterTimeout.<locals>.FakeProtocolc S s t � | _d S r7 r� r3 r+ r+ r, r; @ s zOResolverTests.test_differentProtocolAfterTimeout.<locals>.FakeProtocol.__init__r< Nc s � � | � ��d�S �Nr �r> r� r� �r� r( r+ r, rA C s zLResolverTests.test_differentProtocolAfterTimeout.<locals>.FakeProtocol.query)r< Nr� r+ r� r+ r, r� ? s r� r� r� )r rP r �failr r� r r$ r r� r� r� rA r� rW rX r� r� r+ r� r, �"test_differentProtocolAfterTimeout5 s �z0ResolverTests.test_differentProtocolAfterTimeoutc s| t jdgd�}g � t�� �G � �fdd�dt�}||_|�t�d�� | � � d j j � �� t� � � | �� d j j � dS )z� After the L{Deferred} returned by L{DNSDatagramProtocol.query} is called back, the L{DNSDatagramProtocol} is disconnected from its transport. r� r� c s$ e Zd Zdd� Zd� �fdd� ZdS )z9ResolverTests.test_protocolShutDown.<locals>.FakeProtocolc S s t � | _d S r7 r� r3 r+ r+ r, r; W s zBResolverTests.test_protocolShutDown.<locals>.FakeProtocol.__init__r< Nc s � � | � �S r7 �r> r� �r� r@ r+ r, rA Z s z?ResolverTests.test_protocolShutDown.<locals>.FakeProtocol.query)r< Nr� r+ r� r+ r, r� V s r� r� r N)r rP r r= r� r� rA r r� � assertFalser: r2 r� r� rD r� r+ r� r, �test_protocolShutDownL s z#ResolverTests.test_protocolShutDownc s� t jdgd�}g � t�� }t�t�td���|g�G � �fdd�dt�}||_ |� t � d�� | � � d jj� |�t �� � | �� d jj� dS )z� The L{DNSDatagramProtocol} created when an interim timeout occurs is also disconnected from its transport after the Deferred returned by its query method completes. r� r� Nc s$ e Zd Zdd� Zd� �fdd� ZdS )zEResolverTests.test_protocolShutDownAfterTimeout.<locals>.FakeProtocolc S s t � | _d S r7 r� r3 r+ r+ r, r; s s zNResolverTests.test_protocolShutDownAfterTimeout.<locals>.FakeProtocol.__init__r< Nc s � � | � ��d�S r� r� r� r� r+ r, rA v s zKResolverTests.test_protocolShutDownAfterTimeout.<locals>.FakeProtocol.query)r< Nr� r+ r� r+ r, r� r s r� r� rU )r rP r r= r� r r� r r� r� rA r r� r� r: r2 r� r� rD )r% rF r@ r� r+ r� r, �!test_protocolShutDownAfterTimeoutf s �z/ResolverTests.test_protocolShutDownAfterTimeoutc s� G dd� dt �}tjdgd�}g � t�� �G � �fdd�dt�}||_|�t� d��}| � � d j j � �� t�|� �� | �� d j j � | �||�S ) z� If the L{Deferred} returned by L{DNSDatagramProtocol.query} fires with a failure, the L{DNSDatagramProtocol} is still disconnected from its transport. c @ s e Zd ZdS )zJResolverTests.test_protocolShutDownAfterFailure.<locals>.ExpectedExceptionNr� r+ r+ r+ r, r� � s r� r� r� c s$ e Zd Zdd� Zd� �fdd� ZdS )zEResolverTests.test_protocolShutDownAfterFailure.<locals>.FakeProtocolc S s t � | _d S r7 r� r3 r+ r+ r, r; � s zNResolverTests.test_protocolShutDownAfterFailure.<locals>.FakeProtocol.__init__r< Nc s � � | � �S r7 r� r� r� r+ r, rA � s zKResolverTests.test_protocolShutDownAfterFailure.<locals>.FakeProtocol.query)r< Nr� r+ r� r+ r, r� � s r� r� r )r� r rP r r= r� r� rA r r� r� r: r2 r� r r� rD r� )r% r� rF r� r� r+ r� r, �!test_protocolShutDownAfterFailure� s z/ResolverTests.test_protocolShutDownAfterFailurec C sN t jdgd�}|j�d�}|�d� | �||j� |�d� | �||j� dS )z� When a TCP DNS protocol associated with a Resolver disconnects, it is removed from the Resolver's connection list. r� r� N) r rP �factoryZ buildProtocolZmakeConnectionZassertInZ connectionsZconnectionLostZ assertNotIn)r% rF r� r+ r+ r, �(test_tcpDisconnectRemovesFromConnections� s z6ResolverTests.test_tcpDisconnectRemovesFromConnectionsc C sv t �� }tjdg|d�}|�t�d��}|jd \}}}}}G dd� dt�} |� |j d t � | � �� | � || � dS )z� The deferred returned by L{client.Resolver.queryTCP} will errback when the TCP connection attempt fails. The reason for the connection failure is passed as the argument to errback. �z 192.0.2.100rp r� r� r c @ s e Zd ZdS )zVResolverTests.test_singleTCPQueryErrbackOnConnectionFailure.<locals>.SentinelExceptionNr� r+ r+ r+ r, �SentinelException� s r� N)r r� r rP �queryTCPr r� � tcpClientsr� �clientConnectionFailed� connectorsr r� �failureResultOf) r% rg rF r� r� r� r� r' � bindAddressr� r+ r+ r, �-test_singleTCPQueryErrbackOnConnectionFailure� s � �z;ResolverTests.test_singleTCPQueryErrbackOnConnectionFailurec C s� t �� }tjdg|d�}|�t�d��}|�t�d��}|jd \}}}}} G dd� dt�} |� |j d t � | � �� | � || �} | � || �} | �| | � dS ) z� All pending L{resolver.queryTCP} C{deferred}s will C{errback} with the same C{Failure} if the connection attempt fails. r� r� r� z example.netr c @ s e Zd ZdS )zXResolverTests.test_multipleTCPQueryErrbackOnConnectionFailure.<locals>.SentinelExceptionNr� r+ r+ r+ r, r� � s r� N)r r� r rP r� r r� r� r� r� r� r r� r� rH ) r% rg rF Zd1Zd2r� r� r� r' r� r� �f1�f2r+ r+ r, �/test_multipleTCPQueryErrbackOnConnectionFailure� s � � z=ResolverTests.test_multipleTCPQueryErrbackOnConnectionFailurec s t �� }tjdg|d��t�d�� ��� �}� �fdd�}|�|� | �t |j �d� | �t |j �d� |j d \}}}}}t � t� �} |�|j d | � | �t |j �d� | �t |j �d� | �|� t � t� �} |�|j d | � | �|t�} | �| | � d S ) z{ An errback on the deferred returned by L{client.Resolver.queryTCP} may trigger another TCP query. )r iE' r� r� c s | � t� ��� �S r7 )�trapr r� )�e��qrF r+ r, �reissue� s zOResolverTests.test_reentrantTCPQueryErrbackOnConnectionFailure.<locals>.reissuerU r r� N)r r� r rP r r� r� � addErrbackrW rX r� r� r r� r r� ZassertNoResultr� rH ) r% rg r� r� r� r� r� r' r� r� r� �fr+ r� r, �0test_reentrantTCPQueryErrbackOnConnectionFailure� s6 � � � z>ResolverTests.test_reentrantTCPQueryErrbackOnConnectionFailurec C s� t �� }tjdg|d�}|�t�d��}|jd \}}}}}|j} | � t | �d� G dd� dt �} |� |j d t�| � �� | �|| � | �|j| � | � t | �d� dS ) z� When the TCP connection attempt fails, the L{client.Resolver.pending} list is emptied in place. It is not replaced with a new empty list. r� r� r� r rU c @ s e Zd ZdS )zJResolverTests.test_pendingEmptiedInPlaceOnError.<locals>.SentinelExceptionNr� r+ r+ r+ r, r� # s r� N)r r� r rP r� r r� r� �pendingrW rX r� r� r� r r� r� rH ) r% rg rF r� r� r� r� r' r� Z prePendingr� r+ r+ r, �!test_pendingEmptiedInPlaceOnError s" � � z/ResolverTests.test_pendingEmptiedInPlaceOnErrorN) r. r/ r0 r5 rz r{ r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r+ r+ r+ r, ry � s6 #$ 1ry c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;S )<� ClientTestsc C s t � t_d| _d| _dS )z: Replace the resolver with a FakeResolver � example.comr N)r r � theResolver�hostname�hostnameForGetHostByNamer3 r+ r+ r, �setUp1 s zClientTests.setUpc C s dt _dS )zp By setting the resolver to None, it will be recreated next time a name lookup is done. N)r r� r3 r+ r+ r, �tearDown9 s zClientTests.tearDownc C s6 |\}}}|d }| � |jj| j� | � |j|� dS )zT Verify that the result is the same query type as what is expected. r N)rW r r� r! )r% r( r& r� r) r* r@ r+ r+ r, � checkResult@ s zClientTests.checkResultc C s | � |d� dS )zR Test that the getHostByName query returns the 127.0.0.1 address. r Nr� �r% r@ r+ r+ r, �checkGetHostByNameI s z ClientTests.checkGetHostByNamec C s t �| j�}|�| j� |S )zM do a getHostByName of a value that should return 127.0.0.1. )r Z getHostByNamer� r� r� �r% r� r+ r+ r, �test_getHostByNameO s z ClientTests.test_getHostByNamec C s t �| j�}|�| jtj� |S )z� Do a lookup and test that the resolver will issue the correct type of query type. We do this by checking that FakeResolver returns a result record with the same query type as what we issued. )r � lookupAddressr� r� r� r r# r r+ r+ r, �test_lookupAddressW s z ClientTests.test_lookupAddressc C s t �| j�}|�| jtj� |S �z+ See L{test_lookupAddress} )r ZlookupIPV6Addressr� r� r� r ZAAAAr r+ r+ r, �test_lookupIPV6Addressa s z"ClientTests.test_lookupIPV6Addressc C s t �| j�}|�| jtj� |S r )r ZlookupAddress6r� r� r� r r� r r+ r+ r, �test_lookupAddress6i s z ClientTests.test_lookupAddress6c C s t �| j�}|�| jtj� |S r )r ZlookupNameserversr� r� r� r ZNSr r+ r+ r, �test_lookupNameserversq s z"ClientTests.test_lookupNameserversc C s t �| j�}|�| jtj� |S r )r ZlookupCanonicalNamer� r� r� r ZCNAMEr r+ r+ r, �test_lookupCanonicalNamey s z$ClientTests.test_lookupCanonicalNamec C s t �| j�}|�| jtj� |S r )r ZlookupAuthorityr� r� r� r ZSOAr r+ r+ r, �test_lookupAuthority� s z ClientTests.test_lookupAuthorityc C s t �| j�}|�| jtj� |S r )r Z lookupMailBoxr� r� r� r ZMBr r+ r+ r, �test_lookupMailBox� s z ClientTests.test_lookupMailBoxc C s t �| j�}|�| jtj� |S r )r ZlookupMailGroupr� r� r� r ZMGr r+ r+ r, �test_lookupMailGroup� s z ClientTests.test_lookupMailGroupc C s t �| j�}|�| jtj� |S r )r ZlookupMailRenamer� r� r� r ZMRr r+ r+ r, �test_lookupMailRename� s z!ClientTests.test_lookupMailRenamec C s t �| j�}|�| jtj� |S r )r Z lookupNullr� r� r� r ZNULLr r+ r+ r, �test_lookupNull� s zClientTests.test_lookupNullc C s t �| j�}|�| jtj� |S r )r ZlookupWellKnownServicesr� r� r� r ZWKSr r+ r+ r, � test_lookupWellKnownServices� s z(ClientTests.test_lookupWellKnownServicesc C s t �| j�}|�| jtj� |S r )r Z lookupPointerr� r� r� r ZPTRr r+ r+ r, �test_lookupPointer� s z ClientTests.test_lookupPointerc C s t �| j�}|�| jtj� |S r )r ZlookupHostInfor� r� r� r ZHINFOr r+ r+ r, �test_lookupHostInfo� s z ClientTests.test_lookupHostInfoc C s t �| j�}|�| jtj� |S r )r ZlookupMailboxInfor� r� r� r ZMINFOr r+ r+ r, �test_lookupMailboxInfo� s z"ClientTests.test_lookupMailboxInfoc C s t �| j�}|�| jtj� |S r )r ZlookupMailExchanger� r� r� r ZMXr r+ r+ r, �test_lookupMailExchange� s z#ClientTests.test_lookupMailExchangec C s t �| j�}|�| jtj� |S r )r Z lookupTextr� r� r� r ZTXTr r+ r+ r, �test_lookupText� s zClientTests.test_lookupTextc C s t �| j�}|�| jtj� |S r )r ZlookupSenderPolicyr� r� r� r ZSPFr r+ r+ r, �test_lookupSenderPolicy� s z#ClientTests.test_lookupSenderPolicyc C s t �| j�}|�| jtj� |S r )r ZlookupResponsibilityr� r� r� r ZRPr r+ r+ r, �test_lookupResponsibility� s z%ClientTests.test_lookupResponsibilityc C s t �| j�}|�| jtj� |S r )r ZlookupAFSDatabaser� r� r� r ZAFSDBr r+ r+ r, �test_lookupAFSDatabase� s z"ClientTests.test_lookupAFSDatabasec C s t �| j�}|�| jtj� |S r )r Z lookupServicer� r� r� r ZSRVr r+ r+ r, �test_lookupService� s z ClientTests.test_lookupServicec C s t �| j�}|�| jtj� |S r )r Z lookupZoner� r� r� r ZAXFRr r+ r+ r, �test_lookupZone� s zClientTests.test_lookupZonec C s t �| j�}|�| jtj� |S r )r ZlookupAllRecordsr� r� r� r Z ALL_RECORDSr r+ r+ r, �test_lookupAllRecords s z!ClientTests.test_lookupAllRecordsc C s t �| j�}|�| jtj� |S r )r Z lookupNamingAuthorityPointerr� r� r� r ZNAPTRr r+ r+ r, �!test_lookupNamingAuthorityPointer s z-ClientTests.test_lookupNamingAuthorityPointerc C s. t �| jt j�}t�|�}|�| jt j� |S )a L{client.query} accepts a L{dns.Query} instance and dispatches it to L{client.theResolver}.C{query}, which in turn dispatches to an appropriate C{lookup*} method of L{client.theResolver}, based on the L{dns.Query} type. )r r� r� r# r rA r� r� )r% r� r� r+ r+ r, � test_query s zClientTests.test_queryN) r. r/ r0 r� r� r� r� r r r r r r r r r r r r r r r r r r r r r r r r r r+ r+ r+ r, r� / s: r� c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )�FilterAnswersTestsz| Test L{twisted.names.client.Resolver.filterAnswers}'s handling of various error conditions it might encounter. c C s t jdgd�| _d S )N)z0.0.0.0r r� )r rP rF r3 r+ r+ r, r� ( s zFilterAnswersTests.setUpc sV t jdd�� � �d� � �fdd�}|�j_�j�� �}|��jdgdgdgf� |S ) zf Test that a truncated message results in an equivalent request made via TCP. T)Ztruncr� c s8 �� | � j� t�� }dg|_dg|_dg|_t�|�S )Nr� r) r* ) rW r9 r r� r� r) r* r r$ )r9 r� ��mr% r+ r, r� 6 s z:FilterAnswersTests.test_truncatedMessage.<locals>.queryTCPr� r) r* )r r� ZaddQueryrF r� � filterAnswersr� rW )r% r� r� r+ r r, �test_truncatedMessage. s �z(FilterAnswersTests.test_truncatedMessagec C s& t j|d�}| j�|�}|�|� d S )N)ZrCode)r r� rF r r� )r% Zrcode�excr �errr+ r+ r, � _rcodeTestD s z FilterAnswersTests._rcodeTestc C s | � tjtj�S )z� Test that a message with a result code of C{EFORMAT} results in a failure wrapped around L{DNSFormatError}. )r# r ZEFORMATr ZDNSFormatErrorr3 r+ r+ r, �test_formatErrorJ s z#FilterAnswersTests.test_formatErrorc C s | � tjtj�S )zP Like L{test_formatError} but for C{ESERVER}/L{DNSServerError}. )r# r ZESERVERr ZDNSServerErrorr3 r+ r+ r, �test_serverErrorR s z#FilterAnswersTests.test_serverErrorc C s | � tjtj�S )zL Like L{test_formatError} but for C{ENAME}/L{DNSNameError}. )r# r ZENAMEr Z DNSNameErrorr3 r+ r+ r, �test_nameErrorY s z!FilterAnswersTests.test_nameErrorc C s | � tjtj�S )zX Like L{test_formatError} but for C{ENOTIMP}/L{DNSNotImplementedError}. )r# r ZENOTIMPr ZDNSNotImplementedErrorr3 r+ r+ r, �test_notImplementedError` s z+FilterAnswersTests.test_notImplementedErrorc C s | � tjtj�S )zW Like L{test_formatError} but for C{EREFUSED}/L{DNSQueryRefusedError}. )r# r �EREFUSEDr ZDNSQueryRefusedErrorr3 r+ r+ r, �test_refusedErrorg s z$FilterAnswersTests.test_refusedErrorc C s | � tjd tj�S )zm Like L{test_formatError} but for an unrecognized error code and L{DNSUnknownError}. rU )r# r r( r ZDNSUnknownErrorr3 r+ r+ r, �test_refusedErrorUnknownn s z+FilterAnswersTests.test_refusedErrorUnknownN) r. r/ r0 r5 r� r r# r$ r% r& r' r) r* r+ r+ r+ r, r # s r c @ s& e Zd Zdd� Zd dd�Zdd� ZdS ) �FakeDNSDatagramProtocolc C s g | _ t� | _d S r7 r8 r3 r+ r+ r, r; x s z FakeDNSDatagramProtocol.__init__r< Nc C s$ | j �||||f� t�t�|��S r7 )r9 r> r r� r r )r% r r9 r' r? r+ r+ r, rA | s z FakeDNSDatagramProtocol.queryc C s d S r7 r+ )r% r? r+ r+ r, � removeResend� s z$FakeDNSDatagramProtocol.removeResend)r< N)r. r/ r0 r; rA r, r+ r+ r+ r, r+ w s r+ c @ s4 e Zd ZdZddddgZdd� Zdd � Zd d � Zd S ) �RetryLogicTestszE Tests for query retrying implemented by L{client.Resolver}. z1.2.3.4z4.3.2.1za.b.c.dzz.y.x.wc sN dd� | j D �}tjd|d�}t� � � fdd�|_|�d��| j��| j � �S )a When timeouts occur waiting for responses to queries, the next configured server is issued the query. When the query has been issued to all configured servers, the timeout is increased and the process begins again at the beginning. c S s g | ] }|d f�qS )rp r+ )rR �xr+ r+ r, rT � s z:RetryLogicTests.test_roundRobinBackoff.<locals>.<listcomp>N)rd rs c s � S r7 r+ r+ �r� r+ r, r� � r� z8RetryLogicTests.test_roundRobinBackoff.<locals>.<lambda>r� ) � testServersr rP r+ r� r r� �_cbRoundRobinBackoffr� �_ebRoundRobinBackoff)r% ZaddrsrS r+ r/ r, �test_roundRobinBackoff� s � �z&RetryLogicTests.test_roundRobinBackoffc C s | � d� d S )Nz/Lookup address succeeded, should have timed out)r� r� r+ r+ r, r1 � s z$RetryLogicTests._cbRoundRobinBackoffc C s� |� tj� dD ]|}|jd t| j�� }|jd t| j��= |�� t| j�}|�� t||�D ],\\}}}} } | � || df� | � ||� q^q| � |j� d S )N)rU r� � �- rp ) r� r � TimeoutErrorr9 rX r0 �sort�list�ziprW r� ) r% r Z fakeProto�tZtriesZexpectedZaddrrA r' r? Z expectedAddrr+ r+ r, r2 � s z$RetryLogicTests._ebRoundRobinBackoffN)r. r/ r0 r5 r0 r3 r1 r2 r+ r+ r+ r, r- � s �r- c @ s e Zd ZdZdd� ZdS )�ThreadedResolverTestsz/ Tests for L{client.ThreadedResolver}. c C sT t �� | j| jgd�}| �|d d d� | �|d d t� | �t|�d� dS )z� L{client.ThreadedResolver} is deprecated. Instantiating it emits a deprecation warning pointing at the code that does the instantiation. )ZoffendingFunctionsr �messagezztwisted.names.client.ThreadedResolver is deprecated since Twisted 9.0, use twisted.internet.base.ThreadedResolver instead.�categoryrU N)r ZThreadedResolverZ flushWarnings�test_deprecatedrW �DeprecationWarningrX )r% �warningsr+ r+ r, r> � s �z%ThreadedResolverTests.test_deprecatedN)r. r/ r0 r5 r> r+ r+ r+ r, r; � s r; )9r5 Z __future__r r Zzope.interface.verifyr r Ztwisted.pythonr Ztwisted.python.filepathr Ztwisted.python.runtimer Ztwisted.internetr Ztwisted.internet.errorr r Ztwisted.internet.interfacesr Z#twisted.internet.test.modulehelpersr Ztwisted.internet.taskr Z twisted.namesr r r r r Ztwisted.names.errorr Ztwisted.names.commonr Z twisted.names.test.test_hostsr Ztwisted.names.testr Z twisted.testr Z twisted.trialr Z isWindowsrL r r� r1 r6 ZTestCaserB rN ry r� r r+ r- r; r+ r+ r+ r, �<module> sH | 7 uT5
Save File
Cancel