File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_loopback.cpython-38.pyc
U
W[�8 � @ s d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z m Z ddl
mZ ddlmZ dd lmZ dd
lmZmZmZ dd l
mZmZ G d d
� d
e j�ZG dd� de�ZG dd� d�Z G dd� de e j �Z G dd� de e j �Z G dd� de e j �Z G dd� de j �Z!dS )z.
Test case for L{twisted.protocols.loopback}.
� )�division�absolute_import)� implementer)�
intToBytes)�unittest)�basic�loopback)�defer)�Protocol)�Deferred)�IAddress�
IPushProducer�
IPullProducer)�reactor�
interfacesc @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�SimpleProtocolc C s t �� | _g | _g | _d S �N)r r �conn�lines�connLost��self� r �</usr/lib/python3/dist-packages/twisted/test/test_loopback.py�__init__ s
zSimpleProtocol.__init__c C s | j �d � d S r )r �callbackr r r r �connectionMade s z SimpleProtocol.connectionMadec C s | j �|� d S r )r �append�r �liner r r � lineReceived! s zSimpleProtocol.lineReceivedc C s | j �|� d S r )r r )r �reasonr r r �connectionLost% s z SimpleProtocol.connectionLostN)�__name__�
__module__� __qualname__r r r r" r r r r r s r c @ s e Zd ZdZdd� ZdS )� DoomProtocolr c C sT | j d7 _ | j dk r,| �dt| j � � t�| |� | jd dkrP| j�� d S )N� � s Hello ���� Hello 3)�i�sendLiner r r r � transport�loseConnectionr r r r r , s
zDoomProtocol.lineReceivedN)r# r$ r% r+ r r r r r r& * s r& c @ s e Zd Zdd� Zdd� ZdS )�LoopbackTestCaseMixinc sR t � �t � � �fdd�}�j�|� � ��fdd�}t��j�� �}|�|� |S )Nc s � � d� � j�� d S )N� THIS IS LINE ONE!)r, r- r. ��result��sr r � sendALine= s
z<LoopbackTestCaseMixin.testRegularFunction.<locals>.sendALinec s8 �� � jdg� �� t�j�d� �� t� j�d� d S )Nr0 r' �� assertEqualr �lenr �Zignored��cr4 r r r �checkB s z8LoopbackTestCaseMixin.testRegularFunction.<locals>.check)r r � addCallbackr �
maybeDeferred� loopbackFunc�r r5 r< �dr r: r �testRegularFunction9 s
z)LoopbackTestCaseMixin.testRegularFunctionc sR t � �t � � �fdd�}�j�|� � ��fdd�}t��j�� �}|�|� |S )Nc s � � d� d S )N� DOOM LINE)r, r1 r3 r r r5 O s z=LoopbackTestCaseMixin.testSneakyHiddenDoom.<locals>.sendALinec sR �� �jdddg� �� � jddddg� �� t�j�d� �� t� j�d� d S )Ns Hello 1s Hello 2r* rC r' r6 r9 r: r r r< S s
�z9LoopbackTestCaseMixin.testSneakyHiddenDoom.<locals>.check)r& r r= r r> r? r@ r r: r �testSneakyHiddenDoomK s
z*LoopbackTestCaseMixin.testSneakyHiddenDoomN)r# r$ r% rB rD r r r r r/ 8 s r/ c @ s� e Zd Zeej�Zdd� Zdd� Zdd� Z dd� Z
d(d
d �Z d d
� Z dd� Z
dd� Zdd� Zd)dd�Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'S )*�LoopbackAsyncTestsc C sD G dd� dt �}|� }|� }t�||� | �|j� | �|j� dS )z}
Test that the client and server protocol both have makeConnection
invoked on them by loopbackAsync.
c @ s e Zd ZdZdd� ZdS )z<LoopbackAsyncTests.test_makeConnection.<locals>.TestProtocolNc S s
|| _ d S r )r- �r r- r r r �makeConnectionj s zKLoopbackAsyncTests.test_makeConnection.<locals>.TestProtocol.makeConnection)r# r$ r% r- rG r r r r � TestProtocolh s rH N)r
r �
loopbackAsyncZassertIsNotNoner- )r rH �server�clientr r r �test_makeConnectionc s z&LoopbackAsyncTests.test_makeConnectionc sj G dd� dt �}|r.|� }t� }|_t � }nt � }|� }t� }|_t�||� � �fdd�}|�|�S )zJ
Test one of the permutations of client/server host/peer.
c @ s e Zd Zdd� ZdS )z6LoopbackAsyncTests._hostpeertest.<locals>.TestProtocolc S s t �| |� | j�|� d S r )r
rG � onConnectionr rF r r r rG y s zELoopbackAsyncTests._hostpeertest.<locals>.TestProtocol.makeConnectionN)r# r$ r% rG r r r r rH x s rH c s t | � �� }��t�|�� d S r )�getattr�
assertTruer Z
providedBy)r- Zhost��getr r r � connected� s z3LoopbackAsyncTests._hostpeertest.<locals>.connected)r
r rM r rI r= )r rQ �
testServerrH rJ rA rK rR r rP r �
_hostpeertestt s z LoopbackAsyncTests._hostpeertestc C s | � dd�S )z�
Test that the server gets a transport with a properly functioning
implementation of L{ITransport.getHost}.
�getHostT�rT r r r r �test_serverHost� s z"LoopbackAsyncTests.test_serverHostc C s | � dd�S )zG
Like C{test_serverHost} but for L{ITransport.getPeer}
�getPeerTrV r r r r �test_serverPeer� s z"LoopbackAsyncTests.test_serverPeerrU c C s | � dd�S )z�
Test that the client gets a transport with a properly functioning
implementation of L{ITransport.getHost}.
rU FrV )r rQ r r r �test_clientHost� s z"LoopbackAsyncTests.test_clientHostc C s | � dd�S )zH
Like C{test_clientHost} but for L{ITransport.getPeer}.
rX FrV r r r r �test_clientPeer� s z"LoopbackAsyncTests.test_clientPeerc sj G dd� dt �}G � fdd�dt �}|rB|� }|� }t� }|_n|� }t� }|_|� }t�||� |S )z�
Test one of the permutations of write/writeSequence client/server.
@param write: The name of the method to test, C{"write"} or
C{"writeSequence"}.
c @ s e Zd ZdZdd� ZdS )z9LoopbackAsyncTests._greetingtest.<locals>.GreeteeProtocol� c S s( | j |7 _ | j dkr$| j�d � d S �N� bytes)�bytes�receivedr �r r_ r r r � dataReceived� s
zFLoopbackAsyncTests._greetingtest.<locals>.GreeteeProtocol.dataReceivedN�r# r$ r% r_ rb r r r r �GreeteeProtocol� s rd c s e Zd Z� fdd�ZdS )z9LoopbackAsyncTests._greetingtest.<locals>.GreeterProtocolc s* � dkr| j �d� n| j �ddg� d S )N�writer^ s byts es)r- re �
writeSequencer �re r r r � s zHLoopbackAsyncTests._greetingtest.<locals>.GreeterProtocol.connectionMadeN�r# r$ r% r r rg r r �GreeterProtocol� s ri )r
r r` r rI )r re rS rd ri rJ rK rA r rg r �
_greetingtest� s z LoopbackAsyncTests._greetingtestc C s | � dd�S )z�
Test that on a connection where the client speaks first, the server
receives the bytes sent by the client.
re F�rj r r r r �test_clientGreeting� s z&LoopbackAsyncTests.test_clientGreetingc C s | � dd�S )zz
Like C{test_clientGreeting}, but use C{writeSequence} instead of
C{write} to issue the greeting.
rf Frk r r r r �test_clientGreetingSequence� s z.LoopbackAsyncTests.test_clientGreetingSequencere c C s | � dd�S )z�
Test that on a connection where the server speaks first, the client
receives the bytes sent by the server.
re Trk )r re r r r �test_serverGreeting� s z&LoopbackAsyncTests.test_serverGreetingc C s | � dd�S )zz
Like C{test_serverGreeting}, but use C{writeSequence} instead of
C{write} to issue the greeting.
rf Trk r r r r �test_serverGreetingSequence� s z.LoopbackAsyncTests.test_serverGreetingSequencec sh t tttdd����G ��fdd�dt�}G � ��fdd�dt�}|� �|� � t� � _t��� � � jS )Nr �
c s e Zd Z� �fdd�ZdS )z;LoopbackAsyncTests._producertest.<locals>.ProducingProtocolc s � t ���| _| j�| j� d S r )�list�producer�startr- r ��
producerClass� toProducer r r � s zJLoopbackAsyncTests._producertest.<locals>.ProducingProtocol.connectionMadeNrh r rt r r �ProducingProtocol� s rw c s e Zd ZdZ� ��fdd�ZdS )z;LoopbackAsyncTests._producertest.<locals>.ReceivingProtocolr\ c s2 | j |7 _ | j d���kr.| j�� �f� d S )Nr\ )r_ �joinr` r �r �data�rK rJ rv r r rb � s zHLoopbackAsyncTests._producertest.<locals>.ReceivingProtocol.dataReceivedNrc r r{ r r �ReceivingProtocol� s r| ) rq �mapr �ranger
r r` r rI )r ru rw r| r )rK ru rJ rv r �
_producertest� s z LoopbackAsyncTests._producertestc s< t t�G dd� dt��}� �|�}� fdd�}|�|� |S )zO
Test a push producer registered against a loopback transport.
c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) z:LoopbackAsyncTests.test_pushProducer.<locals>.PushProducerFc S s
|| _ d S r �rv �r rv r r r r s zCLoopbackAsyncTests.test_pushProducer.<locals>.PushProducer.__init__c S s
d| _ d S �NT)�resumedr r r r �resumeProducing s zJLoopbackAsyncTests.test_pushProducer.<locals>.PushProducer.resumeProducingc S s || _ |�| d� | �� d S r� )�consumer�registerProducer�_produceAndSchedule�r r� r r r rs s z@LoopbackAsyncTests.test_pushProducer.<locals>.PushProducer.startc S s8 | j r*| j�| j �d�� t�d| j� n
| j�� d S �Nr )rv r� re �popr � callLaterr� �unregisterProducerr r r r r� s zNLoopbackAsyncTests.test_pushProducer.<locals>.PushProducer._produceAndScheduleN)r# r$ r% r� r r� rs r� r r r r � PushProducer
s
r� c s | \}}� � |jjd� d S )Nz0Streaming producer should not have been resumed.)Z assertFalserr r� )ZresultsrK rJ r r r �finished! s
�z6LoopbackAsyncTests.test_pushProducer.<locals>.finished)r r
�objectr r= )r r� rA r� r r r �test_pushProducer s
z$LoopbackAsyncTests.test_pushProducerc C s" t t�G dd� dt��}| �|�S )zO
Test a pull producer registered against a loopback transport.
c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z:LoopbackAsyncTests.test_pullProducer.<locals>.PullProducerc S s
|| _ d S r r� r� r r r r 0 s zCLoopbackAsyncTests.test_pullProducer.<locals>.PullProducer.__init__c S s || _ | j �| d� d S )NF)r� r� r� r r r rs 3 s z@LoopbackAsyncTests.test_pullProducer.<locals>.PullProducer.startc S s( | j �| j�d�� | js$| j �� d S r� )r� re rv r� r� r r r r r� 7 s zJLoopbackAsyncTests.test_pullProducer.<locals>.PullProducer.resumeProducingN)r# r$ r% r rs r� r r r r � PullProducer. s r� )r r r� r )r r� r r r �test_pullProducer* s z$LoopbackAsyncTests.test_pullProducerc sT G dd� dt �}G dd� dt �}|� }|� � t�� |�}� �fdd�}|�|� |S )z�
L{loopback.loopbackAsync} does not call a protocol's C{dataReceived}
method while that protocol's transport's C{write} method is higher up
on the stack.
c @ s e Zd Zdd� ZdS )z9LoopbackAsyncTests.test_writeNotReentrant.<locals>.Serverc S s | j �d� d S r] )r- re ra r r r rb E s zFLoopbackAsyncTests.test_writeNotReentrant.<locals>.Server.dataReceivedN)r# r$ r% rb r r r r �ServerD s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) z9LoopbackAsyncTests.test_writeNotReentrant.<locals>.ClientFc S s t �d| j� d S r� )r r� �gor r r r r K s zHLoopbackAsyncTests.test_writeNotReentrant.<locals>.Client.connectionMadec S s | j �d� d| _d S )N� fooT)r- re �readyr r r r r� N s z<LoopbackAsyncTests.test_writeNotReentrant.<locals>.Client.goc S s | j | _| j�� d S r )r� �wasReadyr- r. ra r r r rb R s zFLoopbackAsyncTests.test_writeNotReentrant.<locals>.Client.dataReceivedN)r# r$ r% r� r r� rb r r r r �ClientH s r� c s �� � j� d S r )rO r� r9 �rK r r r �
cbFinishedY s z=LoopbackAsyncTests.test_writeNotReentrant.<locals>.cbFinished)r
r rI r= )r r� r� rJ rA r� r r� r �test_writeNotReentrant> s
z)LoopbackAsyncTests.test_writeNotReentrantc s� g ��fdd�}t � � t � �t��� |�}���g � � j�d� � j�d� �j�d� �j�d� �j�� � ���fdd�}|�|� |S ) z�
The callable passed as the value for the C{pumpPolicy} parameter to
L{loopbackAsync} is called with a L{_LoopbackQueue} of pending bytes
and a protocol to which they should be delivered.
c s* g }| r|� | �� � q� � ||f� d S r )r rQ )�queue�targetr_ )� pumpCallsr r � dummyPolicyf s z7LoopbackAsyncTests.test_pumpPolicy.<locals>.dummyPolicyr� � bar� baz� quuxc s& �� �� ddd gf�ddgfg� d S )Nr� r� r� r� )r7 r9 �rK r� r rJ r r �
cbCompletex s
��z6LoopbackAsyncTests.test_pumpPolicy.<locals>.cbComplete)r
r rI r7 r- re r. r= )r r� r� r� r r� r �test_pumpPolicy_ s
z"LoopbackAsyncTests.test_pumpPolicyc C sX g }t � }|j|_t�� }|�d� |�d� |�d� t�||� | �|ddg� dS )z�
L{identityPumpPolicy} is a pump policy which calls the target's
C{dataReceived} method one for each string in the queue passed to it.
r� r� N)r
r rb r �_LoopbackQueue�putZidentityPumpPolicyr7 �r r_ rK r� r r r �test_identityPumpPolicy� s
z*LoopbackAsyncTests.test_identityPumpPolicyc C sV g }t � }|j|_t�� }|�d� |�d� |�d� t�||� | �|dg� dS )z�
L{collapsingPumpPolicy} is a pump policy which calls the target's
C{dataReceived} only once with all of the strings in the queue passed
to it joined together.
r� r� Ns foobar)r
r rb r r� r� ZcollapsingPumpPolicyr7 r� r r r �test_collapsingPumpPolicy� s
z,LoopbackAsyncTests.test_collapsingPumpPolicyN)rU )re )r# r$ r% � staticmethodr rI r? rL rT rW rY rZ r[ rj rl rm rn ro r r� r� r� r� r� r� r r r r rE _ s&
"
$!$rE c @ s e Zd Zeej�ZdS )�LoopbackTCPTestsN)r# r$ r% r� r Z loopbackTCPr? r r r r r� � s r� c @ s* e Zd Zeej�Ze�e d�dkr&dZ
dS )�LoopbackUNIXTestsNz-Current reactor does not support UNIX sockets) r# r$ r% r� r Z loopbackUNIXr? r Z IReactorUNIXr �skipr r r r r� � s
r� c @ s0 e Zd ZdZG dd� de�Zdd� Zdd� ZdS ) �LoopbackRelayTestz>
Test for L{twisted.protocols.loopback.LoopbackRelay}
c @ s e Zd ZdZdZdd� ZdS )zLoopbackRelayTest.ReceiverzF
Simple Receiver class used for testing LoopbackRelay
r\ c C s | j |7 _ dS )z)Accumulate received data for verificationN)rz ry r r r rb � s z'LoopbackRelayTest.Receiver.dataReceivedN)r# r$ r% �__doc__rz rb r r r r �Receiver� s r� c C sN | � � }t�|�}|�d� |�d� | �|jd� |�� | �|jd� dS )z8Test to verify that the write function works as expecteds abcs defr\ s abcdefN)r� r �
LoopbackRelayre r7 rz � clearBuffer�r ZreceiverZrelayr r r �
test_write� s
z LoopbackRelayTest.test_writec C s\ | � � }t�|�}|�ddddg� |�dddg� | �|jd� |�� | �|jd � d
S ) z@Test to verify that the writeSequence function works as expecteds The s quick s brown s fox s jumps s over s the lazy dogr\ s+ The quick brown fox jumps over the lazy dogN)r� r r� rf r7 rz r� r� r r r �test_writeSequence� s
�� �z$LoopbackRelayTest.test_writeSequenceN)r# r$ r% r� r
r� r� r� r r r r r� � s
r� N)"r� Z
__future__r r Zzope.interfacer Ztwisted.python.compatr Z
twisted.trialr Ztwisted.protocolsr r Ztwisted.internetr Ztwisted.internet.protocolr
Ztwisted.internet.deferr Ztwisted.internet.interfacesr r
r r r Z LineReceiverr r& r/ ZTestCaserE r� r� r� r r r r �<module> s& ' L