← Back
Editing: test_rootresolve.cpython-38.pyc
U W[+d � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z m Z ddl m Z mZmZmZ ddlmZ dd lmZmZ dd lmZ dd lmZmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z& dd l'm(Z(m)Z) dd l*m+Z+ dd� Z,dd� Z-G dd� de �Z.G dd� de/�Z0dd� Z1G dd� de �Z2ddd d d d d d!d"d#d$d%d&g Z3ee�G d'd(� d(e4��Z5eee5� G d)d*� d*e �Z6G d+d,� d,�Z7ej8e9d-d.�Z:d/S )0z. Test cases for Twisted.names' root resolver. � )� implementer)� verifyClass)�msg)�util)�SynchronousTestCase�TestCase)�Deferred�succeed� gatherResults� TimeoutError)�IResolverSimple)�client�root)�Resolver)�IN�HS�A�NS�CNAME�OK�ENAME� Record_CNAME�Name�Query�Message�RRHeader�Record_A� Record_NS)� DNSNameError� ResolverError)� MemoryReactorc C s | \}}}|d j S )z� From the result of a L{Deferred} returned by L{IResolver.lookupAddress}, return the payload of the first record in the answer section. r ��payload��resultsZansZauth�add� r& �E/usr/lib/python3/dist-packages/twisted/names/test/test_rootresolve.py� getOnePayload s r( c C s t | ��� S )z� From the result of a L{Deferred} returned by L{IResolver.lookupAddress}, return the first IPv4 address from the answer section. )r( Z dottedQuad�r$ r& r& r' � getOneAddress# s r* c @ s� e Zd ZdZdd� Zdd� Zdd� Zg g g efdd �Zd&d d �Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%S )'�RootResolverTestsz3 Tests for L{twisted.names.root.Resolver}. c C s� t � }tg |d�}|�tdtt�dgd|�}|j�� \}}|j\\}}t � } | � |� | � | j tdtt�g� | � | j g � | � | jg � | � | jg � g } |�| j� | � | g � | j dd�= d| _| j �tdtd�d�� |j�| �� d� | d S ) aK Invoke L{Resolver._query} and verify that it sends the correct DNS query. Deliver a canned response to the query and return whatever the L{Deferred} returned by L{Resolver._query} fires with. @param filter: The value to pass for the C{filter} parameter to L{Resolver._query}. )�reactor� foo.example.com)�1.1.2.3i )� N� � 5.8.13.21r! r )r r �_queryr r r ZudpPorts�popitemZ _sentPacketsr ZfromStr� assertEqual�queries�answers� authority� additional� addCallback�append�answerr r Z _protocolZdatagramReceivedZtoStr) �self�filterr, �resolver�dZ portNumberZ transportZpacketZaddress�message�responser& r& r' � _queryTest0 s< � � �z RootResolverTests._queryTestc C sJ | � d�\}}}| �|tdtddd�d�g� | �|g � | �|g � dS )a) L{Resolver._query} accepts a L{Query} instance and an address, issues the query, and returns a L{Deferred} which fires with the response to the query. If a true value is passed for the C{filter} parameter, the result is a three-tuple of lists of records. Tr- r1 r �Zttlr! N)rB r4 r r )r< r; r7 r8 r& r& r' �test_filteredQuery\ s � z$RootResolverTests.test_filteredQueryc C sd | � d�}| �|t� | �|jg � | �|jtdtddd�d�g� | �|jg � | �|j g � dS )z� Similar to L{test_filteredQuery}, but for the case where a false value is passed for the C{filter} parameter. In this case, the result is a L{Message} instance. Fr- r1 r rC r! N) rB �assertIsInstancer r4 r5 r6 r r r7 r8 )r< r@ r& r& r' �test_unfilteredQueryk s �z&RootResolverTests.test_unfilteredQueryc C sH t |d�}|j|f|j|f|j|ffD ] \}}|�dd� |D �� q&|S )a� Create a L{Message} suitable for use as a response to a query. @param answers: A C{list} of two-tuples giving data for the answers section of the message. The first element of each tuple is a name for the L{RRHeader}. The second element is the payload. @param authority: A C{list} like C{answers}, but for the authority section of the response. @param additional: A C{list} like C{answers}, but for the additional section of the response. @param rCode: The response code the message will be created with. @return: A new L{Message} initialized with the given values. )�rCodec S s* g | ]"\}}t ||jt|d t�|d��qS )�CLASSr! )r ZTYPE�getattrr )�.0�name�recordr& r& r' � <listcomp>� s ��z.RootResolverTests._respond.<locals>.<listcomp>)r r6 r7 r8 �extend)r< r6 r7 r8 rG rA Zsection�datar& r& r' �_respond{ s � � zRootResolverTests._respond� c s( dg}t ||�}� �fdd�}||_|S )a� Create and return a new L{root.Resolver} modified to resolve queries against the record data represented by C{servers}. @param serverResponses: A mapping from dns server addresses to mappings. The inner mappings are from query two-tuples (name, type) to dictionaries suitable for use as **arguments to L{_respond}. See that method for details. r. c sj t d| j|f � |D ]N}z �| }W n tk r> Y qY nX || jj| jf }t� jf |�� S d S )NzQuery for QNAME %s at %r)r rK �KeyError�typer rP )�queryZserverAddressesZtimeoutr= ZaddrZserverZrecords�r< �serverResponsesr& r' rT � s z-RootResolverTests._getResolver.<locals>.query)r r2 )r< rV ZmaximumQueries�rootsr> rT r& rU r' � _getResolver� s z RootResolverTests._getResolverc C sr dt fdtd�fgdtd�fgd�idt fddtd�fgiid�}| �|�}|�d�}|�t� |�| jd� |S )a L{root.Resolver.lookupAddress} looks up the I{A} records for the specified hostname by first querying one of the root servers the resolver was created with and then following the authority delegations until a result is received. r- � ns1.example.com� 34.55.89.144�r7 r8 r6 �10.0.0.1)�r. �5 )rZ r^ �r r r rX � lookupAddressr9 r* r4 �r< �serversr> r? r& r& r' �test_lookupAddress� s �� ��� z$RootResolverTests.test_lookupAddressc C s� t d�}t|_dtfd|fgdtd�fgdt d�fgd�idtfddt d�fgiid�}| �|�}|�d�}|�t� |�| j t d�� |S ) z� If a response includes a record with a class different from the one in the query, it is ignored and lookup continues until a record with the right class is found. r\ r- rY �10.0.0.2)r6 r7 r8 r6 �10.0.0.3)r] �rd r^ ) r r rH r r rX r` r9 r( r4 )r< ZbadClassrb r> r? r& r& r' �test_lookupChecksClass� s& �� ��� z(RootResolverTests.test_lookupChecksClassc C s| dt fddtd�fgidt fddtd�fgiidt fddtd�fgiid�}| �|�}|�d�}|�t� |�| jd� |S )z� If an intermediate response includes no glue records for the authorities, separate queries are made to find those addresses. r- r7 s ns1.example.orgr6 r\ rd )r] )r\ r^ r_ ra r& r& r' �test_missingGlue� s( � �� ��� z"RootResolverTests.test_missingGluec C s4 ddt fdtiii}| �|�}|�d�}| �|t�S )z� If a name is missing, L{Resolver.lookupAddress} returns a L{Deferred} which fails with L{DNSNameError}. r] r- rG )r r rX r` � assertFailurer ra r& r& r' �test_missingName s ��� z"RootResolverTests.test_missingNamec C s0 ddt fi ii}| �|�}|�d�}| �|t�S )z� If a query is responded to with no answers or nameserver records, the L{Deferred} returned by L{Resolver.lookupAddress} fires with L{ResolverError}. r] � example.com)r rX r` ri r ra r& r& r' �test_answerless s �� z!RootResolverTests.test_answerlessc C sJ ddt fddtd�fgidt fdtiii}| �|�}|�d�}| �|t�S )z� If there is an error resolving the nameserver in a delegation response, the L{Deferred} returned by L{Resolver.lookupAddress} fires with that error. r] rk r7 rY rG )r r r rX r` ri r ra r& r& r' �test_delegationLookupError% s � ��� z,RootResolverTests.test_delegationLookupErrorc C sF ddt fddtd�fgidt fi ii}| �|�}|�d�}| �|t�S )z� If there are no records in the response to a lookup of a delegation nameserver, the L{Deferred} returned by L{Resolver.lookupAddress} fires with L{ResolverError}. r] rk r7 rY )r r rX r` ri r ra r& r& r' �test_delegationLookupEmpty: s � �� z,RootResolverTests.test_delegationLookupEmptyc C sf ddt fdtidtfddtd�fgiii}| �|�}|�d�}dd� }|�|� |�| jtd�� |S )z� L{Resolver.lookupNameservers} is like L{Resolver.lookupAddress}, except it queries for I{NS} records instead of I{A} records. r] rk rG r6 rY c S s | \}}}|d j jS �Nr )r" rK r# r& r& r' � getOneName_ s z<RootResolverTests.test_lookupNameservers.<locals>.getOneName) r r r r rX ZlookupNameserversr9 r4 r )r< rb r> r? rp r& r& r' �test_lookupNameserversN s" � ��� z(RootResolverTests.test_lookupNameserversc C s| ddt fddtd�fdtd�fgiii}| �|�}|�d�}|�dd� � |�| jtdttd�d�tdt td�d�g� |S ) z� If a I{CNAME} record is encountered as the answer to a query for another record type, that record is returned as the answer. r] rk r6 � example.netz10.0.0.7c S s | d S ro r& r) r& r& r' �<lambda>v � z<RootResolverTests.test_returnCanonicalName.<locals>.<lambda>r! � r r r rX r` r9 r4 r r ra r& r& r' �test_returnCanonicalNameg s& ���� ��z*RootResolverTests.test_returnCanonicalNamec C s� ddt fddtd�fgidt fddtd�fgiii}| �|�}|�d�}|�dd� � |�| jtdttd�d�tdt td�d�g� |S ) z� If no record of the requested type is included in a response, but a I{CNAME} record for the query name is included, queries are made to resolve the value of the I{CNAME}. r] rk r6 rr z10.0.0.5c S s | d S ro r& r) r& r& r' rs � rt z<RootResolverTests.test_followCanonicalName.<locals>.<lambda>r! ru ra r& r& r' �test_followCanonicalName~ s* � ��� ��z*RootResolverTests.test_followCanonicalNamec C sH ddt fddtd�fdtd�fgiii}| �|�}|�d�}| �|t�S )z� If there is a cycle between I{CNAME} records in a response, this is detected and the L{Deferred} returned by the lookup method fails with L{ResolverError}. r] rk r6 rr )r r rX r` ri r ra r& r& r' � test_detectCanonicalNameLoop� s ���� z.RootResolverTests.test_detectCanonicalNameLoopc C s� dt fddtd�fgidt fddtd�fgiidt fdtd�fgdtd�fgd�idt fddtd �fgiid �}| �|d �}| �|�d�t�}| �|d �}|�d�}|�t� |�| j td �� t ||g�S ) z� L{Resolver.lookupAddress} won't issue more queries following delegations than the limit passed to its initializer. rk r7 rY r6 rd s ns2.example.comre r[ z10.0.0.4)r] rf )re r^ � � ) r r r rX ri r` r r9 r( r4 r )r< rb ZfailerZfailDZ succeederZsucceedDr& r& r' �test_boundedQueries� s< � �� �� ���! � z%RootResolverTests.test_boundedQueriesN)rQ )�__name__� __module__� __qualname__�__doc__rB rD rF r rP rX rc rg rh rj rl rm rn rq rv rw rx r{ r& r& r& r' r+ , s$ , r+ c @ s e Zd ZdZdd� ZdS )�ResolverFactoryArgumentszf Raised by L{raisingResolverFactory} with the *args and **kwargs passed to that function. c C s || _ || _dS )z� Store the supplied args and kwargs as attributes. @param args: Positional arguments. @param kwargs: Keyword arguments. N��args�kwargs)r< r� r� r& r& r' �__init__� s z!ResolverFactoryArguments.__init__N)r| r} r~ r r� r& r& r& r' r� � s r� c O s t | |��dS )a= Raise a L{ResolverFactoryArguments} exception containing the positional and keyword arguments passed to resolverFactory. @param args: A L{list} of all the positional arguments supplied by the caller. @param kwargs: A L{list} of all the keyword arguments supplied by the caller. N)r� r� r& r& r' �raisingResolverFactory� s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � RootResolverResolverFactoryTestsz6 Tests for L{root.Resolver._resolverFactory}. c C s t dgtd�}| �|jt� dS )z� L{root.Resolver.__init__} accepts a C{resolverFactory} argument and assigns it to C{self._resolverFactory}. N)�hints�resolverFactory)r r� �assertIs�_resolverFactory�r< �rr& r& r' �#test_resolverFactoryArgumentPresent s zDRootResolverResolverFactoryTests.test_resolverFactoryArgumentPresentc C s t dgd�}| �|jtj � dS )z� L{root.Resolver.__init__} sets L{client.Resolver} as the C{_resolverFactory} if a C{resolverFactory} argument is not supplied. N)r� )r r� r� r r� r& r& r' �"test_resolverFactoryArgumentAbsent s zCRootResolverResolverFactoryTests.test_resolverFactoryArgumentAbsentc C sJ t � }tdgt|d�}| �t|jd�}| �d|dgd�f|j|jf� dS )zy L{root.Resolver._resolverFactory} is supplied with C{reactor} and C{servers} keyword arguments. � 192.0.2.101)r� r� r, z example.comr& )r� r^ )r, rb N) �objectr r� Z assertRaisesr� r` r4 r� r� )r< Z dummyReactorr� �er& r& r' �)test_resolverFactoryOnlyExpectedArguments s � � �zJRootResolverResolverFactoryTests.test_resolverFactoryOnlyExpectedArgumentsN)r| r} r~ r r� r� r� r& r& r& r' r� � s r� za.root-servers.netzb.root-servers.netzc.root-servers.netzd.root-servers.netze.root-servers.netzf.root-servers.netzg.root-servers.netzh.root-servers.netzi.root-servers.netzj.root-servers.netzk.root-servers.netzl.root-servers.netzm.root-servers.netc @ s e Zd ZdZdd� Zdd� ZdS )� StubResolverz� An L{IResolverSimple} implementer which traces all getHostByName calls and their deferred results. The deferred results can be accessed and fired synchronously. c C s g | _ g | _dS )z� @type calls: L{list} of L{tuple} containing C{args} and C{kwargs} supplied to C{getHostByName} calls. @type pendingResults: L{list} of L{Deferred} returned by C{getHostByName}. N)�calls�pendingResults)r< r& r& r' r� A s zStubResolver.__init__c O s& | j �||f� t� }| j�|� |S )al A fake implementation of L{IResolverSimple.getHostByName} @param args: A L{list} of all the positional arguments supplied by the caller. @param kwargs: A L{list} of all the keyword arguments supplied by the caller. @return: A L{Deferred} which may be fired later from the test fixture. )r� r: r r� )r< r� r� r? r& r& r' � getHostByNameL s zStubResolver.getHostByNameN)r| r} r~ r r� r� r& r& r& r' r� : s r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�BootstrapTestsz% Tests for L{root.bootstrap} c C s t �t� �}| �|t j� dS )zl L{root.bootstrap} returns an object which is initially a L{root.DeferredResolver}. N)r � bootstrapr� rE ZDeferredResolver)r< �deferredResolverr& r& r' � test_returnsDeferredResolverh s z+BootstrapTests.test_returnsDeferredResolverc C s, t � }t�|� | �|jdd� tD �� dS )z� The L{IResolverSimple} supplied to L{root.bootstrap} is used to lookup the IP addresses of the 13 root name servers. c S s g | ]}|fi f�qS r& r& )rJ �sr& r& r' rM z s z=BootstrapTests.test_resolves13RootServers.<locals>.<listcomp>N)r� r r� r4 r� � ROOT_SERVERS)r< � stubResolverr& r& r' �test_resolves13RootServersq s �z)BootstrapTests.test_resolves13RootServersc C s6 t � }t�|�}|jD ]}|�d� q| �|t� dS )z� The L{root.DeferredResolver} initially returned by L{root.bootstrap} becomes a L{root.Resolver} when the supplied resolver has successfully looked up all root hints. r� N)r� r r� r� �callbackrE r �r< r� r� r? r& r& r' �test_becomesResolver} s z#BootstrapTests.test_becomesResolverc C s> t � }t�|�}|jD ]}|�d� q| �|jdgd � dS )z� The L{root.Resolver} which eventually replaces L{root.DeferredResolver} is supplied with the IP addresses of the 13 root servers. r� � N)r� r r� r� r� r4 r� r� r& r& r' � test_resolverReceivesRootHints� s z-BootstrapTests.test_resolverReceivesRootHintsc s^ t � }t�|�� t|j�}t|�}|D ]}|�d� q&|�t� � � �fdd�}|� |� dS )z� The L{root.Resolver} is eventually created, even if some of the root hint lookups fail. Only the working root hint IP addresses are supplied to the L{root.Resolver}. r� c s �� � jdgd � d S )Nr� � �r4 r� ��res�r� r< r& r' � checkHints� s zFBootstrapTests.test_continuesWhenSomeRootHintsFail.<locals>.checkHintsN) r� r r� �iterr� �nextr� �errbackr �addBoth�r< r� r$ Zd1r? r� r& r� r' �#test_continuesWhenSomeRootHintsFail� s z2BootstrapTests.test_continuesWhenSomeRootHintsFailc sn t � }t�|�� t|j�}t|�}|D ]}|�t� � q&|�t� � � �fdd�}|�|� �� �j t� dS )z� The L{root.Resolver} is eventually created, even if all of the root hint lookups fail. Pending and new lookups will then fail with AttributeError. c s �� � jg � d S �Nr� r� r� r& r' r� � s zEBootstrapTests.test_continuesWhenAllRootHintsFail.<locals>.checkHintsN) r� r r� r� r� r� r� r r� Z addCleanupZflushLoggedErrorsr� r& r� r' �"test_continuesWhenAllRootHintsFail� s z1BootstrapTests.test_continuesWhenAllRootHintsFailc C s< t � }tj|td�}|jD ]}|�d� q| �|jt� dS )z� L{root.bootstrap} accepts a C{resolverFactory} argument which is passed as an argument to L{root.Resolver} when it has successfully looked up root hints. )r� r� N)r� r r� r� r� r� r� r� r� r& r& r' �test_passesResolverFactory� s � �z)BootstrapTests.test_passesResolverFactoryN) r| r} r~ r r� r� r� r� r� r� r� r& r& r& r' r� d s r� c @ s e Zd ZdZdd� ZdS )�StubDNSDatagramProtocolz� A do-nothing stand-in for L{DNSDatagramProtocol} which can be used to avoid network traffic in tests where that kind of thing doesn't matter. c O s t � S r� )r )r< �a�kwr& r& r' rT � s z StubDNSDatagramProtocol.queryN)r| r} r~ r rT r& r& r& r' r� � s r� zbtwisted.names.root.retry is deprecated since Twisted 10.0. Use a Resolver object for retry logic.)�categoryr@ N);r Zzope.interfacer Zzope.interface.verifyr Ztwisted.python.logr Z twisted.trialr Ztwisted.trial.unittestr r Ztwisted.internet.deferr r r r Ztwisted.internet.interfacesr Z twisted.namesr r Ztwisted.names.rootr Ztwisted.names.dnsr r r r r r r r r r r r r r Ztwisted.names.errorr r Z twisted.names.test.test_utilr r( r* r+ � Exceptionr� r� r� r� r� r� r� r� �suppress�DeprecationWarningZ_retrySuppressionr& r& r& r' �<module> sX @ 5,�% l �
Save File
Cancel