File: //lib/python3/dist-packages/twisted/internet/test/__pycache__/test_filedescriptor.cpython-310.pyc
o
�b�
� @ sh d Z ddlmZ ddlmZ ddlmZ ddlmZ G dd� de�Z G dd � d e�Z
G d
d � d e�Z d S )
zA
Whitebox tests for L{twisted.internet.abstract.FileDescriptor}.
� )� 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��reactorr )r �__init__�object�_written�
_freeSpace��self� r �K/usr/lib/python3/dist-packages/twisted/internet/test/test_filedescriptor.pyr s
zMemoryFile.__init__c C � d S �Nr r
r r r � startWriting" � zMemoryFile.startWritingc C r r r r
r r r � stopWriting% r zMemoryFile.stopWritingc C s: t | jt|��}|r| 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 �data� acceptLengthr r r �
writeSomeData( s
zMemoryFile.writeSomeDataN) �__name__�
__module__� __qualname__�__doc__� 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 �fooN)r r
� assertRaises� TypeError�write�r �fileDescriptorr r r �$test_writeWithUnicodeRaisesException: s z8FileDescriptorTests.test_writeWithUnicodeRaisesExceptionc C s$ t t� d�}| �t|jg d�� dS )zN
L{FileDescriptor.writeSequence} doesn't accept unicode data.
r )s foo�bars bazN)r r
r# r$ �
writeSequencer& r r r �,test_writeSequenceWithUnicodeRaisesExceptionA s �z@FileDescriptorTests.test_writeSequenceWithUnicodeRaisesExceptionc C s | � ttt�� dS )zF
L{FileDescriptor} should implement L{IPushProducer}.
N)�
assertTruer r r r
r r r �$test_implementInterfaceIPushProducerJ s z8FileDescriptorTests.test_implementInterfaceIPushProducerN)r r r r r( r+ r- r r r r r! 5 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% � assertIsNone�doWrite)r �
descriptorr r r �test_kernelBufferFullV s
z*WriteDescriptorTests.test_kernelBufferFullN)r r r r r2 r r r r r. Q s r. N) r �zope.interface.verifyr �twisted.internet.abstractr �twisted.internet.interfacesr �twisted.trial.unittestr r r! r. r r r r �<module> s %