0x1949 Team - FAZEMRX - MANAGER
Edit File: test_protocols.cpython-310.pyc
o �b� � @ s� d Z ddlmZmZmZmZ ddlmZmZ ddl m Z ddl 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. � )�address�defer�protocol�reactor)� portforward�wire)� iterbytes)� proto_helpers)�unittestc @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � WireTestsz Test wire protocols. c C �` 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 � zWireTests.test_echoc C �. t �� }t�� }|�|� | �|�� d� dS )z) Test wire.Who protocol. s root N)r r r �Whor r r r r r r �test_who" � zWireTests.test_whoc C r! )z* Test wire.QOTD protocol. s' An apple a day keeps the doctor away. N)r r r �QOTDr r r r r r r � test_QOTD+ r$ zWireTests.test_QOTDc C r )z- Test wire.Discard protocol. r r r r r � N)r r r �Discardr r r r r r r r � test_discard4 r zWireTests.test_discardN)�__name__� __module__� __qualname__�__doc__r r# r& r) r r r r r s r c @ � 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 r1 K s z(TestableProxyClientFactory.buildProtocolN�r* r+ r, r- r1 r r r r r/ C s r/ c @ r. )�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 � � j S �N)�clientFactoryInstancer �r r r �<lambda>h � z4TestableProxyFactory.buildProtocol.<locals>.<lambda>)r � ProxyFactoryr1 r/ r: �clientProtocolFactoryr2 r3 r r; r r1 ` s z"TestableProxyFactory.buildProtocolNr6 r r r r r7 T s r7 c @ r ) �PortforwardingTestsz Test port forwarding. c C s t �� | _t�� | _g | _d S r9 )r r �serverProtocolr �Protocol�clientProtocol� openPortsr; r r r �setUpr s zPortforwardingTests.setUpc C s� z | j jj�� W n ty Y nw z | j jj}|j�� W n ty( Y nw z| jj�� W n ty: Y nw z| jj�� W n tyL Y nw t� dd� | j D ��S )Nc S s g | ]}t �|j��qS r )r � maybeDeferred� stopListening)�.0�pr r r � <listcomp>� s z0PortforwardingTests.tearDown.<locals>.<listcomp>) �proxyServerFactoryr2 � transport�loseConnection�AttributeErrorr: rC rA r � gatherResultsrD )r �pir r r �tearDownw s0 � � � ��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 r8 r9 )rA r r; r r r< � r= z6PortforwardingTests.test_portforward.<locals>.<lambda>r � 127.0.0.1)� interfacei� c sB �� t| �� t���kr ��d���d� � � �d � d S d S )Nr' � x)�extendr �lenr �join�callback)�data��d�nBytes�receivedr r r �testDataReceived� s �z>PortforwardingTests.test_portforward.<locals>.testDataReceivedc s �j j�d� � d S )NrT )rC rL �writer )r\ r r r �testConnectionMade� s z@PortforwardingTests.test_portforward.<locals>.testConnectionMadec r8 r9 )rC r r; r r r< � r= )r � ServerFactoryr � listenTCPrD �appendr7 �getHost�portrK r �DeferredrC r �connectionMade� ClientFactory� connectTCP)r �realServerFactory�realServerPort�proxyServerPortr^ r` � clientFactoryr rZ r �test_portforward� s, �� 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. �TCPrR r � N)r � IPv4Addressr r> r1 r � MemoryReactorr r r r rV � tcpClients�assertIsInstancer0 �assertIs�producer� assertTrue� streaming) r r4 �serverr �serverTransport�hostre rm �timeout�_�client�clientTransportr r r �test_registerProducers� s z*PortforwardingTests.test_registerProducersN)r* r+ r, r- rE rQ rn r� r r r r r@ m s (r@ c @ r. )�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. �fooN)r r � assertRaises� TypeErrorr_ )r �sr r r �test_noUnicode� s z#StringTransportTests.test_noUnicodeN)r* r+ r, r- r� r r r r r� � s r� N)r- �twisted.internetr r r r �twisted.protocolsr r �twisted.python.compatr � twisted.testr � twisted.trialr �TestCaser r0 r/ r> r7 r@ r� r r r r �<module> s 4l