← Back
Editing: 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
Save File
Cancel