File: //lib/python3/dist-packages/twisted/test/__pycache__/test_fdesc.cpython-310.pyc
o
�b � @ s� d Z ddlZddlZddlZzddlZW n ey dZY nw ddlmZ ddl m
Z
ddl m Z G dd� de j
�ZG d d
� d
e j
�ZG d d � d e j
�ZdS )
z&
Tests for L{twisted.internet.fdesc}.
� Nz not supported on this platform)�fdesc)�untilConcludes)�unittestc @ s e Zd ZdZdd� Zdd� ZdS )�NonBlockingTestszE
Tests for L{fdesc.setNonBlocking} and L{fdesc.setBlocking}.
c C sj t �� \}}| �t j|� | �t j|� | �t�|tj�t j@ � t� |� | �
t�|tj�t j@ � dS )zR
L{fdesc.setNonBlocking} sets a file description to non-blocking.
N) �os�pipe�
addCleanup�close� assertFalse�fcntl�F_GETFL�
O_NONBLOCKr �setNonBlocking�
assertTrue��self�r�w� r �9/usr/lib/python3/dist-packages/twisted/test/test_fdesc.py�test_setNonBlocking s
z$NonBlockingTests.test_setNonBlockingc C sZ t �� \}}| �t j|� | �t j|� t�|� t�|� | �t�|tj �t j
@ � dS )zK
L{fdesc.setBlocking} sets a file description to blocking.
N) r r r r r r � setBlockingr
r r r
r r r r �test_setBlocking' s
z!NonBlockingTests.test_setBlockingN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�ReadWriteTestsz<
Tests for L{fdesc.readFromFD}, L{fdesc.writeToFD}.
c C s, t �� \| _| _t�| j� t�| j� dS )zG
Create a non-blocking pipe that can be used in tests.
N)r r r r r r �r r r r �setUp8 s zReadWriteTests.setUpc C sL zt �| j� W n ty Y nw z t �| j� W dS ty% Y dS w )z
Close pipes.
N)r r r �OSErrorr r r r r �tearDown@ s � �zReadWriteTests.tearDownc C s t �| j|�S )z)
Write data to the pipe.
)r � writeToFDr )r �dr r r �writeM s zReadWriteTests.writec C s0 g }t �| j|j�}|du r|r|d S dS |S )z*
Read data from the pipe.
Nr � )r �
readFromFDr �append)r �l�resr r r �readS s zReadWriteTests.readc C sH | � d�}| �|dk� | �� }| �t|�|� | �dd|� |� dS )z�
Test that the number of bytes L{fdesc.writeToFD} reports as written
with its return value are seen by L{fdesc.readFromFD}.
s hellor N)r$ r r* � assertEqual�len)r �n�sr r r �test_writeAndReada s
z ReadWriteTests.test_writeAndReadc C s� dd }| � |�}| �|dk� g }d}d}||k s |dk r9|�| �� � |t|d �7 }|d7 }||k s |dk s d�|�}| �t|�|� | �|d|� |� dS ) z}
Similar to L{test_writeAndRead}, but use a much larger string to verify
the behavior for that case.
s
0123456879i' r �2 ���� r% N)r$ r r' r* r, �joinr+ )r �orig�written�result� resultlength�ir r r �test_writeAndReadLargel s
�
z%ReadWriteTests.test_writeAndReadLargec C s. g }t �| j|j�}| �|g � | �|� dS )z�
Verify that reading from a file descriptor with no data does not raise
an exception and does not result in the callback function being called.
N)r r& r r' r+ � assertIsNone)r r( r6 r r r �test_readFromEmpty� s z!ReadWriteTests.test_readFromEmptyc C �" t �| j� | �| �� tj� dS )z�
Test that using L{fdesc.readFromFD} on a cleanly closed file descriptor
returns a connection done indicator.
N)r r r r+ r* r �CONNECTION_DONEr r r r �test_readFromCleanClose� � z&ReadWriteTests.test_readFromCleanClosec C �$ t �| j� | �| �d�tj� dS )z�
Verify that writing with L{fdesc.writeToFD} when the read end is closed
results in a connection lost indicator.
� sN)r r r r+ r$ r �CONNECTION_LOSTr r r r �test_writeToClosed� � z!ReadWriteTests.test_writeToClosedc C r< )z�
Verify that reading with L{fdesc.readFromFD} when the read end is
closed results in a connection lost indicator.
N)r r r r+ r* r rB r r r r �test_readFromInvalid� r? z#ReadWriteTests.test_readFromInvalidc C r@ )z�
Verify that writing with L{fdesc.writeToFD} when the write end is
closed results in a connection lost indicator.
rA N)r r r r+ r$ r rB r r r r �test_writeToInvalid� rD z"ReadWriteTests.test_writeToInvalidc C sp t j}dd� }|t _z| �| �d�d� W |t _n|t _w dd� }|t _z| �| �d�d� W |t _dS |t _w )z8
Test error path for L{fdesc.writeTod}.
c S � t � }tj|_|��N)r �errno�EAGAIN��fd�data�errr r r � eagainWrite� � z4ReadWriteTests.test_writeErrors.<locals>.eagainWriterA r c S rG rH )r rI �EINTRrK r r r �
eintrWrite� rP z3ReadWriteTests.test_writeErrors.<locals>.eintrWriteN)r r$ r+ )r �
oldOsWriterO rR r r r �test_writeErrors� s z ReadWriteTests.test_writeErrorsN)r r r r r r! r$ r* r/ r9 r; r> rC rE rF rT r r r r r 3 s
r c @ s, e Zd ZdZdZdd� Zdd� Zdd� Zd S )
�CloseOnExecTestszL
Tests for L{fdesc._setCloseOnExec} and L{fdesc._unsetCloseOnExec}.
z�
import os, errno
try:
os.write(%d, b'lul')
except OSError as e:
if e.errno == errno.EBADF:
os._exit(0)
os._exit(5)
except BaseException:
os._exit(10)
else:
os._exit(20)
c C sz t �� }|dkr4zt �tjtjd| j|�� f g� W d S ty3 dd l}|� � t �
d� Y d S w t t j |d�d S )Nr z-c� r2 )
r �fork�execv�sys�
executable�program�fileno�
BaseException� traceback� print_exc�_exitr �waitpid)r �fObj�pidr^ r r r �_execWithFileDescriptor� s
� �z(CloseOnExecTests._execWithFileDescriptorc C sn t | �� d��&}t�|�� � | �|�}| �t�|�� | � t�
|�d� W d � dS 1 s0w Y dS )z�
A file descriptor passed to L{fdesc._setCloseOnExec} is not inherited
by a new process image created with one of the exec family of
functions.
�wbr N) �open�mktempr �_setCloseOnExecr\ rd r r � WIFEXITEDr+ � WEXITSTATUS�r rb �statusr r r �test_setCloseOnExec� s
"�z$CloseOnExecTests.test_setCloseOnExecc C s| t | �� d��-}t�|�� � t�|�� � | �|�}| �t� |�� | �
t� |�d� W d � dS 1 s7w Y dS )z�
A file descriptor passed to L{fdesc._unsetCloseOnExec} is inherited by
a new process image created with one of the exec family of functions.
re � N) rf rg r rh r\ �_unsetCloseOnExecrd r r ri r+ rj rk r r r �test_unsetCloseOnExec� s
"�z&CloseOnExecTests.test_unsetCloseOnExecN)r r r r r[ rd rm rp r r r r rU � s rU )r rI r rY r � ImportError�skip�twisted.internetr �twisted.python.utilr �
twisted.trialr �SynchronousTestCaser r rU r r r r �<module> s �