Your IP : 216.73.216.213
B
;��]5; � @ s* d Z ddlmZmZ ddlZddlZddlZddlZddlm Z m
Z
m Z m Z m
Z
ddlmZmZ ddlmZ ddlmZmZ ddlmZ dd lmZ dd
lmZmZ G d d � d e j �Z G d
d� dej �Z G dd� de j �Z!G dd� de j"�Z#G dd� dej �Z$e �%e
d��sde _&e �'e
d��s&de$_&dS )zK
Tests for implementations of L{IReactorUNIX} and L{IReactorUNIXDatagram}.
� )�division�absolute_importN)�
interfaces�reactor�protocol�error�address)�defer�utils)�lockfile)�_PY3�
networkString)�FilePath)�unittest)�MyServerFactory�MyClientFactoryc @ s e Zd Zdd� Zdd� ZdS )� FailedConnectionClientFactoryc C s
|| _ d S )N)�onFail)�selfr � r �I/opt/alt/python37/lib64/python3.7/site-packages/twisted/test/test_unix.py�__init__ s z&FailedConnectionClientFactory.__init__c C s | j �|� d S )N)r Zerrback)r Z connector�reasonr r r �clientConnectionFailed s z4FailedConnectionClientFactory.clientConnectionFailedN)�__name__�
__module__� __qualname__r r r r r r r s r c @ sj e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
r^de _dd� ZdS )�UnixSocketTestsz
Test unix sockets.
c s� �� � }�� � � t� �t�� }�_t�|��}��|j� t � t j
t j �}��|j � |�
� � |�|� � ��fdd�}|�|� |S )z�
The address passed to the server factory's C{buildProtocol} method and
the address returned by the connected protocol's transport's C{getPeer}
method match the address the client socket is bound to.
c s0 t �� �}���j|g� ��| j�� |� d S )N)r � UNIXAddress� assertEqual�
peerAddresses� transportZgetPeer)�protoZexpected)�peernamer �
serverFactoryr r �
cbConnMade6 s
z1UnixSocketTests.test_peerBind.<locals>.cbConnMade)�mktempr r �Deferred�protocolConnectionMader �
listenUNIX�
addCleanup�
stopListening�socket�AF_UNIX� SOCK_STREAM�close�bind�connect� addCallback)r �filenameZconnMade�unixPortZ
unixSocketr% r )r# r r$ r �
test_peerBind& s
z UnixSocketTests.test_peerBindc s� �� � �t� }t�� }||_t��|�}��|j� t � � t�� }|� _t�
�� � t� ||g�}� ��fdd�}|� |� |S )z�
L{IReactorUNIX.connectUNIX} can be used to connect a client to a server
started with L{IReactorUNIX.listenUNIX}.
c s6 | \}}�� � jt���g� |j�� |j�� d S )N)r r r r r! �loseConnection)�args�serverProtocolZclientProtocol)�
clientFactoryr3 r r r � allConnectedN s
z1UnixSocketTests.test_dumber.<locals>.allConnected)
r& r r r' r( r r) r* r+ r � connectUNIX�
gatherResultsr2 )r r$ �serverConnMader4 �clientConnMade�dr: r )r9 r3 r r � test_dumber> s
zUnixSocketTests.test_dumberc s� �� � �t� }t�� }||_tj�|dd����t� �d �� t
� � t�� }|� _tj �� dd� t� ||g�}� ���fdd�}|�
|� ��fdd �}|�
|� |S )
z�
A lockfile is created and locked when L{IReactorUNIX.listenUNIX} is
called and released when the Deferred returned by the L{IListeningPort}
provider's C{stopListening} method is called back.
T)�wantPIDz.lock� )�checkPIDc s: | \}}�� � jt���g� |j�� |j�� ��� S )N)r r r r r! r6 r+ )r7 r8 Z clientProto)r9 r3 r r4 r r �
_portStuffq s
z0UnixSocketTests.test_pidFile.<locals>._portStuffc s �� t�� d �d� d S )Nz.lock�locked)Z assertFalser �isLocked)�ignored)r3 r r r �_check~ s z,UnixSocketTests.test_pidFile.<locals>._check)r& r r r' r( r r) Z
assertTruer rF r r; r< r2 )r r$ r= r> r? rD rH r )r9 r3 r r4 r � test_pidFile[ s
z UnixSocketTests.test_pidFilec sR | � � � t� �tj� �dd�}| jtjtj� �dd� � �fdd�}|�� �|�S )z�
L{IReactorUNIX.listenUNIX} raises L{error.CannotListenError} if passed
the name of a file on which a server is already listening.
T)rA c s t j� �dd�}|�� S )NT)rA )r r) r+ )�ignr4 )r3 r$ r r �stoppedListening� s z<UnixSocketTests.test_socketLocking.<locals>.stoppedListening) r& r r r) � assertRaisesr �CannotListenErrorr+ r2 )r r4 rK r )r3 r$ r �test_socketLocking� s z"UnixSocketTests.test_socketLockingc C sj | � � | _td| jf �}dttj�tj��� � ji}ttj
�� � j}t j |dd|f|d�}|�
|� |S )Nzmfrom twisted.internet import protocol, reactor
reactor.listenUNIX(%r, protocol.ServerFactory(),wantPID=True)
s
PYTHONPATHs -us -c)�env)r& r3 r
r �os�pathsep�join�sys�pathZ asBytesMode�
executabler
ZgetProcessValuer2 )r �callback�sourcerO ZpyExer? r r r �_uncleanSocketTest� s
z"UnixSocketTests._uncleanSocketTestc s � fdd�}� � |�S )a"
If passed C{True} for the C{wantPID} parameter, a server can be started
listening with L{IReactorUNIX.listenUNIX} when passed the name of a
file on which a previous server which has not exited cleanly has been
listening using the C{wantPID} option.
c s t j� jt� dd�}|�� S )NT)rA )r r) r3 r r+ )rJ �p)r r r �ranStupidChild� s zGUnixSocketTests.test_uncleanServerSocketLocking.<locals>.ranStupidChild)rX )r rZ r )r r � test_uncleanServerSocketLocking� s z/UnixSocketTests.test_uncleanServerSocketLockingc s � fdd�}� � |�S )z�
If passed C{True} for the C{checkPID} parameter, a client connection
attempt made with L{IReactorUNIX.connectUNIX} fails with
L{error.BadFileError}.
c s0 t �� }t|�}tj� j|dd� � �|tj�S )NT)rC ) r r' r r r; r3 Z
assertFailurer Z BadFileError)rJ r? �f)r r r rZ � s zCUnixSocketTests.test_connectToUncleanServer.<locals>.ranStupidChild)rX )r rZ r )r r �test_connectToUncleanServer� s z+UnixSocketTests.test_connectToUncleanServerc sj �� � }t�||��d� |f }��t��|� ��t��|� t��j�}� ��fdd�}|� |� |S )z~
Test the C{__str__} and C{__repr__} implementations of a UNIX port when
used with the given factory.
z
<%s on %r>c s. d� f }�� t��|� �� t��|� d S )Nz<%s (not listening)>)r �repr�str)rJ �unconnectedString)� factoryNamer r4 r r rK � s
z3UnixSocketTests._reprTest.<locals>.stoppedListening)
r& r r) r r^ r_ r �
maybeDeferredr+ r2 )r r$ ra r3 �connectedStringr? rK r )ra r r4 r � _reprTest� s
zUnixSocketTests._reprTestc C s* G dd� d�}| � |tj� | �|� d�S )a
The two string representations of the L{IListeningPort} returned by
L{IReactorUNIX.listenUNIX} contains the name of the classic factory
class being used and the filename on which the port is listening or
indicates that the port is not listening.
c @ s e Zd Zdd� Zdd� ZdS )zCUnixSocketTests.test_reprWithClassicFactory.<locals>.ClassicFactoryc S s d S )Nr )r r r r �doStart� s zKUnixSocketTests.test_reprWithClassicFactory.<locals>.ClassicFactory.doStartc S s d S )Nr )r r r r �doStop� s zJUnixSocketTests.test_reprWithClassicFactory.<locals>.ClassicFactory.doStopN)r r r re rf r r r r �ClassicFactory� s rg z%twisted.test.test_unix.ClassicFactory)�assertIsInstance�types� ClassTyperd )r rg r r r �test_reprWithClassicFactory� s z+UnixSocketTests.test_reprWithClassicFactoryz)Classic classes do not exist on Python 3.c C s* G dd� dt �}| �|t� | �|� d�S )a!
The two string representations of the L{IListeningPort} returned by
L{IReactorUNIX.listenUNIX} contains the name of the new-style factory
class being used and the filename on which the port is listening or
indicates that the port is not listening.
c @ s e Zd Zdd� Zdd� ZdS )zEUnixSocketTests.test_reprWithNewStyleFactory.<locals>.NewStyleFactoryc S s d S )Nr )r r r r re � s zMUnixSocketTests.test_reprWithNewStyleFactory.<locals>.NewStyleFactory.doStartc S s d S )Nr )r r r r rf � s zLUnixSocketTests.test_reprWithNewStyleFactory.<locals>.NewStyleFactory.doStopN)r r r re rf r r r r �NewStyleFactory� s rl z&twisted.test.test_unix.NewStyleFactory)�objectrh �typerd )r rl r r r � test_reprWithNewStyleFactory� s z,UnixSocketTests.test_reprWithNewStyleFactoryN)r r r �__doc__r5 r@ rI rN rX r[ r] rd rk r �skipro r r r r r " s )r c @ s8 e Zd Zd ZZdZdd� Zdd� Zdd� Zd d
� Z dS ) � ClientProtoFNc C s t �� | _t �� | _d S )N)r r' �deferredStarted�deferredGotBack)r r r r r s
zClientProto.__init__c C s
d| _ d S )NT)�stopped)r r r r � stopProtocol s zClientProto.stopProtocolc C s d| _ | j�d � d S )NT)�startedrs rV )r r r r �
startProtocol s zClientProto.startProtocolc C s || _ | j�d � d S )N)�gotbackrt rV )r �datar r r �datagramReceived s z ClientProto.datagramReceived)
r r r rw ru ry r rv rx r{ r r r r rr s rr c @ s< e Zd Zd ZZd ZZdd� Zdd� Zdd� Z d d
� Z
dS ) � ServerProtoFNc C s t �� | _t �� | _d S )N)r r' rs �deferredGotWhat)r r r r r s
zServerProto.__init__c C s
d| _ d S )NT)ru )r r r r rv $ s zServerProto.stopProtocolc C s d| _ | j�d � d S )NT)rw rs rV )r r r r rx ' s zServerProto.startProtocolc C s* || _ | j�d|� || _| j�d � d S )Ns hi back)�gotfromr! �write�gotwhatr} rV )r rz �addrr r r r{ + s z ServerProto.datagramReceived) r r r rw ru r� r~ r rv rx r{ r r r r r| s r| c @ sB e Zd ZdZdd� Zdd� Zdd� Zdd � Zer6d
e_ d d � Z
d
S )�DatagramUnixSocketTestsz%
Test datagram UNIX sockets.
c s� �� � � �� � }t� �t� �t�|��}��|j� tj|�� d�}��|j� t� �j
�j
g�}��fdd�}� ���fdd�}|� |� |� |� |S )zf
Test that a datagram can be sent to and received by a server and vice
versa.
)Z bindAddressc s � j �d� t��j� jg�S )Ns hi)r! r r r< r} rt )rG )�cp�spr r r F s z4DatagramUnixSocketTests.test_exchange.<locals>.writec s. �� d�j� �� � �j� �� d�j� d S )Ns his hi back)r r� r~ ry )rG )�
clientaddrr� r r� r r �_cbTestExchangeK s z>DatagramUnixSocketTests.test_exchange.<locals>._cbTestExchange) r& r| rr r �listenUNIXDatagramr* r+ ZconnectUNIXDatagramr r< rs r2 )r Z
serveraddr�s�cr? r r� r )r� r� r r� r �
test_exchange7 s
z%DatagramUnixSocketTests.test_exchangec C sD | � � }t� }t�||�}| �tjtj||� |�� t� |� dS )z�
L{IReactorUNIXDatagram.listenUNIXDatagram} raises
L{error.CannotListenError} if the unix socket specified is already in
use.
N)
r& r| r r� rL r rM r+ rP �unlink)r r� rY r� r r r �test_cannotListenU s z)DatagramUnixSocketTests.test_cannotListenc sj �� � }t�||��d� |f }��t��|� ��t��|� t��j�}� ��fdd�}|� |� |S )z�
Test the C{__str__} and C{__repr__} implementations of a UNIX datagram
port when used with the given protocol.
z
<%s on %r>c s. d� f }�� t��|� �� t��|� d S )Nz<%s (not listening)>)r r^ r_ )rJ r` )� protocolNamer r4 r r rK r s
z;DatagramUnixSocketTests._reprTest.<locals>.stoppedListening)
r& r r� r r^ r_ r rb r+ r2 )r Z serverProtor� r3 rc Z stopDeferredrK r )r� r r4 r rd e s
z!DatagramUnixSocketTests._reprTestc C s* G dd� d�}| � |tj� | �|� d�S )a0
The two string representations of the L{IListeningPort} returned by
L{IReactorUNIXDatagram.listenUNIXDatagram} contains the name of the
classic protocol class being used and the filename on which the port is
listening or indicates that the port is not listening.
c @ s e Zd Zdd� Zdd� ZdS )zMDatagramUnixSocketTests.test_reprWithClassicProtocol.<locals>.ClassicProtocolc S s d S )Nr )r r! r r r �makeConnection� s z\DatagramUnixSocketTests.test_reprWithClassicProtocol.<locals>.ClassicProtocol.makeConnectionc S s d S )Nr )r r r r rf � s zTDatagramUnixSocketTests.test_reprWithClassicProtocol.<locals>.ClassicProtocol.doStopN)r r r r� rf r r r r �ClassicProtocol� s r� z&twisted.test.test_unix.ClassicProtocol)rh ri rj rd )r r� r r r � test_reprWithClassicProtocolz s z4DatagramUnixSocketTests.test_reprWithClassicProtocolz)Classic classes do not exist on Python 3.c C s* G dd� dt �}| �|t� | �|� d�S )a2
The two string representations of the L{IListeningPort} returned by
L{IReactorUNIXDatagram.listenUNIXDatagram} contains the name of the
new-style protocol class being used and the filename on which the port
is listening or indicates that the port is not listening.
c @ s e Zd Zdd� Zdd� ZdS )zODatagramUnixSocketTests.test_reprWithNewStyleProtocol.<locals>.NewStyleProtocolc S s d S )Nr )r r! r r r r� � s z^DatagramUnixSocketTests.test_reprWithNewStyleProtocol.<locals>.NewStyleProtocol.makeConnectionc S s d S )Nr )r r r r rf � s zVDatagramUnixSocketTests.test_reprWithNewStyleProtocol.<locals>.NewStyleProtocol.doStopN)r r r r� rf r r r r �NewStyleProtocol� s r� z'twisted.test.test_unix.NewStyleProtocol)rm rh rn rd )r r� r r r � test_reprWithNewStyleProtocol� s z5DatagramUnixSocketTests.test_reprWithNewStyleProtocolN) r r r rp r� r� rd r� r rq r� r r r r r� 3 s r� z1This reactor does not support UNIX domain socketsz3This reactor does not support UNIX datagram sockets)(rp �
__future__r r rP rS ri r, Ztwisted.internetr r r r r r r
Ztwisted.pythonr Ztwisted.python.compatr r
Ztwisted.python.filepathr Z
twisted.trialr Ztwisted.test.test_tcpr r Z
ClientFactoryr ZTestCaser ZConnectedDatagramProtocolrr ZDatagramProtocolr| r� Z IReactorUNIXrq ZIReactorUNIXDatagramr r r r �<module> s, fv