File: //proc/self/root/lib/python3/dist-packages/twisted/test/__pycache__/test_ftp.cpython-38.pyc
U
��W[� � @ sf d Z ddlZddlZddlmZ ddlZddlZddlZddlm Z ddl
m Z ddl m
Z
mZmZ ddlmZ ddlmZ dd lmZmZmZmZmZ dd
lmZ dd l m Z dd l m Z m Z m!Z! dd
l"m#Z# ddl$m%Z% ddl&m'Z' ddl m(Z(m)Z) e!j*�+� �r
dZ,ndZ,G dd� de j-�Z.G dd� dej/�Z0dXdd�Z1G dd� de'j2�Z3G d d � d e3�Z4G d d � d e3�Z5G d d!� d!e3�Z6G d"d#� d#e3�Z7G d$d%� d%e7�Z8G d&d'� d'e'j2�Z9G d(d)� d)e'j2�Z:G d*d+� d+ej/�Z;G d,d-� d-e(j<�Z=G d.d/� d/e'j2�Z>G d0d1� d1e'j2�Z?G d2d3� d3e'j2�Z@G d4d5� d5e'j2�ZAG d6d7� d7e'j2�ZBG d8d9� d9e'j2�ZCG d:d;� d;e'j2�ZDG d<d=� d=e'j2�ZEG d>d?� d?e'j2�ZFG d@dA� dAe'j2�ZGG dBdC� dCe'j2�ZHG dDdE� dE�ZIG dFdG� dGe'j2eI�ZJe e�G dHdI� dIeK��ZLG dJdK� dKeK�ZMG dLdM� dM�ZNG dNdO� dOe'j2eN�ZOe e(jP�G dPdQ� dQ��ZQG dRdS� dS�ZRG dTdU� dUe'j2�ZSG dVdW� dWe'j2�ZTdS )Yz
FTP tests.
� N)�BytesIO)� implementer)� verifyClass)�portal�checkers� credentials)�UnauthorizedLogin)�IRealm)�reactor�task�protocol�defer�error)� IConsumer)�basic)�failure�filepath�runtime)�range)�
proto_helpers)�unittest)�ftp�loopbackzCannot run on Windowsc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � ZdS ) �DummyNc C s g | _ g | _d S �N)�lines�rawData��self� r �7/usr/lib/python3/dist-packages/twisted/test/test_ftp.py�__init__) s zDummy.__init__c C s | j | _d S r )�factory�fr r r r �connectionMade. s zDummy.connectionMadec C s | j �|� d S r )r �append�r �liner r r � lineReceived2 s zDummy.lineReceivedc C s | j �|� d S r )r r% �r �datar r r �rawDataReceived6 s zDummy.rawDataReceivedc C s d S r r r& r r r �lineLengthExceeded: s zDummy.lineLengthExceeded) �__name__�
__module__� __qualname__Zlognamer! r$ r( r+ r, r r r r r ' s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_BufferingProtocolc C s d| _ t�� | _d S �N� )�bufferr
�Deferred�dr r r r r$ @ s z!_BufferingProtocol.connectionMadec C s | j |7 _ d S r �r3 r) r r r � dataReceivedE s z _BufferingProtocol.dataReceivedc C s | j �| � d S r )r5 �callback)r �reasonr r r �connectionLostI s z!_BufferingProtocol.connectionLostN)r- r. r/ r$ r7 r: r r r r r0 ? s r0 � 127.0.0.1�90 c C s dt �||�f }|�| j�S )z�
Construct a passive mode message with the correct encoding
@param protocol: the FTP protocol from which to base the encoding
@param host: the hostname
@param port: the port
@return: the passive mode message
z 227 Entering Passive Mode (%s).)r �encodeHostPort�encode� _encoding)r �host�port�msgr r r �passivemode_msgN s rC c @ sF e Zd ZdZejZdZdd� Zddd�Z ddd �Z
d
d � Z d d
� Z dS )�FTPServerTestCasezw
Simple tests for an FTP server with the default settings.
@ivar clientFactory: class used as ftp client.
� anonymousc s> g ��� � �_t��j� t��j��_t�t j
�j�jd��}|� t �
� tj� t �� }d�_d�_|��j�j� |� |tj� t j|�jd��_tjd�jdd� �_}��|j � �jj � t � � �� ���fdd �}|�j_ |� � j}t!�"t�j#�}|�$d|�}��fd
d �}|�%|� ��t&j't�� t �(�|g�S ) N)Z
anonymousRootZuserHomez test-userz
test-password)r �
userAnonymousr r; �Z interfacec sD �j `� | �}|j�_�fdd�}��|� ��d � ��|� |S )Nc s � j jd k r� j j�� d S r ��serverProtocol� transport�loseConnectionr r r r �
cleanupServer� s zQFTPServerTestCase.setUp.<locals>._rememberProtocolInstance.<locals>.cleanupServer)r" �
buildProtocol�wrappedProtocolrI �
addCleanupr8 r% )�addrr rL �rM �d1� protocolsr r r �_rememberProtocolInstance� s
z:FTPServerTestCase.setUp.<locals>._rememberProtocolInstancec s& | �_ ���j jj� � ��j � d S r )�clientrO rJ rK r% )rU )rS r r r � gotClient� s z*FTPServerTestCase.setUp.<locals>.gotClient))�mktemp� directory�os�mkdirr �FilePath�dirPathr ZPortalr �FTPRealm�registerCheckerr ZAllowAnonymousAccessr Z
IAnonymous�'InMemoryUsernamePasswordDatabaseDontUse�username�passwordZaddUser�IUsernamePassword�
FTPFactoryrF r" r
� listenTCPrA rO �
stopListeningrM r
r4 �getHostr �
ClientCreator�
clientFactory�
connectTCP� addCallbackr ZwaitUntilAllDisconnected�
gatherResults) r �pZ
users_checkerrA rT �portNumZ
clientCreator�d2rV r rQ r �setUpe sN
�
�� �
�zFTPServerTestCase.setUpNc s, |dkrt �d�}� ��fdd�}|�|�S )z�
Asserts that a sending an FTP command receives the expected
response.
Returns a Deferred. Optionally accepts a deferred to chain its actions
to.
Nc s$ �j �� �}��fdd�}|�|�S )Nc s �� � | � d S r �� assertEqual��
responseLines)�expectedResponseLinesr r r � gotResponse� s zRFTPServerTestCase.assertCommandResponse.<locals>.queueCommand.<locals>.gotResponse�rU �queueStringCommandrj )�ignoredr5 ru ��commandrt r r r � queueCommand� s z=FTPServerTestCase.assertCommandResponse.<locals>.queueCommand)r
�succeedrj )r rz rt �
chainDeferredr{ r ry r �assertCommandResponse� s
z'FTPServerTestCase.assertCommandResponsec sP |d krt �d �}� �fdd�}|�|� ��|tj� ��fdd�}|�|�S )Nc s �j �� �S r �rU rw �rx �rz r r r r{ � s z;FTPServerTestCase.assertCommandFailed.<locals>.queueCommandc s � d k r�� � | jd � d S �Nr �rq �args)� exception)�expectedResponser r r �failed� s
�z5FTPServerTestCase.assertCommandFailed.<locals>.failed)r
r| rj �
assertFailurer �
CommandFailed)r rz r� r} r{ r� r )rz r� r r �assertCommandFailed� s
z%FTPServerTestCase.assertCommandFailedc C s | � ddg�}| j ddg|d�S )N�USER anonymous�8331 Guest login ok, type your email address as password.�PASS test@twistedmatrix.com�2230 Anonymous login ok, access restrictions apply.�r} �r~ �r r5 r r r �_anonymousLogin� s ��z!FTPServerTestCase._anonymousLoginc C s2 | � d| j d| j g�}| j d| j dg|d�S )zx
Authenticates the FTP client using the test account.
@return: L{Deferred} of command response
zUSER %sz 331 Password required for %s.zPASS %sz230 User logged in, proceedr� )r~ r` ra r� r r r �
_userLogin� s
��z FTPServerTestCase._userLogin)N)NN)
r- r. r/ �__doc__r �FTPClientBasicrh rF ro r~ r� r� r� r r r r rD \ s A �
�
rD c @ s e Zd ZdZdZdd� ZdS )�FTPAnonymousTestsz�
Simple tests for an FTP server with different anonymous username.
The new anonymous username used in this test case is "guest"
Zguestc C s | � ddg�}| j ddg|d�S )a
Tests whether the changing of the anonymous username is working or not.
The FTP server should not comply about the need of password for the
username 'guest', letting it login as anonymous asking just an email
address as password.
z
USER guestr� r� r� r� r� r� r r r �test_anonymousLogin� s ��z%FTPAnonymousTests.test_anonymousLoginN)r- r. r/ r� rF r� r r r r r� � s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6S )7�BasicFTPServerTestsz,
Basic functionality of FTP server.
c sL d� j _t�tt�}|�d� j�� j�}|j dd� �}|j � fdd��}|S )zn
When the connection limit is reached, the server should send an
appropriate response
� r; c S s | j S r )r5 ��protor r r rV s z>BasicFTPServerTests.test_tooManyConnections.<locals>.gotClientc s � � d| j� d S )Ns; 421 Too many users right now, try again in a few minutes.
�rq r3 r� r r r �onConnectionLost s �zEBasicFTPServerTests.test_tooManyConnections.<locals>.onConnectionLost)
r" ZconnectionLimitr rg r
r0 ri rA rf rj )r �ccr5 rV r� r r r �test_tooManyConnections� s
z+BasicFTPServerTests.test_tooManyConnectionsc
s� ddddddddd d
g
}d g}� fd d
�}� fdd�}g }|D ]4}� j �|�}� �|tj� |�||� |�|� q>|D ]&}� j �|�}|�||� |�|� qxtj|dd�S )z�
When not logged in, most commands other than USER and PASS should
get NOT_LOGGED_IN errors, but some can be called before USER and PASS.
ZCDUPZCWD�LISTZMODE�PASV�PWDZRETRZSTRU�SYSTZTYPE�FEATc s0 | j d }� �|d �d�d||d f � d S )Nr ����530z'%s - Response didn't start with 530: %r)r� �
assertTrue�
startswith)r� rz ZfailureResponseLinesr r r �checkFailResponse s
��zDBasicFTPServerTests.test_NotLoggedInReply.<locals>.checkFailResponsec s& | d } � � | �d�d|| f � d S )Nr r� z %s - Response start with 530: %r)� assertFalser� )�resultrz r r r �checkPassResponse& s ��zDBasicFTPServerTests.test_NotLoggedInReply.<locals>.checkPassResponseT�ZfireOnOneErrback) rU rw r� r r� rj r% r
� DeferredList)r ZloginRequiredCommandListZloginNotRequiredCommandListr� r� Z deferredsrz �deferredr r r �test_NotLoggedInReply s*
� z)BasicFTPServerTests.test_NotLoggedInReplyc C s | � ddg�S )z@
Issuing PASS before USER should give an error.
zPASS fooz=503 Incorrect sequence of commands: USER required before PASS�r� r r r r �test_PASSBeforeUSER; s �z'BasicFTPServerTests.test_PASSBeforeUSERc C s | � ddg�S )zD
Issuing USER without a username is a syntax error.
ZUSERz,500 Syntax error: USER requires an argument.r� r r r r �test_NoParamsForUSERE s �z(BasicFTPServerTests.test_NoParamsForUSERc C s | j �d�}| jddg|d�S )zD
Issuing PASS without a password is a syntax error.
�USER fooZPASSz,500 Syntax error: PASS requires an argument.r� )rU rw r� r� r r r �test_NoParamsForPASSN s �z(BasicFTPServerTests.test_NoParamsForPASSc sF dd� }|� j j_� j�d�}� jddg|d� |j� fdd��}|S ) zI
Unexpected exceptions from the login handler are caught
c _ s t �td��S )Nztest exception�r
�fail�AssertionError)r� �kwargsr r r �_fake_loginhandler] s z?BasicFTPServerTests.test_loginError.<locals>._fake_loginhandlerr� zPASS barz5550 Requested action not taken: internal server errorr� c s. � � � }� �dt|�� � �|d jt� d S )Nr� r ��flushLoggedErrorsrq �len�assertIsInstance�valuer� �r� �logsr r r � checkLogsg s z6BasicFTPServerTests.test_loginError.<locals>.checkLogs)rI r ZloginrU rw r� rj )r r� r5 r� r r r �test_loginErrorY s
�z#BasicFTPServerTests.test_loginErrorc C s | � � S )z/
Login with userid 'anonymous'
)r� r r r r �test_AnonymousLoginp s z'BasicFTPServerTests.test_AnonymousLoginc C s | � � }| jddg|d�S )zm
Issuing QUIT should return a 221 message.
@return: L{Deferred} of command response
ZQUITz 221 Goodbye.r� �r� r~ r� r r r � test_Quitw s �z BasicFTPServerTests.test_Quitc C sX d| j _t�� }| j j�|tj� | �ddg�}| j ddg|d�}| j ddg|d�}|S ) z�
Reconfigure the server to disallow anonymous access, and to have an
IUsernamePassword checker that always rejects.
@return: L{Deferred} of command response
Fr� z$331 Password required for anonymous.r� z!530 Sorry, Authentication failed.r� r� z$530 Please login with USER and PASS.)
r" ZallowAnonymousr r_ r r^ r rb r~ r� )r ZdenyAlwaysCheckerr5 r r r �test_AnonymousLoginDenied� s(
����z-BasicFTPServerTests.test_AnonymousLoginDeniedc C s | � � }| jddg|d�S )z�
When an anonymous user attempts to edit the server-side filesystem, they
will receive a 550 error with a descriptive message.
@return: L{Deferred} of command response
z
MKD newdirz:550 Anonymous users are forbidden to change the filesystemr� �r� r� r� r r r �test_anonymousWriteDenied� s �z-BasicFTPServerTests.test_anonymousWriteDeniedc C s | � � }| jddg|d�S )z\
Send an invalid command.
@return: L{Deferred} of command response
Z GIBBERISHz'502 Command 'GIBBERISH' not implementedr� r� r� r r r �test_UnknownCommand� s �z'BasicFTPServerTests.test_UnknownCommandc C s | � � }| jddg|d�S )zb
Send RETR before sending PORT.
@return: L{Deferred} of command response
�RETR foozE503 Incorrect sequence of commands: PORT or PASV required before RETRr� r� r� r r r �test_RETRBeforePORT� s �z'BasicFTPServerTests.test_RETRBeforePORTc C s | � � }| jddg|d�S )zb
Send STOR before sending PORT.
@return: L{Deferred} of command response
zSTOR foozE503 Incorrect sequence of commands: PORT or PASV required before STORr� r� r� r r r �test_STORBeforePORT� s �z'BasicFTPServerTests.test_STORBeforePORTc C s0 | � � }| jddg|d� | jddg|d� |S )zd
Send command with bad arguments.
@return: L{Deferred} of command response
zMODE zz&504 Not implemented for parameter 'z'.r� zSTRU Iz&504 Not implemented for parameter 'I'.r� r� r r r �test_BadCommandArgs� s ��z'BasicFTPServerTests.test_BadCommandArgsc C s\ | � t�d�d� td�}td�D ]4}t|�}d||< d�tt|��}| �t tj|� q"dS )z%
Decode a host port.
z25,234,129,22,100,23)z
25.234.129.22id � � �,N)
rq r �decodeHostPortr �list�join�map�str� assertRaises�
ValueError)r Znums�iZbadValue�sr r r �test_DecodeHostPort� s � z'BasicFTPServerTests.test_DecodeHostPortc sF � � � }|�� fdd�� � fdd�}|�|� |�� fdd�� |S )z�
When the client sends the command C{PASV}, the server responds with a
host and port, and is listening on that port.
c s � j �d�S )Nr� r ��_r r r �<lambda> r2 z/BasicFTPServerTests.test_PASV.<locals>.<lambda>c s4 t �| d dd� �\}}� �|� jj�� j� dS )z�
Extract the host and port from the resonse, and
verify the server is listening of the port it claims to be.
r� � N)r r� rq rI ZdtpPortrf rA )rs r@ rA r r r �cb s z)BasicFTPServerTests.test_PASV.<locals>.cbc s � j j�� S r rH r� r r r r� r2 �r� rj �r r5 r� r r r � test_PASV� s
z BasicFTPServerTests.test_PASVc C s | � � }| jddg|d� |S )z?
SYST command will always return UNIX Type: L8
r� z215 UNIX Type: L8r� r� r� r r r � test_SYST s
�z BasicFTPServerTests.test_SYSTc sp � j �� j��� � j �� j��d��� � �� }� jddg|d� � jddg|d� � fdd�}|�|� |S ) z�
Sending the RNFR command followed by RNTO, with valid filenames, will
perform a successful rename operation.
�foo�RNFR foo�6350 Requested file action pending further information.r� zRNTO bar�&250 Requested File Action Completed OKc s" � � � j�� j��d��� � | S �N�bar)r� r\ �childr` �exists�r� r r r � check_rename- s �z:BasicFTPServerTests.test_RNFRandRNTO.<locals>.check_rename)r\ r� r` �createDirectory�touchr� r~ rj )r r5 r� r r r �test_RNFRandRNTO s ��
z$BasicFTPServerTests.test_RNFRandRNTOc C s0 | � � }| jddg|d� | jddg|d� |S )z�
Sending the RNFR command followed by any command other than RNTO
should return an error informing users that RNFR should be followed
by RNTO.
r� r� r� zOTHER don-tcarez<503 Incorrect sequence of commands: RNTO required after RNFR)r� r~ r� r� r r r �test_RNFRwithoutRNTO6 s ��z(BasicFTPServerTests.test_RNFRwithoutRNTOc C s* dd� }|| j _| �t| j jt�� � dS )z�
Exceptions other than L{error.CannotListenError} which are raised by
C{listenFactory} should be raised to the caller of L{FTP.getDTPPort}.
c S s
t � �d S r )� RuntimeError�Z
portNumberr" r r r �
listenFactoryM s zEBasicFTPServerTests.test_portRangeForwardError.<locals>.listenFactoryN)rI r� r� r� �
getDTPPortr �Factory)r r� r r r �test_portRangeForwardErrorH s
�z.BasicFTPServerTests.test_portRangeForwardErrorc C s� dd� }|| j _| j �t�� �}| �|d� tdd�| j _| j �t�� �}| �|d� tdd�| j _| �t j
| j jt�� � dS )a3
L{FTP.passivePortRange} should determine the ports which
L{FTP.getDTPPort} attempts to bind. If no port from that iterator can
be bound, L{error.CannotListenError} should be raised, otherwise the
first successful result from L{FTP.listenFactory} should be returned.
c S s | dkrt �d| d��| S )N)�V iV iV Z localhostr )r �CannotListenErrorr� r r r r� \ s z9BasicFTPServerTests.test_portRange.<locals>.listenFactoryr r� i iV N) rI r� r� r r� rq r �passivePortRanger� r r� )r r� rA r r r �test_portRangeU s �z"BasicFTPServerTests.test_portRangec C s2 t dd�}|| j_| j�d�}| �||jj� dS )z�
The L{FTP} instances created by L{ftp.FTPFactory.buildProtocol} have
their C{passivePortRange} attribute set to the same object the
factory's C{passivePortRange} attribute is set to.
i� i� N)r r" r� rM rq rN )r Z portRanger r r r �"test_portRangeInheritedFromFactoryo s
z6BasicFTPServerTests.test_portRangeInheritedFromFactoryc s" � j �d�}� fdd�}|�|�S )z�
When the server receives 'FEAT', it should report the list of supported
features. (Additionally, ensure that the server reports various
particular features that are supported by all Twisted FTP servers.)
r� c sT � � d| d � � �d| � � �d| � � �d| � � �d| � � � d| d � d S ) Nz
211-Features:r z MDTMz PASVz TYPE A;Iz SIZEz211 Endr� )rq �assertInrr r r r ru � s z2BasicFTPServerTests.test_FEAT.<locals>.gotResponserv )r r5 ru r r r � test_FEAT{ s z BasicFTPServerTests.test_FEATc C s | � � }| jddg|d� |S )z�
When the server receives 'OPTS something', it should report
that the FTP server does not support the option called 'something'.
zOPTS somethingz'502 Option 'something' not implemented.r� r� r� r r r � test_OPTS� s �z BasicFTPServerTests.test_OPTSc st � j �� j��� � j �� j��d��� � �� }� fdd�}� fdd�}|�|� |�|� � jddg|d� |S ) zl
Any FTP error raised inside STOR while opening the file is returned
to the client.
Zfolderc s � j �d�S )zE
Send the PASV command required before port.
r� r r� r r r �sendPASV� s zCBasicFTPServerTests.test_STORreturnsErrorFromOpen.<locals>.sendPASVc s � j jj�d� t� � j _| S )z�
Fake an incoming connection and create a mock DTPInstance so
that PORT command will start processing the request.
N)rI �
dtpFactoryr� r8 �object� dtpInstancer� r r r �mockDTPInstance� s
zJBasicFTPServerTests.test_STORreturnsErrorFromOpen.<locals>.mockDTPInstancez STOR folderz550 folder: is a directoryr� )r\ r� r` r� r� rj r� )r r5 r r r r r � test_STORreturnsErrorFromOpen� s
�z1BasicFTPServerTests.test_STORreturnsErrorFromOpenc sj �� � }dd� � �fdd�}� �fdd�}�fdd�}|�|� |�|� �jd d
g|d � |�|� |S ) z�
Any non FTP error raised inside STOR while opening the file is
converted into FileNotFound error and returned to the client together
with the path.
The unknown error is logged.
c S s t �t� �S )z�
Override openForWriting.
@param ignore: ignored, used for callback
@return: an error
r� )�ignorer r r �failingOpenForWriting� s z[BasicFTPServerTests.test_STORunknownErrorBecomesFileNotFound.<locals>.failingOpenForWritingc s � j �d�S )z
Send the PASV command required before port.
@param result: parameter used in L{Deferred}
r� r r� r r r r � s zNBasicFTPServerTests.test_STORunknownErrorBecomesFileNotFound.<locals>.sendPASVc s( �j jj�d� t� �j _� �j j_| S )z�
Fake an incoming connection and create a mock DTPInstance so
that PORT command will start processing the request.
@param result: parameter used in L{Deferred}
N)rI r r� r8 r r �shell�openForWritingr� �r r r r r � s
zUBasicFTPServerTests.test_STORunknownErrorBecomesFileNotFound.<locals>.mockDTPInstancec s. � � � }� �dt|�� � �|d jt� dS )zy
Check that unknown errors are logged.
@param result: parameter used in L{Deferred}
r� r Nr� r� r r r r� � s zOBasicFTPServerTests.test_STORunknownErrorBecomesFileNotFound.<locals>.checkLogszSTOR somethingz)550 something: No such file or directory.r� )r� rj r� )r r5 r r r� r r r �(test_STORunknownErrorBecomesFileNotFound� s
�
z<BasicFTPServerTests.test_STORunknownErrorBecomesFileNotFoundN) r- r. r/ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r� � s6 %
%r� c @ s6 e Zd ZdZejZdd� Zdd� Zdd� Z dd � Z
d
S ) � FTPServerAdvancedClientTestsz:
Test FTP server with the L{ftp.FTPClient} class.
c s4 � fdd�}� j �d�\}}|�|� t�||g�S )zl
Try to make an STOR as anonymous, and check that we got a permission
denied error.
c s( | � tj� � �| jjd d d� d S )Nr z550 foo: Permission denied.)�trapr r� rq r� r� ��resr r r �eb s �z;FTPServerAdvancedClientTests.test_anonymousSTOR.<locals>.ebr� )rU � storeFile�
addErrbackr
rk )r r rR rn r r r �test_anonymousSTOR s
z/FTPServerAdvancedClientTests.test_anonymousSTORc sb G dd� dt j�� � fdd�}��t jd|� �fdd�}�j�d�\}}|�|� t�||g�S ) zm
Any FTP error raised by STOR while transferring the file is returned
to the client.
c @ s e Zd Zdd� ZdS )zXFTPServerAdvancedClientTests.test_STORtransferErrorIsReturned.<locals>.FailingFileWriterc S s t �t�d��S )N� failing_file�r
r� r �IsADirectoryErrorr r r r �receive s z`FTPServerAdvancedClientTests.test_STORtransferErrorIsReturned.<locals>.FailingFileWriter.receiveN�r- r. r/ r r r r r �FailingFileWriter s r c s t �� d ��S r �r
r| ��a�b�r r r � failingSTOR s zRFTPServerAdvancedClientTests.test_STORtransferErrorIsReturned.<locals>.failingSTORr
c sT | � tj� � �� }� �dt|�� � �|d jtj� � �| jj d d d� d S )Nr� r z 550 failing_file: is a directory)
r r r� r� rq r� r� r� r r� �r r� r r r r " s �zIFTPServerAdvancedClientTests.test_STORtransferErrorIsReturned.<locals>.ebr � r Z _FileWriter�patch�FTPAnonymousShellrU r r r
rk �r r r rR rn r �r r r � test_STORtransferErrorIsReturned s
z=FTPServerAdvancedClientTests.test_STORtransferErrorIsReturnedc sb G dd� dt j�� � fdd�}��t jd|� �fdd�}�j�d�\}}|�|� t�||g�S ) z�
Any non FTP error raised by STOR while transferring the file is
converted into a critical error and transfer is closed.
The unknown error is logged.
c @ s e Zd Zdd� ZdS )zaFTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.FailingFileWriterc S s t �t� �S r r� r r r r r 7 s ziFTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.FailingFileWriter.receiveNr r r r r r 6 s r c s t �� d ��S r r r r r r r : s z[FTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.failingSTORr
c sR | � tj� � �� }� �dt|�� � �|d jt� � �| jj d d d� d S )Nr� r z.426 Transfer aborted. Data connection closed.)
r r r� r� rq r� r� r� r� r� r! r r r r A s �zRFTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.ebr r" r% r r& r �)test_STORunknownTransferErrorBecomesAbort/ s
zFFTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbortc s\ G dd� dt j�� � fdd�}��t jd|� �fdd�}t� }�j�d|�}|�|� |S ) zj
Any errors during reading a file inside a RETR should be returned to
the client.
c @ s e Zd Zdd� ZdS )zJFTPServerAdvancedClientTests.test_RETRreadError.<locals>.FailingFileReaderc S s t �t�d��S )NZblahr )r �consumerr r r �sendU s zOFTPServerAdvancedClientTests.test_RETRreadError.<locals>.FailingFileReader.sendN)r- r. r/ r* r r r r �FailingFileReaderT s r+ c s t �� d ��S r r r )r+ r r � failingRETRX s zDFTPServerAdvancedClientTests.test_RETRreadError.<locals>.failingRETR�openForReadingc sH � � � | �tj� � �| jjd d d� � �| jjd d d� d S )Nr z3125 Data connection already open, starting transferr� z550 blah: is a directory)r� r r r� rq r� r� �r r r r �check_response_ s ��zGFTPServerAdvancedClientTests.test_RETRreadError.<locals>.check_responser )r Z _FileReaderr# r$ r0 rU � retrieveFiler )r r, r/ r� r5 r )r+ r r �test_RETRreadErrorN s
z/FTPServerAdvancedClientTests.test_RETRreadErrorN) r- r. r/ r� r � FTPClientrh r r' r( r1 r r r r r
� s r
c @ s� e Zd ZdZd!dd�Zd"dd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � ZdS )#� FTPServerPasvDataConnectionTestsz
PASV data connection.
Nc C s | j �d�}dd� }|�|�S )z�
Establish a passive data connection (i.e. client connecting to
server).
@param ignored: ignored
@return: L{Deferred.addCallback}
r� c S s2 t �| d dd � �\}}t�tt�}|�d|�S )Nr� r� r; )r r� r rg r
r0 ri )rs r@ rA r� r r r �gotPASV} s zEFTPServerPasvDataConnectionTests._makeDataConnection.<locals>.gotPASVrv )r rx r5 r4 r r r �_makeDataConnectiont s z4FTPServerPasvDataConnectionTests._makeDataConnectionc sH |dkrt �d�}|��j� � �fdd�}|�|� dd� }|�|�S )z�
Download file.
@param command: command to run
@param chainDeferred: L{Deferred} used to queue commands.
@return: L{Deferred} of command response
Nc s �j �� �}| j}t�||g�S r )rU rw r5 r
rk )�
downloaderrR rn r� r r r{ � s z@FTPServerPasvDataConnectionTests._download.<locals>.queueCommandc S s | \}}|j S r r6 )r� rx r6 r r r � downloadDone� s z@FTPServerPasvDataConnectionTests._download.<locals>.downloadDone)r
r| rj r5 )r rz r} r{ r7 r r� r � _download� s
z*FTPServerPasvDataConnectionTests._downloadc s, � � � }� jd|d� � fdd�}|�|�S )zM
When listing empty folders, LIST returns an empty response.
r� r� c s � � d| � d S r1 rp r� r r r �
checkEmpty� s zCFTPServerPasvDataConnectionTests.test_LISTEmpty.<locals>.checkEmpty�r� r8 rj )r r5 r9 r r r �test_LISTEmpty� s z/FTPServerPasvDataConnectionTests.test_LISTEmptyc sX t �t j�� jd�� t �t j�� jd�� � �� }� jd|d� � fdd�}|�|�S )z}
LIST ignores requests for folder with names like '-al' and will list
the content of current folder.
r� r� zLIST -aLr� c sR g }| � � D ]}|�|�d�d � q � �dt|�� � �d|� � �d|� d S )N� r� � � foo� bar)�
splitlinesr% �splitrq r� r� )�download�namesr' r r r �
checkDownload� s zOFTPServerPasvDataConnectionTests.test_LISTWithBinLsFlags.<locals>.checkDownload�rY rZ �pathr� rX r� r8 rj �r r5 rD r r r �test_LISTWithBinLsFlags� s z8FTPServerPasvDataConnectionTests.test_LISTWithBinLsFlagsc s� t �t j�� jd�� t �t j�� jd�� � �� }� jd|d� � fdd�}|�|� � jd|d� � fdd�}|�|� � jd |d� � fd
d�}|�|� � fd d �}|�|� � jd|d� � fd
d�}|�|�S )z�
LIST returns all folder's members, each member listed on a separate
line and with name and other details.
r� r� r� r� c s" � � dt| d d� �d��� d S )Nr= ����
)rq r� rA �rB r r r rD � s zLFTPServerPasvDataConnectionTests.test_LISTWithContent.<locals>.checkDownload�NLST c s. | d d� � d�}|�� � �ddg|� d S )NrI rJ r? r> �rA �sortrq �rB � filenamesr r r rD � s zLIST fooc s � � d| � d S r1 rp rK r r r rD � s c s � j �d�S )NzCWD foor r� r r r �chdir� s zDFTPServerPasvDataConnectionTests.test_LISTWithContent.<locals>.chdirc s � � d| � d S r1 rp rK r r r rD � s rE )r r5 rD rQ r r r �test_LISTWithContent� s"
z5FTPServerPasvDataConnectionTests.test_LISTWithContentc s\ �� � }d
�fdd� ���fdd�}|�|� �jd|f |d� � ��fdd �}|�|�S ) a5
Exercise handling by the implementation of I{LIST} or I{NLST} of certain
return values and types from an L{IFTPShell.list} implementation.
This will issue C{command} and assert that if the L{IFTPShell.list}
implementation includes C{listOutput} as one of the file entries then
the result given to the client is matches C{expectedOutput}.
@param command: Either C{b"LIST"} or C{b"NLST"}
@type command: L{bytes}
@param listOutput: A value suitable to be used as an element of the list
returned by L{IFTPShell.list}. Vary the values and types of the
contents to exercise different code paths in the server's handling
of this result.
@param expectedOutput: A line of output to expect as a result of
C{listOutput} being transformed into a response to the command
issued.
@type expectedOutput: L{bytes}
@return: A L{Deferred} which fires when the test is done, either with an
L{Failure} if the test failed or with a function object if it
succeeds. The function object is the function which implements
L{IFTPShell.list} (and is useful to make assertions about what
warnings might have been emitted).
@rtype: L{Deferred}
r c s t �� g�S r r )Zsegments�keys)�
listOutputr r � patchedList s zEFTPServerPasvDataConnectionTests._listTestHelper.<locals>.patchedListc s � �j j_| S r )rI r r� r� )rU r r r �loggedIn s
zBFTPServerPasvDataConnectionTests._listTestHelper.<locals>.loggedInz %s somethingr� c s �� � | � �S r rp rK )�expectedOutputrU r r r rD s zGFTPServerPasvDataConnectionTests._listTestHelper.<locals>.checkDownload)r )r� rj r8 )r rz rT rW r5 rV rD r )rW rT rU r r �_listTestHelper� s
z0FTPServerPasvDataConnectionTests._listTestHelperc C s& | � ddddt�d�ddddffd�S ) z�
Unicode filenames returned from L{IFTPShell.list} are encoded using
UTF-8 before being sent with the response.
r� �
my resumér r� � �user�group�L drwxrwxrwx 0 user group 0 Jan 01 1970 my resumé
�rX r Z Permissionsr r r r �test_LISTUnicode! s ��z1FTPServerPasvDataConnectionTests.test_LISTUnicodec C s& | � ddddt�d�ddddffd�S ) z�
When LIST receive a filename as byte string from L{IFTPShell.list}
it will just pass the data to lower level without any change.
@return: L{_listTestHelper}
r� �
my resumér r� rZ r[ r\ r] r^ r r r r �test_LISTNonASCIIBytes. s ��z7FTPServerPasvDataConnectionTests.test_LISTNonASCIIBytesc
s� � � � }tddd�D ]h}ttj�� jd|f �d��}|�d| � W 5 Q R X � jd|f |d� |f� fd d
� }|� |� q|S ) zJ
Download many large files.
@return: L{Deferred}
i�� i�� i� z%d.txt�wb� xz RETR %d.txtr� c s � � |t| �� d S r �rq r� )rB �sizer r r rD M s zOFTPServerPasvDataConnectionTests.test_ManyLargeDownloads.<locals>.checkDownload)
r� r �openrY rF r� rX �writer8 rj )r r5 re ZfObjrD r r r �test_ManyLargeDownloads= s �� z8FTPServerPasvDataConnectionTests.test_ManyLargeDownloadsc sf � j �d��� � �� }|�� j� � fdd�}|�|� dd� }|�|� � fdd�}|�|� |S )zo
When RETR is called for a folder, it will fail complaining that
the path is a folder.
r� c s | j �� � j�d�}|S )Nr� )rJ rK rU rw )r6 r� r r r �
retrFolder^ s
zHFTPServerPasvDataConnectionTests.test_downloadFolder.<locals>.retrFolderc S s t d��d S )Nz(Downloading a folder should not succeed.)r� r� r r r �
failOnSuccessd s zKFTPServerPasvDataConnectionTests.test_downloadFolder.<locals>.failOnSuccessc s@ | � tj� � �dg| jjd � � �� }� �dt|�d� d S )Nz550 foo: is a directoryr z6No errors should be logged while downloading a folder.)r r r� rq r� r� r� r� )r Zcurrent_errorsr r r �
checkErrorh s
� �zHFTPServerPasvDataConnectionTests.test_downloadFolder.<locals>.checkError)r\ r� r� r� rj r5 r )r r5 ri rj rk r r r �test_downloadFolderS s
z4FTPServerPasvDataConnectionTests.test_downloadFolderc sL � � � }� j�d��� � j�d��� � jd|d� � fdd�}|�|�S )zp
NLST with no argument returns the directory listing for the current
working directory.
�test.txtr� rL r� c s. | d d� � d�}|�� � �ddg|� d S )NrI rJ r> � test.txtrM rO r r r rD � s zFFTPServerPasvDataConnectionTests.test_NLSTEmpty.<locals>.checkDownload)r� r\ r� r� r� r8 rj rG r r r �test_NLSTEmptyt s z/FTPServerPasvDataConnectionTests.test_NLSTEmptyc s, � � � }� jd|d� � fdd�}|�|�S )zH
NLST on a non-existent file/directory returns nothing.
zNLST nonexistent.txtr� c s � � d| � d S r1 rp rK r r r rD � s zLFTPServerPasvDataConnectionTests.test_NLSTNonexistent.<locals>.checkDownloadr: rG r r r �test_NLSTNonexistent� s z5FTPServerPasvDataConnectionTests.test_NLSTNonexistentc C s& | � ddddt�d�ddddffd�S ) zs
NLST will receive Unicode filenames for IFTPShell.list, and will
encode them using UTF-8.
�NLSTrY r r� rZ r[ r\ � my resumé
r^ r r r r �test_NLSTUnicode� s ��z1FTPServerPasvDataConnectionTests.test_NLSTUnicodec C s& | � ddddt�d�ddddffd�S ) zJ
NLST will just pass the non-Unicode data to lower level.
rq r` r r� rZ r[ r\ rr r^ r r r r �test_NLSTNonASCIIBytes� s ��z7FTPServerPasvDataConnectionTests.test_NLSTNonASCIIBytesc s< � � � }� j�d��� � jd|d� � fdd�}|�|�S )zN
NLST on an existent file returns only the path to that file.
rm z
NLST test.txtr� c s$ | d d� � d�}� �dg|� d S )NrI rJ rn )rA rq rO r r r rD � s zMFTPServerPasvDataConnectionTests.test_NLSTOnPathToFile.<locals>.checkDownload)r� r\ r� r� r8 rj rG r r r �test_NLSTOnPathToFile� s
z6FTPServerPasvDataConnectionTests.test_NLSTOnPathToFile)N)N)r- r. r/ r� r5 r8 r; rH rR rX r_ ra rh rl ro rp rs rt ru r r r r r3 p s
-1
!
r3 c @ s6 e Zd Zdd� Zd dd�Zdd� Zdd � Zd
d � ZdS )
� FTPServerPortDataConnectionTestsc C s g | _ t�| �S r )� dataPortsr3 ro r r r r ro � s z&FTPServerPortDataConnectionTests.setUpNc sb t �� � G � fdd�dtj�}tjd|� dd�}| j�|� dt� d|�
� j � }| j �
|� � S )Nc s e Zd ZeZ� fdd�ZdS )zIFTPServerPortDataConnectionTests._makeDataConnection.<locals>.DataFactoryc s" t j�| |�}t�d� j|� |S r� )r �
ServerFactoryrM r
Z callLaterr8 )r rP rl �r� r r rM � s zWFTPServerPortDataConnectionTests._makeDataConnection.<locals>.DataFactory.buildProtocolN)r- r. r/ r0 r rM r ry r r � DataFactory� s rz r r; rG �PORT )r
r4 r rx r
rd rw r% r r= rf rA rU rw )r rx rz ZdataPort�cmdr ry r r5 � s � z4FTPServerPortDataConnectionTests._makeDataConnectionc C s6 dd� | j D �}t�tj| �}|�|� tj|dd�S )zS
Tear down the connection.
@return: L{defer.DeferredList}
c S s g | ]}t �|j��qS r )r
�
maybeDeferredre )�.0rA r r r �
<listcomp>� s �z=FTPServerPortDataConnectionTests.tearDown.<locals>.<listcomp>Tr� )rw r
r} r3 �tearDownr% r� )r �lr5 r r r r� � s � �
z)FTPServerPortDataConnectionTests.tearDownc s0 � � � }dd� }|�|� � fdd�}|�|�S )z�
Listen on a port, and immediately stop listening as a way to find a
port number that is definitely closed.
c s< t jdt�� dd�}|�� j� |�� }|�� fdd�� |S )Nr r; rG c s � S r r r� �rm r r r� � r2 z[FTPServerPortDataConnectionTests.test_PORTCannotConnect.<locals>.loggedIn.<locals>.<lambda>)r
rd r r� rf rA re rj )rx rA r5 r r� r rV � s �
zIFTPServerPortDataConnectionTests.test_PORTCannotConnect.<locals>.loggedInc s � � dt�d| � dg�S )Nr{ r; z 425 Can't open data connection.)r� r r= r� r r r �
gotPortNum� s �zKFTPServerPortDataConnectionTests.test_PORTCannotConnect.<locals>.gotPortNumr� )r r5 rV r� r r r �test_PORTCannotConnect� s
z7FTPServerPortDataConnectionTests.test_PORTCannotConnectc s\ � j �d��� � j �d��� � j �d��� � �� }� jd|d� � fdd�}|�|�S )zv
When Unix shell globbing is used with NLST only files matching the
pattern will be returned.
rm zceva.txtzno.matchz
NLST *.txtr� c s. | d d� � d�}|�� � �ddg|� d S )NrI rJ s ceva.txtrn rM rO r r r rD s zIFTPServerPortDataConnectionTests.test_nlstGlobbing.<locals>.checkDownload)r\ r� r� r� r8 rj rG r r r �test_nlstGlobbing s z2FTPServerPortDataConnectionTests.test_nlstGlobbing)N)r- r. r/ ro r5 r� r� r� r r r r rv � s
rv c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�DTPFactoryTestsz&
Tests for L{ftp.DTPFactory}.
c C s: t �� | _G dd� dt�}|� | _t�| jd| j�| _dS )zf
Create a fake protocol interpreter and a L{ftp.DTPFactory} instance to
test.
c @ s e Zd ZdZdS )z2DTPFactoryTests.setUp.<locals>.ProtocolInterpreterN�r- r. r/ r r r r r �ProtocolInterpreter s r� N)r �Clockr
r �protocolInterpreterr �
DTPFactoryr" �r r� r r r ro s
�zDTPFactoryTests.setUpc C sl g }| � | jjtj�}|�|j� | j�d� | j� d� | �
|� | j� d� | � |� | �
| jj � dS )z�
L{ftp.DTPFactory.setTimeout} uses the reactor passed to its initializer
to set up a timed event to time out the DTP setup after the specified
number of seconds.
r� � r� N)
r� r" r� r �PortConnectionErrorrj r% �
setTimeoutr
�advancer� r� �calls�r Zfinishedr5 r r r �test_setTimeout( s
z DTPFactoryTests.test_setTimeoutc C s0 | j �d�}| �|tj� | �| j �d�� dS )z�
A L{ftp.DTPFactory} instance's C{buildProtocol} method can be used once
to create a L{ftp.DTP} instance.
N)r" rM r� r �DTP� assertIsNone)r r r r r �test_buildProtocolOnceF s z&DTPFactoryTests.test_buildProtocolOncec C s* | j �d� | j �d� | �| jj� dS )z�
If a timeout has been set up using L{ftp.DTPFactory.setTimeout}, it is
cancelled by L{ftp.DTPFactory.buildProtocol}.
�
N)r" r� rM r� r
r� r r r r �test_timeoutAfterConnectionR s z+DTPFactoryTests.test_timeoutAfterConnectionc C s@ | � | jjtj�}| j�d� | j�d� | �| j� d�� |S )z�
If L{ftp.DTPFactory.buildProtocol} is called after the timeout
specified by L{ftp.DTPFactory.setTimeout} has elapsed, L{None} is
returned.
r� N)
r� r" r� r r� r� r
r� r� rM r� r r r �test_connectionAfterTimeout] s
z+DTPFactoryTests.test_connectionAfterTimeoutc C s` g }| � | jjtj�}|�|j� | j�d� | �|� | j� dd� | �
|� | j � d� |S )a
L{ftp.DTPFactory.deferred} fails with L{PortConnectionError} when
L{ftp.DTPFactory.clientConnectionFailed} is called. If the timeout
specified with L{ftp.DTPFactory.setTimeout} expires after that, nothing
additional happens.
r� N)
r� r" r� r r� rj r% r� r� �clientConnectionFailedr� r
r� r� r r r �!test_timeoutAfterConnectionFailedr s
z1DTPFactoryTests.test_timeoutAfterConnectionFailedc C sB | � | jjtj�}| j�d� | j�d� | j�dt �
d�� |S )z�
If L{ftp.DTPFactory.clientConnectionFailed} is called after the timeout
specified by L{ftp.DTPFactory.setTimeout} has elapsed, nothing beyond
the normal timeout before happens.
r� Nr� ) r� r" r� r r� r� r
r� r� r
� TimeoutErrorr� r r r �!test_connectionFailedAfterTimeout� s
z1DTPFactoryTests.test_connectionFailedAfterTimeoutN) r- r. r/ r� ro r� r� r� r� r� r� r r r r r� s r� c @ s e Zd ZdZdd� Zdd� ZdS )�DTPTestszx
Tests for L{ftp.DTP}.
The DTP instances in these tests are generated using
DTPFactory.buildProtocol()
c C sD t �� | _G dd� dt�}|� | _t�| jd| j�| _t� � | _
dS )z�
Create a fake protocol interpreter, a L{ftp.DTPFactory} instance,
and dummy transport to help with tests.
c @ s e Zd ZdZdS )z+DTPTests.setUp.<locals>.ProtocolInterpreterNr� r r r r r� � s r� N) r r� r
r r� r r� r" r � StringTransportWithDisconnectionrJ r� r r r ro � s
�zDTPTests.setUpc C sD | j �d�}|�| j� d}|�|� | j�� }| �|d |� dS )zr
L{ftp.DTP.sendLine} writes the line passed to it plus a line delimiter
to its transport.
Ns line contentrJ )r" rM �makeConnectionrJ �sendLiner� rq )r r Z lineContentZdataSentr r r �test_sendLineNewline� s
z DTPTests.test_sendLineNewlineN)r- r. r/ r� ro r� r r r r r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )�
PrintLinesz3
Helper class used by FTPFileListingTests.
c C s
|| _ d S r )�_lines)r r r r r r! � s zPrintLines.__init__c C s0 | j D ]}| j�|�d�d � q| j�� d S )Nzlatin-1rJ )r� rJ rg r> rK r& r r r r$ � s
zPrintLines.connectionMadeN)r- r. r/ r� r! r$ r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�MyFTPFileListProtocolc C s g | _ tj�| � d S r )�otherr �FTPFileListProtocolr! r r r r r! � s z MyFTPFileListProtocol.__init__c C s | j �|� d S r )r� r% r& r r r � unknownLine� s z!MyFTPFileListProtocol.unknownLineN)r- r. r/ r! r� r r r r r� � s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�FTPFileListingTestsc s, t � � t�t|�� �}|�� fdd�� |S )Nc s � j � jfS r )�filesr� r� ��fileListr r r� � r2 z6FTPFileListingTests.getFilesForLines.<locals>.<lambda>)r� r �
loopbackAsyncr� rj )r r r5 r r� r �getFilesForLines� s z$FTPFileListingTests.getFilesForLinesc s" d}� fdd�}� � |g��|�S )z�
This example line taken from the docstring for FTPFileListProtocol
@return: L{Deferred} of command response
z<-rw-r--r-- 1 root other 531 Jan 29 03:26 READMEc s� | \\}}� � |dt|�f � � �|d dkd� � �|d dkd� � �|d d kd� � �|d
d kd� � �|d d
kd� � �|d dkd� � �|d dkd� � �|d dkd� � � |d d� d S )Nz unexpect unparsable lines: %s�filetype�-�misparsed fileitem�perms� rw-r--r--�misparsed perms�owner�rootr\ r� re � �date� Jan 29 03:26�filename�README�nlinksr� �misparsed nlinks�
linktarget�misparsed linktarget)r� �reprr� )Z fileOther�filer� r r r �check� s
�z/FTPFileListingTests.test_OneLine.<locals>.check�r� rj )r r' r� r r r � test_OneLine� s
z FTPFileListingTests.test_OneLinec s. d}d}d}� fdd�}� � |||g��|�S )z
Variant lines.
z7drw-r--r-- 2 root other 531 Jan 9 2003 Az<lrw-r--r-- 1 root other 1 Jan 29 03:26 B -> Azwoohoo! c s� | \\}}\}� � |dkd� � � |d dkd� � � |d dkd� � � |d d
kd � � � |d d
kd� � � |d dkd� � � |d dkd� � � |d dkd� � � |d dkd� � �|d d � � � |d d kd� � � |d dkd� � � |d d
kd � � � |d d
kd� � � |d d kd� � � |d d kd� � � |d d kd� � � |d d kd� � � |d dkd � d S )!Nz woohoo!
zincorrect other liner� r5 r� r� r� r� r� r� zmisparsed ownerr\ r� zmisparsed groupre r� zmisparsed sizer� z Jan 9 2003zmisparsed dater� �A�misparsed filenamer� r= r� r� r� r� r� r� �B)r� r� )r� Zfile1Zfile2r� r r r r� s( z4FTPFileListingTests.test_VariantLines.<locals>.checkr� )r �line1�line2Zline3r� r r r �test_VariantLines s
z%FTPFileListingTests.test_VariantLinesc s � fdd�}� � ddg��|�S )z
Unknown lines.
c s4 | \}}� � |d� � �|ddgkdt|� � d S )Nzunexpected file entrieszABC
z not a file
z incorrect unparsable lines: %s)r� r� r� �r� r� Zothersr r r r� & s
�z3FTPFileListingTests.test_UnknownLine.<locals>.check�ABCz
not a filer� )r r� r r r �test_UnknownLine" s z$FTPFileListingTests.test_UnknownLinec s( d}d}� fdd�}� � ||g��|�S )ze
Will parse filenames and linktargets containing unescaped
space characters.
z9drw-r--r-- 2 root other 531 Jan 9 2003 A BzDlrw-r--r-- 1 root other 1 Jan 29 03:26 B A -> D C/A Bc s\ | \}}� � g |d� � � d|d d d� � � d|d d d� � � d|d d d
� d S � Nzunexpected others entrieszA Br r� r� zB Ar� zD C/A Br� r� rp r� r r r r� 9 s"
�
�
�zBFTPFileListingTests.test_filenameWithUnescapedSpace.<locals>.checkr� �r r� r� r� r r r � test_filenameWithUnescapedSpace. s
� z3FTPFileListingTests.test_filenameWithUnescapedSpacec s( d}d}� fdd�}� � ||g��|�S )zc
Will parse filenames and linktargets containing escaped
space characters.
z:drw-r--r-- 2 root other 531 Jan 9 2003 A\ BzElrw-r--r-- 1 root other 1 Jan 29 03:26 B A -> D\ C/A Bc s\ | \}}� � g |d� � � d|d d d� � � d|d d d� � � d|d d d
� d S r� rp r� r r r r� P s"
�
�
�z@FTPFileListingTests.test_filenameWithEscapedSpace.<locals>.checkr� r� r r r � test_filenameWithEscapedSpaceE s
� z1FTPFileListingTests.test_filenameWithEscapedSpacec sH t �� �d� G � fdd�dtj�}��fdd�}t�|� ��}|�|�S )z{
This example derived from bug description in issue 514.
@return: L{Deferred} of command response
s< -rw-r--r-- 1 root other 531 Jan 29 2003 README
c s e Zd Z� fdd�ZdS )z0FTPFileListingTests.test_Year.<locals>.PrintLinec s | j �� � | j �� d S r )rJ rg rK r �� exampleLiner r r$ f s z?FTPFileListingTests.test_Year.<locals>.PrintLine.connectionMadeN)r- r. r/ r$ r r� r r � PrintLinee s r� c sJ � j d }��|d dkd� ��|d dkd� ��|d dkd� d S ) Nr re r� r� r� z Jan 29 2003r� r� )r� r� )rx r� )r� r r r r� j s
��z,FTPFileListingTests.test_Year.<locals>.check)r r� r �Protocolr r� rj )r r� r� r5 r )r� r� r r � test_Year\ s �z FTPFileListingTests.test_YearN)
r- r. r/ r� r� r� r� r� r� r� r r r r r� � s ! r� c @ s e Zd ZdZdd� Zdd� ZdS )�1FTPClientFailedRETRAndErrbacksUponDisconnectTestsz:
FTP client fails and RETR fails and disconnects.
c s� t �� }d|_tjd|dd�}| �|j� |�� j}ddddd|d ? |d
@ f d g� � fd d
�|_ t j
tt j dd�}|�
d|�}dd� }|�|� | �|t j�S )z
RETR fails.
r r; rG z1220 ready, dude (vsFTPd 1.0.0: beat me, break me)z 331 Please specify the password.z 230 Login successful. Have fun.z200 Binary it is, then.z+227 Entering Passive Mode (127,0,0,1,%d,%d)� � z550 Failed to open file.c s t � �S r )r� )rP �Z responsesr r r� � r2 zSFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_FailedRETR.<locals>.<lambda>r� )�passivec S s t �� }| �d|�S )Nz/file/that/doesnt/exist)r r� r0 )rU rl r r r rV � s zTFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_FailedRETR.<locals>.gotClient)r r� Znoisyr
rd rO re rf rA rM rg r r2 ri rj r� r� )r r# rA rm r� r5 rV r r� r �test_FailedRETRz s(
��
zAFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_FailedRETRc st t �� }t�� }|�|� ||_|�dt� �}g � � fdd�}|�|� ddl m
} |� t �
|�� | �� � � |S )zl
Test the ftp command errbacks when a connection lost happens during
the operation.
z some pathc s � � | � d S r �r% r. ��mr r �_eb� s
zZFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_errbacksUponDisconnect.<locals>._ebr )�CONNECTION_LOST)r r2 r r� r� r r� r r Ztwisted.internet.mainr� r: r �Failurer� )r � ftpClient�trr5 r� r� r r� r �test_errbacksUponDisconnect� s
zMFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_errbacksUponDisconnectN)r- r. r/ r� r� r� r r r r r� v s !r� c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLdM� Z)dNdO� Z*dPdQ� Z+dRdS� Z,dTS )U�FTPClientTestsz,
Test advanced FTP client commands.
c C s0 t �� | _t�� | _| j�| j� | j| j_dS )zG
Create a FTP client and connect it to fake transport.
N)r r2 rU r r� rJ r� r r r r r ro � s
zFTPClientTests.setUpc C s | j �t�� � dS )z�
Deliver disconnection notification to the client so that it can
perform any cleanup which may be required.
N)rU r: r �ConnectionLostr r r r r� � s zFTPClientTests.tearDownc C sr | � | j�� d� | j�d� | � | j�� d� | j�� | j�d� | � | j�� d� | j�� | j�d� dS )z&
Test the login part.
r2 s8 331 Guest login ok, type your email address as password.s USER anonymous
s2 230 Anonymous login ok, access restrictions apply.s TYPE I
s 200 Type set to I.N)rq rJ r� rU r( �clearr r r r �
_testLogin� s �
�
zFTPClientTests._testLoginc C sz | � | j�� d� | j�d� | � | j�� d� | j�d� | � | j�� d� | j�� | j�d� | � | j�� d� dS )z9
Test encoding behaviour of sendLine
r2 N� rJ � és �
)rq rJ r� rU r� r� r r r r �
test_sendLine� s
z FTPClientTests.test_sendLinec sP � fdd�}� � � � j�� �|�}� �� j�� d� � j�� � j�d� |S )a
Test the CDUP command.
L{ftp.FTPClient.cdup} should return a Deferred which fires with a
sequence of one element which is the string the server sent
indicating that the command was executed successfully.
(XXX - This is a bad API)
c s � � | d d� d S �Nr r� rp r r r r �cbCdup� s z(FTPClientTests.test_CDUP.<locals>.cbCdup� CDUP
�& 250 Requested File Action Completed OK) r� rU �cduprj rq rJ r� r� r( )r r� r5 r r r � test_CDUP� s
zFTPClientTests.test_CDUPc C sL | � � | j�� }| �|tj� | �| j�� d� | j� � | j�
d� |S )z�
Test L{ftp.FTPClient.cdup}'s handling of a failed CDUP command.
When the CDUP command fails, the returned Deferred should errback
with L{ftp.CommandFailed}.
r� s! 550 ..: No such file or directory) r� rU r� r� r r� rq rJ r� r� r( r� r r r �test_failedCDUP� s
z FTPClientTests.test_failedCDUPc sF � fdd�}� � � � j�� �|�}� �� j�� d� � j�d� |S )a
Test the PWD command.
L{ftp.FTPClient.pwd} should return a Deferred which fires with a
sequence of one element which is a string representing the current
working directory on the server.
(XXX - This is a bad API)
c s � � t�| d �d� d S )Nr �/bar/baz)rq r ZparsePWDResponser r r r �cbPwd s z&FTPClientTests.test_PWD.<locals>.cbPwd� PWD
� 257 "/bar/baz")r� rU �pwdrj rq rJ r� r( )r r� r5 r r r �test_PWD
s
zFTPClientTests.test_PWDc C sB | � � | j�� }| �|tj� | �| j�� d� | j� d� |S )z�
Test a failure in PWD command.
When the PWD command fails, the returned Deferred should errback
with L{ftp.CommandFailed}.
r� �' 550 /bar/baz: No such file or directory)
r� rU r� r� r r� rq rJ r� r( r� r r r �test_failedPWD s
z FTPClientTests.test_failedPWDc sH � fdd�}� � � � j�d��|�}� �� j�� d� � j�d� |S )a
Test the CWD command.
L{ftp.FTPClient.cwd} should return a Deferred which fires with a
sequence of one element which is the string the server sent
indicating that the command was executed successfully.
(XXX - This is a bad API)
c s � � | d d� d S r� rp r r r r �cbCwd7 s z&FTPClientTests.test_CWD.<locals>.cbCwd�bar/foo�
CWD bar/foo
r� )r� rU �cwdrj rq rJ r� r( )r r� r5 r r r �test_CWD- s
zFTPClientTests.test_CWDc C sD | � � | j�d�}| �|tj� | �| j�� d� | j� d� |S )z�
Test a failure in CWD command.
When the PWD command fails, the returned Deferred should errback
with L{ftp.CommandFailed}.
r� r� s& 550 bar/foo: No such file or directory)
r� rU r� r� r r� rq rJ r� r( r� r r r �test_failedCWDA s z FTPClientTests.test_failedCWDc s� � fdd�}� fdd�}|� j _� �� t� }� j �d|�}|�||� � �� j�� d� � j� � � j �
t � j �� � �� j�� d� � j� � � j �
d� |S ) aN
Test the RETR command in passive mode: get a file and verify its
content.
L{ftp.FTPClient.retrieveFile} should return a Deferred which fires
with the protocol instance passed to it after the download has
completed.
(XXX - This API should be based on producers and consumers)
c s � � |jd� d S �N�� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr� �r r� r r r �cbRetr[ s z/FTPClientTests.test_passiveRETR.<locals>.cbRetrc sd � � | d� � � |d� |�| |f�}|�t�� � � j�d� |�d� |�t �
t � d��� d S )Nr; r< �4 150 File status okay; about to open data connection.r r� �
rq rM r� r �StringTransportrU r( r7 r: r r� r �ConnectionDone�r@ rA r" r� r r r � cbConnect^ s �
z2FTPClientTests.test_passiveRETR.<locals>.cbConnect�spam� PASV
� RETR spam
� 226 Transfer Complete.) rU �connectFactoryr� r0 r0 rj rq rJ r� r� r( rC )r r r r� r5 r r r �test_passiveRETRP s
z FTPClientTests.test_passiveRETRc s� d� j _dd� }� fdd�}|� j _� �� t� }� j �d|�}|�||� � �� j� � dt
� dd �f � � j j
�� � j�� � j �d
� � �� j� � d � � j�� � j �d � |S )
z�
Test the RETR command in non-passive mode.
Like L{test_passiveRETR} but in the configuration where the server
establishes the data connection to the client, rather than the other
way around.
Fc S sL dt �dd�f | _| j�t�� � | j�d� | j�t �
t � d��� d S )N�PORT %sr; �& r r� )
r r= �textr r� r r r7 r: r r� r r ��portCmdr r r � generatePort� s �z.FTPClientTests.test_RETR.<locals>.generatePortc s � � |jd� d S r r� r r r r r � s z(FTPClientTests.test_RETR.<locals>.cbRetrr � PORT %s
r; r � 200 PORT OKr r )rU r� �generatePortCommandr� r0 r0 rj rq rJ r� r r= r> r? r� r( )r r r r� r5 r r r � test_RETRv s, ���
zFTPClientTests.test_RETRc s� � fdd�}|� j _� �� t� }� j �d|�}� �|tj� � �� j �
� d� � j � � � j � t
� j �� � �� j �
� d� � j � � � j � d� |S )z�
Try to RETR an unexisting file.
L{ftp.FTPClient.retrieveFile} should return a Deferred which
errbacks with L{ftp.CommandFailed} if the server indicates the file
cannot be transferred for some reason.
c sZ � � | d� � � |d� |�| |f�}|�t�� � � j�d� |�t� t
� d��� d S �Nr; r< r r� � rq rM r� r r rU r( r: r r� r r r
r r r r � s �z1FTPClientTests.test_failedRETR.<locals>.cbConnectr r
r s# 550 spam: No such file or directory)rU r r� r0 r0 r� r r� rq rJ r� r� r( rC )r r r� r5 r r r �test_failedRETR� s
z FTPClientTests.test_failedRETRc s� d| j _g � � fdd�}|| j _| �� t� }| j �d|�}| �| j�� dt �
dd�f � | j j �� | j�
� | j �d� | �| j�� d � | �� � � d
�� | j�� | �|t j� |S ) z�
Try a RETR, but disconnect during the transfer.
L{ftp.FTPClient.retrieveFile} should return a Deferred which
errbacks with L{ftp.ConnectionLost)
Fc sJ dt �dd�f | _t�� }| j�|� | j|_| j�d� � �|� d S )Nr r; r s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx) r r= r r r� r r� r7 r% )r r� �r� r r r � s z2FTPClientTests.test_lostRETR.<locals>.generatePortr r r; r r r r )rU r� r r� r0 r0 rq rJ r� r r= r> r? r� r( r� rK r� r� )r r r� r5 r r r �
test_lostRETR� s( ��
z FTPClientTests.test_lostRETRc s� t �� �� fdd�}� �fdd�}� �fdd�}|� j_� �� � j�d�\}}|�|� |�|� � �� j� � d� � j�
� � j� t � j�� � �� j� � d � � j�
� � j� d
� t
�||g�S ) a�
Test the STOR command: send a file and verify its content.
L{ftp.FTPClient.storeFile} should return a two-tuple of Deferreds.
The first of which should fire with a protocol instance when the
data connection has been established and is responsible for sending
the contents of the file. The second of which should fire when the
upload has completed, the data connection has been closed, and the
server has acknowledged receipt of the file.
(XXX - storeFile should take a producer as an argument, instead, and
only return a Deferred which fires when the upload has succeeded or
failed).
c s: � j �d� | j�d� | �� | �t�t� d��� d S �Nr r r� �
rU r( rJ rg �finishr: r r� r r �Zsenderr r r �cbStore� s � z0FTPClientTests.test_passiveSTOR.<locals>.cbStorec s � � ��� d� d S r �rq r� �Zign�r r� r r �cbFinish� s z1FTPClientTests.test_passiveSTOR.<locals>.cbFinishc s4 � � | d� � � |d� |�| |f�}|��� d S �Nr; r< �rq rM r� r
r( r r r � s z2FTPClientTests.test_passiveSTOR.<locals>.cbConnectr r
� STOR spam
r )r r rU r r� r rj rq rJ r� r� r( rC r
rk )r r% r) r rR rn r r( r �test_passiveSTOR� s
z FTPClientTests.test_passiveSTORc s� t �� �� fdd�}� �fdd�}|� j_� �� � j�d�\}}|�|� � �|tj � � �
� j � � d� � j �
� � j�t� j�� � �
� j � � d� � j �
� � j�d� t�||g�S ) z�
Test a failure in the STOR command.
If the server does not acknowledge successful receipt of the
uploaded file, the second Deferred returned by
L{ftp.FTPClient.storeFile} should errback with L{ftp.CommandFailed}.
c s: � j �d� | j�d� | �� | �t�t� d��� d S r! r" r$ r r r r% s � z/FTPClientTests.test_failedSTOR.<locals>.cbStorec s4 � � | d� � � |d� |�| |f�}|��� d S r* r+ r
r( r r r s z1FTPClientTests.test_failedSTOR.<locals>.cbConnectr r
r, s. 426 Transfer aborted. Data connection closed.)r r rU r r� r rj r� r r� rq rJ r� r� r( rC r
rk )r r% r rR rn r r( r �test_failedSTOR s"
�z FTPClientTests.test_failedSTORc sx t �� �d� j_�fdd�}� fdd�}� �fdd�}|� j_� �� � j�d�\}}|�|� |�|� t� ||g�S ) z�
Test the STOR command in non-passive mode.
Like L{test_passiveSTOR} but in the configuration where the server
establishes the data connection to the client, rather than the other
way around.
Fc s" dt �dd� | _| j�� � d S )Nr{ r; r )r r= r r r� r )r� r r r ; s z.FTPClientTests.test_STOR.<locals>.generatePortc s� � � � j�� dt�dd�f �� jj�� � j�� � j� d� � � � j�� d� � j�� � j� d� | j�
d� | � � | � t
�t�d��� � j� d � d S )
Nr r; r r r, r r r� r )rq rJ r� r r= r> rU r? r� r( rg r# r: r r� r r r$ r r r r% ? s ��
� z)FTPClientTests.test_STOR.<locals>.cbStorec s � � ��� d� d S r r& r' r( r r r) N s z*FTPClientTests.test_STOR.<locals>.cbFinishr )
r r rU r� r r� r rj r
rk )r r r% r) rR rn r r( r � test_STOR1 s
zFTPClientTests.test_STORc s� � fdd�}� fdd�}|� j _� �� t�� }� j �d|��||�}� �� j� � d� � j�
� � j � t � j �� � �� j� � d� � j � d� |S ) a{
Test the LIST command.
L{ftp.FTPClient.list} should return a Deferred which fires with a
protocol instance which was passed to list after the command has
succeeded.
(XXX - This is a very unfortunate API; if my understanding is
correct, the results are always at least line-oriented, so allowing
a per-line parser function to be specified would make this simpler,
but a default implementation should really be provided which knows
how to deal with all the formats used in real servers, so
application developers never have to care about this insanity. It
would also be nice to either get back a Deferred of a list of
filenames or to be able to consume the files as they are received
(which the current API does allow, but in a somewhat inconvenient
fashion) -exarkun)
c s: dd� |j D �}dddg}|�� |�� � �||� d S )Nc S s g | ] }|d �qS �r� r �r~ r# r r r r m s zCFTPClientTests.test_passiveLIST.<locals>.cbList.<locals>.<listcomp>r� r� �baz�r� rN rq �r r� �fls�expectedr r r �cbListl s
z/FTPClientTests.test_passiveLIST.<locals>.cbListc sx � � | d� � � |d� |�| |f�}|�t�� � � j�d� dddg}|D ]}|�|� qN|�t �
t � d��� d S )Nr; r< r �8 -rw-r--r-- 0 spam egg 100 Oct 10 2006 foo
�8 -rw-r--r-- 3 spam egg 100 Oct 10 2006 bar
�8 -rw-r--r-- 4 spam egg 100 Oct 10 2006 baz
r� r )r@ rA r" r� �sendingr� r r r r s s �� z2FTPClientTests.test_passiveLIST.<locals>.cbConnect�foo/barr
� LIST foo/bar
r )
rU r r� r r� r� rj rq rJ r� r� r( rC )r r7 r r� r5 r r r �test_passiveLISTY s
z FTPClientTests.test_passiveLISTc s� d� j _� fdd�}� fdd�}|� j _� �� t�� }� j �d|��||�}� �� j �
� dt� dd �f � � j j
�� � j �� � j �d
� � �� j �
� d � � j �� � j �d � |S )
z�
Test the LIST command in non-passive mode.
Like L{test_passiveLIST} but in the configuration where the server
establishes the data connection to the client, rather than the other
way around.
Fc sl dt �dd�f | _| j�t�� � � j�d� dddg}|D ]}| j� |� q>| j�
t � t
�d��� d S ) Nr r; r r r8 r9 r: r� �r r= r r r� r r rU r( r7 r: r r� r r )r r; r� r r r r � s ���z.FTPClientTests.test_LIST.<locals>.generatePortc s: dd� |j D �}dddg}|�� |�� � �||� d S )Nc S s g | ] }|d �qS r0 r r1 r r r r � s z<FTPClientTests.test_LIST.<locals>.cbList.<locals>.<listcomp>r� r� r2 r3 r4 r r r r7 � s
z(FTPClientTests.test_LIST.<locals>.cbListr< r r; r r r= r )rU r� r r� r r� r� rj rq rJ r� r= r> r? r� r( )r r r7 r� r5 r r r � test_LIST� s$ ��
zFTPClientTests.test_LISTc s� � fdd�}|� j _� �� t�� }� j �d|�}� �|tj� � �� j �
� d� � j � � � j � t
� j �� � �� j �
� d� � j � d� |S )z�
Test a failure in LIST command.
L{ftp.FTPClient.list} should return a Deferred which fails with
L{ftp.CommandFailed} if the server indicates the indicated path is
invalid for some reason.
c sZ � � | d� � � |d� |�| |f�}|�t�� � � j�d� |�t� t
� d��� d S r r r
r r r r � s �z1FTPClientTests.test_failedLIST.<locals>.cbConnectr< r
r= �& 550 foo/bar: No such file or directory)rU r r� r r� r� r� r� rq rJ r� r� r( rC )r r r� r5 r r r �test_failedLIST� s
z FTPClientTests.test_failedLISTc s� d� j _� fdd�}� fdd�}|� j _� �� t� }� j �d|��||�}� �� j� � dt
� dd �f � � j j
�� � j�� � j �d
� � �� j� � d � � j �d � |S )
z�
Test the NLST command in non-passive mode.
L{ftp.FTPClient.nlst} should return a Deferred which fires with a
list of filenames when the list command has completed.
Fc sp dt �dd�f | _| j�t�� � � j�d� | j� d� | j� d� | j� d� | j�
t � t
�d��� d S ) Nr r; r r � foo
� bar
� baz
r� r? r r r r r � s � �z.FTPClientTests.test_NLST.<locals>.generatePortc s> |j �� jj��� }dddg}|�� |�� � �||� d S )Nr� r� r2 )r3 �decoderU r? r@ rN rq �r r� r5 r6 r r r r7 � s
z(FTPClientTests.test_NLST.<locals>.cbListr< r r; r r � NLST foo/bar
r )rU r� r r� r0 �nlstrj rq rJ r� r r= r> r? r� r( )r r r7 �lstprotor5 r r r � test_NLST� s" ��
zFTPClientTests.test_NLSTc s� � fdd�}� fdd�}|� j _� �� t� }� j �d|��||�}� �� j�� d� � j� � � j �
t � j �� � �� j�� d� � j �
d� |S ) z�
Test the NLST command.
Like L{test_passiveNLST} but in the configuration where the server
establishes the data connection to the client, rather than the other
way around.
c s4 |j �� }dddg}|�� |�� � �||� d S )Nr> r? s baz)r3 r@ rN rq rG r r r r7 s
z/FTPClientTests.test_passiveNLST.<locals>.cbListc sx � � | d� � � |d� |�| |f�}|�t�� � � j�d� |�d� |�d� |�d� |�t �
t � d��� d S )Nr; r< r rC rD rE r� r r
r r r r s �
z2FTPClientTests.test_passiveNLST.<locals>.cbConnectr< r
rH r ) rU r r� r0 rI rj rq rJ r� r� r( rC )r r7 r rJ r5 r r r �test_passiveNLST s
z FTPClientTests.test_passiveNLSTc s� t �� �� �fdd�}|� j_� �� t� }� j�d|�}� �|tj � � �
� j � � d� � j �
� � j�t� j�� � �
� j � � d� � j�d� |S )z�
Test a failure in NLST command.
L{ftp.FTPClient.nlst} should return a Deferred which fails with
L{ftp.CommandFailed} if the server indicates the indicated path is
invalid for some reason.
c sV � � | d� � � |d� |�| |f�}|��� � j�d� |�t�t� d��� d S r )
rq rM r� rU r( r: r r� r r r
r( r r r 3 s
�z1FTPClientTests.test_failedNLST.<locals>.cbConnectr< r
rH rA )r r rU r r� r0 rI r� r r� rq rJ r� r� r( rC )r r rJ r5 r r( r �test_failedNLST* s
z FTPClientTests.test_failedNLSTc C s� | � � | j�dd�}| �| j�� d� | j�� d}| j�|�| jj �� | �| j�� d� d}| j�|�| jj �� |�
| j|g|gf� |S )z�
L{ftp.FTPClient.rename} issues I{RNTO} and I{RNFR} commands and returns
a L{Deferred} which fires when a file has successfully been renamed.
�/spam�/ham� RNFR /spam
z8350 Requested file action pending further information.
� RNTO /ham
r� ) r� rU �renamerq rJ r� r� r( r> r? rj )r r5 Z fromResponseZ
toResponser r r �test_renameFromToI s
��z FTPClientTests.test_renameFromToc C sL | � � d}d}| j�||� | j�d� | j�d� | �| j�� d� dS )z�
L{ftp.FTPClient.rename} issues I{RNTO} and I{RNFR} commands with paths
escaped according to U{http://cr.yp.to/ftp/filesystem.html}.
z
/foo/ba
r/bazz/qu
uxs 350 s 250 s! RNFR /foo/ba r/baz
RNTO /qu ux
N)r� rU rR r( rq rJ r� )r ZfromFileZtoFiler r r � test_renameFromToEscapesPaths` s �z,FTPClientTests.test_renameFromToEscapesPathsc C s^ | � � | j�dd�}| �| j�� d� | j�� | j�d� | �| j�� d� | �|t j
�S )z�
The L{Deferred} returned by L{ftp.FTPClient.rename} is errbacked with
L{CommandFailed} if the I{RNFR} command receives an error response code
(for example, because the file does not exist).
rN rO rP �! 550 Requested file unavailable.
r2 � r� rU rR rq rJ r� r� r( r� r r� r� r r r �$test_renameFromToFailingOnFirstErrorr s
z3FTPClientTests.test_renameFromToFailingOnFirstErrorc C sj | � � | j�dd�}| �| j�� d� | j�� | j�d� | �| j�� d� | j�d� | �|t j
�S )z�
The L{Deferred} returned by L{ftp.FTPClient.rename} is errbacked with
L{CommandFailed} if the I{RNTO} command receives an error response code
(for example, because the destination directory does not exist).
rN rO rP s8 350 Requested file action pending further information.
rQ rU rV r� r r r �"test_renameFromToFailingOnRenameTo� s
� z1FTPClientTests.test_renameFromToFailingOnRenameToc C sB | � � | j�d�}| �| j�� d� | j�d� |�| jdg�S )z�
L{ftp.FTPClient.makeDirectory} issues a I{MKD} command and returns a
L{Deferred} which is called back with the server's response if the
directory is created.
rN � MKD /spam
s 257 "/spam" created.z257 "/spam" created.)r� rU �
makeDirectoryrq rJ r� r( rj r� r r r �test_makeDirectory� s
z!FTPClientTests.test_makeDirectoryc C s6 | � � | j�d�}| �| j�� d� | j�d� |S )z�
L{ftp.FTPClient.makeDirectory} escapes the path name it sends according
to U{http://cr.yp.to/ftp/filesystem.html}.
z/sp
ams MKD /sp am
s 257 win)r� rU rZ rq rJ r� r( r� r r r � test_makeDirectoryPathEscape� s
z+FTPClientTests.test_makeDirectoryPathEscapec C s@ | � � | j�d�}| �| j�� d� | j�d� | �|tj �S )z�
L{ftp.FTPClient.makeDirectory} returns a L{Deferred} which is errbacked
with L{CommandFailed} if the server returns an error response code.
rN rY s 550 PERMISSION DENIED)
r� rU rZ rq rJ r� r( r� r r� r� r r r �test_failedMakeDirectory� s
z'FTPClientTests.test_failedMakeDirectoryc sF � fdd�}� � � � j�� �|�}� �� j�� d� � j�d� |S )z�
Test the getDirectory method.
L{ftp.FTPClient.getDirectory} should return a Deferred which fires with
the current directory on the server. It wraps PWD command.
c s � � | d� d S )Nr� rp r r r r �cbGet� s z/FTPClientTests.test_getDirectory.<locals>.cbGetr� r� )r� rU � getDirectoryrj rq rJ r� r( )r r^ r5 r r r �test_getDirectory� s z FTPClientTests.test_getDirectoryc C sB | � � | j�� }| �|tj� | �| j�� d� | j� d� |S )zj
Test a failure in getDirectory method.
The behaviour should be the same as PWD.
r� r� �
r� rU r_ r� r r� rq rJ r� r( r� r r r �test_failedGetDirectory� s
z&FTPClientTests.test_failedGetDirectoryc C sB | � � | j�� }| �|tj� | �| j�� d� | j� d� |S )z�
Test a different failure in getDirectory method.
The response should be quoted to be parsed, so it returns an error
otherwise.
r� s 257 /bar/bazra r� r r r � test_anotherFailedGetDirectory� s
z-FTPClientTests.test_anotherFailedGetDirectoryc C sP | � � | j�d�}| �| j�� d� d}| j�|�| jj�� |� | j|g�S )z�
L{ftp.FTPClient.removeFile} sends a I{DELE} command to the server for
the indicated file and returns a Deferred which fires after the server
sends a 250 response code.
� /tmp/test� DELE /tmp/test
�*250 Requested file action okay, completed.)
r� rU �
removeFilerq rJ r� r( r> r? rj �r r5 �responser r r �test_removeFile� s z FTPClientTests.test_removeFilec sf �� � �j�d�}���j�� d� d� �j�� ��jj�� �� |t
j �}|� � �fdd�� |S )a
If the server returns a response code other than 250 in response to a
I{DELE} sent by L{ftp.FTPClient.removeFile}, the L{Deferred} returned
by C{removeFile} is errbacked with a L{Failure} wrapping a
L{CommandFailed}.
rd re �,501 Syntax error in parameters or arguments.c s �� | j� gf�S r r� ��exc�ri r r r r�
r2 z6FTPClientTests.test_failedRemoveFile.<locals>.<lambda>)
r� rU rg rq rJ r� r( r> r? r� r r� rj r� r rn r �test_failedRemoveFile� s z$FTPClientTests.test_failedRemoveFilec sT �� � �j�d�}d� �j�� ��jj�� ��|tj�}|� � �fdd�� |S )z�
If the server returns a response line which cannot be parsed, the
L{Deferred} returned by L{ftp.FTPClient.removeFile} is errbacked with a
L{BadResponse} containing the response.
rd �765 blah blah blahc s �� | j� gf�S r r� rl rn r r r�
r2 zBFTPClientTests.test_unparsableRemoveFileResponse.<locals>.<lambda>)
r� rU rg r( r> r? r� r � BadResponserj r� r rn r �!test_unparsableRemoveFileResponse
s z0FTPClientTests.test_unparsableRemoveFileResponsec C s8 | � � | j�d�}| j�d� | j�d� |�| j�S )z�
If the server returns multiple response lines, the L{Deferred} returned
by L{ftp.FTPClient.removeFile} is still fired with a true value if the
ultimate response code is 250.
rd � 250-perhaps a progress report� 250 okay)r� rU rg r( rj r� r� r r r � test_multilineRemoveFileResponse
s
z/FTPClientTests.test_multilineRemoveFileResponsec C sP | � � | j�d�}| �| j�� d� d}| j�|�| jj�� |� | j|g�S )z�
L{ftp.FTPClient.removeDirectory} sends a I{RMD} command to the server
for the indicated directory and returns a Deferred which fires after
the server sends a 250 response code.
rd � RMD /tmp/test
rf )
r� rU �removeDirectoryrq rJ r� r( r> r? rj rh r r r �test_removeDirectory+
s z#FTPClientTests.test_removeDirectoryc sf �� � �j�d�}���j�� d� d� �j�� ��jj�� �� |t
j �}|� � �fdd�� |S )a
If the server returns a response code other than 250 in response to a
I{RMD} sent by L{ftp.FTPClient.removeDirectory}, the L{Deferred}
returned by C{removeDirectory} is errbacked with a L{Failure} wrapping
a L{CommandFailed}.
rd rv rk c s �� | j� gf�S r r� rl rn r r r� F
r2 z;FTPClientTests.test_failedRemoveDirectory.<locals>.<lambda>)
r� rU rw rq rJ r� r( r> r? r� r r� rj r� r rn r �test_failedRemoveDirectory9
s z)FTPClientTests.test_failedRemoveDirectoryc sT �� � �j�d�}d� �j�� ��jj�� ��|tj�}|� � �fdd�� |S )z�
If the server returns a response line which cannot be parsed, the
L{Deferred} returned by L{ftp.FTPClient.removeDirectory} is errbacked
with a L{BadResponse} containing the response.
rd rp c s �� | j� gf�S r r� rl rn r r r� U
r2 zGFTPClientTests.test_unparsableRemoveDirectoryResponse.<locals>.<lambda>)
r� rU rw r( r> r? r� r rq rj r� r rn r �&test_unparsableRemoveDirectoryResponseJ
s z5FTPClientTests.test_unparsableRemoveDirectoryResponsec C s8 | � � | j�d�}| j�d� | j�d� |�| j�S )z�
If the server returns multiple response lines, the L{Deferred} returned
by L{ftp.FTPClient.removeDirectory} is still fired with a true value
if the ultimate response code is 250.
rd rs rt )r� rU rw r( rj r� r� r r r �%test_multilineRemoveDirectoryResponseY
s
z4FTPClientTests.test_multilineRemoveDirectoryResponseN)-r- r. r/ r� ro r� r� r� r� r� r� r� r r r r r r r- r. r/ r> r@ rB rK rL rM rS rT rW rX r[ r\ r] r` rb rc rj ro rr ru rx ry rz r{ r r r r r� � sT
&& #.%(6. ('
r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�FTPClientBasicTestsz
FTP client
c C s& t �� }|�d� | �dg|j� dS )z?
The first response is captured as a greeting.
� 220 Imaginary FTP.z220 Imaginary FTP.N�r r� r( rq Zgreeting�r r� r r r �
test_greetingk
s
z!FTPClientBasicTests.test_greetingc C s& t �� }|�d� | �dg|j� dS )z�
Responses with no message are still valid, i.e. three digits
followed by a space is complete response.
s 220 z220 Nr~ r r r r �test_responseWithNoMessaget
s
z.FTPClientBasicTests.test_responseWithNoMessagec C s� t �� }t�� |_|�d� |�d�}g }|�|j� |� | j
� |�d� | � g |� |�d� | � g |� |�d� | � g |� |�d� | � g |� |�d� | � g |� |�d� | � g |� |�d � | � d
d d d
dddg|d � dS )z$
Multiline response
r} ZBLAHs 210-First line.s 123-Second line.s Just some text.s Hir2 s 321s 210 Done.z210-First line.z123-Second line.zJust some text.ZHir� Z321z 210 Done.r N) r r� r r rJ r( rw rj r% r r� rq )r r� r� r� r r r �test_MultilineResponse~
s>
��z*FTPClientBasicTests.test_MultilineResponsec C sd t �� }t�� |_|�d� |�dd� | �d|j�� � |j� � |�d� | �d|j�� � dS )zO
Passing None as the password avoids sending the PASS command.
� 220 Welcome to Imaginary FTP.�bobN�
USER bob
s 200 Hello bob.r2 �
r r� r r rJ r( Z
queueLoginrq r� r� r r r r �test_noPasswordGiven�
s
z(FTPClientBasicTests.test_noPasswordGivenc C sd t �� }t�� |_|�d� |�dd� | �d|j�� � |j� � |�d� | �d|j�� � dS )zQ
Receiving a 230 response to USER prevents PASS from being sent.
r� r� Zsecretr� s# 230 Hello bob. No password needed.r2 Nr� r r r r �test_noPasswordNeeded�
s
z)FTPClientBasicTests.test_noPasswordNeededN) r- r. r/ r� r� r� r� r� r� r r r r r| g
s
3r| c @ s e Zd ZdZdd� ZdS )�PathHandlingTestsz
Handling paths.
c
C s ddgfddgfdg fddddgfddddgfddgfd dgffD ] \}}| � t�g |�|� qBdddgfd
ddgfd dgfd dgfd
dddgfddddgfdddgfdddgffD ] \}}| � t�dg|�|� q�dg fddgfddgfdddgffD ] \}}| � t�g |�|� q�dg fddgfddddgffD ] \}}| � t�dg|�|� �q4dg fdg fddgfdg fd dgfd dgfd ddgfd ddgfd dgfd!dgfd"g fd#g ff D ] \}}| � t�dg|�|� �q�d$D ]}| �tjtjg |� �q�d%D ] }| �tjtjdg|� �q�d&S )'z"
Normalize paths.
r z/a�/za/b/cr �cz/a/b/cz/a/za/zb/z/bz/b/zb/czb/c/z/b/cz/b/c/z//z//aza//za//bz//bzb//c�..�../za/..�xz/a/..z/a/b/..z/a/b/../z /a/b/../cz
/a/b/../c/z /a/b/../../cz
/a/b/../../c/z/a/b/../../c/..z/a/b/../../c/../) r� r� za/../..za/../../z/..z/../z/a/../..z /a/../../z
/a/b/../../..)z../..z../../z
../a/../..N)rq r Z
toSegmentsr� Z InvalidPath)r ZinpZoutpr r r �test_Normalizer�
s` �
�
� �
� z!PathHandlingTests.test_NormalizerN)r- r. r/ r� r� r r r r r� �
s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �IsGlobbingExpressionTestsz;
Tests for _isGlobbingExpression utility function.
c C s2 | � t�� � | � t�g �� | � t�d�� dS )z^
_isGlobbingExpression will return False for None, or empty
segments.
N�r� r �_isGlobbingExpressionr r r r �&test_isGlobbingExpressionEmptySegments" s z@IsGlobbingExpressionTests.test_isGlobbingExpressionEmptySegmentsc C s, | � t�ddg�� | � t�ddg�� dS )z�
_isGlobbingExpression will return False for plain segments.
Also, it only checks the last segment part (filename) and will not
check the path name.
r �expr�*.txtNr� r r r r � test_isGlobbingExpressionNoGlob, s z9IsGlobbingExpressionTests.test_isGlobbingExpressionNoGlobc C s@ | � t�ddg�� | � t�ddg�� | � t�ddg�� dS )z�
_isGlobbingExpression will return True for segments which contains
globbing characters in the last segment part (filename).
r r� z [a-b].txtzfil?.txtN)r� r r� r r r r � test_isGlobbingExpressionGlob7 s z7IsGlobbingExpressionTests.test_isGlobbingExpressionGlobN)r- r. r/ r� r� r� r� r r r r r� s
r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �BaseFTPRealmTestsz�
Tests for L{ftp.BaseFTPRealm}, a base class to help define L{IFTPShell}
realms with different user home directory policies.
c C s | � tttj�� dS )z;
L{ftp.BaseFTPRealm} implements L{IRealm}.
N)r� r r r � BaseFTPRealmr r r r �test_interfaceG s z BaseFTPRealmTests.test_interfacec sl t �| �� ��g � G � �fdd�dtj�}|| �� �}|�ddtj�\}}}| �|tj� | � |j
�� dS )z�
L{ftp.BaseFTPRealm} calls its C{getHomeDirectory} method with the
avatarId being requested to determine the home directory for that
avatar.
c s e Zd Z� �fdd�ZdS )z:BaseFTPRealmTests.test_getHomeDirectory.<locals>.TestRealmc s � � |� �S r r� )r ZavatarId�Zavatarsr� r r �getHomeDirectoryW s
zKBaseFTPRealmTests.test_getHomeDirectory.<locals>.TestRealm.getHomeDirectoryN)r- r. r/ r� r r� r r � TestRealmV s r� �alice@example.comN) r r[ rW r r� �
requestAvatar� IFTPShellr� �FTPShellrq �filesystemRoot)r r� �realm�iface�avatar�logoutr r� r �test_getHomeDirectoryN s �
z'BaseFTPRealmTests.test_getHomeDirectoryc C sP | � � }t�|�}|�tjdtj�\}}}| �|tj� | � |j
t � |�� dS )zy
L{ftp.BaseFTPRealm} returns an L{ftp.FTPAnonymousShell} instance for
anonymous avatar requests.
N)
rW r r� r� r Z ANONYMOUSr� r� r$ rq r� r r[ )r rE r� r� r� r� r r r �test_anonymousb s
�
z BaseFTPRealmTests.test_anonymousc C s$ t �| �� �}| �t|jt� � dS )z�
L{ftp.BaseFTPRealm.getHomeDirectory} should be overridden by a subclass
and raises L{NotImplementedError} if it is not.
N)r r� rW r� �NotImplementedErrorr� r )r r� r r r �test_notImplementedo s z%BaseFTPRealmTests.test_notImplementedN)r- r. r/ r� r� r� r� r� r r r r r� B s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�
FTPRealmTestsz$
Tests for L{ftp.FTPRealm}.
c C s: d}t �| �� |�}|�d�}| �t�|��d�|� dS )z�
L{ftp.FTPRealm} accepts an extra directory to its initializer and treats
the avatarId passed to L{ftp.FTPRealm.getHomeDirectory} as a single path
segment to construct a child of that directory.
z
/path/to/homer� N)r r] rW r� rq r r[ r� )r �baser� �homer r r r� } s
�z#FTPRealmTests.test_getHomeDirectoryc C s. t �| �� �}|�d�}| �t�d�|� dS )z�
If no extra directory is passed to L{ftp.FTPRealm}, it uses C{"/home"}
as the base directory containing all user home directories.
r� z/home/alice@example.comN)r r] rW r� rq r r[ )r r� r� r r r �test_defaultHomeDirectory� s
z'FTPRealmTests.test_defaultHomeDirectoryN)r- r. r/ r� r� r� r r r r r� y s
r� c @ s$ e Zd ZdZeZdd� Zdd� ZdS )�SystemFTPRealmTestsz*
Tests for L{ftp.SystemFTPRealm}.
c C sJ t �� }ddl}|�|�j}t�| �� �}|�|�}| � |t
� |�� dS )z�
L{ftp.SystemFTPRealm.getHomeDirectory} treats the avatarId passed to it
as a username in the underlying platform and returns that account's home
directory.
r N) �getpassZgetuserr� �getpwnam�pw_dirr �SystemFTPRealmrW r� rq r r[ )r r[ r� r6 r� r� r r r r� � s
z)SystemFTPRealmTests.test_getHomeDirectoryc C sF t �tj�d�dd� td�D �� }t�| �� �}| � t
|j |� dS )z�
L{ftp.SystemFTPRealm.getHomeDirectory} raises L{UnauthorizedLogin} when
passed a username which has no corresponding home directory in the
system's accounts database.
r� c s s | ]}t �tjtj �V qd S r )�random�choice�string�
ascii_lettersZdigits)r~ r� r r r � <genexpr>� s �
�z6SystemFTPRealmTests.test_noSuchUser.<locals>.<genexpr>r� N) r� r� r� r� r� r r r� rW r� r r� )r r[ r� r r r �test_noSuchUser� s
�
z#SystemFTPRealmTests.test_noSuchUserN)r- r. r/ r� � nonPOSIXSkip�skipr� r� r r r r r� � s r� c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�ErrnoToFailureTestsz9
Tests for L{ftp.errnoToFailure} errno checking.
c C s t �tjd�}| �|t j�S )zS
C{errno.ENOENT} should be translated to L{ftp.FileNotFoundError}.
r� )r �errnoToFailure�errnoZENOENTr� �FileNotFoundErrorr� r r r �
test_notFound� s z!ErrnoToFailureTests.test_notFoundc C s t �tjd�}| �|t j�S )zV
C{errno.EPERM} should be translated to L{ftp.PermissionDeniedError}.
r� )r r� r� ZEPERMr� �PermissionDeniedErrorr� r r r �test_permissionDenied� s z)ErrnoToFailureTests.test_permissionDeniedc C s t �tjd�}| �|t j�S )zW
C{errno.EACCES} should be translated to L{ftp.PermissionDeniedError}.
r� )r r� r� ZEACCESr� r� r� r r r �test_accessDenied� s z%ErrnoToFailureTests.test_accessDeniedc C s t �tjd�}| �|t j�S )zW
C{errno.ENOTDIR} should be translated to L{ftp.IsNotADirectoryError}.
r� )r r� r� ZENOTDIRr� �IsNotADirectoryErrorr� r r r �test_notDirectory� s z%ErrnoToFailureTests.test_notDirectoryc C s t �tjd�}| �|t j�S )zQ
C{errno.EEXIST} should be translated to L{ftp.FileExistsError}.
r� )r r� r� ZEEXISTr� �FileExistsErrorr� r r r �test_fileExists� s z#ErrnoToFailureTests.test_fileExistsc C s t �tjd�}| �|t j�S )zS
C{errno.EISDIR} should be translated to L{ftp.IsADirectoryError}.
r� )r r� r� ZEISDIRr� r r� r r r �test_isDirectory� s z$ErrnoToFailureTests.test_isDirectoryc C s6 z t d��W n$ t�dd�}| �|t � Y S X dS )z�
If an unknown errno is passed to L{ftp.errnoToFailure}, it should let
the originating exception pass through.
r� r� r� N)r� r r� r� r� r r r �test_passThrough� s
z$ErrnoToFailureTests.test_passThroughN) r- r. r/ r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�AnonymousFTPShellTestsz*
Test anonymous shell properties.
c C s& t �d�}|�d�}| �|t j� |S )zt
Check that L{ftp.FTPAnonymousShell} returns an error when trying to
open it in write mode.
r� �r� )r r$ r
r� r� )r r r5 r r r �test_anonymousWrite s
z*AnonymousFTPShellTests.test_anonymousWriteN)r- r. r/ r� r� r r r r r� � s r� c @ s2 e Zd ZdZdd� Zdd� Zdd� ZdLd d
�Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKS )M�IFTPShellTestsMixinz7
Generic tests for the C{IFTPShell} interface.
c C s
t � �dS )a
Test if the directory exists at C{path}.
@param path: the relative path to check.
@type path: C{str}.
@return: C{True} if C{path} exists and is a directory, C{False} if
it's not the case
@rtype: C{bool}
N�r� �r rF r r r �directoryExists s z#IFTPShellTestsMixin.directoryExistsc C s
t � �dS )z�
Create a directory in C{path}.
@param path: the relative path of the directory to create, with one
segment.
@type path: C{str}
Nr� r� r r r r� # s z#IFTPShellTestsMixin.createDirectoryc C s
t � �dS )a
Test if the file exists at C{path}.
@param path: the relative path to check.
@type path: C{str}.
@return: C{True} if C{path} exists and is a file, C{False} if it's not
the case.
@rtype: C{bool}
Nr� r� r r r �
fileExists. s z IFTPShellTestsMixin.fileExistsr2 c C s
t � �dS )a
Create a file named C{path} with some content.
@param path: the relative path of the file to create, without
directory.
@type path: C{str}
@param fileContent: the content of the file.
@type fileContent: C{str}
Nr� �r rF � fileContentr r r �
createFile< s z IFTPShellTestsMixin.createFilec C s. | � | �d�� | �d� | �| �d�� dS )z�
C{directoryExists} should report correctly about directory existence,
and C{createDirectory} should create a directory detectable by
C{directoryExists}.
r� N)r� r� r� r� r r r r �test_createDirectoryJ s
z(IFTPShellTestsMixin.test_createDirectoryc C s. | � | �d�� | �d� | �| �d�� dS )z�
C{fileExists} should report correctly about file existence, and
C{createFile} should create a file detectable by C{fileExists}.
�file.txtN)r� r� r� r� r r r r �test_createFileU s
z#IFTPShellTestsMixin.test_createFilec s" � j �d�}� fdd�}|�|�S )zI
Create a directory and check it ends in the filesystem.
r� c s � � � �d�� d S )Nr� )r� r� r� r r r r� d s z2IFTPShellTestsMixin.test_makeDirectory.<locals>.cb)r rZ rj r� r r r r[ _ s z&IFTPShellTestsMixin.test_makeDirectoryc C s$ | � d� | j�d�}| �|tj�S )zm
Creating a directory that already exists should fail with a
C{ftp.FileExistsError}.
r� r� )r� r rZ r� r r� r� r r r �test_makeDirectoryErrori s
z+IFTPShellTestsMixin.test_makeDirectoryErrorc s, � � d� � j�d�}� fdd�}|�|�S )zW
Try to remove a directory and check it's removed from the filesystem.
r� �r� c s � � � �d�� d S r� )r� r� r� r r r r� y s z4IFTPShellTestsMixin.test_removeDirectory.<locals>.cb)r� r rw rj r� r r r rx s s
z(IFTPShellTestsMixin.test_removeDirectoryc C s$ | � d� | j�d�}| �|tj�S )zn
removeDirectory should not work in file and fail with a
C{ftp.IsNotADirectoryError}.
r� �r� )r� r rw r� r r� r� r r r �test_removeDirectoryOnFile~ s
z.IFTPShellTestsMixin.test_removeDirectoryOnFilec C s | j �d�}| �|tj�S )zl
Removing directory that doesn't exist should fail with a
C{ftp.FileNotFoundError}.
r� )r rw r� r r� r� r r r � test_removeNotExistingDirectory� s z3IFTPShellTestsMixin.test_removeNotExistingDirectoryc s0 � � d� � j�d�}� fdd�}|�|� |S )zR
Try to remove a file and check it's removed from the filesystem.
r� r� c s � � � �d�� d S �Nr� )r� r� r r r r r� � s z/IFTPShellTestsMixin.test_removeFile.<locals>.cb)r� r rg rj r� r r r rj � s
z#IFTPShellTestsMixin.test_removeFilec C s$ | � d� | j�d�}| �|tj�S )z:
removeFile should not work on directory.
�ned�r� )r� r rg r� r r r� r r r �test_removeFileOnDirectory� s
z.IFTPShellTestsMixin.test_removeFileOnDirectoryc C s | j �d�}| �|tj�S )zl
Try to remove a non existent file, and check it raises a
L{ftp.FileNotFoundError}.
r� )r rg r� r r� r� r r r �test_removeNotExistingFile� s z.IFTPShellTestsMixin.test_removeNotExistingFilec s6 � � d� � �d� � j�d�}� fdd�}|�|�S )z6
Check the output of the list method.
r� r� ��.c s$ | � � � �| dg fdg fg� d S )Nr� r� �rN rq r r r r r� � s �z)IFTPShellTestsMixin.test_list.<locals>.cb�r� r� r r� rj r� r r r � test_list� s
z IFTPShellTestsMixin.test_listc s8 � � d� � �d� � j�dd�}� fdd�}|�|�S )z<
Check the output of list with asked stats.
r� r� r� )re Z permissionsc st | � � � �t| �d� � �| d d d� � �| d d d� � �t| d d �d� � �t| d d �d� d S )Nr= r r� r� r� )rN rq r� r r r r r� � s z1IFTPShellTestsMixin.test_listWithStat.<locals>.cbr� r� r r r �test_listWithStat� s
z%IFTPShellTestsMixin.test_listWithStatc C s$ | � d� | j�dd�}| �|t�S )�P
Querying an invalid stat should result to a C{AttributeError}.
r� r� �re Z whateverstat)r� r r� r� �AttributeErrorr� r r r �test_listWithInvalidStat� s
z,IFTPShellTestsMixin.test_listWithInvalidStatc s, � � d� � j�d�}� fdd�}|�|�S )z@
Check the output of the list method on a file.
r� r� c s | � � � �| dg fg� d S r� r� r r r r r� � s �z-IFTPShellTestsMixin.test_listFile.<locals>.cb)r� r r� rj r� r r r �
test_listFile� s
z!IFTPShellTestsMixin.test_listFilec C s | j �d�}| �|tj�S )zm
list on a directory that doesn't exist should fail with a
L{ftp.FileNotFoundError}.
r� )r r� r� r r� r� r r r � test_listNotExistingDirectory� s z1IFTPShellTestsMixin.test_listNotExistingDirectoryc C s | � d� | j�d�}|S )z+
Try to access a resource.
r� r� )r� r �accessr� r r r � test_access� s
z IFTPShellTestsMixin.test_accessc C s | j �d�}| �|tj�S )zF
access should fail on a resource that doesn't exist.
r� )r r� r� r r� r� r r r �test_accessNotFound� s z'IFTPShellTestsMixin.test_accessNotFoundc s0 � � d� � j�d�}� fdd�}|�|� |S )zY
Check that openForReading returns an object providing C{ftp.IReadFile}.
r� r� c s � � tj�| �� d S r )r� r Z IReadFile�
providedByr r r r r�
s z3IFTPShellTestsMixin.test_openForReading.<locals>.cb)r� r r- rj r� r r r �test_openForReading
s
z'IFTPShellTestsMixin.test_openForReadingc C s | j �d�}| �|tj�S )zr
openForReading should fail with a C{ftp.FileNotFoundError} on a file
that doesn't exist.
r� )r r- r� r r� r� r r r �test_openForReadingNotFound
s z/IFTPShellTestsMixin.test_openForReadingNotFoundc C s$ | � d� | j�d�}| �|tj�S )z>
openForReading should not work on directory.
r� r� )r� r r- r� r r r� r r r � test_openForReadingOnDirectory
s
z2IFTPShellTestsMixin.test_openForReadingOnDirectoryc s4 �j �d�}� �fdd�}�fdd�� |�|� |S )zZ
Check that openForWriting returns an object providing C{ftp.IWriteFile}.
r� c s �� tj�| �� | �� �� �S r )r� r �
IWriteFiler� r rj r ��cb2r r r �cb1#
s z4IFTPShellTestsMixin.test_openForWriting.<locals>.cb1c s � � t�| �� d S r )r� r r� r r r r r� &
s z4IFTPShellTestsMixin.test_openForWriting.<locals>.cb2)r r
rj )r r5 r� r r� r �test_openForWriting
s
z'IFTPShellTestsMixin.test_openForWritingc C s$ | � d� | j�d�}| �|tj�S )zd
openForWriting should not be able to open a directory that already
exists.
r� r� )r� r r
r� r r r� r r r �$test_openForWritingExistingDirectory,
s
z8IFTPShellTestsMixin.test_openForWritingExistingDirectoryc C s$ | � d� | j�d�}| �|tj�S )z�
openForWring should fail with a L{ftp.FileNotFoundError} if you specify
a file in a directory that doesn't exist.
r� )r� Z idonotexistr� )r� r r
r� r r� r� r r r �)test_openForWritingInNotExistingDirectory6
s
z=IFTPShellTestsMixin.test_openForWritingInNotExistingDirectoryc s: d� �� d� � �j�dd�}� �fdd�}|�|� |S )z@
Check the output of the stat method on a file.
� wobble
r� r� �re rX c s& �� | d t� �� ��| d � d S )Nr r� )rq r� r� r �r� r r r r� G
s z-IFTPShellTestsMixin.test_statFile.<locals>.cb)r� r �statrj r� r r� r �
test_statFile@
s
z!IFTPShellTestsMixin.test_statFilec s2 � � d� � j�dd�}� fdd�}|�|� |S )zE
Check the output of the stat method on a directory.
r� r� r� c s � � | d � d S )Nr� )r� r r r r r� T
s z2IFTPShellTestsMixin.test_statDirectory.<locals>.cb�r� r r� rj r� r r r �test_statDirectoryN
s
z&IFTPShellTestsMixin.test_statDirectoryc s2 � � d� � j�dd�}� fdd�}|�|� |S )z3
Check the owner and groups stats.
r� r� �r� r\ c s � � t| �d� d S )Nr= rd r r r r r� `
s z3IFTPShellTestsMixin.test_statOwnerGroup.<locals>.cbr r� r r r �test_statOwnerGroupZ
s
z'IFTPShellTestsMixin.test_statOwnerGroupc sV | j j� dd� �� �fdd�}|| j _| �d� | j �dd�}| �| �|�dg� d S )
z�
If L{twisted.python.filepath.FilePath.getNumberOfHardLinks} is not
implemented, the number returned is 0
c S s t �d S r r� r r r r �raiseNotImplementedm
s zQIFTPShellTestsMixin.test_statHardlinksNotImplemented.<locals>.raiseNotImplementedc s � | �}�|_ |S r )ZgetNumberOfHardLinks�rF r# �ZpathFuncr r r �notImplementedFilePathp
s zTIFTPShellTestsMixin.test_statHardlinksNotImplemented.<locals>.notImplementedFilePathr� r� )Z hardlinksr N�r �_pathr� r� rq ZsuccessResultOf�r r r5 r r r � test_statHardlinksNotImplementedf
s
z4IFTPShellTestsMixin.test_statHardlinksNotImplementedc sX | j j� dd� �� �fdd�}|| j _| �d� | j �dd�}| �| �|�ddg� d S )
z�
If L{twisted.python.filepath.FilePath.getUserID} or
L{twisted.python.filepath.FilePath.getGroupID} are not implemented,
the owner returned is "0" and the group is returned as "0"
c S s t �d S r r� r r r r r �
s zRIFTPShellTestsMixin.test_statOwnerGroupNotImplemented.<locals>.raiseNotImplementedc s � | �}�|_ �|_|S r )Z getUserIDZ
getGroupIDr r r r r �
s zUIFTPShellTestsMixin.test_statOwnerGroupNotImplemented.<locals>.notImplementedFilePathr� r� r �0Nr r r r r �!test_statOwnerGroupNotImplemented|
s
z5IFTPShellTestsMixin.test_statOwnerGroupNotImplementedc C s | j �dd�}| �|tj�S )zf
stat should fail with L{ftp.FileNotFoundError} on a file that doesn't
exist.
r� r� )r r� r� r r� r� r r r �test_statNotExisting�
s z(IFTPShellTestsMixin.test_statNotExistingc C s$ | � d� | j�dd�}| �|t�S )r� r� r� r� )r� r r� r� r� r� r r r �test_invalidStat�
s
z$IFTPShellTestsMixin.test_invalidStatc s. � � d� � j�dd�}� fdd�}|�|�S )z,
Try to rename a directory.
r� r� r� c s$ � � � �d�� � �� �d�� d S )Nr� r� )r� r� r� r r r r r� �
s z+IFTPShellTestsMixin.test_rename.<locals>.cb)r� r rR rj r� r r r � test_rename�
s
z IFTPShellTestsMixin.test_renamec C s | j �dd�}| �|tj�S )zl
Renaming a directory that doesn't exist should fail with
L{ftp.FileNotFoundError}.
r� r� )r rR r� r r� r� r r r �test_renameNotExisting�
s z*IFTPShellTestsMixin.test_renameNotExistingN)r2 )(r- r. r/ r� r� r� r� r� r� r� r[ r� rx r� r� rj r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r r r r r� sJ
r� c @ s: e Zd ZdZdd� Zdd� Zdd� Zdd � Zdd d �Zd
S )�
FTPShellTestsz/
Tests for the C{ftp.FTPShell} object.
c C s, t �| �� �| _| j�� t�| j�| _dS )zB
Create a root directory and instantiate a shell.
N)r r[ rW r� r� r r� r r r r r ro �
s
zFTPShellTests.setUpc C s | j �|��� S )z:
Test if the directory exists at C{path}.
)r� r� �isdirr� r r r r� �
s z FTPShellTests.directoryExistsc C s | j �|��� S )z0
Create a directory in C{path}.
)r� r� r� r� r r r r� �
s z FTPShellTests.createDirectoryc C s | j �|��� S )z5
Test if the file exists at C{path}.
)r� r� �isfiler� r r r r� �
s zFTPShellTests.fileExistsr2 c C s | j �|��|�S )z@
Create a file named C{path} with some content.
)r� r� �
setContentr� r r r r� �
s zFTPShellTests.createFileN)r2 ) r- r. r/ r� ro r� r� r� r� r r r r r �
s r c @ s, e Zd ZdZdZdd� Zdd� Zdd� ZdS ) � TestConsumerz�
A simple consumer for tests. It only works with non-streaming producers.
@ivar producer: an object providing
L{twisted.internet.interfaces.IPullProducer}.
Nc C s( | j dkst�g | _|| _ | j �� dS )z�
Simple register of producer, checks that no register has happened
before.
@param producer: pull producer to use
@param streaming: unused
N)�producerr� r3 �resumeProducing)r r Z streamingr r r �registerProducer�
s z TestConsumer.registerProducerc C s | j dk st�d| _ dS )zN
Unregister the producer, it should be done after a register.
N)r r� r r r r �unregisterProducer s z TestConsumer.unregisterProducerc C s | j �|� | j�� dS )zN
Save the data received.
@param data: data to append
N)r3 r% r r r) r r r rg s zTestConsumer.write)r- r. r/ r� r r r rg r r r r r �
s
r c @ s e Zd ZdZdd� Zdd� ZdS )� TestProducerz
A dumb producer.
c C s || _ || _dS )z�
@param toProduce: data to write
@type toProduce: C{str}
@param consumer: the consumer of data.
@type consumer: C{IConsumer}
N)� toProducer) )r r r) r r r r! s zTestProducer.__init__c C s | j �| j� dS )z+
Send the data to consume.
N)r) rg r r r r r �start# s zTestProducer.startN)r- r. r/ r� r! r r r r r r s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�IReadWriteTestsMixinzJ
Generic tests for the C{IReadFile} and C{IWriteFile} interfaces.
c C s
t � �dS )z
Return an object providing C{IReadFile}, ready to send data C{content}.
@param content: data to send
Nr� �r �contentr r r �
getFileReader0 s z"IReadWriteTestsMixin.getFileReaderc C s
t � �dS )zR
Return an object providing C{IWriteFile}, ready to receive data.
Nr� r r r r �
getFileWriter9 s z"IReadWriteTestsMixin.getFileWriterc C s
t � �dS )z6
Return the content of the file used.
Nr� r r r r �getFileContent@ s z#IReadWriteTestsMixin.getFileContentc s8 d�t � �� �fdd�}���fdd�� �����|�S )z�
Test L{ftp.IReadFile}: the implementation should have a send method
returning a C{Deferred} which fires when all the data has been sent
to the consumer, and the data should be correctly send to the consumer.
r� c s | � ���� �S r )r* rj )�reader)�cbSendr) r r r^ O s z-IReadWriteTestsMixin.test_read.<locals>.cbGetc s �� d�� j��� d S r1 )rq r� r3 r )r) r! r r r r& Q s z.IReadWriteTestsMixin.test_read.<locals>.cbSend)r r" rj �r r^ r )r& r) r! r r � test_readG s
z IReadWriteTestsMixin.test_readc s: d��fdd�}� �fdd����fdd�� �� � �|�S )a
Test L{ftp.IWriteFile}: the implementation should have a receive
method returning a C{Deferred} which fires with a consumer ready to
receive data to be written. It should also have a close() method that
returns a Deferred.
s elbbow
c s | � � �� | �S r )r rj )�writer)� cbReceiver r r^ ^ s z.IReadWriteTestsMixin.test_write.<locals>.cbGetc s4 t �| �}| �d d� |�� | �� |�� �� �S )NT)r r r r �closerj )r) r) r )�cbCloser! r r r* ` s
z2IReadWriteTestsMixin.test_write.<locals>.cbReceivec s �� ��� � � d S r )rq r$ r� )r! r r r r, f s z0IReadWriteTestsMixin.test_write.<locals>.cbClose)r# rj r' r )r, r* r! r r �
test_writeV s
z IReadWriteTestsMixin.test_writeN) r- r. r/ r� r" r# r$ r( r- r r r r r + s r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �FTPReadWriteTestsz�
Tests for C{ftp._FileReader} and C{ftp._FileWriter}, the objects returned
by the shell in C{openForReading}/C{openForWriting}.
c C s2 t �| �� �| _| j�� t�| j�| _d| _dS )z5
Create a temporary file used later.
r� N) r r[ rW r� r� r r� r r� r r r r ro r s
zFTPReadWriteTests.setUpc C s$ | j �| j��|� | j�| jf�S )zV
Return a C{ftp._FileReader} instance with a file opened for reading.
)r� r� r� r r r- r r r r r" | s z FTPReadWriteTests.getFileReaderc C s | j �| jf�S )zV
Return a C{ftp._FileWriter} instance with a file opened for writing.
)r r
r� r r r r r# � s z FTPReadWriteTests.getFileWriterc C s | j �| j��� S )z;
Return the content of the temporary file.
)r� r� r� Z
getContentr r r r r$ � s z FTPReadWriteTests.getFileContentN)r- r. r/ r� ro r" r# r$ r r r r r. l s
r. c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�CloseTestWriterz"
Close writing to a file.
Fc C s t � | _t�| j�}t�|�S )z>
Receive bytes.
@return: L{Deferred}
)r r3 r Z FileConsumerr
r| )r Zfcr r r r � s zCloseTestWriter.receivec C s d| _ | jS )z<
Close bytes.
@return: L{Deferred}
T)� closeStartedr5 r r r r r+ � s zCloseTestWriter.closeN)r- r. r/ r� r0 r r+ r r r r r/ � s r/ c @ s e Zd ZdZdd� ZdS )�CloseTestShellz
Close writing shell.
c C s t �| j�S r )r
r| r) )r Zsegsr r r r
� s z CloseTestShell.openForWritingN)r- r. r/ r� r
r r r r r1 � s r1 c @ s e Zd ZdZdd� ZdS )�
FTPCloseTestsz8
Tests that the server invokes IWriteFile.close
c C s� t �� }dg|_t� |_t� |j_t�� |jj_ t �
� |_ d|j _ |�
t� � t �� }t �|�|_ ||_|�
d� g }|�d�}|�|j� | �|jjjd� |�d� | �|jjjd� |�d� | �|jjjd� | �|� |jjj �d� | �|� |S ) z�
Confirm that FTP uploads (i.e. ftp_STOR) correctly call and wait
upon the IWriteFile object's close() method
r� NrF zclose() called earlys some data herezreason is ignoredzclose() not calledzallow close() to finish)r ZFTPZworkingDirectoryr1 r r/ r) r
r4 r5 rc r" ZtimeOutr� r r� r� r Zftp_STORrj r% r� r0 r7 r: r� r8 )r r# ZdiZ stor_doner5 r r r r- � s0
zFTPCloseTests.test_writeN)r- r. r/ r� r- r r r r r2 � s r2 c @ s e Zd ZdZdd� ZdS )�FTPResponseCodeTestsz4
Tests relating directly to response codes.
c C sr t tj�}t � }tt��� D ]P\}}t|t�r |�� r | �||d||f � | � ||d||f � |�
|� q dS )z�
All of the response code globals (for example C{RESTART_MARKER_REPLY} or
C{USR_NAME_OK_NEED_PASS}) have unique values and are present in the
C{RESPONSE} dictionary.
z0Code %r with value %r missing from RESPONSE dictz Duplicate code %r with value %rN) �setr ZRESPONSE�vars�items�
isinstancer� �isupperr� Z assertNotIn�add)r Z allValuesZ
seenValues�keyr� r r r � test_unique� s$
��
�z FTPResponseCodeTests.test_uniqueN)r- r. r/ r� r; r r r r r3 � s r3 )r; r< )Ur� rY r� �ior r� r� r� Zzope.interfacer Zzope.interface.verifyr Z twisted.credr r r Ztwisted.cred.errorr Ztwisted.cred.portalr Ztwisted.internetr
r r r
r Ztwisted.internet.interfacesr Ztwisted.protocolsr Ztwisted.pythonr r r Ztwisted.python.compatr Z twisted.testr Z
twisted.trialr r r �platformZ isWindowsr� Z LineReceiverr r� r0 rC ZTestCaserD r� r� r
r3 rv r� r� r� r� r� r� r� r� r| r� r� r� r� r� r� r� r� r r r r r r. r� r/ r1 r2 r3 r r r r �<module> s�
s RU * ; =x>%7 (B /++A' *