File: //usr/lib/python3/dist-packages/twisted/names/test/__pycache__/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