Your IP : 216.73.216.213
B
;��]1 � @ sP d dd� Z ddlmZ ddlmZ G dd� d�ZG dd � d �ZG d
d � d e�ZG d d
� d
ej�Z G dd� d�Z
G dd� d�Z G dd� de
ej�Z G dd� de
ej�Z
G dd� de ej�ZG dd� de ej�ZG dd� dej�ZG d d � d ej�ZG d d � d �ZG d d!� d!eej�ZG d"d#� d#eej�ZG d$d%� d%ej�ZG d&d'� d'ej�Zd(S ))z$Revision: 1.5 $� ���� )�unittest)�pcpc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �DummyTransportz A dumb transport to wrap around.c C s
g | _ d S )N)�_writes)�self� r �H/opt/alt/python37/lib64/python3.7/site-packages/twisted/test/test_pcp.py�__init__% s zDummyTransport.__init__c C s | j �|� d S )N)r �append)r �datar r r
�write( s zDummyTransport.writec C s d� | j�S )N� )�joinr )r r r r
�getvalue+ s zDummyTransport.getvalueN)�__name__�
__module__� __qualname__�__doc__r r r r r r r
r " s r c @ s8 e Zd ZdZdZdZdd� Zdd� Zdd� Zdd � Z d
S ) �
DummyProducerFc C s
|| _ d S )N)�consumer)r r r r r
r 3 s zDummyProducer.__init__c C s d| _ d| _d S )NTF)�resumed�paused)r r r r
�resumeProducing6 s z DummyProducer.resumeProducingc C s
d| _ d S )NT)r )r r r r
�pauseProducing: s z DummyProducer.pauseProducingc C s
d| _ d S )NT)�stopped)r r r r
�
stopProducing= s zDummyProducer.stopProducingN)
r r r r r r r r r r r r r r
r . s r c @ s0 e Zd ZdZdZdZdd� Zdd� Zdd � ZdS )
�
DummyConsumerNFTc C s ||f| _ d S )N)�producer)r r Z streamingr r r
�registerProducerF s z DummyConsumer.registerProducerc C s
d| _ d S )NT)� unregistered)r r r r
�unregisterProducerI s z DummyConsumer.unregisterProducerc C s
d| _ d S )NT)�finished)r r r r
�finishL s zDummyConsumer.finish) r r r r r# r! r r" r$ r r r r
r A s r c @ s" e Zd ZejZdd� Zdd� ZdS )�TransportInterfaceTestsc C s t � | _| �| j�| _d S )N)r �
underlying�
proxyClass� transport)r r r r
�setUpR s z TransportInterfaceTests.setUpc C s | j �d� d S )Nz
some bytes)r( r )r r r r
� testWriteV s z!TransportInterfaceTests.testWriteN)r r r r �BasicProducerConsumerProxyr' r) r* r r r r
r% O s r% c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �ConsumerInterfaceTestz�Test ProducerConsumerProxy as a Consumer.
Normally we have ProducingServer -> ConsumingTransport.
If I am to go between (Server -> Shaper -> Transport), I have to
play the role of Consumer convincingly for the ProducingServer.
c C s& t � | _| �| j�| _t| j�| _d S )N)r r& r' r r r )r r r r
r) b s zConsumerInterfaceTest.setUpc C s" | j �| jd� | �| jj� d S )NT)r r r � assertFalser )r r r r
�testRegisterPushg s z&ConsumerInterfaceTest.testRegisterPushc C s> | j �| jd� | j �� d| j_| j �� | �| jj� d S )NF)r r r r" r r r- )r r r r
�testUnregisters s
z$ConsumerInterfaceTest.testUnregisterc C s> | j �| jd� | j �� d| j_| j �� | �| jj� d S )NF)r r r r$ r r r- )r r r r
�
testFinish~ s
z ConsumerInterfaceTest.testFinishN)r r r r r) r. r/ r0 r r r r
r, Y s
r, c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�ProducerInterfaceTestz�Test ProducerConsumerProxy as a Producer.
Normally we have ProducingServer -> ConsumingTransport.
If I am to go between (Server -> Shaper -> Transport), I have to
play the role of Producer convincingly for the ConsumingTransport.
c C s t � | _| �| j�| _d S )N)r r r' r )r r r r
r) � s zProducerInterfaceTest.setUpc C s | � | jjd | j� d S )Nr )� assertEqualr r )r r r r
�testRegistersProducer� s z+ProducerInterfaceTest.testRegistersProducerc C s, | j �� | j �d� | �| j�� d� d S )Nz yakkity yakz*Paused producer should not have sent data.)r r r r- r r )r r r r
� testPause� s
z ProducerInterfaceTest.testPausec C s6 | j �� | j �� | j �d� | �| j�� d� d S )Nz yakkity yak)r r r r r2 r r )r r r r
�
testResume� s
z ProducerInterfaceTest.testResumec C s. | j �� | j �� | �t| jj�dd� d S )Nr z Resume triggered an empty write.)r r r r2 �lenr r )r r r r
�testResumeNoEmptyWrite� s
z,ProducerInterfaceTest.testResumeNoEmptyWritec C s6 | j �� | j �d� | j �� | �| j�� d� d S )Nz buffer this)r r r r r2 r r )r r r r
�testResumeBuffer� s
z&ProducerInterfaceTest.testResumeBufferc C s, | j �� | j �d� | �| j�� d� d S )Nz yakkity yakz+Stopped producer should not have sent data.)r r r r- r r )r r r r
�testStop� s
z ProducerInterfaceTest.testStopN) r r r r r) r3 r4 r5 r7 r8 r9 r r r r
r1 � s r1 c @ s e Zd ZejZdS )�PCP_ConsumerInterfaceTestsN)r r r r r+ r' r r r r
r: � s r: c @ s e Zd ZejZdS )� PCPII_ConsumerInterfaceTestsN)r r r r �ProducerConsumerProxyr' r r r r
r; � s r; c @ s e Zd ZejZdS )�PCP_ProducerInterfaceTestsN)r r r r r+ r' r r r r
r= � s r= c @ s e Zd ZejZdS )� PCPII_ProducerInterfaceTestsN)r r r r r< r' r r r r
r> � s r> c @ s& e Zd ZdZejZdd� Zdd� ZdS )�ProducerProxyTestszFProducer methods on me should be relayed to the Producer I proxy.
c C s, | � d �| _t| j�| _| j�| jd� d S )NT)r' �proxyr �parentProducerr )r r r r
r) � s zProducerProxyTests.setUpc C s | j �� | �| jj� d S )N)r@ r �
assertTruerA r )r r r r
r9 � s
zProducerProxyTests.testStopN) r r r r r r+ r' r) r9 r r r r
r? � s r? c @ s6 e Zd ZdZejZdd� Zdd� Zdd� Z dd � Z
d
S ) �ConsumerProxyTestszFConsumer methods on me should be relayed to the Consumer I proxy.
c C s t � | _| �| j�| _d S )N)r r& r' r )r r r r
r) � s zConsumerProxyTests.setUpc C s" | j �d� | �| j�� d� d S )Nz
some bytes)r r r2 r& r )r r r r
r* � s z ConsumerProxyTests.testWritec C s | j �� | �| jj� d S )N)r r$ rB r& r# )r r r r
r0 � s
z ConsumerProxyTests.testFinishc C s | j �� | �| jj� d S )N)r r" rB r& r! )r r r r
r/ � s
z!ConsumerProxyTests.testUnregisterN) r r r r r r+ r' r) r* r0 r/ r r r r
rC � s rC c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �PullProducerTestc C s6 t � | _| �| j�| _t| j�| _| j�| jd� d S )NT)r r& r' r@ r rA r )r r r r
r) � s zPullProducerTest.setUpc C s" | j �d� | �| j�� d� d S )N�helloz+Pulling Consumer got data before it pulled.)r@ r r- r& r )r r r r
�testHoldWrites� s z PullProducerTest.testHoldWritesc C s, | j �d� | j �� | �| j�� d� d S )NrE )r@ r r r2 r& r )r r r r
�testPull� s
zPullProducerTest.testPullc C sX | j �d� | j �d� | j �� t| jj�}| �|dd|f � | �| j�� d� d S )Nzhello Zsunshine� z(Pull resulted in %d writes instead of 1.zhello sunshine)r@ r r r6 r& r r2 r )r Znwritesr r r
�testMergeWrites� s
z PullProducerTest.testMergeWritesc C s, | j �� | j �d� | �| j�� d� d S )Nr
)r@ r r r2 r& r )r r r r
�
testLateWrite s
z PullProducerTest.testLateWriteN)r r r r) rF rG rI rJ r r r r
rD � s
rD c @ s e Zd ZG dd� dej�ZdS )�PCP_PullProducerTestsc @ s e Zd ZdZdS )z PCP_PullProducerTests.proxyClassFN)r r r � iAmStreamingr r r r
r' s r' N)r r r r r+ r' r r r r
rK s rK c @ s e Zd ZG dd� dej�ZdS )�PCPII_PullProducerTestsc @ s e Zd ZdZdS )z"PCPII_PullProducerTests.proxyClassFN)r r r rL r r r r
r' s r' N)r r r r r< r' r r r r
rM s rM c @ sF e Zd ZdZejZdd� Zdd� Zdd� Z dd � Z
d
d � Z d d
� Z dS )�BufferedConsumerTestsz=As a consumer, ask the producer to pause after too much data.c C s> t � | _| �| j�| _d| j_t| j�| _| j�| jd� d S )N�d T)r r& r' r@ �
bufferSizer rA r )r r r r
r) s
zBufferedConsumerTests.setUpc C s" | j �| jd� | �| jj� d S )NF)r@ r rA rB r )r r r r
�testRegisterPull" s z&BufferedConsumerTests.testRegisterPullc C s | j �� | �| jj� d S )N)r@ r r- rA r )r r r r
�testPauseIntercept' s
z(BufferedConsumerTests.testPauseInterceptc C s& | j �� | j �� | �| jj� d S )N)r@ r r r- rA r )r r r r
�testResumeIntercept+ s
z)BufferedConsumerTests.testResumeInterceptc C sT | j �� | �| jjd� | j �d� | �| jjd� | j �d� | �| jj� dS )zMake sure I say "when."zdon't pause yetZ3xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxN)r@ r r- rA r r rB )r r r r
�testTriggerPause3 s
z&BufferedConsumerTests.testTriggerPausec C sR | j �� | j �d� | �| jjd� | j �� | �| jjd� | �| j j� dS )z3Make sure I resumeProducing when my buffer empties.Zfxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxzshould be pausedz Producer should have resumed.N) r@ r r rB rA r r r- ZproducerPaused)r r r r
�testTriggerResume> s
z'BufferedConsumerTests.testTriggerResumeN)
r r r r r r< r' r) rQ rR rS rT rU r r r r
rN s rN c @ s6 e Zd ZG dd� dej�Zdd� Zdd� Zdd� Zd S )
�BufferedPullTestsc @ s e Zd ZdZdd� ZdS )z BufferedPullTests.proxyClassFc C s$ t j�| |d d� � tt|�d�S )NrO )r r< �_writeSomeData�minr6 )r r
r r r
rW N s z+BufferedPullTests.proxyClass._writeSomeDataN)r r r rL rW r r r r
r' K s r' c C s> t � | _| �| j�| _d| j_t| j�| _| j�| jd� d S )NrO F)r r& r' r@ rP r rA r )r r r r
r) R s
zBufferedPullTests.setUpc C s$ d| j _| j�� | �| j j� d S )NF)rA r r@ r rB )r r r r
�testResumePullZ s
z BufferedPullTests.testResumePullc C s> | j �� | j �d� | �| j�� d� | �| j jdg� d S )NZidatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumZddatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumZdatum)r@ r r r2 r& r �_buffer)r r r r
�testLateWriteBufferinga s
z(BufferedPullTests.testLateWriteBufferingN) r r r r r< r' r) rY r[ r r r r
rV J s rV N)� __version__Z
twisted.trialr Ztwisted.protocolsr r r r ZTestCaser% r, r1 r: r; r= r> r? rC rD rK rM rN rV r r r r
�<module> s&
./#5