File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_pcp.cpython-38.pyc
U
W[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 �7/usr/lib/python3/dist-packages/twisted/test/test_pcp.py�__init__% s zDummyTransport.__init__c C s | j �|� d S r )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 r )�consumer)r
r r r r r
3 s zDummyProducer.__init__c C s d| _ d| _d S )NTF)�resumed�pausedr 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 r )�stoppedr 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 r )�producer)r
r# Z streamingr r r �registerProducerF s z DummyConsumer.registerProducerc C s
d| _ d S r )� unregisteredr r r r �unregisterProducerI s z DummyConsumer.unregisterProducerc C s
d| _ d S r )�finishedr 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 r )r" �
underlying�
proxyClass� transportr 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 r )r" r* r+ r r r# r r r r r- b s zConsumerInterfaceTest.setUpc C s" | j �| jd� | �| jj� d S r )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 r2 r r r r �testUnregisters s
z$ConsumerInterfaceTest.testUnregisterc C s> | j �| jd� | j �� d| j_| j �� | �| jj� d S r4 )r r$ r# r( r r r2 r r r r �
testFinish~ s
z ConsumerInterfaceTest.testFinishN)r r r r r- r3 r5 r6 r r r r r1 Y s
r1 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 r )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 )N� yakkity yakz*Paused producer should not have sent data.)r# r r r2 r r r r r r � testPause� s
�z ProducerInterfaceTest.testPausec C s6 | j �� | j �� | j �d� | �| j�� d� d S )Nr: )r# r r r r8 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 r8 �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 r8 r r r r r r �testResumeBuffer� s
z&ProducerInterfaceTest.testResumeBufferc C s, | j �� | j �d� | �| j�� d� d S )Nr: z+Stopped producer should not have sent data.)r# r! r r2 r r r r r r �testStop� s
�z ProducerInterfaceTest.testStopN) r r r r r- r9 r; r< r> r? r@ r r r r r7 � s r7 c @ s e Zd ZejZdS )�PCP_ConsumerInterfaceTestsN�r r r r r0 r+ r r r r rA � s rA c @ s e Zd ZejZdS )� PCPII_ConsumerInterfaceTestsN�r r r r �ProducerConsumerProxyr+ r r r r rC � s rC c @ s e Zd ZejZdS )�PCP_ProducerInterfaceTestsNrB r r r r rF � s rF c @ s e Zd ZejZdS )� PCPII_ProducerInterfaceTestsNrD r r r r rG � s rG 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 r )r+ �proxyr �parentProducerr$ r r r r r- � s zProducerProxyTests.setUpc C s | j �� | �| jj� d S r )rI r! �
assertTruerJ r r r r r r@ � s
zProducerProxyTests.testStopN) r r r r r r0 r+ r- r@ r r r r rH � s rH 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 r )r" r* r+ r r r r r r- � s zConsumerProxyTests.setUpc C s" | j �d� | �| j�� d� d S r. )r r r8 r* r r r r r r/ � s z ConsumerProxyTests.testWritec C s | j �� | �| jj� d S r )r r( rK r* r' r r r r r6 � s
z ConsumerProxyTests.testFinishc C s | j �� | �| jj� d S r )r r& rK r* r% r r r r r5 � s
z!ConsumerProxyTests.testUnregisterN) r r r r r r0 r+ r- r/ r6 r5 r r r r rL � s rL c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �PullProducerTestc C s6 t � | _| �| j�| _t| j�| _| j�| jd� d S r )r" r* r+ rI r rJ 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.)rI r r2 r* r r r r r �testHoldWrites� s �z PullProducerTest.testHoldWritesc C s, | j �d� | j �� | �| j�� d� d S )NrN )rI r r r8 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)rI r r r= r* r r8 r )r
Znwritesr r r �testMergeWrites� s
�z PullProducerTest.testMergeWritesc C s, | j �� | j �d� | �| j�� d� d S )Nr )rI r r r8 r* r r r r r �
testLateWrite s
z PullProducerTest.testLateWriteN)r r r r- rO rP rR rS r r r r rM � s
rM 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+ NrB r r r r rT s rT c @ s e Zd ZG dd� dej�ZdS )�PCPII_PullProducerTestsc @ s e Zd ZdZdS )z"PCPII_PullProducerTests.proxyClassFNrU r r r r r+ s r+ NrD r r r r rW s rW 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+ rI Z
bufferSizer rJ r$ r r r r r- s
zBufferedConsumerTests.setUpc C s" | j �| jd� | �| jj� d S r4 )rI r$ rJ rK r r r r r �testRegisterPull" s z&BufferedConsumerTests.testRegisterPullc C s | j �� | �| jj� d S r )rI r r2 rJ r r r r r �testPauseIntercept' s
z(BufferedConsumerTests.testPauseInterceptc C s& | j �� | j �� | �| jj� d S r )rI r r r2 rJ 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)rI r r2 rJ r r rK 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) rI r r rK rJ r r r2 ZproducerPausedr r r r �testTriggerResume> s
�z'BufferedConsumerTests.testTriggerResumeN)
r r r r r rE r+ r- r[ r\ r] r^ r_ r r r r rX s rX 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 )NrY )r rE �_writeSomeData�minr= r r r r ra N s z+BufferedPullTests.proxyClass._writeSomeDataN)r r r rV ra r r r r r+ K s r+ c C s> t � | _| �| j�| _d| j_t| j�| _| j�| jd� d S )NrY FrZ r r r r r- R s
zBufferedPullTests.setUpc C s$ d| j _| j�� | �| j j� d S r4 )rJ r rI r rK r r r r �testResumePullZ s
z BufferedPullTests.testResumePullc C s> | j �� | j �d� | �| j�� d� | �| j jdg� d S )NZidatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumZddatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumZdatum)rI r r r8 r* r Z_bufferr r r r �testLateWriteBufferinga s
z(BufferedPullTests.testLateWriteBufferingN) r r r r rE r+ r- rc rd r r r r r` J s r` N)� __version__Z
twisted.trialr Ztwisted.protocolsr r r r" ZTestCaser) r1 r7 rA rC rF rG rH rL rM rT rW rX r` r r r r �<module> s&
./#5