File: //lib/python3/dist-packages/twisted/internet/test/__pycache__/test_filedescriptor.cpython-38.pyc
U
W[�
� @ sx d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z G dd� de�Z G d d
� d
e �Z
G d d � d e �Zd
S )zA
Whitebox tests for L{twisted.internet.abstract.FileDescriptor}.
� )�division�absolute_import)� verifyClass)�FileDescriptor)�
IPushProducer)�SynchronousTestCasec @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) �
MemoryFilea-
A L{FileDescriptor} customization which writes to a Python list in memory
with certain limitations.
@ivar _written: A C{list} of C{bytes} which have been accepted as written.
@ivar _freeSpace: A C{int} giving the number of bytes which will be accepted
by future writes.
Tc C s t j| t� d� g | _d| _d S )N�Zreactorr )r �__init__�object�_written�
_freeSpace��self� r �K/usr/lib/python3/dist-packages/twisted/internet/test/test_filedescriptor.pyr
s zMemoryFile.__init__c C s d S �Nr r r r r � startWriting$ s zMemoryFile.startWritingc C s d S r r r r r r � stopWriting( s zMemoryFile.stopWritingc C s: t | jt|��}|r6| j|8 _| j�|d|� � |S )z�
Copy at most C{self._freeSpace} bytes from C{data} into C{self._written}.
@return: A C{int} indicating how many bytes were copied from C{data}.
N)�minr
�lenr �append)r �dataZ acceptLengthr r r �
writeSomeData, s
zMemoryFile.writeSomeDataN) �__name__�
__module__� __qualname__�__doc__Z connectedr
r r r r r r r r s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �FileDescriptorTestsz&
Tests for L{FileDescriptor}.
c C s t t� d�}| �t|jd� dS )zF
L{FileDescriptor.write} doesn't accept unicode data.
r ZfooN)r r � assertRaises� TypeError�write�r ZfileDescriptorr r r �$test_writeWithUnicodeRaisesException> s z8FileDescriptorTests.test_writeWithUnicodeRaisesExceptionc C s& t t� d�}| �t|jdddg� dS )zN
L{FileDescriptor.writeSequence} doesn't accept unicode data.
r s fooZbars bazN)r r r r Z
writeSequencer" r r r �,test_writeSequenceWithUnicodeRaisesExceptionF s �z@FileDescriptorTests.test_writeSequenceWithUnicodeRaisesExceptionc C s | � ttt�� dS )zF
L{FileDescriptor} should implement L{IPushProducer}.
N)Z
assertTruer r r r r r r �$test_implementInterfaceIPushProducerO s z8FileDescriptorTests.test_implementInterfaceIPushProducerN)r r r r r# r$ r% r r r r r : s r c @ s e Zd ZdZdd� ZdS )�WriteDescriptorTestszN
Tests for L{FileDescriptor}'s implementation of L{IWriteDescriptor}.
c C s" t � }|�d� | �|�� � dS )z�
When L{FileDescriptor.writeSomeData} returns C{0} to indicate no more
data can be written immediately, L{FileDescriptor.doWrite} returns
L{None}.
s hello, worldN)r r! Z assertIsNoneZdoWrite)r Z
descriptorr r r �test_kernelBufferFull[ s
z*WriteDescriptorTests.test_kernelBufferFullN)r r r r r' r r r r r& W s r& N)r Z
__future__r r Zzope.interface.verifyr Ztwisted.internet.abstractr Ztwisted.internet.interfacesr Ztwisted.trial.unittestr r r r& r r r r �<module> s (