File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_loopback.cpython-310.pyc
o
�b�7 � @ s� d Z ddlmZ ddlmZmZmZ ddlmZ ddl m
Z
m Z m Z ddl
mZ ddlmZmZ ddlmZ G d d
� d
ej�ZG d d � d e�ZG d
d� d�ZG dd� deej�ZG dd� deej�ZG dd� deej�ZG dd� dej�Z dS )z.
Test case for L{twisted.protocols.loopback}.
� )� implementer)�defer�
interfaces�reactor)�Deferred)�IAddress�
IPullProducer�
IPushProducer)�Protocol)�basic�loopback)�unittestc @ 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 � z SimpleProtocol.connectionMadec C � | j �|� d S r )r �append�r �liner r r � lineReceived r zSimpleProtocol.lineReceivedc C r r )r r )r �reasonr r r �connectionLost r z SimpleProtocol.connectionLostN)�__name__�
__module__� __qualname__r r r r! r r r r r s
r c @ � e Zd ZdZdd� ZdS )� DoomProtocolr c C sV | j d7 _ | j dk r| �d| j f � t�| |� | jd dkr)| j�� d S d S )N� � s Hello %d���� Hello 3)�i�sendLiner r r � transport�loseConnectionr r r r r & s
�zDoomProtocol.lineReceivedN)r" r# r$ r+ r r r r r r& # � r& c @ s e Zd Zdd� Zdd� ZdS )�LoopbackTestCaseMixinc �R 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 � sendALine6 s
z<LoopbackTestCaseMixin.testRegularFunction.<locals>.sendALinec s8 �� � jdg� �� t�j�d� �� t� j�d� d S )Nr2 r' �� assertEqualr �lenr ��ignored��cr6 r r r �check< s z8LoopbackTestCaseMixin.testRegularFunction.<locals>.check)r r � addCallbackr �
maybeDeferred� loopbackFunc�r r7 r? �dr r= r �testRegularFunction2 s
z)LoopbackTestCaseMixin.testRegularFunctionc r1 )Nc s � � d� d S )N� DOOM LINE)r, r3 r5 r r r7 I s z=LoopbackTestCaseMixin.testSneakyHiddenDoom.<locals>.sendALinec sL �� �jg d�� �� � jg d�� �� t�j�d� �� t� j�d� d S )N)� Hello 1� Hello 2r* )rF rG rH r* r' r8 r; r= r r r? N s
�z9LoopbackTestCaseMixin.testSneakyHiddenDoom.<locals>.check)r& r r@ r rA rB rC r r= r �testSneakyHiddenDoomE s
z*LoopbackTestCaseMixin.testSneakyHiddenDoomN)r" r# r$ rE rI r r r r r0 1 s r0 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 �
|| _ d S r )r- �r r- r r r �makeConnectiong �
zKLoopbackAsyncTests.test_makeConnection.<locals>.TestProtocol.makeConnection)r" r# r$ r- rM r r r r � TestProtocold r/ rO N)r
r �
loopbackAsync�assertIsNotNoner- )r rO �server�clientr r r �test_makeConnection^ s z&LoopbackAsyncTests.test_makeConnectionc sj G dd� dt �}|r|� }t� }|_t � }n t � }|� }t� }|_t�||� � �fdd�}|�|�S )zJ
Test one of the permutations of client/server host/peer.
c @ � e Zd Zdd� ZdS )z6LoopbackAsyncTests._hostpeertest.<locals>.TestProtocolc S s t �| |� | j�|� d S r )r
rM � onConnectionr rL r r r rM v s zELoopbackAsyncTests._hostpeertest.<locals>.TestProtocol.makeConnectionN)r" r# r$ rM r r r r rO u � rO c s t | � �� }��t�|�� d S r )�getattr�
assertTruer �
providedBy)r- �host��getr r r � connected� s z3LoopbackAsyncTests._hostpeertest.<locals>.connected)r
r rV r rP r@ )r r] �
testServerrO rR rD rS r^ r r\ r �
_hostpeertestp s
z LoopbackAsyncTests._hostpeertestc C � | � dd�S )z�
Test that the server gets a transport with a properly functioning
implementation of L{ITransport.getHost}.
�getHostT�r` r r r r �test_serverHost� � z"LoopbackAsyncTests.test_serverHostc C ra )zG
Like C{test_serverHost} but for L{ITransport.getPeer}
�getPeerTrc r r r r �test_serverPeer� � z"LoopbackAsyncTests.test_serverPeerrb c C ra )z�
Test that the client gets a transport with a properly functioning
implementation of L{ITransport.getHost}.
rb Frc )r r] r r r �test_clientHost� re z"LoopbackAsyncTests.test_clientHostc C ra )zH
Like C{test_clientHost} but for L{ITransport.getPeer}.
rf Frc r r r r �test_clientPeer� rh z"LoopbackAsyncTests.test_clientPeerc sj G dd� dt �}G � fdd�dt �}|r!|� }|� }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 @ r% )z9LoopbackAsyncTests._greetingtest.<locals>.GreeteeProtocol� c S s, | j |7 _ | j dkr| j�d � d S d S �N� bytes)�bytes�receivedr �r rn r r r � dataReceived� s
�zFLoopbackAsyncTests._greetingtest.<locals>.GreeteeProtocol.dataReceivedN�r" r# r$ rn rq r r r r �GreeteeProtocol� r/ rs c s e Zd Z� fdd�ZdS )z9LoopbackAsyncTests._greetingtest.<locals>.GreeterProtocolc s, � dkr | j �d� d S | j �ddg� d S )N�writerm s byts es)r- rt �
writeSequencer �rt r r r � s zHLoopbackAsyncTests._greetingtest.<locals>.GreeterProtocol.connectionMadeN�r" r# r$ r r rv r r �GreeterProtocol� s rx )r
r ro r rP )r rt r_ rs rx rR rS rD r rv r �
_greetingtest� s z LoopbackAsyncTests._greetingtestc C ra )z�
Test that on a connection where the client speaks first, the server
receives the bytes sent by the client.
rt F�ry r r r r �test_clientGreeting� re z&LoopbackAsyncTests.test_clientGreetingc C ra )zz
Like C{test_clientGreeting}, but use C{writeSequence} instead of
C{write} to issue the greeting.
ru Frz r r r r �test_clientGreetingSequence� re z.LoopbackAsyncTests.test_clientGreetingSequencert c C ra )z�
Test that on a connection where the server speaks first, the client
receives the bytes sent by the server.
rt Trz )r rt r r r �test_serverGreeting� re z&LoopbackAsyncTests.test_serverGreetingc C ra )zz
Like C{test_serverGreeting}, but use C{writeSequence} instead of
C{write} to issue the greeting.
ru Trz r r r r �test_serverGreetingSequence� re z.LoopbackAsyncTests.test_serverGreetingSequencec sh dd� t dd�D ��G ��fdd�dt�}G � ��fdd�dt�}|� �|� � t� � _t��� � � jS ) Nc S s g | ]}d |f �qS )s %dr )�.0r+ r r r �
<listcomp>� s z4LoopbackAsyncTests._producertest.<locals>.<listcomp>r �
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.connectionMadeNrw r r� r r �ProducingProtocol� s r� c s e Zd ZdZ� ��fdd�ZdS )z;LoopbackAsyncTests._producertest.<locals>.ReceivingProtocolrk c s6 | j |7 _ | j d���kr| j�� �f� d S d S )Nrk )rn �joinro r �r �data�rS rR r� r r rq � s �zHLoopbackAsyncTests._producertest.<locals>.ReceivingProtocol.dataReceivedNrr r r� r r �ReceivingProtocol� s r� )�ranger
r ro r rP )r r� r� r� r )rS r� rR r� r �
_producertest� s z LoopbackAsyncTests._producertestc s: t t�G dd� d��}� �|�}� 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 rK r �r� �r r� r r r r rN zCLoopbackAsyncTests.test_pushProducer.<locals>.PushProducer.__init__c S s
d| _ d S �NT)�resumedr r r r �resumeProducing rN zJLoopbackAsyncTests.test_pushProducer.<locals>.PushProducer.resumeProducingc S s || _ |�| d� | �� d S r� )�consumer�registerProducer�_produceAndSchedule�r r� r r r r�
s z@LoopbackAsyncTests.test_pushProducer.<locals>.PushProducer.startc S s: | j r| j�| j �d�� t�d| j� d S | j�� d S �Nr )r� r� rt �popr � callLaterr� �unregisterProducerr r r r r� s zNLoopbackAsyncTests.test_pushProducer.<locals>.PushProducer._produceAndScheduleN)r" r# r$ r� r r� r� r� r r r r � PushProducer s r� c s | \}}� � |jjd� d S )Nz0Streaming producer should not have been resumed.)� assertFalser� r� )�resultsrS rR r r r �finished s
�z6LoopbackAsyncTests.test_pushProducer.<locals>.finished)r r r� r@ )r r� rD r� r r r �test_pushProducer� s
z$LoopbackAsyncTests.test_pushProducerc C s t t�G dd� d��}| �|�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 rK r r� r� r r r r ) rN zCLoopbackAsyncTests.test_pullProducer.<locals>.PullProducer.__init__c S s || _ | j �| d� d S )NF)r� r� r� r r r r� , s z@LoopbackAsyncTests.test_pullProducer.<locals>.PullProducer.startc S s, | j �| j�d�� | js| j �� d S d S r� )r� rt r� r� r� r r r r r� 0 s �zJLoopbackAsyncTests.test_pullProducer.<locals>.PullProducer.resumeProducingN)r" r# r$ r r� r� r r r r � PullProducer' s 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 @ rU )z9LoopbackAsyncTests.test_writeNotReentrant.<locals>.Serverc S s | j �d� d S rl )r- rt rp r r r rq ? r zFLoopbackAsyncTests.test_writeNotReentrant.<locals>.Server.dataReceivedN)r" r# r$ rq r r r r �Server> rW 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 E s zHLoopbackAsyncTests.test_writeNotReentrant.<locals>.Client.connectionMadec S s | j �d� d| _d S )N� fooT)r- rt �readyr r r r r� H s
z<LoopbackAsyncTests.test_writeNotReentrant.<locals>.Client.goc S s | j | _| j�� d S r )r� �wasReadyr- r. rp r r r rq L s zFLoopbackAsyncTests.test_writeNotReentrant.<locals>.Client.dataReceivedN)r" r# r$ r� r r� rq r r r r �ClientB s
r� c s �� � j� d S r )rY r� r; �rS r r r �
cbFinishedT r z=LoopbackAsyncTests.test_writeNotReentrant.<locals>.cbFinished)r
r rP r@ )r r� r� rR rD r� r r� r �test_writeNotReentrant7 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
|� | �� � | s� � ||f� d S r )r r] )�queue�targetrn )� pumpCallsr r � dummyPolicyb s
�z7LoopbackAsyncTests.test_pumpPolicy.<locals>.dummyPolicyr� � bar� baz� quuxc s$ �� �� g d�f�ddgfg� d S )N)r� r� Nr� r� )r9 r; �rS r� r rR r r �
cbCompletet s �z6LoopbackAsyncTests.test_pumpPolicy.<locals>.cbComplete)r
r rP r9 r- rt r. r@ )r r� r� r� r r� r �test_pumpPolicyZ 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 rq r �_LoopbackQueue�put�identityPumpPolicyr9 �r rn rS 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 rq r r� r� �collapsingPumpPolicyr9 r� r r r �test_collapsingPumpPolicy� s
z,LoopbackAsyncTests.test_collapsingPumpPolicyN)rb rv )r" r# r$ � staticmethodr rP rB rT r` rd rg ri rj ry r{ r| r} r~ r� r� r� r� r� r� r� r r r r rJ [ s(
#
'#% rJ c @ s e Zd Zeej�ZdS )�LoopbackTCPTestsN)r" r# r$ r� r � loopbackTCPrB r r r r r� � s r� c @ s. e Zd Zeej�Ze�e d�du rdZ
dS dS )�LoopbackUNIXTestsNz-Current reactor does not support UNIX sockets) r" r# r$ r� r � loopbackUNIXrB r � 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
rk c C s | j |7 _ dS )z)Accumulate received data for verificationN)r� r� r r r rq � s z'LoopbackRelayTest.Receiver.dataReceivedN)r" r# r$ �__doc__r� rq 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 defrk s abcdefN)r� r �
LoopbackRelayrt r9 r� � clearBuffer�r �receiver�relayr r r �
test_write� s
z LoopbackRelayTest.test_writec C sV | � � }t�|�}|�g d�� |�g d�� | �|jd� |�� | �|jd� dS )z@Test to verify that the writeSequence function works as expected)s The s quick s brown s fox )s jumps s over s the lazy dogrk s+ The quick brown fox jumps over the lazy dogN)r� r r� ru r9 r� 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� �zope.interfacer �twisted.internetr r r �twisted.internet.deferr �twisted.internet.interfacesr r r �twisted.internet.protocolr
�twisted.protocolsr r �
twisted.trialr
� LineReceiverr r& r0 �TestCaserJ r� r� r� r r r r �<module> s" * J