File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_pcp.cpython-310.pyc
o
�b�0 � @ 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 $� ���� )�pcp)�unittestc @ 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__% �
zDummyTransport.__init__c C s | j �|� d S r )r �append�r
�datar r r �write( � 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
4 r zDummyProducer.__init__c C s d| _ d| _d S )NTF)�resumed�pausedr r r r �resumeProducing7 s
z DummyProducer.resumeProducingc C �
d| _ d S �NT)r r r r r �pauseProducing; r z DummyProducer.pauseProducingc C r r! )�stoppedr r r r �
stopProducing> r 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& � streamingr r r �registerProducerG s z DummyConsumer.registerProducerc C r r! )� unregisteredr r r r �unregisterProducerJ r z DummyConsumer.unregisterProducerc C r r! )�finishedr r r r �finishM r zDummyConsumer.finish) r r r r& r+ r) r( r* r, r r r r r% B s r% c @ s" e Zd ZejZdd� Zdd� ZdS )�TransportInterfaceTestsc C � t � | _| �| j�| _d S r )r% �
underlying�
proxyClass� transportr r r r �setUpT � z TransportInterfaceTests.setUpc C s | j �d� d S �Nz
some bytes)r1 r r r r r � testWriteX r z!TransportInterfaceTests.testWriteN)r r r r �BasicProducerConsumerProxyr0 r2 r5 r r r r r- Q 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/ r0 r r r& r r r r r2 e s zConsumerInterfaceTest.setUpc C �" | j �| jd� | �| jj� d S r! )r r( r& � assertFalser r r r r �testRegisterPushj � z&ConsumerInterfaceTest.testRegisterPushc C �> | j �| jd� | j �� d| j_| j �� | �| jj� d S �NF)r r( r& r* r r r9 r r r r �testUnregisterv s
z$ConsumerInterfaceTest.testUnregisterc C r< r= )r r( r& r, r r r9 r r r r �
testFinish� s
z ConsumerInterfaceTest.testFinishN)r r r r r2 r: r> r? r r r r r7 \ s r7 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 r. r )r% r r0 r& r r r r r2 � r3 zProducerInterfaceTest.setUpc C s | � | jjd | j� d S )Nr )� assertEqualr r& r r r r �testRegistersProducer� s z+ProducerInterfaceTest.testRegistersProducerc C �, | j �� | j �d� | �| j�� d� d S )N� yakkity yakz*Paused producer should not have sent data.)r& r" r r9 r r r r r r � testPause� �
�z ProducerInterfaceTest.testPausec C s6 | j �� | j �� | j �d� | �| j�� d� d S )NrD )r& r" r r rA 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 rA �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 rA r r r r r r �testResumeBuffer� s
z&ProducerInterfaceTest.testResumeBufferc C rC )NrD z+Stopped producer should not have sent data.)r& r$ r r9 r r r r r r �testStop� rF z ProducerInterfaceTest.testStopN) r r r r r2 rB rE rG rI rJ rK r r r r r@ � s r@ c @ � e Zd ZejZdS )�PCP_ConsumerInterfaceTestsN�r r r r r6 r0 r r r r rM � �
rM c @ rL )� PCPII_ConsumerInterfaceTestsN�r r r r �ProducerConsumerProxyr0 r r r r rP � rO rP c @ rL )�PCP_ProducerInterfaceTestsNrN r r r r rS � rO rS c @ rL )� PCPII_ProducerInterfaceTestsNrQ r r r r rT � rO rT c @ s& e Zd ZdZejZdd� Zdd� ZdS )�ProducerProxyTestszAProducer methods on me should be relayed to the Producer I proxy.c C s, | � d �| _t| j�| _| j�| jd� d S r! )r0 �proxyr �parentProducerr( r r r r r2 � s zProducerProxyTests.setUpc C � | j �� | �| jj� d S r )rV r$ �
assertTruerW r# r r r r rK � �
zProducerProxyTests.testStopN) r r r r r r6 r0 r2 rK r r r r rU � s
rU c @ s6 e Zd ZdZejZdd� Zdd� Zdd� Z dd � Z
d
S ) �ConsumerProxyTestszAConsumer methods on me should be relayed to the Consumer I proxy.c C r. r )r% r/ r0 r r r r r r2 � r3 zConsumerProxyTests.setUpc C s" | j �d� | �| j�� d� d S r4 )r r rA r/ r r r r r r5 � s z ConsumerProxyTests.testWritec C rX r )r r, rY r/ r+ r r r r r? � rZ z ConsumerProxyTests.testFinishc C rX r )r r* rY r/ r) r r r r r> � rZ z!ConsumerProxyTests.testUnregisterN) r r r r r r6 r0 r2 r5 r? r> r r r r r[ � s r[ 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/ r0 rV r rW r( r r r r r2 � s zPullProducerTest.setUpc C s" | j �d� | �| j�� d� d S )N�helloz+Pulling Consumer got data before it pulled.)rV r r9 r/ r r r r r �testHoldWrites� s
�z PullProducerTest.testHoldWritesc C s, | j �d� | j �� | �| j�� d� d S )Nr] )rV r r rA 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 �sunshine� z(Pull resulted in %d writes instead of 1.zhello sunshine)rV r r rH r/ r rA r )r
�nwritesr r r �testMergeWrites s
�z PullProducerTest.testMergeWritesc C s, | j �� | j �d� | �| j�� d� d S )Nr )rV r r rA r/ r r r r r �
testLateWrite s
z PullProducerTest.testLateWriteN)r r r r2 r^ r_ rc rd r r r r r\ � s
r\ c @ � e Zd ZG dd� dej�ZdS )�PCP_PullProducerTestsc @ � e Zd ZdZdS )z PCP_PullProducerTests.proxyClassFN�r r r � iAmStreamingr r r r r0 � r0 NrN r r r r rf � rf c @ re )�PCPII_PullProducerTestsc @ rg )z"PCPII_PullProducerTests.proxyClassFNrh r r r r r0 rj r0 NrQ r r r r rl rk rl 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 �> t � | _| �| j�| _d| j_t| j�| _| j�| jd� d S )N�d T�r% r/ r0 rV �
bufferSizer rW r( r r r r r2 ) �
zBufferedConsumerTests.setUpc C r8 r= )rV r( rW rY r r r r r �testRegisterPull1 r; z&BufferedConsumerTests.testRegisterPullc C rX r )rV r" r9 rW r r r r r �testPauseIntercept6 rZ z(BufferedConsumerTests.testPauseInterceptc C s& | j �� | j �� | �| jj� d S r )rV r" r r9 rW 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 yet�3xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxN)rV r" r9 rW r r rY r r r r �testTriggerPauseB s
z&BufferedConsumerTests.testTriggerPausec C sR | j �� | j �d� | �| jjd� | j �� | �| jjd� | �| j j� dS )z3Make sure I resumeProducing when my buffer empties.�fxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxzshould be pausedz Producer should have resumed.N) rV r" r rY rW r r r9 �producerPausedr r r r �testTriggerResumeM s
z'BufferedConsumerTests.testTriggerResumeN)
r r r r r rR r0 r2 rs rt ru rw rz r r r r rm $ s rm 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 rR �_writeSomeData�minrH r r r r r| ] s z+BufferedPullTests.proxyClass._writeSomeDataN)r r r ri r| r r r r r0 Z s r0 c C rn )Nro Frp r r r r r2 a rr zBufferedPullTests.setUpc C s$ d| j _| j�� | �| j j� d S r= )rW r rV r rY r r r r �testResumePulli s
z BufferedPullTests.testResumePullc C s> | j �� | j �d� | �| j�� d� | �| j jdg� d S )N�idatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatum�ddatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatumdatum�datum)rV r r rA r/ r �_bufferr r r r �testLateWriteBufferingp s
z(BufferedPullTests.testLateWriteBufferingN) r r r r rR r0 r2 r~ r� r r r r r{ Y s
r{ N)� __version__�twisted.protocolsr �
twisted.trialr r r r% �TestCaser- r7 r@ rM rP rS rT rU r[ r\ rf rl rm r{ r r r r �<module> s(
.2%5