← Back
Editing: 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 ]
Save File
Cancel