0x1949 Team - FAZEMRX - MANAGER
Edit File: 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>