← Back
Editing: test_srvconnect.cpython-38.pyc
U W[�' � @ s� d Z ddlmZmZ ddlZddlmZ ddlmZ ddl m Z m Z ddl m Z mZ ddlmZ dd lmZmZmZ dd lmZ dd lmZ dd lmZ dd lm Z G dd� de�Z G dd� de j �Z G dd� dej �Z!dS )z- Test cases for L{twisted.names.srvconnect}. � )�absolute_import�divisionN)� verifyObject)�unittest)�defer�protocol)�DNSLookupError�ServiceNameUnknownError)� IConnector)�client�dns� srvconnect)� ResolverBase)� DNSNameError)� nativeString)� MemoryReactorc @ s" e Zd ZdZddd�Zdd� ZdS )� FakeResolvera� Resolver that only gives out one given result. Either L{results} or L{failure} must be set and will be used for the return value of L{_lookup} @ivar results: List of L{dns.RRHeader} for the desired result. @type results: C{list} @ivar failure: Failure with an exception from L{twisted.names.error}. @type failure: L{Failure<twisted.python.failure.Failure>} Nc C s || _ || _g | _d S �N)�results�failure�lookups)�selfr r � r �D/usr/lib/python3/dist-packages/twisted/names/test/test_srvconnect.py�__init__' s zFakeResolver.__init__c C s@ | j �||||f� | jdk r0t�| jg g f�S t�| j�S dS )z9 Return the result or failure on lookup. N)r �appendr r ZsucceedZfailr )r �name�clsZqtypeZtimeoutr r r �_lookup, s zFakeResolver._lookup)NN)�__name__� __module__� __qualname__�__doc__r r r r r r r s r c @ s e Zd ZdZdd� Zdd� ZdS )� DummyFactoryzL Dummy client factory that stores the reason of connection failure. c C s d | _ d S r ��reason�r r r r r < s zDummyFactory.__init__c C s || _ d S r r$ )r � connectorr% r r r �clientConnectionFailed? s z#DummyFactory.clientConnectionFailedN)r r r! r" r r( r r r r r# 8 s r# 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 S )!�SRVConnectorTestsz/ Tests for L{srvconnect.SRVConnector}. c C sF | � tdt� � t� | _t� | _t�| jdd| j�| _ g | _ g | _ d S )N� theResolver� xmpp-server� example.org) �patchr r r �reactorr# �factoryr � SRVConnectorr' � randIntArgs�randIntResultsr& r r r �setUpI s �zSRVConnectorTests.setUpc C s | j �||f� | j�d�S )a� Fake randint. Returns the first element of L{randIntResults} and records the arguments passed to it in L{randIntArgs}. @param min: Lower bound of the random number. @type min: L{int} @param max: Higher bound of the random number. @type max: L{int} @return: Fake random number from L{randIntResults}. @rtype: L{int} r )r1 r r2 �pop)r �min�maxr r r �_randintS s zSRVConnectorTests._randintc C s t t| j� dS )zF L{srvconnect.SRVConnector} implements L{IConnector}. N)r r r' r& r r r �test_interfaceg s z SRVConnectorTests.test_interfacec C sf t jdddd�}t jdt jt jd|d�gtj_| j� � | � | j j � | � | jj�� dd� d � dS ) zS Test connectTCP gets called with the address from the SRV record. �} �host.example.org�< �Zport�target�ttlr, �r �typer r> �payloadN� )r: r9 )r � Record_SRV�RRHeader�SRV�INr r* r r' �connect� assertIsNoner/ r% � assertEqualr. � tcpClientsr4 �r rA r r r �test_SRVPresentn s � �z!SRVConnectorTests.test_SRVPresentc C sD t d�tj_| j�� | �| jj� | � | j j � � dd� d� dS )zS Test connectTCP gets called with fallback parameters on NXDOMAIN. r, NrB �r, r+ ) r r r* r r' rG rH r/ r% rI r. rJ r4 r& r r r �test_SRVNotPresent~ s �z$SRVConnectorTests.test_SRVNotPresentc C s@ g t j_| j�� | �| jj� | �| j j � � dd� d� dS )zW Test connectTCP gets called with fallback parameters on empty result. NrB rM ) r r* r r' rG rH r/ r% rI r. rJ r4 r& r r r �test_SRVNoResult� s �z"SRVConnectorTests.test_SRVNoResultc C s\ t j| jdd| jdd�| _t� tj_| j� � | � | jj � | � | jj �� dd� d� dS )zY connectTCP gets called with default port if the service is not defined. �thisbetternotexistr, �f )Z defaultPortNrB )r, rQ )r r0 r. r/ r' r r r* r rG rH r% rI rJ r4 r& r r r �)test_SRVNoResultUnknownServiceDefaultPort� s � �z;SRVConnectorTests.test_SRVNoResultUnknownServiceDefaultPortc C sB t �| jdd| j�| _t� tj_| j� � | � | jj � t�� dS )zR Connect fails on no result, unknown service and no default port. rP r, N) r r0 r. r/ r' r r r* r rG Z assertTruer% Zcheckr& r r r �+test_SRVNoResultUnknownServiceNoDefaultPort� s � z=SRVConnectorTests.test_SRVNoResultUnknownServiceNoDefaultPortc C sV t jdt jt jddd�gtj_| j�� | � | j j � | � | j j�� dd� d� dS )zU Test connectTCP gets called with fallback parameters on bad result. r, r; Nr? rB rM )r rD ZCNAMErF r r* r r' rG rH r/ r% rI r. rJ r4 r& r r r �test_SRVBadResult� s � �z#SRVConnectorTests.test_SRVBadResultc C sh t jdddd�}t jdt jt jd|d�gtj_| j� � | � | j j � | j j � t� | �| jjg � dS )zH Test that connecting fails when no service is present. i� � .r; r< r, r? N)r rC rD rE rF r r* r r' rG ZassertIsNotNoner/ r% Ztrapr rI r. rJ rK r r r �test_SRVNoService� s � z#SRVConnectorTests.test_SRVNoServicec C s6 g t j_| j�� t jjd d }| �td�|� dS )zW The lookup name is a native string from service, protocol and domain. ���r z _xmpp-server._tcp.example.orgN)r r* r r' rG r rI r )r r r r r �test_SRVLookupName� s z$SRVConnectorTests.test_SRVLookupNamec C s* t �| jdd| j�| _| �d| jj� dS )zq L{srvconnect.SRVConnector} automatically encodes unicode domain using C{idna} encoding. z xmpp-clientu échec.example.orgzxn--chec-9oa.example.orgN)r r0 r. r/ r' rI Zdomainr& r r r �test_unicodeDomain� s �z$SRVConnectorTests.test_unicodeDomainc C s� t �dddd�}t �dddd�}||g| j_g | j_| �td| j� ddg| _| j� � | � | j d d � | j� � | � | j d d � ddg| _| j� � | � | j d d � | j� � | � | j d d� dS )a pickServer calculates running sum of weights and calls randint. This exercises the server selection algorithm specified in RFC 2782 by preparing fake L{random.randint} results and checking the values it was called with. � rQ �host1.example.org� �host2.example.org�randint� r )r � � )r rZ rB � )r r\ N) r rC r' �orderedServers�serversr- �randomr7 r2 � pickServerrI r1 �r Zrecord1Zrecord2r r r �test_pickServerWeights� s z(SRVConnectorTests.test_pickServerWeightsc C sv t �dddd�}t �dddd�}||g| j_g | j_| �td| j� ddg| _| � d| j� � � | � d | j� � � d S ) zR Two records with equal priorities compare on weight (ascending). rZ rQ r[ r\ r] r^ r �r[ rQ �r] rQ N� r rC r' rc rd r- re r7 r2 rI rf rg r r r � test_pickServerSamePriorities s ��z/SRVConnectorTests.test_pickServerSamePrioritiesc C sv t �dddd�}t �dddd�}||g| j_g | j_| �td| j� ddg| _| � d| j� � � | � d | j� � � d S ) zX Two records with differing priorities compare on priority (ascending). rZ r rQ r[ r\ r] r^ ri rj Nrk rg r r r �test_srvDifferentPriorities s ��z-SRVConnectorTests.test_srvDifferentPrioritiesN)r r r! r" r3 r7 r8 rL rN rO rR rS rT rV rX rY rh rl rm r r r r r) D s "r) )"r" Z __future__r r re Zzope.interface.verifyr Z twisted.trialr Ztwisted.internetr r Ztwisted.internet.errorr r Ztwisted.internet.interfacesr Z twisted.namesr r r Ztwisted.names.commonr Ztwisted.names.errorr Ztwisted.python.compatr Ztwisted.test.proto_helpersr r Z ClientFactoryr# ZTestCaser) r r r r �<module> s
Save File
Cancel