← Back
Editing: test_protocols.cpython-38.pyc
U W[ � @ s� d Z ddlmZ ddlmZmZ ddlmZ ddlm Z m Z m Z m Z ddl mZ G dd� dej�ZG d d � d ej�ZG d d � d ej�ZG d d� dej�ZG dd� dej�ZdS )z+ Test cases for twisted.protocols package. � )�unittest)�wire� portforward)� iterbytes)�reactor�defer�address�protocol)� proto_helpersc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � WireTestsz Test wire protocols. c C s` t �� }t�� }|�|� |�d� |�d� |�d� |�d� |�d� | �|�� d� dS )zT Test wire.Echo protocol: send some data and check it send it back. � hello� world� how� are� yous helloworldhowareyouN)r �StringTransportr �Echo�makeConnection� dataReceived� assertEqual�value��self�t�a� r �=/usr/lib/python3/dist-packages/twisted/test/test_protocols.py� test_echo s zWireTests.test_echoc C s. t �� }t�� }|�|� | �|�� d� dS )z) Test wire.Who protocol. s root N)r r r ZWhor r r r r r r �test_who# s zWireTests.test_whoc C s. t �� }t�� }|�|� | �|�� d� dS )z* Test wire.QOTD protocol. s' An apple a day keeps the doctor away. N)r r r ZQOTDr r r r r r r � test_QOTD- s �zWireTests.test_QOTDc C s` t �� }t�� }|�|� |�d� |�d� |�d� |�d� |�d� | �|�� d� dS )z- Test wire.Discard protocol. r r r r r � N)r r r ZDiscardr r r r r r r r � test_discard8 s zWireTests.test_discardN)�__name__� __module__� __qualname__�__doc__r r r r! r r r r r s r c @ s e Zd ZdZdd� ZdS )�TestableProxyClientFactoryz� Test proxy client factory that keeps the last created protocol instance. @ivar protoInstance: the last instance of the protocol. @type protoInstance: L{portforward.ProxyClient} c C s t j�| |�}|| _|S )zE Create the protocol instance and keeps track of it. )r �ProxyClientFactory� buildProtocol� protoInstance�r �addr�protor r r r( P s z(TestableProxyClientFactory.buildProtocolN�r" r# r$ r% r( r r r r r& H s r& c @ s e Zd ZdZdd� ZdS )�TestableProxyFactoryal Test proxy factory that keeps the last created protocol instance. @ivar protoInstance: the last instance of the protocol. @type protoInstance: L{portforward.ProxyServer} @ivar clientFactoryInstance: client factory used by C{protoInstance} to create forward connections. @type clientFactoryInstance: L{TestableProxyClientFactory} c s. t j�� |�}t� � _� fdd�|_|� _|S )z� Create the protocol instance, keeps track of it, and makes it use C{clientFactoryInstance} as client factory. c s � j S �N)�clientFactoryInstancer �r r r �<lambda>n r z4TestableProxyFactory.buildProtocol.<locals>.<lambda>)r � ProxyFactoryr( r& r0 ZclientProtocolFactoryr) r* r r1 r r( f s z"TestableProxyFactory.buildProtocolNr- r r r r r. Z s r. c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �PortforwardingTestsz Test port forwarding. c C s t �� | _t�� | _g | _d S r/ )r r �serverProtocolr ZProtocol�clientProtocol� openPortsr1 r r r �setUpy s zPortforwardingTests.setUpc C s� z| j jj�� W n tk r& Y nX z| j jj}|j�� W n tk rT Y nX z| jj�� W n tk rz Y nX z| jj�� W n tk r� Y nX t� dd� | j D ��S )Nc S s g | ]}t �|j��qS r )r Z maybeDeferredZ stopListening)�.0�pr r r � <listcomp>� s z0PortforwardingTests.tearDown.<locals>.<listcomp>) �proxyServerFactoryr) � transportZloseConnection�AttributeErrorr0 r6 r5 r Z gatherResultsr7 )r Zpir r r �tearDown s( �z PortforwardingTests.tearDownc s� t �� }�fdd�|_ tjd|dd�}�j�|� td|�� j��_ tjd�j dd�}�j�|� d�g �t � � � � ���fdd�}|�j _ ��fd d �}|�j _t �� }�fd d�|_ t�d|�� j|� � S ) z= Test port forwarding through Echo protocol. c s � j S r/ )r5 r r1 r r r2 � r z6PortforwardingTests.test_portforward.<locals>.<lambda>r � 127.0.0.1)Z interfacei� c s> �� t| �� t���kr:��d���d� � � �d � d S )Nr � x)�extendr �lenr �join�callback)�data��d�nBytesZreceivedr r r �testDataReceived� s z>PortforwardingTests.test_portforward.<locals>.testDataReceivedc s �j j�d� � d S )NrA )r6 r= �writer )rI r r r �testConnectionMade� s z@PortforwardingTests.test_portforward.<locals>.testConnectionMadec s � j S r/ )r6 r r1 r r r2 � r )r Z ServerFactoryr Z listenTCPr7 �appendr. ZgetHost�portr< r ZDeferredr6 r ZconnectionMadeZ ClientFactoryZ connectTCP)r ZrealServerFactoryZrealServerPortZproxyServerPortrJ rL � clientFactoryr rG r �test_portforward� s8 � � � �z$PortforwardingTests.test_portforwardc C s� t �ddd�}t�dd��|�}t�� }||_t�� }|� |� | � t |j �d� |j d \}}}}} | � |tj� |�|�} t�� } | � | � | �| j|� | �|j| � | �| j� | �|j� dS )zm The proxy client registers itself as a producer of the proxy server and vice versa. ZTCPr@ r � N)r Z IPv4Addressr r3 r( r Z MemoryReactorr r r r rC Z tcpClientsZassertIsInstancer' ZassertIsZproducerZ assertTrueZ streaming) r r+ Zserverr ZserverTransportZhostrN rO Ztimeout�_ZclientZclientTransportr r r �test_registerProducers� s z*PortforwardingTests.test_registerProducersN)r" r# r$ r% r8 r? rP rS r r r r r4 t s )r4 c @ s e Zd ZdZdd� ZdS )�StringTransportTestszA Test L{proto_helpers.StringTransport} helper behaviour. c C s t �� }| �t|jd� dS )zY Test that L{proto_helpers.StringTransport} doesn't accept unicode data. ZfooN)r r Z assertRaises� TypeErrorrK )r �sr r r �test_noUnicode� s z#StringTransportTests.test_noUnicodeN)r" r# r$ r% rW r r r r rT � s rT N)r% Z twisted.trialr Ztwisted.protocolsr r Ztwisted.python.compatr Ztwisted.internetr r r r Z twisted.testr ZTestCaser r' r& r3 r. r4 rT r r r r �<module> s 9o
Save File
Cancel