File: //lib/python3/dist-packages/twisted/internet/test/__pycache__/test_protocol.cpython-38.pyc
U
W['I � @ s d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z ddl m
Z
mZmZmZ ddlmZmZmZmZmZmZ dd lmZmZ dd
lm Z dd l m Z dd l m Z m!Z! dd
l"m#Z# G dd� de#�Z$G dd� de#�Z%G dd� de#�Z&G dd� de#�Z'G dd� de#�Z(dS )z)
Tests for L{twisted.internet.protocol}.
� )�division�absolute_import)�BytesIO)� implementer)� verifyObject)�CancelledError)� IProtocol�ILoggingContext�IProtocolFactory� IConsumer)�Protocol�
ClientCreator�Factory�ProtocolToConsumerAdapter�ConsumerToProtocolAdapter� FileWrapper)�LogLevel�globalLogPublisher)�_PY3)�Failure)�MemoryReactorClock�StringTransport)�TestCasec @ 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 )#�ClientCreatorTestsz?
Tests for L{twisted.internet.protocol.ClientCreator}.
c C sD G dd� dt �}t� }t||�}|||�}|�d�}| �||� dS )a
Helper for implementing a test to verify that one of the I{connect}
methods of L{ClientCreator} passes the right arguments to the right
reactor method.
@param check: A function which will be invoked with a reactor and a
L{ClientCreator} instance and which should call one of the
L{ClientCreator}'s I{connect} methods and assert that all of its
arguments except for the factory are passed on as expected to the
reactor. The factory should be returned.
c @ s e Zd ZdS )z:ClientCreatorTests._basicConnectTest.<locals>.SomeProtocolN��__name__�
__module__� __qualname__� r r �E/usr/lib/python3/dist-packages/twisted/internet/test/test_protocol.py� SomeProtocol- s r N)r r r
�
buildProtocol�assertIsInstance)�self�checkr �reactor�cc�factory�protocolr r r �_basicConnectTest! s
z$ClientCreatorTests._basicConnectTestc s � fdd�}� � |� dS )z�
L{ClientCreator.connectTCP} calls C{reactor.connectTCP} with the host
and port information passed to it, and with a factory which will
construct the protocol passed to L{ClientCreator.__init__}.
c sX |� dddd� | j�� \}}}}}� �|d� � �|d� � �|d� � �|d� |S )N� example.com�� �� )z1.2.3.4i�& )�
connectTCP�
tcpClients�pop� assertEqual)r% r& �host�portr' �timeout� bindAddress�r# r r r$ = s z1ClientCreatorTests.test_connectTCP.<locals>.checkN�r) �r# r$ r r5 r �test_connectTCP7 s z"ClientCreatorTests.test_connectTCPc s � fdd�}� � |� dS )z�
L{ClientCreator.connectUNIX} calls C{reactor.connectUNIX} with the
filename passed to it, and with a factory which will construct the
protocol passed to L{ClientCreator.__init__}.
c sF |� ddd� | j�� \}}}}� �|d� � �|d� � �|� |S )N�/foo/bar�{ T)� connectUNIX� unixClientsr/ r0 �
assertTrue)r% r& �addressr' r3 ZcheckPIDr5 r r r$ N s
z2ClientCreatorTests.test_connectUNIX.<locals>.checkNr6 r7 r r5 r �test_connectUNIXH s z#ClientCreatorTests.test_connectUNIXc s � fdd�}� � |� dS )z�
L{ClientCreator.connectSSL} calls C{reactor.connectSSL} with the host,
port, and context factory passed to it, and with a factory which will
construct the protocol passed to L{ClientCreator.__init__}.
c sn t � }|�dd|dd� | j�� \}}}}}}� �|d� � �|d� � �||� � �|d� � �|d� |S )Nr* r+ r, )z4.3.2.1i. )�object�
connectSSL�
sslClientsr/ r0 �assertIs) r% r& ZexpectedContextFactoryr1 r2 r' �contextFactoryr3 r4 r5 r r r$ ^ s z1ClientCreatorTests.test_connectSSL.<locals>.checkNr6 r7 r r5 r �test_connectSSLX s
z"ClientCreatorTests.test_connectSSLc C sN t � }t|t�}||�}|j�� }| �|j� |�� | �|j� | � |t
�S )a
Helper for implementing a test to verify that cancellation of the
L{Deferred} returned by one of L{ClientCreator}'s I{connect} methods is
implemented to cancel the underlying connector.
@param connect: A function which will be invoked with a L{ClientCreator}
instance as an argument and which should call one its I{connect}
methods and return the result.
@return: A L{Deferred} which fires when the test is complete or fails if
there is a problem.
) r r
r �
connectorsr/ Z assertFalse�
_disconnected�cancelr= �
assertFailurer �r# �connectr% r& �d� connectorr r r �_cancelConnectTestk s
z%ClientCreatorTests._cancelConnectTestc C s dd� }| � |�S )��
The L{Deferred} returned by L{ClientCreator.connectTCP} can be cancelled
to abort the connection attempt before it completes.
c S s | � dd�S �Nr* r+ )r- �r& r r r rK � s z9ClientCreatorTests.test_cancelConnectTCP.<locals>.connect�rN �r# rK r r r �test_cancelConnectTCP� s z(ClientCreatorTests.test_cancelConnectTCPc C s dd� }| � |�S )rO c S s
| � d�S �Nr9 )r; rQ r r r rK � s z:ClientCreatorTests.test_cancelConnectUNIX.<locals>.connectrR rS r r r �test_cancelConnectUNIX� s z)ClientCreatorTests.test_cancelConnectUNIXc C s dd� }| � |�S )rO c S s | � ddt� �S rP )rA r@ rQ r r r rK � s z9ClientCreatorTests.test_cancelConnectSSL.<locals>.connectrR rS r r r �test_cancelConnectSSL� s z(ClientCreatorTests.test_cancelConnectSSLc C sh t � }t|t�}|||�}|j�� }| �t|�� �d� |�� | �|�� g � | � |j
� | � |t �S )z�
Like L{_cancelConnectTest}, but for the case where the L{Deferred} is
cancelled after the connection is set up but before it is fired with the
resulting protocol instance.
� )
r r
r rF r/ r0 �len�getDelayedCallsrH r= rG rI r rJ r r r �_cancelConnectTimeoutTest� s
z,ClientCreatorTests._cancelConnectTimeoutTestc C s dd� }| � |�S )a�
L{ClientCreator.connectTCP} inserts a very short delayed call between
the time the connection is established and the time the L{Deferred}
returned from one of its connect methods actually fires. If the
L{Deferred} is cancelled in this interval, the established connection is
closed, the timeout is cancelled, and the L{Deferred} fails with
L{CancelledError}.
c
S s> |� dd�}| j�� \}}}}}|�d �}t� } |�| � |S rP )r- r. r/ r! r �makeConnection)
r% r& rL r1 r2 r' r3 r4 r( � transportr r r rK � s
z@ClientCreatorTests.test_cancelConnectTCPTimeout.<locals>.connect�r[ rS r r r � test_cancelConnectTCPTimeout� s z/ClientCreatorTests.test_cancelConnectTCPTimeoutc C s dd� }| � |�S )a�
L{ClientCreator.connectUNIX} inserts a very short delayed call between
the time the connection is established and the time the L{Deferred}
returned from one of its connect methods actually fires. If the
L{Deferred} is cancelled in this interval, the established connection is
closed, the timeout is cancelled, and the L{Deferred} fails with
L{CancelledError}.
c S s: |� d�}| j�� \}}}}|�d �}t� }|�|� |S rU )r; r< r/ r! r r\ ) r% r& rL r> r' r3 r4 r( r] r r r rK � s
zAClientCreatorTests.test_cancelConnectUNIXTimeout.<locals>.connectr^ rS r r r � test_cancelConnectUNIXTimeout� s z0ClientCreatorTests.test_cancelConnectUNIXTimeoutc C s dd� }| � |�S )a�
L{ClientCreator.connectSSL} inserts a very short delayed call between
the time the connection is established and the time the L{Deferred}
returned from one of its connect methods actually fires. If the
L{Deferred} is cancelled in this interval, the established connection is
closed, the timeout is cancelled, and the L{Deferred} fails with
L{CancelledError}.
c S sD |� ddt� �}| j�� \}}}}}}|�d �} t� }
| �|
� |S rP )rA r@ rB r/ r! r r\ ) r% r& rL r1 r2 r' rD r3 � bindADdressr( r] r r r rK � s
z@ClientCreatorTests.test_cancelConnectSSLTimeout.<locals>.connectr^ rS r r r � test_cancelConnectSSLTimeout� s z/ClientCreatorTests.test_cancelConnectSSLTimeoutc C st t � }t|t�}|||�\}}|j�� }|�|ttd��� | �t |�
� �d� |� � | �|�
� g � | � |t
�S )z�
Like L{_cancelConnectTest}, but for the case where the L{Deferred} is
cancelled after the connection attempt has failed but before it is fired
with the resulting failure.
zSimulated failurerX )r r
r rF r/ ZclientConnectionFailedr � Exceptionr0 rY rZ rH rI r )r# rK r% r& rL r' rM r r r � _cancelConnectFailedTimeoutTest� s
�z2ClientCreatorTests._cancelConnectFailedTimeoutTestc C s dd� }| � |�S )zz
Similar to L{test_cancelConnectTCPTimeout}, but for the case where the
connection attempt fails.
c S s( |� dd�}| j�� \}}}}}||fS rP )r- r. r/ )r% r& rL r1 r2 r' r3 r4 r r r rK s zFClientCreatorTests.test_cancelConnectTCPFailedTimeout.<locals>.connect�rd rS r r r �"test_cancelConnectTCPFailedTimeout s z5ClientCreatorTests.test_cancelConnectTCPFailedTimeoutc C s dd� }| � |�S )z{
Similar to L{test_cancelConnectUNIXTimeout}, but for the case where the
connection attempt fails.
c S s$ |� d�}| j�� \}}}}||fS rU )r; r< r/ )r% r& rL r> r' r3 r4 r r r rK s
zGClientCreatorTests.test_cancelConnectUNIXFailedTimeout.<locals>.connectre rS r r r �#test_cancelConnectUNIXFailedTimeout s z6ClientCreatorTests.test_cancelConnectUNIXFailedTimeoutc C s dd� }| � |�S )zz
Similar to L{test_cancelConnectSSLTimeout}, but for the case where the
connection attempt fails.
c S s. |� ddt� �}| j�� \}}}}}}||fS rP )rA r@ rB r/ ) r% r& rL r1 r2 r' rD r3 ra r r r rK + s zFClientCreatorTests.test_cancelConnectSSLFailedTimeout.<locals>.connectre rS r r r �"test_cancelConnectSSLFailedTimeout& s z5ClientCreatorTests.test_cancelConnectSSLFailedTimeoutN)r r r �__doc__r) r8 r? rE rN rT rV rW r[ r_ r` rb rd rf rg rh r r r r r s"
r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �
ProtocolTestsz:
Tests for L{twisted.internet.protocol.Protocol}.
c C s* t � }| �tt|�� | �tt|�� dS )zT
L{Protocol} instances provide L{IProtocol} and L{ILoggingContext}.
N)r r= r r r )r# �protor r r �test_interfaces7 s z ProtocolTests.test_interfacesc C s& G dd� dt �}| �d|� �� � dS )zJ
L{Protocol.logPrefix} returns the protocol class's name.
c @ s e Zd ZdS )z/ProtocolTests.test_logPrefix.<locals>.SomeThingNr r r r r � SomeThingD s rm N)r r0 � logPrefix)r# rm r r r �test_logPrefix@ s z ProtocolTests.test_logPrefixc s@ g � G � fdd�dt �}t� }|� }|�|� | �� |g� dS )zz
L{Protocol.makeConnection} sets the given transport on itself, and
then calls C{connectionMade}.
c s e Zd Z� fdd�ZdS )z7ProtocolTests.test_makeConnection.<locals>.SomeProtocolc s � � | j� d S �N)�appendr] r5 ��resultr r �connectionMadeP s zFProtocolTests.test_makeConnection.<locals>.SomeProtocol.connectionMadeN)r r r rt r rr r r r O s r N)r r@ r\ r0 )r# r r] r( r rr r �test_makeConnectionI s
z!ProtocolTests.test_makeConnectionN)r r r ri rl ro ru r r r r rj 3 s rj c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� FactoryTestsz(
Tests for L{protocol.Factory}.
c C s* t � }| �tt|�� | �tt|�� dS )zg
L{Factory} instances provide both L{IProtocolFactory} and
L{ILoggingContext}.
N)r r= r r
r )r# r' r r r rl _ s z FactoryTests.test_interfacesc C s& G dd� dt �}| �d|� �� � dS )zM
L{Factory.logPrefix} returns the name of the factory class.
c @ s e Zd ZdS )z6FactoryTests.test_logPrefix.<locals>.SomeKindOfFactoryNr r r r r �SomeKindOfFactorym s rw N)r r0 rn )r# rw r r r ro i s zFactoryTests.test_logPrefixc C sD G dd� dt �}t� }||_|�d�}| �||� | �|j|� dS )z�
L{Factory.buildProtocol} by default constructs a protocol by calling
its C{protocol} attribute, and attaches the factory to the result.
c @ s e Zd ZdS )z<FactoryTests.test_defaultBuildProtocol.<locals>.SomeProtocolNr r r r r r x s r N)r r r( r! r" rC r' )r# r �fr( r r r �test_defaultBuildProtocols s
z&FactoryTests.test_defaultBuildProtocolc C sT G dd� dt �}|jtdddd�}| �|jt� | �|jd� | �|jddi� d S )
z�
L{Factory.forProtocol} constructs a Factory, passing along any
additional arguments, and sets its C{protocol} attribute to the given
Protocol subclass.
c @ s e Zd Zdd� ZdS )z7FactoryTests.test_forProtocol.<locals>.ArgTakingFactoryc _ s || | _ | _d S rp )�args�kwargs)r# rz r{ r r r �__init__� s z@FactoryTests.test_forProtocol.<locals>.ArgTakingFactory.__init__N)r r r r| r r r r �ArgTakingFactory� s r} rX � � )�foo)rX r~ r� N)r Z forProtocolr r0 r( rz r{ )r# r} r' r r r �test_forProtocol� s
z FactoryTests.test_forProtocolc sr g � t �� j� | �� fdd�� t� }|�� | �� d d |� | �� d d tj � | �� d d d� dS ) z�
L{Factory.doStart} logs that it is starting a factory, followed by
the L{repr} of the L{Factory} instance that is being started.
c s t �� j�S rp �r ZremoveObserverrq r �Zeventsr r �<lambda>� � z;FactoryTests.test_doStartLoggingStatement.<locals>.<lambda>r r' � log_level�
log_formatz Starting factory {factory!r}N)
r � addObserverrq �
addCleanupr ZdoStartrC r0 r �info)r# rx r r� r � test_doStartLoggingStatement� s
��z)FactoryTests.test_doStartLoggingStatementc s� g � t �� j� | �� fdd�� G dd� dt�}|� }|�� | �� d d |� | �� d d tj � | �� d d d � d
S ) z�
L{Factory.doStop} logs that it is stopping a factory, followed by
the L{repr} of the L{Factory} instance that is being stopped.
c s t �� j�S rp r� r r� r r r� � r� z:FactoryTests.test_doStopLoggingStatement.<locals>.<lambda>c @ s e Zd ZdZdS )z;FactoryTests.test_doStopLoggingStatement.<locals>.MyFactoryrX N)r r r ZnumPortsr r r r � MyFactory� s r� r r' r� r� z Stopping factory {factory!r}N)
r r� rq r� r ZdoStoprC r0 r r� )r# r� rx r r� r �test_doStopLoggingStatement� s
��z(FactoryTests.test_doStopLoggingStatementN)
r r r ri rl ro ry r� r� r� r r r r rv Z s
rv c @ s e Zd ZdZdd� Zdd� ZdS )� AdapterTestszR
Tests for L{ProtocolToConsumerAdapter} and L{ConsumerToProtocolAdapter}.
c C sB g }t � }|j|_t|�}|�d� | �|dg� | �|t� dS )z}
L{IProtocol} providers can be adapted to L{IConsumer} providers using
L{ProtocolToConsumerAdapter}.
� helloN)r rq � dataReceivedr �writer0 r" r )r# rs �pZconsumerr r r �test_protocolToConsumer� s
z$AdapterTests.test_protocolToConsumerc sV g � t t�G � fdd�dt��}|� }t|�}|�d� | �� dg� | �|t� dS )z}
L{IConsumer} providers can be adapted to L{IProtocol} providers using
L{ProtocolToConsumerAdapter}.
c s e Zd Z� fdd�ZdS )z6AdapterTests.test_consumerToProtocol.<locals>.Consumerc s � � |� d S rp )rq )r# rL rr r r r� � s z<AdapterTests.test_consumerToProtocol.<locals>.Consumer.writeN)r r r r� r rr r r �Consumer� s r� r� N)r r r@ r r� r0 r" r )r# r� �cr( r rr r �test_consumerToProtocol� s
z$AdapterTests.test_consumerToProtocolN)r r r ri r� r� r r r r r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )�FileWrapperTestsz2
L{twisted.internet.protocol.FileWrapper}
c C sP t t� �}|�d� | �|j�� d� t t� �}|�d� | �|j�� d� dS )z@
L{twisted.internet.protocol.FileWrapper.write}
� test1ZstuffN)r r r� r0 �file�getvalueZassertNotEqual�r# �wrapperr r r �
test_write� s
zFileWrapperTests.test_writec C sx t t� �}|�ddg� | �|j�� d� t t� �}trN| �t|jddg� n&|�ddg� | � t
|j�� �dk� dS )zH
L{twisted.internet.protocol.FileWrapper.writeSequence}
r� s test2s
test1test2Ztest3Ztest4r N) r r Z
writeSequencer0 r� r� r Z assertRaises� TypeErrorr= rY r� r r r �test_writeSequence� s
�z#FileWrapperTests.test_writeSequenceN)r r r ri r� r� r r r r r� � s r� N))ri Z
__future__r r �ior Zzope.interfacer Zzope.interface.verifyr Ztwisted.internet.deferr Ztwisted.internet.interfacesr r r
r Ztwisted.internet.protocolr r
r r r r Ztwisted.loggerr r Ztwisted.python.compatr Ztwisted.python.failurer Ztwisted.test.proto_helpersr r Ztwisted.trial.unittestr r rj rv r� r� r r r r �<module> s& '`%