File: //proc/self/root/lib/python3/dist-packages/twisted/test/__pycache__/test_tpfile.cpython-38.pyc
U
W[A � @ sz d dl mZ d dlmZ d dlmZ d dlmZmZ d dlm Z G dd� dej
�Z G dd � d ej
�Z G d
d � d ej
�Zd S )
� )�unittest)�loopback)�basic)�protocol�abstract)�BytesIOc @ s e Zd ZdZdd� ZdS )�BufferingServer� c C s | j |7 _ d S �N)�buffer)�self�data� r �:/usr/lib/python3/dist-packages/twisted/test/test_tpfile.py� dataReceived s z BufferingServer.dataReceivedN)�__name__�
__module__� __qualname__r r r r r r r
s r c @ s e Zd Zdd� Zdd� ZdS )�FileSendingClientc C s
|| _ d S r
)�f)r r r r r �__init__ s zFileSendingClient.__init__c s4 t �� }|�� j� jdd� �}|�� fdd�� d S )Nc S s | S r
r ��xr r r �<lambda> r z2FileSendingClient.connectionMade.<locals>.<lambda>c s
� j �� S r
)� transportZloseConnection)�r�r r r r r )r �
FileSender�beginFileTransferr r � addCallback)r �s�dr r r �connectionMade s z FileSendingClient.connectionMadeN)r r r r r" r r r r r s r c @ s e Zd Zdd� Zdd� ZdS )�FileSenderTestsc s< d�t � � tt���}t�� |�}|�� ��fdd�� |S )Ns� xyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzxyzabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabcabc123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123123c s �� � j��S r
)Z assertEqualr r �r r ZtestStrr r r $ r z1FileSenderTests.testSendingFile.<locals>.<lambda>)r r r r Z loopbackTCPr )r �cr! r r$ r �testSendingFile s z FileSenderTests.testSendingFilec C sN t �� }t�� }d|_td�}|�||dd� �}| �|j� | � |j
d� d S )N� r c S s | S r
r r r r r r . r z6FileSenderTests.testSendingEmptyFile.<locals>.<lambda>z0producer unregistered with deferred being called) r r r ZFileDescriptorZ connectedr r Z assertIsNoneZproducerZ
assertTrueZcalled)r Z
fileSenderZconsumerZ emptyFiler! r r r �testSendingEmptyFile( s �z$FileSenderTests.testSendingEmptyFileN)r r r r& r( r r r r r# s
r# N)Z
twisted.trialr Ztwisted.protocolsr r Ztwisted.internetr r �ior ZProtocolr r ZTestCaser# r r r r �<module> s