← Back
Editing: test_unix.cpython-38.pyc
U W[�� � @ s� d Z ddlmZmZ ddlmZ ddlmZmZmZm Z m Z ddl m Z m Z ddlmZmZmZmZmZ ddlmZ ddlmZ dd lmZ zdd lmZ W n ek r� d ZY nX dd lm Z dd l m Z m Z ddl m!Z! ddl"m#Z#m$Z$m%Z% ddl&m'Z'm(Z( ddl)m*Z*m+Z+m,Z, ddl-m.Z.m/Z/m0Z0m1Z1 ddl2m3Z3 ddl2m4Z4m5Z5 ddl6m7Z7 ddl8m9Z9 ddl:m;Z; ddl<m=Z= ddl>m?Z?m@Z@mAZAmBZB ddl8mCZC ddl8mDZD dd l8mEZE dd l8mFZF dd lGmHZHmIZImJZJ dd lKmLZL dd lMmNZNmOZOmPZP dd!lQmRZR dd"lSmTZT dd#lUmVZV eTd$�d k �r.d ZWnd%ZWG d&d'� d'eX�ZYd(d)� ZZG d*d+� d+e9�Z[G d,d-� d-eC�Z\e e.�G d.d/� d/eC��Z]G d0d1� d1eYe;eD�Z^G d2d3� d3eYe;�Z_G d4d5� d5eX�Z`G d6d7� d7eX�ZaG d8d9� d9eX�ZbG d:d;� d;eaebe;e=e?�ZcG d<d=� d=e`ebe;e=e?�ZdG d>d?� d?e@e;�Zeef� �ge^�h� � ef� �ge_�h� � ef� �gec�h� � ef� �ged�h� � ef� �gee�h� � G d@dA� dAe;eE�Zief� �gei�h� � d S )Bz/ Tests for implementations of L{IReactorUNIX}. � )�division�absolute_import)�S_IMODE)�stat�close�urandom�unlink�fstat)�mktemp�mkstemp)�AF_INET� SOCK_STREAM� SOL_SOCKET�socket�error)�pformat)�md5)�pack)�AF_UNIXN)� implementer)� interfaces�base)� UNIXAddress)�Deferred�fail� gatherResults)�UNIXServerEndpoint�UNIXClientEndpoint)�ConnectionClosed�FileDescriptorOverrun�CannotListenError)�IFileDescriptorReceiver� IReactorUNIX�IReactorSocket� IReactorFDSet)�DatagramProtocol)� ServerFactory� ClientFactory)� LoopingCall)�EndpointCreator)�ReactorBuilder)�ObjectModelIntegrationMixin)�StreamTransportTestsMixin�WriteSequenceTestsMixin�MyClientFactory�MyServerFactory)�ConnectableProtocol)�ConnectionTestsMixin)�StreamClientTestsMixin)�runProtocolsWithReactor)� nativeString�_PY3� iteritems)�Failure)� addObserver�removeObserver�err)�platform)� requireModule)�_coerceToFilesystemEncodingztwisted.python.sendmsgz>sendmsg extension unavailable, extended UNIX features disabledc @ s e Zd ZdZdd� ZdS )�UNIXFamilyMixinzK Test-helper defining mixin for things related to AF_UNIX sockets. c C sB d}| � � }t||�|||d�}|�� | �tt|�j�|� dS )z} Assert that the mode of the created unix socket is set to the mode specified to the reactor method. i� )�modeN)� buildReactor�getattrZ stopListening� assertEqualr r �st_mode)�selfZ methodName�path�factoryr? �reactorZunixPort� rH �A/usr/lib/python3/dist-packages/twisted/internet/test/test_unix.py� _modeTest@ s zUNIXFamilyMixin._modeTestN)�__name__� __module__� __qualname__�__doc__rJ rH rH rH rI r> < s r> c C s t td���� S )zI Return a new, unique abstract namespace path to be listened on. �d )r r Z hexdigest)ZcaserH rH rI � _abstractPathL s rP c @ s( e Zd ZdZejfZdd� Zdd� ZdS )� UNIXCreatorz( Create UNIX socket end points. c C s t ddd�}t||�S )z3 Construct a UNIX server endpoint. �.sock�.��suffix�dir)r r )rD rG rE rH rH rI �serverZ s zUNIXCreator.serverc C s t ||j�S )z3 Construct a UNIX client endpoint. )r �name)rD rG Z serverAddressrH rH rI �clientc s zUNIXCreator.clientN) rK rL rM rN r r" �requiredInterfacesrW rY rH rH rH rI rQ T s rQ c @ s, e Zd ZdZdZdd� Zdd� Zdd� ZdS ) �SendFileDescriptorz� L{SendFileDescriptorAndBytes} sends a file descriptor and optionally some normal bytes and then closes its connection. @ivar reason: The reason the connection was lost, after C{connectionLost} is called. Nc C s || _ || _dS )z� @param fd: A C{int} giving a file descriptor to send over the connection. @param data: A C{str} giving data to send over the connection, or L{None} if no data is to be sent. N)�fd�data)rD r\ r] rH rH rI �__init__u s zSendFileDescriptor.__init__c C s0 | j �| j� | jr"| j �| j� | j �� dS )zn Send C{self.fd} and, if it is not L{None}, C{self.data}. Then close the connection. N)� transport�sendFileDescriptorr\ r] �write�loseConnection�rD rH rH rI �connectionMade� s z!SendFileDescriptor.connectionMadec C s t �| |� || _d S �N)r0 �connectionLost�reason�rD rg rH rH rI rf � s z!SendFileDescriptor.connectionLost)rK rL rM rN rg r^ rd rf rH rH rH rI r[ k s r[ c @ s8 e Zd ZdZdZdZdd� Zdd� Zdd� Zd d � Z dS ) �ReceiveFileDescriptora} L{ReceiveFileDescriptor} provides an API for waiting for file descriptors to be received. @ivar reason: The reason the connection was lost, after C{connectionLost} is called. @ivar waiting: A L{Deferred} which fires with a file descriptor once one is received, or with a failure if the connection is lost with no descriptor arriving. Nc C s& | j dkrt� | _| jS t| j �S dS )z� Return a L{Deferred} which will fire with the next file descriptor received, or with a failure if the connection is or has already been lost. N)rg r �waitingr rc rH rH rI �waitForDescriptor� s z'ReceiveFileDescriptor.waitForDescriptorc C s | j �|� d| _ dS )z� Fire the waiting Deferred, initialized by C{waitForDescriptor}, with the file descriptor just received. N)rj �callback)rD � descriptorrH rH rI �fileDescriptorReceived� s z,ReceiveFileDescriptor.fileDescriptorReceivedc C s. | j dk r*| j �ttd|f ��� d| _ dS )a_ Fail the waiting Deferred, if it has not already been fired by C{fileDescriptorReceived}. The bytes sent along with a file descriptor are guaranteed to be delivered to the protocol's C{dataReceived} method only after the file descriptor has been delivered to the protocol's C{fileDescriptorReceived}. Nz&Received bytes (%r) before descriptor.)rj �errbackr7 � Exception�rD r] rH rH rI � dataReceived� s �z"ReceiveFileDescriptor.dataReceivedc C s2 t �| |� | jdk r(| j�|� d| _|| _dS )zj Fail the waiting Deferred, initialized by C{waitForDescriptor}, if there is one. N)r0 rf rj ro rg rh rH rH rI rf � s z$ReceiveFileDescriptor.connectionLost) rK rL rM rN rg rj rk rn rr rf rH rH rH rI ri � s ri c @ s, e Zd ZdZefZe� Zdd� Zdd� Z e � � s6de _ dd� Z d d � Ze � � sTde_ d d � Zd d� Zedk rree_ dd� Zedk r�ee_ dd� Zedk r�ee_ dd� Zdd� Zedk r�ee_ dd� Ze �� r�de_ nedk r�ee_ dd � Zedk r�ee_ d d � Zedk �ree_ d d � Zedk �r(ee_ dS )!�UNIXTestsBuilderz= Builder defining tests relating to L{IReactorUNIX}. c C s | � d| �� t� � dS )zs The UNIX socket created by L{IReactorUNIX.listenUNIX} is created with the mode specified. � listenUNIXN)rJ r r& rc rH rH rI � test_mode� s zUNIXTestsBuilder.test_modec C s> t | �}| �� }|�d| t� �}| �|�� td| �� dS )z� On Linux, a UNIX socket path may begin with C{' '} to indicate a socket in the abstract namespace. L{IReactorUNIX.listenUNIX} accepts such a path. � N)rP r@ rt r& rB �getHostr �rD rE rG �portrH rH rI �#test_listenOnLinuxAbstractNamespace� s z4UNIXTestsBuilder.test_listenOnLinuxAbstractNamespace�8Abstract namespace UNIX sockets only supported on Linux.c C sH dd� }| � tjd|� | �� }| �t�� |�dt� � W 5 Q R X dS )z� L{IReactorUNIX.listenUNIX} raises L{CannotListenError} if the underlying port's createInternetSocket raises a socket error. c S s t d��d S )Nz FakeBasePort forced socket.error)r rc rH rH rI �raiseSocketError� s z=UNIXTestsBuilder.test_listenFailure.<locals>.raiseSocketErrorZcreateInternetSocketznot-usedN)�patchr ZBasePortr@ Z assertRaisesr rt r& )rD r| rG rH rH rI �test_listenFailure� s z#UNIXTestsBuilder.test_listenFailurec C s> t | �}| �� }|�d| t� �}| �|�� td| �� dS )zc L{IReactorUNIX.connectUNIX} also accepts a Linux abstract namespace path. rv N)rP r@ � connectUNIXr' rB ZgetDestinationr )rD rE rG Z connectorrH rH rI �$test_connectToLinuxAbstractNamespace s z5UNIXTestsBuilder.test_connectToLinuxAbstractNamespacec C s` G dd� dt �}|� }|� }t| ||| j� | �|jd |jd � | �|jd |jd � dS )z� A client's transport's C{getHost} and C{getPeer} return L{UNIXAddress} instances which have the filesystem path of the host and peer ends of the connection. c @ s e Zd Zdd� ZdS )z4UNIXTestsBuilder.test_addresses.<locals>.SaveAddressc S s" t |�� |�� d�| _|�� d S )N)�host�peer)�dictrw �getPeer� addressesrb )rD r_ rH rH rI �makeConnection s �zCUNIXTestsBuilder.test_addresses.<locals>.SaveAddress.makeConnectionN)rK rL rM r� rH rH rH rI � SaveAddress s r� r� r� N)r0 r3 � endpointsrB r� )rD r� rW rY rH rH rI �test_addresses s z UNIXTestsBuilder.test_addressesc s� ddl m� t � ���d� t��� d��t� }|�� }� ��fdd�}|�|� |�t d� |� �fdd �� t ��|�j � d S ) z� L{IUNIXTransport.sendFileDescriptor} accepts an integer file descriptor and sends a copy of it to the process reading from the connection. r )�fromfd)� r � junkc s@ � | t t�}t| � ����� |�� � ����� |�� � d S re )r r r rB Z getsocknameZassertNotEqual�fileno)rm Zreceived)r� �srD rH rI �checkDescriptor3 s zAUNIXTestsBuilder.test_sendFileDescriptor.<locals>.checkDescriptorz-Sending file descriptor encountered a problemc s � j �� S re �r_ rb �Zignored�rW rH rI �<lambda>B � z:UNIXTestsBuilder.test_sendFileDescriptor.<locals>.<lambda>N) r r� �bindr[ r� ri rk � addCallback� addErrbackr: �addBothr3 r� )rD rY �dr� rH )r� r� rD rW rI �test_sendFileDescriptor& s z(UNIXTestsBuilder.test_sendFileDescriptorNc C sT G dd� dt �}G dd� dt �}|� }|� }||_t| ||| j� | �|jd� dS )z� If a L{IUNIXTransport.sendFileDescriptor} call fills up the send buffer, any registered producer is paused. c @ s e Zd Zdd� ZdS )zSUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing.<locals>.DoesNotReadc S s | j �� d S re )r_ �pauseProducingrc rH rH rI rd O s zbUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing.<locals>.DoesNotRead.connectionMadeN�rK rL rM rd rH rH rH rI � DoesNotReadN s r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) z`UNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing.<locals>.SendsManyFileDescriptorsFc sP t � � _ � j�� d� � fdd�}t|�� _� jj� j_� j�d��t d� d S )NTc s" � j �� j�� � � j �d� d S )N� x)r_ r` r r� ra rH rc rH rI �senderX s zUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing.<locals>.SendsManyFileDescriptors.connectionMade.<locals>.senderr zSend loop failure) r r_ ZregisterProducerr( �taskrG Zclock�startr� r: )rD r� rH rc rI rd U s zoUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing.<locals>.SendsManyFileDescriptors.connectionMadec S s | � � d S re �� _disconnectrc rH rH rI � stopProducing_ s znUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing.<locals>.SendsManyFileDescriptors.stopProducingc S s | � � d S re r� rc rH rH rI �resumeProducingb s zpUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing.<locals>.SendsManyFileDescriptors.resumeProducingc S s d| _ | j�� | �� d S )NT)�pausedr_ ZunregisterProducerr� rc rH rH rI r� e s zoUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing.<locals>.SendsManyFileDescriptors.pauseProducingc S s$ | j �� | j�� | jj�� d S re )r� �stopr_ ZabortConnection�otherrc rH rH rI r� j s zlUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing.<locals>.SendsManyFileDescriptors._disconnectN) rK rL rM r� rd r� r� r� r� rH rH rH rI �SendsManyFileDescriptorsR s r� z*sendFileDescriptor producer was not pausedN)r0 r� r3 r� � assertTruer� )rD r� r� rW rY rH rH rI �-test_sendFileDescriptorTriggersPauseProducingI s �z>UNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducingc s� t � }t|�� d�� t� }g }|�� }|�|j� |�� fdd�� t| � || j� | � |d t � |d � t � | � � j jt� dS )ag If L{IUNIXTransport.sendFileDescriptor} is used to queue a greater number of file descriptors than the number of bytes sent using L{ITransport.write}, the connection is closed and the protocol connected to the transport has its C{connectionLost} method called with a failure wrapping L{FileDescriptorOverrun}. Nc s � j �� S re r� r� r� rH rI r� � r� z=UNIXTestsBuilder.test_fileDescriptorOverrun.<locals>.<lambda>r )r r[ r� ri rk r� �appendr3 r� �assertIsInstancer7 Ztrapr rg �valuer )rD �cargorY �resultr� rH r� rI �test_fileDescriptorOverrunz s z+UNIXTestsBuilder.test_fileDescriptorOverrunc s> ddl m} ddlm} ddlm} dd� � tt�G � fdd�dt��}G d d � d |�}|t t �\}}| � |j � | � |j � |� } ||| �} t � \} } t � \} }| � t| � | � t|� d }| | g}||�\}}||||� | �� | �t| j�|� | �t|��t| j��� | j�r:� fd d �|D �}| �|| j� dS )a� Drive _SendmsgMixin via sendmsg socket calls to check that L{IFileDescriptorReceiver.fileDescriptorReceived} is called once for each file descriptor received in the ancillary messages. @param ancillaryPacker: A callable that will be given a list of two file descriptors and should return a two-tuple where: The first item is an iterable of zero or more (cmsg_level, cmsg_type, cmsg_data) tuples in the same order as the given list for actual sending via sendmsg; the second item is an integer indicating the expected number of FDs to be received. r �� socketpair)� _SendmsgMixin��sendmsgc S s t | �}|j|jfS re )r �st_dev�st_ino)r\ ZfsrH rH rI �deviceInodeTuple� s zTUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.deviceInodeTuplec s e Zd Zdd� Z� fdd�ZdS )zPUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.FakeProtocolc S s g | _ g | _d S re )�fds�deviceInodesReceivedrc rH rH rI r^ � s zYUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.FakeProtocol.__init__c s( | j �|� | j�� |�� t|� d S re )r� r� r� r )rD r\ �r� rH rI rn � s zgUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.FakeProtocol.fileDescriptorReceivedN)rK rL rM r^ rn rH r� rH rI � FakeProtocol� s r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) zPUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.FakeReceiver� c S s || _ || _d S re )r Zprotocol)rD �skt�protorH rH rI r^ � s zYUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.FakeReceiver.__init__c S s d S re rH rq rH rH rI � _dataReceived� s z^UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.FakeReceiver._dataReceivedc S s d S re rH rc rH rH rI rw � s zXUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.FakeReceiver.getHostc S s d S re rH rc rH rH rI r� � s zXUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.FakeReceiver.getPeerc S s d S re rH )rD �orH rH rI � _getLogPrefix� s z^UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.FakeReceiver._getLogPrefixN) rK rL rM Z bufferSizer^ r� rw r� r� rH rH rH rI � FakeReceiver� s r� s some data needs to be sentc s g | ] }� |��qS rH rH ��.0r\ r� rH rI � <listcomp>� s zNUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver.<locals>.<listcomp>N)r r� Ztwisted.internet.unixr� �twisted.python.sendmsgr� r r! r0 r r � addCleanupr r r ZdoReadrB �lenr� Z assertFalse�set� intersectionr� )rD �ancillaryPackerr� r� r� r� r� Z sendSocketZ recvSocketr� ZreceiverZ fileOneFDZ fileOneNameZ fileTwoFDZ fileTwoNameZ dataToSend� fdsToSend� ancillary� expectedCountZdeviceInodesSentrH r� rI �)_sendmsgMixinFileDescriptorReceivedDriver� s4 z:UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriverc s& ddl m� � fdd�}| �|� dS )z� _SendmsgMixin handles multiple file descriptors per recvmsg, calling L{IFileDescriptorReceiver.fileDescriptorReceived} once per received file descriptor. Scenario: single CMSG with two FDs. r �� SCM_RIGHTSc s t � td| �� fg}d}||fS )N�ii� )r� �r r �r� r� r� r� rH rI r� � s z[UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgOneCMSG.<locals>.ancillaryPackerN�r� r� r� �rD r� rH r� rI �1test_multiFileDescriptorReceivedPerRecvmsgOneCMSG� s zBUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgOneCMSGc s& ddl m� � fdd�}| �|� dS )z� _SendmsgMixin handles multiple file descriptors per recvmsg, calling L{IFileDescriptorReceiver.fileDescriptorReceived} once per received file descriptor. Scenario: two CMSGs with one FD each. r r� c s � fdd�| D �}d}||fS )Nc s g | ]}t � td |�f�qS )�ir� r� r� rH rI r� s �zpUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGs.<locals>.ancillaryPacker.<locals>.<listcomp>r� rH r� r� rH rI r� s �z\UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGs.<locals>.ancillaryPackerNr� r� rH r� rI �2test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGs s zCUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGsz=Multi control message ancillary sendmsg not supported on Mac.c s~ ddl m� dd� }�fdd�}g }t|j� | �t|j� | ��d|� | �|� d� t� fd d �|D ��}| � |d � d S ) z� _SendmsgMixin handles multiple file descriptors per recvmsg, calling L{IFileDescriptorReceiver.fileDescriptorReceived} once per received file descriptor. Scenario: unsupported CMSGs. r r� c S s g }d}||fS )Nr rH r� rH rH rI r� ) s z[UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG.<locals>.ancillaryPackerc s d}dg}d}� � |||�S )Ns some data)NNr� r )ZRecievedMessage)r� �args�kwargsr] r� �flagsr� rH rI � fakeRecvmsgUnsupportedAncillary. s zkUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG.<locals>.fakeRecvmsgUnsupportedAncillaryZrecvmsgz#received unsupported ancillary datac 3 s | ]}� |d kV qdS )�formatNrH )r� �e)�expectedMessagerH rI � <genexpr>= s zUUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG.<locals>.<genexpr>z+Expected message not found in logged eventsN) Ztwisted.pythonr� r8 r� r� r9 r} r� �anyr� )rD r� r� �events�foundrH )r� r� rI �1test_multiFileDescriptorReceivedPerRecvmsgBadCMSG s zBUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSGc C s� ddl m} |� \}}g }t|j� | �t|j� G dd� dt�}||�� d�}t� }t | ||| j � |� � |� d� | � d|�d�� d }d } t|j|j| |d �} |D ],} t| �D ]\} } | | �| �kr� q�q� q�q�| �d | t|�f � d S )z� If associated with a protocol which does not provide L{IFileDescriptorReceiver}, file descriptors received by the L{IUNIXTransport} implementation are closed and a warning is emitted. r r� c @ s e Zd Zdd� ZdS )zRUNIXTestsBuilder.test_avoidLeakingFileDescriptors.<locals>.RecordEndpointAddressesc S s& | j �� | _| j �� | _t�| � d S re )r_ rw � hostAddressr� � peerAddressr[ rd rc rH rH rI rd V s zaUNIXTestsBuilder.test_avoidLeakingFileDescriptors.<locals>.RecordEndpointAddresses.connectionMadeNr� rH rH rH rI �RecordEndpointAddressesU s r� r� Fr� r� z�%(protocolName)s (on %(hostAddress)r) does not provide IFileDescriptorReceiver; closing file descriptor received (from %(peerAddress)r).r0 )r� r� Z protocolNamer� z3Expected event (%s) not found in logged events (%s)N)r r� r8 r� r� r9 r[ r� r0 r3 r� r � setblockingrB Zrecvr� r� r� r6 �getr r )rD r� Z probeClientZ probeServerr� r� rW rY r� ZclsNameZ expectedEventZlogEvent�k�vrH rH rI � test_avoidLeakingFileDescriptorsC s> �� ��z1UNIXTestsBuilder.test_avoidLeakingFileDescriptorsc s� t t�G � fdd�dt��}t� }t|�� d�}|� }t� ||� j� � �t |j d � t rx� �dt |j dd� �� n � �dd� |j dd� �� dS )z� L{IUNIXTransport.sendFileDescriptor} sends file descriptors before L{ITransport.write} sends normal bytes. c s( e Zd Zdd� Z� fdd�Zdd� ZdS )zJUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes.<locals>.RecordEventsc S s t �| � g | _d S re )r0 rd r� rc rH rH rI rd � s zYUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes.<locals>.RecordEvents.connectionMadec s � � t|� | j�t|�� d S re )r� r r� r� �type)Z innerSelfrm rc rH rI rn � s zaUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes.<locals>.RecordEvents.fileDescriptorReceivedc S s | j �|� d S re )r� �extendrq rH rH rI rr � s zWUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes.<locals>.RecordEvents.dataReceivedN)rK rL rM rd rn rr rH rc rH rI � RecordEvents� s r� r� r � Nr� )r r! r0 r r[ r� r3 r� rB �intr� r5 �bytes�join)rD r� r� rW rY rH rc rI �#test_descriptorDeliveredBeforeBytes� s z4UNIXTestsBuilder.test_descriptorDeliveredBeforeBytes)rK rL rM rN r" rZ rQ r� ru rz r; �isLinux�skipr~ r� r� r� � sendmsgSkipr� r� r� r� r� ZisMacOSXr� r� r� rH rH rH rI rs � sT � � -]�$A rs c @ s6 e Zd ZdZejfZdd� Zdd� Ze � � s2de_ dS )�UNIXDatagramTestsBuilderzE Builder defining tests relating to L{IReactorUNIXDatagram}. c C s | � d| �� t� � dS )z� The UNIX socket created by L{IReactorUNIXDatagram.listenUNIXDatagram} is created with the mode specified. �listenUNIXDatagramN)rJ r r% rc rH rH rI �test_listenMode� s z(UNIXDatagramTestsBuilder.test_listenModec C s> t | �}| �� }|�d| t� �}| �|�� td| �� dS )z� On Linux, a UNIX socket path may begin with C{' '} to indicate a socket in the abstract namespace. L{IReactorUNIX.listenUNIXDatagram} accepts such a path. rv N)rP r@ r� r% rB rw r rx rH rH rI rz � s z<UNIXDatagramTestsBuilder.test_listenOnLinuxAbstractNamespacer{ N) rK rL rM rN r ZIReactorUNIXDatagramrZ r� rz r; r� r� rH rH rH rI r� � s �r� c @ s( e Zd ZdZeefZdd� Zdd� ZdS )�SocketUNIXMixinzb Mixin which uses L{IReactorSocket.adoptStreamPort} to hand out listening UNIX ports. c C sZ t t�}tddd�}|�|� |�d� |�d� z|�|�� |j |�W �S |�� X dS )zj Get a UNIX port from a reactor, wrapping an already-initialized file descriptor. rR rS rT � FN) r r r r� �listenr� r ZadoptStreamPortr� Zfamily)rD rG rF ZportSockrE rH rH rI �getListeningPort� s �z SocketUNIXMixin.getListeningPortc C s |� |j|�S �aZ Connect to a listening UNIX socket. @param reactor: The reactor under test. @type reactor: L{IReactorUNIX} @param address: The listening's address. @type address: L{UNIXAddress} @param factory: The client factory. @type factory: L{ClientFactory} @return: The connector �r rX �rD rG �addressrF rH rH rI �connectToListener� s z!SocketUNIXMixin.connectToListenerN) rK rL rM rN r" r# rZ r r rH rH rH rI r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )�ListenUNIXMixinzZ Mixin which uses L{IReactorTCP.listenUNIX} to hand out listening UNIX ports. c C s t ddd�}|�||�S )z0 Get a UNIX port from a reactor rR rS rT )r rt )rD rG rF rE rH rH rI r � s z ListenUNIXMixin.getListeningPortc C s |� |j|�S r r r rH rH rI r s z!ListenUNIXMixin.connectToListenerN)rK rL rM rN r r rH rH rH rI r � s r c @ s" e Zd ZefZdd� Zdd� ZdS )�UNIXPortTestsMixinc C s d|t |�� j�f S )zZ Get the message expected to be logged when a UNIX port starts listening. z%s starting on %r�r4 rw rX )rD ry rF rH rH rI �#getExpectedStartListeningLogMessage s �z6UNIXPortTestsMixin.getExpectedStartListeningLogMessagec C s dt |�� j�f S )zJ Get the expected connection lost message for a UNIX port z(UNIX Port %s Closed)r )rD ry rH rH rI � getExpectedConnectionLostLogMsg$ s z2UNIXPortTestsMixin.getExpectedConnectionLostLogMsgN)rK rL rM r" rZ r r rH rH rH rI r s r c @ s e Zd ZdZdS )�UNIXPortTestsBuilderz. Tests for L{IReactorUNIX.listenUnix} N�rK rL rM rN rH rH rH rI r , s r c @ s e Zd ZdZdS )�UNIXFDPortTestsBuilderz3 Tests for L{IReactorUNIX.adoptStreamPort} Nr rH rH rH rI r 4 s r c @ s. e Zd ZeeefZdd� Zdd� Zdd� Z dS )�%UNIXAdoptStreamConnectionTestsBuilderc C s~ | � � }ddlm} G dd� dt�}|tt�\}}|�d� | �|j� | �|j� |� � }|� }|� |t|�}| � |� dS )z� {IReactorSocket.adoptStreamConnection} returns None if the given factory's buildProtocol returns None. r r� c @ s e Zd Zdd� ZdS )zXUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNone.<locals>.NoneFactoryc S s d S re rH )rD r rH rH rI � buildProtocolN s zfUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNone.<locals>.NoneFactory.buildProtocolN)rK rL rM r rH rH rH rI � NoneFactoryM s r FN) r@ r r� r& r r r� r� r r� �adoptStreamConnectionZ assertIsNone) rD rG r� r �s1�s2Zs1FDrF r� rH rH rI � test_buildProtocolReturnsNone? s zCUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNonec s< � fdd�}� � � }� j|ddd�}|�|� � �|� dS )z> Helper method to test UNIX server addresses. c s� | \}}}zftd|�� j�}� �d|j j|f t|j �� � �d|j j|f |j j� |j j d }� � |t � W 5 |j �� X d S )Nr� z <AccumulatingProtocol #%s on %s>zAccumulatingProtocol,%s,%sr ) r_ rb r= rw rX rB Z sessionno�strZlogstrrF Z peerAddressesr� r )� protocolsrY rW ry ZportPathr� rc rH rI � connecteda s$ �� �� zOUNIXAdoptStreamConnectionTestsBuilder.test_ServerAddressUNIX.<locals>.connectedN)� interface� addressFamily)r@ �getConnectedClientAndServerr� Z runReactor)rD r rG r� rH rc rI �test_ServerAddressUNIX\ s z<UNIXAdoptStreamConnectionTestsBuilder.test_ServerAddressUNIXc s� t � }t� |_t � �t� �_t� �_t� }t� |_t� |_tddd�}��||����fdd�}|j�|� t|j�jg�}�fdd�} |� | � t� � � � | � t|j�jg�} � �fdd �} | �| � �� �� � j |� � S ) a0 Return a L{Deferred} firing with a L{MyClientFactory} and L{MyServerFactory} connected pair, and the listening C{Port}. The particularity is that the server protocol has been obtained after doing a C{adoptStreamConnection} against the original server connection. rR rS rT c s0 � � | j� � �| j� � �| j�� t�� d S re )Z removeReaderr_ Z removeWriterr r� r )r� )rG rW rH rI �firstServerConnected� s �z_UNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer.<locals>.firstServerConnectedc s � j r� �� | S re )�runningr� )r� )rG rH rI r� � s zOUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer.<locals>.stopc s | \}}� � ||�f� d S re )rl )r rY rW )�deferredry rH rI r� � s zPUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer.<locals>.start)r/ r ZprotocolConnectionMadeZprotocolConnectionLostr. r rt r� r r� r� r rw rX ) rD rG r r Z firstServerrY rE r Z lostDeferredr� Z startDeferredr� rH )r ry rG rW rI r { s4 � � zAUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServerN) rK rL rM r$ r# r" rZ r r r rH rH rH rI r < s r c @ s6 e Zd ZdZefZdZedd� �Zdd� Z dd� Z dS ) �UnixClientTestsBuilderz7 Define tests for L{IReactorUNIX.connectUNIX}. Nc C s | j dkrt| �| _ | j S )z� Return a path usable by C{connectUNIX} and C{listenUNIX}. @return: A path instance, built with C{_abstractPath}. N)�_pathrP rc rH rH rI rE � s zUnixClientTestsBuilder.pathc C s |� | j|�S )z� Start an UNIX server with the given C{factory}. @param reactor: The reactor to create the UNIX port in. @param factory: The server factory. @return: A UNIX port instance. )rt rE �rD rG rF rH rH rI r � s z UnixClientTestsBuilder.listenc C s |� | j|�S )z� Start an UNIX client with the given C{factory}. @param reactor: The reactor to create the connection in. @param factory: The client factory. @return: A UNIX connector instance. )r rE r# rH rH rI �connect� s z UnixClientTestsBuilder.connect) rK rL rM rN r" rZ r" �propertyrE r r$ rH rH rH rI r! � s r! )jrN Z __future__r r r r �osr r r r Ztempfiler r r r r r r Zpprintr Zhashlibr Zstructr r � ImportErrorZzope.interfacer Ztwisted.internetr r Ztwisted.internet.addressr Ztwisted.internet.deferr r r Ztwisted.internet.endpointsr r Ztwisted.internet.errorr r r Ztwisted.internet.interfacesr! r" r# r$ Ztwisted.internet.protocolr% r& r' Ztwisted.internet.taskr( Z&twisted.internet.test.connectionmixinsr) Z#twisted.internet.test.reactormixinsr* Z twisted.internet.test.test_corer+ Z twisted.internet.test.test_tcpr, r- r. r/ r0 r1 r2 r3 Ztwisted.python.compatr4 r5 r6 Ztwisted.python.failurer7 Ztwisted.python.logr8 r9 r: Ztwisted.python.runtimer; Ztwisted.python.reflectr< Ztwisted.python.filepathr= r� �objectr> rP rQ r[ ri rs r� r� r r r r r �globals�updateZmakeTestCaseClassesr! rH rH rH rI �<module> s� �'A Z!,! � �u/
Save File
Cancel