File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_policies.cpython-310.pyc
o
�b%� � @ s� d Z ddlZddlmZ ddlmZmZmZ ddlm Z m
Z
m Z m Z m
Z
ddlmZ ddlmZmZ ddlmZ G d d
� d
e j�ZG d d � d e j�ZG d
d� de j�ZG dd� de j�ZG dd� dej �Z G dd� dej �Z G dd� dej �Z!G dd� dej"�Z"G dd� dej �Z#G dd � d ej �Z$G d d � d ej �Z%G d d � d e jej&�Z'G d!d"� d"ej �Z(G d#d$� d$ej �Z)G d%d&� d&e�Z*G d'd(� d(ej+�Z,G d)d*� d*ej �Z-dS )+z
Test code for policies.
� N)�StringIO)� Interface�
implementedBy� implementer)�address�defer�protocol�reactor�task)�policies)�StringTransport� StringTransportWithDisconnection)�unittestc @ s8 e Zd Zd ZZdZdd� Zdd� Zdd� Zd d
� Z d S ) �SimpleProtocolr � c C s t �� | _t �� | _d S �N)r �Deferred�
dConnected�
dDisconnected��self� r �</usr/lib/python3/dist-packages/twisted/test/test_policies.py�__init__ s
zSimpleProtocol.__init__c C � d| _ | j�d� d S �N� � )� connectedr �callbackr r r r �connectionMade � z SimpleProtocol.connectionMadec C r r )� disconnectedr r �r �reasonr r r �connectionLost! r! z SimpleProtocol.connectionLostc C s | j |7 _ d S r )�buffer�r �datar r r � dataReceived% � zSimpleProtocol.dataReceivedN)
�__name__�
__module__� __qualname__r r" r& r r r% r) r r r r r s r c @ � e Zd Zdd� Zdd� ZdS )� SillyFactoryc C s
|| _ d S r ��p)r r1 r r r r * �
zSillyFactory.__init__c C s | j S r r0 )r �addrr r r �
buildProtocol- s zSillyFactory.buildProtocolN)r+ r, r- r r4 r r r r r/ ) � r/ c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � EchoProtocolFc C �
d| _ d S �NT��pausedr r r r �pauseProducing4 r2 zEchoProtocol.pauseProducingc C r8 )NFr: r r r r �resumeProducing7 r2 z EchoProtocol.resumeProducingc C s d S r r r r r r �
stopProducing: � zEchoProtocol.stopProducingc C s | j �|� d S r )� transport�writer' r r r r) = s zEchoProtocol.dataReceivedN)r+ r, r- r; r< r= r> r) r r r r r7 1 s r7 c @ s e Zd ZdZeZdS )�Serverz8
A simple server factory using L{EchoProtocol}.
N)r+ r, r- �__doc__r7 r r r r r rB A s rB c @ � e Zd ZdZdd� Zdd� ZdS )�TestableThrottlingFactoryzH
L{policies.ThrottlingFactory} using a L{task.Clock} for tests.
c O �$ t jj| g|�R i |�� || _dS �z�
@param clock: object providing a callLater method that can be used
for tests.
@type clock: C{task.Clock} or alike.
N)r �ThrottlingFactoryr �clock�r rI �args�kwargsr r r r N �
z"TestableThrottlingFactory.__init__c C � | j �||�S �z0
Forward to the testable clock.
�rI � callLater�r �period�funcr r r rQ W � z#TestableThrottlingFactory.callLaterN�r+ r, r- rC r rQ r r r r rE I � rE c @ rD )�TestableTimeoutFactoryzE
L{policies.TimeoutFactory} using a L{task.Clock} for tests.
c O rF rG )r �TimeoutFactoryr rI rJ r r r r c rM z TestableTimeoutFactory.__init__c C rN rO rP rR r r r rQ l rU z TestableTimeoutFactory.callLaterNrV r r r r rX ^ rW rX c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"S )#� WrapperTestsz>
Tests for L{WrappingFactory} and L{ProtocolWrapper}.
c C s8 t � }t�|�}|�t�ddd��}| �|jj|� dS )zf
Make sure protocol.factory is the wrapped factory, not the wrapping
factory.
�TCP� 127.0.0.1�# N) rB r �WrappingFactoryr4 r � IPv4Address�assertIs�wrappedProtocol�factory)r �f�wfr1 r r r � test_protocolFactoryAttributex �
z*WrapperTests.test_protocolFactoryAttributec C sl G dd� dt �}t|�G dd� d��}ttj� t�� }t�t�d�|�}|�|� � | � |�
|j �� dS )z�
The transport wrapper passed to the wrapped protocol's
C{makeConnection} provides the same interfaces as are provided by the
original transport.
c @ � e Zd ZdS )z=WrapperTests.test_transportInterfaces.<locals>.IStubTransportN�r+ r, r- r r r r �IStubTransport� � ri c @ rg )z<WrapperTests.test_transportInterfaces.<locals>.StubTransportNrh r r r r �
StubTransport� s rk N) r r r r �ProtocolWrapperr �Protocolr^ �makeConnection�
assertTrue�
providedByr@ )r ri rk �proto�wrapperr r r �test_transportInterfaces� s
z%WrapperTests.test_transportInterfacesc C s$ t � }t�|�}| �d|�� � dS )z�
L{WrappingFactory.logPrefix} is customized to mention both the original
factory and the wrapping factory.
zServer (WrappingFactory)N)rB r r^ � assertEqual� logPrefix)r �serverrb r r r �test_factoryLogPrefix� s
z"WrapperTests.test_factoryLogPrefixc C s2 G dd� d�}|� }t �|�}| �d|�� � dS )z�
If the wrapped factory doesn't have a L{logPrefix} method,
L{WrappingFactory.logPrefix} falls back to the factory class name.
c @ rg )z=WrapperTests.test_factoryLogPrefixFallback.<locals>.NoFactoryNrh r r r r � NoFactory� rj rx zNoFactory (WrappingFactory)N)r r^ rt ru )r rx rv rb r r r � test_factoryLogPrefixFallback� s
z*WrapperTests.test_factoryLogPrefixFallbackc C s8 t � }t�|�}|�t�ddd��}| �d|�� � dS )z{
L{ProtocolWrapper.logPrefix} is customized to mention both the original
protocol and the wrapper.
r[ r\ r] z EchoProtocol (ProtocolWrapper)N)rB r r^ r4 r r_ rt ru )r rv rb r r r r �test_protocolLogPrefix� rf z#WrapperTests.test_protocolLogPrefixc C sL G dd� d�}t � }||_t�|�}|�t�ddd��}| �d|�� � dS )z�
If the wrapped protocol doesn't have a L{logPrefix} method,
L{ProtocolWrapper.logPrefix} falls back to the protocol class name.
c @ rg )z?WrapperTests.test_protocolLogPrefixFallback.<locals>.NoProtocolNrh r r r r �
NoProtocol� rj r{ r[ r\ r] z NoProtocol (ProtocolWrapper)N) rB r r r^ r4 r r_ rt ru )r r{ rv rb r r r r � test_protocolLogPrefixFallback� s
z+WrapperTests.test_protocolLogPrefixFallbackc C s, t �t �t� �t�� �}t� }|�|� |S )zm
Return L{policies.ProtocolWrapper} that has been connected to a
L{StringTransport}.
)r rl r^ rB r rm r rn �r rr r@ r r r � _getWrapper� s �
zWrapperTests._getWrapperc C �" | � � }| �|�� |j�� � dS )zk
L{policies.ProtocolWrapper.getHost} calls C{getHost} on the underlying
transport.
N)r~ rt �getHostr@ �r rr r r r � test_getHost� � zWrapperTests.test_getHostc C r )zk
L{policies.ProtocolWrapper.getPeer} calls C{getPeer} on the underlying
transport.
N)r~ rt �getPeerr@ r� r r r � test_getPeer� r� zWrapperTests.test_getPeerc C s< | � � }t� }|�|d� | �|jj|� | �|jj� dS )z}
L{policies.ProtocolWrapper.registerProducer} calls C{registerProducer}
on the underlying transport.
TN)r~ �object�registerProducerr` r@ �producerro � streaming�r rr r� r r r �test_registerProducer� s
z"WrapperTests.test_registerProducerc C sB | � � }t� }|�|d� |�� | �|jj� | �|jj� dS )z�
L{policies.ProtocolWrapper.unregisterProducer} calls
C{unregisterProducer} on the underlying transport.
TN)r~ r� r� �unregisterProducer� assertIsNoner@ r� r� r� r r r �test_unregisterProducer� s z$WrapperTests.test_unregisterProducerc s6 | � � }g � � fdd�|j_|�� | �� dg� dS )zw
L{policies.ProtocolWrapper.stopConsuming} calls C{stopConsuming} on
the underlying transport.
c s
� � d�S r9 ��appendr ��resultr r �<lambda> s
z1WrapperTests.test_stopConsuming.<locals>.<lambda>TN)r~ r@ �
stopConsumingrt r� r r� r �test_stopConsuming s
z WrapperTests.test_stopConsumingc sD g � G � fdd�d�}t �|� �}t� }|�|� | �� |g� dS )z}
L{policies.WrappingFactory.startedConnecting} calls
C{startedConnecting} on the underlying factory.
c � e Zd Z� fdd�ZdS )z4WrapperTests.test_startedConnecting.<locals>.Factoryc s � � |� d S r r� )r � connectorr� r r �startedConnecting s zFWrapperTests.test_startedConnecting.<locals>.Factory.startedConnectingN)r+ r, r- r� r r� r r �Factory � r� N)r r^ r� r� rt )r r� rr r� r r� r �test_startedConnecting s
z#WrapperTests.test_startedConnectingc �P g � G � fdd�d�}t �|� �}t� }t� }|�||� | �� ||fg� dS )z�
L{policies.WrappingFactory.clientConnectionLost} calls
C{clientConnectionLost} on the underlying factory.
c r� )z7WrapperTests.test_clientConnectionLost.<locals>.Factoryc � � � ||f� d S r r� �r r� r$ r� r r �clientConnectionLost$ r* zLWrapperTests.test_clientConnectionLost.<locals>.Factory.clientConnectionLostN)r+ r, r- r� r r� r r r� # r� r� N)r r^ r� r� rt �r r� rr r� r$ r r� r �test_clientConnectionLost � z&WrapperTests.test_clientConnectionLostc r� )z�
L{policies.WrappingFactory.clientConnectionFailed} calls
C{clientConnectionFailed} on the underlying factory.
c r� )z9WrapperTests.test_clientConnectionFailed.<locals>.Factoryc r� r r� r� r� r r �clientConnectionFailed5 r* zPWrapperTests.test_clientConnectionFailed.<locals>.Factory.clientConnectionFailedN)r+ r, r- r� r r� r r r� 4 r� r� N)r r^ r� r� rt r� r r� r �test_clientConnectionFailed- r� z(WrapperTests.test_clientConnectionFailedc C sR t �t �t� �t�� �}t� }||_|�|� | �|j � |�
� | � |j � dS )z�
L{policies.ProtocolWrapper.connectionLost} sets C{wrappedProtocol} to
C{None} in order to break reference cycle between wrapper and wrapped
protocols.
:return:
N) r rl r^ rB r rm r
rn �assertIsNotNonera �loseConnectionr� r} r r r �test_breakReferenceCycle> s �
z%WrapperTests.test_breakReferenceCycleN)r+ r, r- rC re rs rw ry rz r| r~ r� r� r� r� r� r� r� r� r� r r r r rZ s s$
rZ c @ r. )r^ c C s |S r r )r rc r1 r r r r R r? zWrappingFactory.protocolc C s t j�| � | j�d � d S r )r r^ � startFactory�deferredr r r r r r� U s z WrappingFactory.startFactoryN)r+ r, r- r r� r r r r r^ Q r5 r^ c @ r6 ) �ThrottlingTestsz2
Tests for L{policies.ThrottlingFactory}.
c s t � }dd� td�D �\� ���t�|d��t��}t�� |_tj d|dd����
� j �� ���fdd �}� ����fd
d �}� fd d
�}��fdd�}��fdd�}���fdd�}|j� |� |j� |� |j� |� |j� |� |j� |� |j� |� |jS )z�
Full test using a custom server limiting number of connections.
FIXME: https://twistedmatrix.com/trac/ticket/10012
This is a flaky test.
c s s � | ]}t � V qd S r )r )�.0�ir r r � <genexpr>g s � z-ThrottlingTests.test_limit.<locals>.<genexpr>� � r r\ )� interfacec sD t �d�t� �� � j���fdd�� �j���fdd�� �jS )Nr\ c � t �d�t� ��S �Nr\ �r �
connectTCPr/ ��r)�c2�nr r r� s � zAThrottlingTests.test_limit.<locals>._connect123.<locals>.<lambda>c r� r� r� r� )�c3r� r r r� v r� )r r� r/ r � addCallbackr ��results)�c1r� r� r� r r � _connect123p s � �z/ThrottlingTests.test_limit.<locals>._connect123c sZ �� dd� � ��fD �g d�� �� dd� � ��fD �g d�� �� t�j�� �d� | S )Nc S � g | ]}|j �qS r )r �r� �cr r r �
<listcomp>{ r� zAThrottlingTests.test_limit.<locals>._check123.<locals>.<listcomp>)r r r c S r� r )r" r� r r r r� | r� )r r r r� )rt �len� protocols�keysr� )r� r� r� r �tServerr r � _check123z s z-ThrottlingTests.test_limit.<locals>._check123c s � j �� � jS r )r@ r� r r� )r� r r �_lose1� s
z*ThrottlingTests.test_limit.<locals>._lose1c s t �d�t� �� � jS r� )r r� r/ r r� )�c4r� r r � _connect4� s z-ThrottlingTests.test_limit.<locals>._connect4c s �� � jd� �� � jd� | S )Nr r )rt r r" r� )r� r r r �_check4� s z+ThrottlingTests.test_limit.<locals>._check4c s4 � �fD ]}|j �� qt�t��j�� j�jg�S r )r@ r� r � DeferredList�
maybeDeferred�
stopListeningr )r� r� )r� r� r1 r r �_cleanup� s
��z,ThrottlingTests.test_limit.<locals>._cleanup)
rB �ranger rH r^ r r r� r � listenTCPr� �portr� ) r rv � wrapTServerr� r� r� r� r� r� r )r� r� r� r� r� r1 r r� r �
test_limit_ s(
zThrottlingTests.test_limitc C sp t � }tt�� |�}|�t�ddd��}t� }||_|� |� |�
dgd � | � |� � d� | � |j
d� dS ) zZ
L{ThrottlingProtocol.writeSequence} is called on the underlying factory.
r[ r\ r � bytesr� s bytesbytesbytesbytes� N)rB rE r
�Clockr4 r r_ r
r rn �
writeSequencert �value�writtenThisSecond)r rv r� r r@ r r r �test_writeSequence� s
z"ThrottlingTests.test_writeSequencec C s� t � }tt�� |dd�}|�t�ddd��}t� }||_|� |� |j
|_ |� d� |� d� | �
|�� d� | �
|jd � | �|j
j� |j�d
� | �
|jd� | �|j
j� |j�d
� | �
|jd� | �|j
j� d S ) ze
Check the writeLimit parameter: write data, and check for the pause
status.
�
)�
writeLimitr[ r\ r �
0123456789�
abcdefghij� 0123456789abcdefghijr� ��������?N)rB rE r
r� r4 r r_ r
r rn ra r� r) rt r� r� � assertFalser; rI �advancero �r rv r� r� �trr r r �test_writeLimit� s$
z ThrottlingTests.test_writeLimitc C sR t � }tt�� |dd�}|�t�ddd��}t� }||_|� |� |�
d� |�
d� | � |� � d� | � |j
d � |j�d
� | � |j
d� | � |jd � |j�d
� | � |j
d� | � |jd � |�� |�
d� |�
d� | � |� � d� | � |j
d � |j�d
� | � |j
d� | � |jd � |j�d
� | � |j
d� | � |jd � d
S )zb
Check the readLimit parameter: read data and check for the pause
status.
r� )� readLimitr[ r\ r r� r� r� r� r� r; � producingN)rB rE r
r� r4 r r_ r
r rn r) rt r� �readThisSecondrI r� �
producerState�clearr� r r r �test_readLimit� s6
z ThrottlingTests.test_readLimitN)r+ r, r- rC r� r� r� r� r r r r r� Z s B r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�TimeoutProtocolTestsz0
Tests for L{policies.TimeoutProtocol}.
c C sT t �� }t�� }t|_t||d�}|�t�ddd��}t � }||_|�
|� ||fS )z�
Helper to set up an already connected protocol to be tested.
@return: A new protocol with its attached clock.
@rtype: Tuple of (L{policies.TimeoutProtocol}, L{task.Clock})
Nr[ r\ �90 ) r
r� r �
ServerFactoryr rX r4 r r_ r
rn )r rI �wrappedFactoryrb rq r@ r r r �getProtocolAndClock� s
z(TimeoutProtocolTests.getProtocolAndClockc C sj | � � \}}|�d� | �|j� | �|jj� |�d� |�� | � |j� |�d� | �|jj� dS )z2
Will cancel the ongoing timeout.
� r N)
r� �
setTimeoutr� � timeoutCallr� ra r" r� �
cancelTimeoutr� �r �sutrI r r r �test_cancelTimeout s
z'TimeoutProtocolTests.test_cancelTimeoutc C s2 | � � \}}| �|j� |�� | �|jj� dS )z<
Does nothing if no timeout is already set.
N)r� r� r� r� r� ra r" r� r r r �test_cancelTimeoutNoTimeout$ s z0TimeoutProtocolTests.test_cancelTimeoutNoTimeoutc C s> | � � \}}|j}|�d� |�d� | �|j� |�� dS )z@
Does nothing if no timeout is already reached.
r� N)r� ra r� r� ro r" r� )r r� rI � wrappedProtor r r � test_cancelTimeoutAlreadyCalled0 s
z4TimeoutProtocolTests.test_cancelTimeoutAlreadyCalledc C s: | � � \}}|�d� |j�� |�� | �|jj� dS )zm
Does nothing if the timeout is cancelled from another part.
Ex from another thread.
r� N)r� r� r� �cancelr� r� ra r" r� r r r �"test_cancelTimeoutAlreadyCancelled> s
z7TimeoutProtocolTests.test_cancelTimeoutAlreadyCancelledN) r+ r, r- rC r� r� r� r� r� r r r r r� � s r� c @ r6 ) �TimeoutFactoryTestsz/
Tests for L{policies.TimeoutFactory}.
c C sn t �� | _t�� }t|_t| j|d�| _| j�t �
ddd��| _ t � | _
| j | j
_| j �| j
� | j j| _dS )zq
Create a testable, deterministic clock, and a set of
server factory/protocol/transport.
r� r[ r\ r� N)r
r� rI r r� r rX rb r4 r r_ rq r
r@ rn ra r� )r r� r r r �setUpS s
�
zTimeoutFactoryTests.setUpc C s@ | j �g d�� | �| jj� | j �ddg� | �| jj� dS )z�
Make sure that when a TimeoutFactory accepts a connection, it will
time out that connection if no data is read or written within the
timeout period.
)� � �?� �?r g�������?r 皙�����?N)rI �pumpr� r� r" ro r r r r � test_timeoutd s z TimeoutFactoryTests.test_timeoutc C s� | j �g d�� | �| jj� | j�d� | j �g d�� | �| jj� | j�dgd � | j �g d�� | �| jj� | j �ddg� | �| jj� dS ) z�
Make sure that writing data to a transport from a protocol
constructed by a TimeoutFactory resets the timeout countdown.
)r r r � bytes bytes bytes�r r r r� r� r g @N) rI r r� r� r" rq rA r� ro r r r r �test_sendAvoidsTimeoutr s z*TimeoutFactoryTests.test_sendAvoidsTimeoutc C sj | j �g d�� | �| jj� | j�d� | j �g d�� | �| jj� | j �g d�� | �| jj� dS )zR
Make sure that receiving data also resets the timeout countdown.
)r r r r r N)rI r r� r� r" rq r) ro r r r r �test_receiveAvoidsTimeout� s z-TimeoutFactoryTests.test_receiveAvoidsTimeoutN)r+ r, r- rC r r r r
r r r r r� N s r� c @ sJ e Zd ZdZdZdZdd� Zdd� Zdd � Zdd d �Z d
d� Z
dd� Z d
S )�
TimeoutTesterz�
A testable protocol with timeout facility.
@ivar timedOut: set to C{True} if a timeout has been detected.
@type timedOut: C{bool}
r� Fc C s
|| _ dS )zF
Initialize the protocol with a C{task.Clock} object.
N)rI )r rI r r r r � �
zTimeoutTester.__init__c C s | � | j� dS )z3
Upon connection, set the timeout.
N)r� �timeOutr r r r r � s z TimeoutTester.connectionMadec C s | � � tj�| |� dS )z,
Reset the timeout on data.
N)� resetTimeoutr rm r) r' r r r r) � s zTimeoutTester.dataReceivedNc C s | � d� dS )zD
On connection lost, cancel all timeout operations.
N)r� r# r r r r% � rU z TimeoutTester.connectionLostc C s
d| _ dS )zX
Flags the timedOut variable to indicate the timeout of the connection.
TN)�timedOutr r r r �timeoutConnection� r z TimeoutTester.timeoutConnectionc O s | j j||g|�R i |��S )zD
Override callLater to use the deterministic clock.
rP )r �timeoutrT rK rL r r r rQ � s zTimeoutTester.callLaterr ) r+ r, r- rC r
r r r r) r% r rQ r r r r r � s
r c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�TimeoutMixinTestsz-
Tests for L{policies.TimeoutMixin}.
c C s t �� | _t| j�| _dS )zY
Create a testable, deterministic clock and a C{TimeoutTester} instance.
N)r
r� rI r rq r r r r r � s
zTimeoutMixinTests.setUpc C s$ | j �d� | �t| jj�d� dS )z�
Test that the callLater of the clock is used instead of
L{reactor.callLater<twisted.internet.interfaces.IReactorTime.callLater>}
r� r N)rq r� rt r� rI �callsr r r r �test_overriddenCallLater� s z*TimeoutMixinTests.test_overriddenCallLaterc C sN | j �t� � | j�g d�� | �| j j� | j�ddg� | �| j j� dS )zq
Check that the protocol does timeout at the time specified by its
C{timeOut} attribute.
�r r r r r r N)rq rn r rI r r� r ro r r r r r � s
z TimeoutMixinTests.test_timeoutc C sx | j �t� � | j�g d�� | �| j j� | j �d� | j�g d�� | �| j j� | j�ddg� | �| j j� dS )zV
Check that receiving data is delaying the timeout of the connection.
r s hello there)r r r r r r N) rq rn r rI r r� r r) ro r r r r �test_noTimeout� s z TimeoutMixinTests.test_noTimeoutc C sr d| j _| j �t� � | j �d� | �| j jd� | j�ddg� | �| j j � | j�ddg� | �
| j j � dS )zy
Check that setting a new value for timeout cancel the previous value
and install a new timeout.
Nr r g�������?r ) rq r
rn r r� rt rI r r� r ro r r r r �test_resetTimeout s z#TimeoutMixinTests.test_resetTimeoutc C sR d| j _| j �t� � | j �d� | �| j j� | j�g d�� | �| j j � dS )zO
Setting the timeout to L{None} cancel any timeout operations.
� N)r r r r )
rq r
rn r r� r� rI r r� r r r r r r� s z$TimeoutMixinTests.test_cancelTimeoutc C sb d| j _| �| j �d�d� | �| j �d�d� | �| j �d�� | �| j jd� | j �d� dS )zM
setTimeout should return the value of the previous timeout.
r r� Nr )rq r
rt r� r� r r r r �test_setTimeoutReturn s z'TimeoutMixinTests.test_setTimeoutReturnc C sJ | j �d� | j�� d �� | �| j j� | j �d� | �| j j� dS )z�
When the timeout was already cancelled from an external place,
calling setTimeout with C{None} to explicitly cancel it will clean
up the timeout without raising any exception.
r� r N)rq r� rI �getDelayedCallsr� r� r
r� r r r r �%test_setTimeoutCancleAlreadyCancelled. s
z7TimeoutMixinTests.test_setTimeoutCancleAlreadyCancelledN) r+ r, r- rC r r r r r r� r r r r r r r � s
r c @ rD )�!LimitTotalConnectionsFactoryTestsz/Tests for policies.LimitTotalConnectionsFactoryc C s� t �� }tj|_| �d|j� |�d �}| �d|j� |�d �}| �d|j� |�d � | �d|j� |�d � | �d|j� d S )Nr r r� )r � LimitTotalConnectionsFactoryr rm rt �connectionCountr4 r% )r rb �p1�p2r r r �testConnectionCountingB s
z8LimitTotalConnectionsFactoryTests.testConnectionCountingc s� t �� � tj� _d� _� �d �}| �|� | �d� j� | � � �d �� | �d� j� G � fdd�dtj�}|� _
d� _ � �d �}|� d � | �
� j � | �d� j� |�d � | �d� j� |�d � | �d� j� d S )Nr c r� )zRLimitTotalConnectionsFactoryTests.testConnectionLimiting.<locals>.OverflowProtocolc s
d� _ d S r9 )�
overflowedr �rb r r r h r2 zaLimitTotalConnectionsFactoryTests.testConnectionLimiting.<locals>.OverflowProtocol.connectionMadeN)r+ r, r- r r r# r r �OverflowProtocolg r� r$ Fr� r )r r r rm �connectionLimitr4 r� rt r r� �overflowProtocolr" rn ro r% )r r1 r$ �opr r# r �testConnectionLimitingV s&
z8LimitTotalConnectionsFactoryTests.testConnectionLimitingN)r+ r, r- rC r! r( r r r r r ? s r c @ s e Zd Zdd� ZdS )�WriteSequenceEchoProtocolc C s0 |� d�dkr| j�|g� d S t�| |� d S )Ns vector!���)�findr@ r� r7 r) )r �bytesr r r r) ~ s z&WriteSequenceEchoProtocol.dataReceivedN)r+ r, r- r) r r r r r) } s r) c @ s e Zd ZdZdd� ZdS )�TestLoggingFactoryNc C s | j d u s J d��t� | _ | j S )Nz open() called too many times)�openFiler )r �namer r r �open� s zTestLoggingFactory.open)r+ r, r- r. r0 r r r r r- � s r- c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �LoggingFactoryTestsz6
Tests for L{policies.TrafficLoggingFactory}.
c C s t � }t|_t� }t|d�}|�d�}||_|�|� |j�� }| � d|� | �
|� � � |� d� |j�� }| � d�
d�|� | � d�
d�|� | �|� � d� |�� |� d� |j�� }| � d�
dg�|� | �|� � d� |�� |j�� }| � d |� d
S ) zQ
Check the output produced by L{policies.TrafficLoggingFactory}.
�test��1.2.3.4i. �*s here are some bytesz C 1: {!r}z S 1: {!r}s" prepare for vector! to the extremez
SV 1: {!r}�ConnectionDoneN)rB r) r r
r- r4 rn r. �getvalue�assertInr� r� r) �formatrt r� r� )r r� �trc r1 �vr r r �test_thingsGetLogged� s.
z(LoggingFactoryTests.test_thingsGetLoggedc C sn t � }t|d�}| �|jd� |�d� | �|jd� d|_|�d� | �|jd� |�� | �|jd� dS )zG
Test counter management with the resetCounter method.
r2 r r3 r N)r4 i/ r� )rB r- rt �_counterr4 r. � resetCounter)r r� rc r r r � test_counter� s
z LoggingFactoryTests.test_counterc s� g � g �� �fdd�}| � td|� t�� }t|_t�|d�}|�t� ddd��}|�t� ddd��}d i f}d
i f}| �
||g� � | �
|j |j g�� d S ) z�
When the L{policies.TrafficLoggingFactory} builds a protocol, it
automatically opens a unique log file for that protocol and attaches
the logfile to the built protocol.
c s, � � | |f� t� }| d |_�� |� |S )zX
Mock for the open call to prevent actually opening a log file.
r )r� r r/ )rK rL �io��
open_calls� open_rvaluesr r � mocked_open� s
zULoggingFactoryTests.test_loggingFactoryOpensLogfileAutomatically.<locals>.mocked_openr0 r2 r[ r\ r� i:0 )ztest-1�w)ztest-2rE N) �patch�builtinsr r� r r �TrafficLoggingFactoryr4 r r_ rt �logfile)r rD r� rb � first_proto� second_proto�
first_call� second_callr rA r �,test_loggingFactoryOpensLogfileAutomatically� s"
� �z@LoggingFactoryTests.test_loggingFactoryOpensLogfileAutomaticallyN)r+ r, r- rC r< r? rN r r r r r1 � s
" r1 ).rC rG r@ r �zope.interfacer r r �twisted.internetr r r r r
�twisted.protocolsr �twisted.test.proto_helpersr r
�
twisted.trialr rm r �
ClientFactoryr/ r7 r� rB rH rE rY rX �TestCaserZ r^ r� r� r� � TimeoutMixinr r r r) rH r- r1 r r r r �<module> s6 _ XU1k>