File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_posixprocess.cpython-38.pyc
U
W[�+ � @ s� d Z ddlmZmZ ddlZddlZddlZz ddlZW n ek
rP dZ Y nX ddl
m Z dZ ddl m
Z
ddlmZ G dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZdS )z;
Tests for POSIX-based L{IReactorProcess} implementations.
� )�division�absolute_importNznon-POSIX platform)�process)�range)�TestCasec @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �FakeFilez>
A dummy file object which records when it is closed.
c C s || _ || _d S �N)�testcase�fd)�selfr r
� r �I/usr/lib/python3/dist-packages/twisted/internet/test/test_posixprocess.py�__init__ s zFakeFile.__init__c C s | j j�| j� d S r )r �_files�remover
�r r r r
�close! s zFakeFile.closec C s | S r r r r r r
� __enter__% s zFakeFile.__enter__c C s | � � d S r )r )r �exc_type� exc_value� tracebackr r r
�__exit__) s zFakeFile.__exit__N)�__name__�
__module__� __qualname__�__doc__r r r r r r r r
r s
r c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�FakeResourceModulez�
Fake version of L{resource} which hard-codes a particular rlimit for maximum
open files.
@ivar _limit: The value to return for the hard limit of number of open files.
� c C s
|| _ d S r )�_limit�r �limitr r r
r 7 s zFakeResourceModule.__init__c C s || j krd| jgS ddgS )zX
A fake of L{resource.getrlimit} which returns a pre-determined result.
r �{ i� )�
RLIMIT_NOFILEr )r Znor r r
� getrlimit; s
z FakeResourceModule.getrlimitN)r r r r r" r r# r r r r
r . s r c @ s� e Zd ZdZeZdZdZdZdd� Z dd� Z
dd� Z d d
� Z d d � Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �FDDetectorTestsa�
Tests for _FDDetector class in twisted.internet.process, which detects
which function to drop in place for the _listOpenFDs method.
@ivar devfs: A flag indicating whether the filesystem fake will indicate
that /dev/fd exists.
@ivar accurateDevFDResults: A flag indicating whether the /dev/fd fake
returns accurate open file information.
@ivar procfs: A flag indicating whether the filesystem fake will indicate
that /proc/<pid>/fd exists.
Fc C s dS )z>
Fake os.getpid, always return the same thing
r! r r r r r
�getpidZ s zFDDetectorTests.getpidc C sT t t| j�}| jr(|d| �� f kr(|S | jrJ|dkrJ| jr@|S dddgS t� �dS )z�
Fake os.listdir, depending on what mode we're in to simulate behaviour.
@param arg: the directory to list
z /proc/%d/fdz/dev/fd�0�1�2N)�map�strr �procfsr% �devfs�accurateDevFDResults�OSError)r �argZaccurater r r
�listdira s
zFDDetectorTests.listdirc C s2 t | tttd��t| j� ��}| j�|j� |S )aO
This is a mock for L{open}. It keeps track of opened files so extra
descriptors can be returned from the mock for L{os.listdir} when used on
one of the list-of-filedescriptors directories.
A L{FakeFile} is returned which can be closed to remove the new
descriptor from the open list.
� )r �min�setr r �appendr
)r �fname�mode�fr r r
�openfileq s
zFDDetectorTests.openfilec C s dt jd< dS )zp
Make the L{resource} module unimportable for the remainder of the
current test method.
N�resource)�sys�modulesr r r r
�hideResourceModule� s z"FDDetectorTests.hideResourceModulec C s t |�tjd< dS )a
Make a L{FakeResourceModule} instance importable at the L{resource}
name.
@param limit: The value which will be returned for the hard limit of
number of open files by the fake resource module's C{getrlimit}
function.
r9 N)r r: r; r r r r
�revealResourceModule� s z$FDDetectorTests.revealResourceModulec C s: |dkr,z t jd= W q6 tk
r( Y q6X n
|t jd< dS )zI
Restore the original resource module to L{sys.modules}.
Nr9 )r: r; �KeyError)r �valuer r r
�replaceResourceModule� s z%FDDetectorTests.replaceResourceModulec C sN t �� | _| j| j_| j| j_| j| j_dddg| _| �| jt j
� d�� dS )z�
Set up the tests, giving ourselves a detector object to play with and
setting up its testable knobs to refer to our mocked versions.
r r � r9 N) r � _FDDetector�detectorr0 r% r8 r Z
addCleanupr@ r: r; �getr r r r
�setUp� s
�zFDDetectorTests.setUpc s@ dd� }dd� }� fdd�}|||g� j _� �|� j �� � dS )z�
L{FDDetector._getImplementation} returns the first method from its
C{_implementations} list which returns results which reflect a newly
opened file descriptor.
c S s t d��d S )NzThis does not work)�
ValueErrorr r r r
�failWithException� s zBFDDetectorTests.test_selectFirstWorking.<locals>.failWithExceptionc S s
dddgS �Nr r rA r r r r r
�failWithWrongResults� s zEFDDetectorTests.test_selectFirstWorking.<locals>.failWithWrongResultsc s � j d d � S r )r r r r r
�correct� s z8FDDetectorTests.test_selectFirstWorking.<locals>.correctN�rC Z_implementationsZassertIsZ_getImplementation)r rG rI rJ r r r
�test_selectFirstWorking� s �z'FDDetectorTests.test_selectFirstWorkingc C s2 dd� }dd� }||g| j _| �|| j �� � dS )z�
L{FDDetector._getImplementation} returns the last method from its
C{_implementations} list if none of the implementations manage to return
results which reflect a newly opened file descriptor.
c S s
dddgS )N� � � r r r r r
rI � s z=FDDetectorTests.test_selectLast.<locals>.failWithWrongResultsc S s
dddgS rH r r r r r
�failWithOtherWrongResults� s zBFDDetectorTests.test_selectLast.<locals>.failWithOtherWrongResultsNrK )r rI rP r r r
�test_selectLast� s � �z FDDetectorTests.test_selectLastc C sL t �� }|jj}|�� |jj}|�� |jj}| �||� | �||� dS )z�
Check that the identity of _listOpenFDs changes after running
_listOpenFDs the first time, but not after the second time it's run.
In other words, check that the monkey patching actually works.
N)r rB � _listOpenFDsr ZassertNotEqual� assertEqual)r rC �first�secondZthirdr r r
�!test_identityOfListOpenFDsChanges� s z1FDDetectorTests.test_identityOfListOpenFDsChangesc C s> d| _ | �t| jj� d| _ d| _| �dddg| j�� � dS )z�
L{_FDDetector._devFDImplementation} raises L{OSError} if there is no
I{/dev/fd} directory, otherwise it returns the basenames of its children
interpreted as integers.
FTr r rA N)r, � assertRaisesr. rC Z_devFDImplementationr- rS r r r r
�test_devFDImplementation� s
z(FDDetectorTests.test_devFDImplementationc C s8 d| _ | �t| jj� d| _ | �dddg| j�� � dS )z�
L{_FDDetector._procFDImplementation} raises L{OSError} if there is no
I{/proc/<pid>/fd} directory, otherwise it returns the basenames of its
children interpreted as integers.
FTr r rA N)r+ rW r. rC Z_procFDImplementationrS r r r r
�test_procFDImplementation� s z)FDDetectorTests.test_procFDImplementationc C sT | � d� | �ttd��t| j�� �� | � d� | �ttd��t| j�� �� dS )z�
L{_FDDetector._fallbackFDImplementation} uses the L{resource} module if
it is available, returning a range of integers from 0 to the
minimum of C{1024} and the hard I{NOFILE} limit.
i i r1 N)r= rS �listr rC �_fallbackFDImplementationr r r r
� test_resourceFDImplementation s
�
�z-FDDetectorTests.test_resourceFDImplementationc C s* | � � | �ttd��t| j�� �� dS )z�
L{_FDDetector._fallbackFDImplementation}, the implementation of last
resort, succeeds with a fixed range of integers from 0 to 1024 when the
L{resource} module is not importable.
r1 N)r< rS rZ r rC r[ r r r r
� test_fallbackFDImplementation s �z-FDDetectorTests.test_fallbackFDImplementationN)r r r r � platformSkip�skipr, r- r+ r% r0 r8 r< r= r@ rE rL rQ rV rX rY r\ r] r r r r
r$ E s&
r$ c @ s$ e Zd ZdZeZdd� Zdd� ZdS )�FileDescriptorTestsz<
Tests for L{twisted.internet.process._listOpenFDs}
c C sd t �� D ]V}zt�|tj� W q tk
r\ } z | �tj|jd||jf � W 5 d}~X Y qX qdS )z�
File descriptors returned by L{_listOpenFDs} are mostly open.
This test assumes that zero-legth writes fail with EBADF on closed
file descriptors.
z2fcntl(%d, F_GETFL) failed with unexpected errno %dN)r rR �fcntlZF_GETFL�IOErrorrS �errnoZEBADF)r r
�errr r r
� test_openFDs' s ��z FileDescriptorTests.test_openFDsc C s� t tj�}t�� }| �|�� |� t�|�� �}| �||�� kd� z |� � | �|t�� � W 5 t� |� X | �
|t�� � dS )zB
L{_listOpenFDs} lists expected file descriptors.
z>Expected duplicate file descriptor to be greater than originalN) �open�os�devnullr rR ZassertIn�fileno�dupZ
assertTruer Z assertNotIn)r r7 Zopenfdsr
r r r
�test_expectedFDs8 s
� z$FileDescriptorTests.test_expectedFDsN)r r r r r^ r_ re rk r r r r
r` ! s r` )r Z
__future__r r rc rg r: ra � ImportErrorr^ Ztwisted.internetr Ztwisted.python.compatr Ztwisted.trial.unittestr �objectr r r$ r` r r r r
�<module> s
]