Your IP : 216.73.216.213
B
;��]� � @ 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 �H/opt/alt/python37/lib64/python3.7/site-packages/twisted/test/test_ftp.py�__init__) s zDummy.__init__c C s | j | _d S )N)�factory�f)r r r r �connectionMade. s zDummy.connectionMadec C s | j �|� d S )N)r �append)r �liner r r � lineReceived2 s zDummy.lineReceivedc C s | j �|� d S )N)r r# )r �datar r r �rawDataReceived6 s zDummy.rawDataReceivedc C s d S )Nr )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�d)r r r r r" @ s z!_BufferingProtocol.connectionMadec C s | j |7 _ d S )N)r. )r r& r r r � dataReceivedE s z _BufferingProtocol.dataReceivedc C s | j �| � d S )N)r0 �callback)r �reasonr r r �connectionLostI s z!_BufferingProtocol.connectionLostN)r) r* r+ r" r1 r4 r r r r r, ? s r, � 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 r= 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 z 127.0.0.1)� interfacec sD �j `� | �}|j�_�fdd�}��|� ��d � ��|� |S )Nc s � j jd k r� j j�� d S )N)�serverProtocol� transport�loseConnectionr )r r r �
cleanupServer� s zQFTPServerTestCase.setUp.<locals>._rememberProtocolInstance.<locals>.cleanupServer)r �
buildProtocol�wrappedProtocolrB �
addCleanupr2 r# )�addrr rE )rF �d1� protocolsr r r �_rememberProtocolInstance� s
z:FTPServerTestCase.setUp.<locals>._rememberProtocolInstancec s& | �_ ���j jj� � ��j � d S )N)�clientrH rC rD r# )rM )rK 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�
FTPFactoryr@ r r
� listenTCPr; rH �
stopListeningrF r
r/ �getHostr �
ClientCreator�
clientFactory�
connectTCP� addCallbackr ZwaitUntilAllDisconnected�
gatherResults) r �pZ
users_checkerr; rL �portNumZ
clientCreator�d2rN r )rF rJ rK r r �setUpe s>
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 )N)� assertEqual)�
responseLines)�expectedResponseLinesr r r � gotResponse� s zRFTPServerTestCase.assertCommandResponse.<locals>.queueCommand.<locals>.gotResponse)rM �queueStringCommandrb )�ignoredr0 rk )�commandrj r r r � queueCommand� s z=FTPServerTestCase.assertCommandResponse.<locals>.queueCommand)r
�succeedrb )r rn rj �
chainDeferredro r )rn rj r r �assertCommandResponse� s
z'FTPServerTestCase.assertCommandResponsec sP |d krt �d �}� �fdd�}|�|� ��|tj� ��fdd�}|�|�S )Nc s �j �� �S )N)rM rl )rm )rn r r r ro � s z;FTPServerTestCase.assertCommandFailed.<locals>.queueCommandc s � d k r�� � | jd � d S )Nr )rh �args)� exception)�expectedResponser r r �failed� s z5FTPServerTestCase.assertCommandFailed.<locals>.failed)r
rp rb �
assertFailurer �
CommandFailed)r rn ru rq ro rv r )rn ru r r �assertCommandFailed� s
z%FTPServerTestCase.assertCommandFailedc C s | � ddg�}| j ddg|d�S )NzUSER anonymousz8331 Guest login ok, type your email address as password.zPASS test@twistedmatrix.comz2230 Anonymous login ok, access restrictions apply.)rq )rr )r r0 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, proceed)rq )rr rX rY )r r0 r r r �
_userLogin� s z FTPServerTestCase._userLogin)N)NN)
r) r* r+ �__doc__r �FTPClientBasicr` r@ rg rr ry rz r{ r r r r r> \ s A
r> 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 guestz8331 Guest login ok, type your email address as password.zPASS test@twistedmatrix.comz2230 Anonymous login ok, access restrictions apply.)rq )rr )r r0 r r r �test_anonymousLogin� s z%FTPAnonymousTests.test_anonymousLoginN)r) r* r+ r| r@ 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
� z 127.0.0.1c S s | j S )N)r0 )�protor r r rN 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.
)rh r. )r� )r r r �onConnectionLost s zEBasicFTPServerTests.test_tooManyConnections.<locals>.onConnectionLost)
r ZconnectionLimitr r_ r
r, ra r; r^ rb )r �ccr0 rN r� r )r r �test_tooManyConnections� s z+BasicFTPServerTests.test_tooManyConnectionsc
s� ddddddddd d
g
}d g}� fd d
�}� fdd�}g }x<|D ]4}� j �|�}� �|tj� |�||� |�|� q@W x.|D ]&}� j �|�}|�||� |�|� q~W tj|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)rs �
assertTrue�
startswith)rt rn ZfailureResponseLines)r 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� )�resultrn )r r r �checkPassResponse& s zDBasicFTPServerTests.test_NotLoggedInReply.<locals>.checkPassResponseT)�fireOnOneErrback) rM rl rw r rx rb r# r
� DeferredList)r ZloginRequiredCommandListZloginNotRequiredCommandListr� r� Z deferredsrn �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)ry )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.)ry )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.
zUSER fooZPASSz,500 Syntax error: PASS requires an argument.)rq )rM rl ry )r r0 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)rs �kwargsr r r �_fake_loginhandler] s z?BasicFTPServerTests.test_loginError.<locals>._fake_loginhandlerzUSER foozPASS barz5550 Requested action not taken: internal server error)rq c s. � � � }� �dt|�� � �|d jt� d S )Nr� r )�flushLoggedErrorsrh �len�assertIsInstance�valuer� )r� �logs)r r r � checkLogsg s z6BasicFTPServerTests.test_loginError.<locals>.checkLogs)rB r ZloginrM rl ry rb )r r� r0 r� r )r r �test_loginErrorY s
z#BasicFTPServerTests.test_loginErrorc C s | � � S )z/
Login with userid 'anonymous'
)rz )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.)rq )rz rr )r r0 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
FzUSER anonymousz$331 Password required for anonymous.zPASS test@twistedmatrix.comz!530 Sorry, Authentication failed.)rq r� z$530 Please login with USER and PASS.)
r ZallowAnonymousr rW r rV r rZ rr ry )r ZdenyAlwaysCheckerr0 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 filesystem)rq )rz ry )r r0 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 implemented)rq )rz ry )r r0 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
zRETR foozE503 Incorrect sequence of commands: PORT or PASV required before RETR)rq )rz ry )r r0 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 STOR)rq )rz ry )r r0 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'.)rq zSTRU Iz&504 Not implemented for parameter 'I'.)rz ry )r r0 r r r �test_BadCommandArgs� s z'BasicFTPServerTests.test_BadCommandArgsc C s` | � t�d�d� td�}x@td�D ]4}t|�}d||< d�tt|��}| �t tj|� q$W dS )z%
Decode a host port.
z25,234,129,22,100,23)z
25.234.129.22id � � �,N)
rh r �decodeHostPortr �list�join�map�str� assertRaises�
ValueError)r �nums�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� )rM rl )�_)r r r �<lambda> r- 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� rh rB ZdtpPortr^ r; )ri r: r; )r r r �cb s z)BasicFTPServerTests.test_PASV.<locals>.cbc s � j j�� S )N)rB rC rD )r� )r r r r� r- )rz rb )r r0 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: L8)rq )rz rr )r r0 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.
�foozRNFR fooz6350 Requested file action pending further information.)rq zRNTO barz&250 Requested File Action Completed OKc s" � � � j�� j��d��� � | S )N�bar)r� rT �childrX �exists)r� )r r r � check_rename- s z:BasicFTPServerTests.test_RNFRandRNTO.<locals>.check_rename)rT r� rX �createDirectory�touchr{ rr rb )r r0 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.
zRNFR fooz6350 Requested file action pending further information.)rq zOTHER don-tcarez<503 Incorrect sequence of commands: RNTO required after RNFR)rz rr ry )r r0 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 )N)� RuntimeError)�
portNumberr r r r �
listenFactoryM s zEBasicFTPServerTests.test_portRangeForwardError.<locals>.listenFactoryN)rB 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)iV iV iV Z localhostr )r �CannotListenError)r� r r r r r� \ s z9BasicFTPServerTests.test_portRange.<locals>.listenFactoryr iV i iV N) rB r� r� r r� rh r �passivePortRanger� r r� )r r� r; 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� rF rh rG )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� )rh �assertIn)ri )r r r rk � s z2BasicFTPServerTests.test_FEAT.<locals>.gotResponse)rM rl rb )r r0 rk 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.)rq )rz ry )r r0 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� )rM rl )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)rB �
dtpFactoryr� r2 �object� dtpInstance)r� )r r r �mockDTPInstance� s
zJBasicFTPServerTests.test_STORreturnsErrorFromOpen.<locals>.mockDTPInstancez STOR folderz550 folder: is a directory)rq )rT r� rX r� r{ rb ry )r r0 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
r� 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� )rM rl )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)rB r� r� r2 r� r� �shell�openForWriting)r� )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 N)r� rh r� r� r� r� )r� r� )r r r r� � s zOBasicFTPServerTests.test_STORunknownErrorBecomesFileNotFound.<locals>.checkLogszSTOR somethingz)550 something: No such file or directory.)rq )r{ rb ry )r r0 r� 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 rx rh r� rs )�res)r r r �eb s z;FTPServerAdvancedClientTests.test_anonymousSTOR.<locals>.ebr� )rM � storeFile�
addErrbackr
rc )r r� rJ rf 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 �IsADirectoryError)r 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 )N)r
rp )�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 rx r� rh r� r� r� r� rs )r� r� )r r r r� " s zIFTPServerAdvancedClientTests.test_STORtransferErrorIsReturned.<locals>.ebr� ) r � _FileWriter�patch�FTPAnonymousShellrM r� r� r
rc )r r� r� rJ rf 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 )N)r
r� r� )r r r r r� 7 s ziFTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.FailingFileWriter.receiveN)r) r* r+ r� r r r r r� 6 s r� c s t �� d ��S )N)r
rp )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 rx r� rh r� r� r� r� rs )r� r� )r r r r� A s zRFTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.ebr� ) r r� r� r� rM r� r� r
rc )r r� r� rJ rf 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 )NZblah)r
r� r r� )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 )N)r
rp )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 rx rh r� rs )r )r r r �check_response_ s zGFTPServerAdvancedClientTests.test_RETRreadError.<locals>.check_responser� )r Z _FileReaderr� r� r, rM � retrieveFiler� )r r� r� r� r0 r )r� r r �test_RETRreadErrorN s
z/FTPServerAdvancedClientTests.test_RETRreadErrorN) r) r* r+ r| r � FTPClientr` r� r� r� r� 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� z 127.0.0.1)r r� r r_ r
r, ra )ri r: r; r� r r r �gotPASV} s zEFTPServerPasvDataConnectionTests._makeDataConnection.<locals>.gotPASV)rM rl rb )r rm r0 r� 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 )N)rM rl r0 r
rc )�
downloaderrJ rf )rn r r r ro � s z@FTPServerPasvDataConnectionTests._download.<locals>.queueCommandc S s | \}}|j S )N)r. )r� rm r r r r � downloadDone� s z@FTPServerPasvDataConnectionTests._download.<locals>.downloadDone)r
rp rb r )r rn rq ro r r )rn r r � _download� s
z*FTPServerPasvDataConnectionTests._downloadc s, � � � }� jd|d� � fdd�}|�|�S )zM
When listing empty folders, LIST returns an empty response.
r� )rq c s � � d| � d S )Nr- )rh )r� )r r r �
checkEmpty� s zCFTPServerPasvDataConnectionTests.test_LISTEmpty.<locals>.checkEmpty)rz r rb )r r0 r 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 -aL)rq c sV g }x$| � � D ]}|�|�d�d � qW � �dt|�� � �d|� � �d|� d S )N� r� � s foos bar)�
splitlinesr# �splitrh r� r� )�download�namesr$ )r r r �
checkDownload� s zOFTPServerPasvDataConnectionTests.test_LISTWithBinLsFlags.<locals>.checkDownload)rQ rR �pathr� rP rz r rb )r r0 r 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� )rq c s" � � dt| d d� �d��� d S )Nr ���s
)rh r� r )r
)r r r r � s zLFTPServerPasvDataConnectionTests.test_LISTWithContent.<locals>.checkDownloadzNLST c s. | d d� � d�}|�� � �ddg|� d S )Nr s
s bars foo)r �sortrh )r
� filenames)r r r r � s zLIST fooc s � � d| � d S )Nr- )rh )r
)r r r r � s c s � j �d�S )NzCWD foo)rM rl )rm )r r r �chdir� s zDFTPServerPasvDataConnectionTests.test_LISTWithContent.<locals>.chdirc s � � d| � d S )Nr- )rh )r
)r r r r � s )rQ rR r
r� rP rz r rb )r r0 r r 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 )N)r
rp )�segments�keys)�
listOutputr r � patchedList s zEFTPServerPasvDataConnectionTests._listTestHelper.<locals>.patchedListc s � �j j_| S )N)rB r� r� )r� )r r r r �loggedIn s
zBFTPServerPasvDataConnectionTests._listTestHelper.<locals>.loggedInz %s something)rq c s �� � | � �S )N)rh )r
)�expectedOutputr r r r r s zGFTPServerPasvDataConnectionTests._listTestHelper.<locals>.checkDownload)r )rz rb r )r rn r r r0 r r r )r r r 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� u
my resumér r� i� �user�groupsL drwxrwxrwx 0 user group 0 Jan 01 1970 my resumé
)r r � Permissions)r 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� s
my resumér r� i� r r sL drwxrwxrwx 0 user group 0 Jan 01 1970 my resumé
)r r r )r r r r �test_LISTNonASCIIBytes. s
z7FTPServerPasvDataConnectionTests.test_LISTNonASCIIBytesc
s� � � � }xxtddd�D ]h}ttj�� jd|f �d��}|�d| � W dQ R X � jd|f |d � |f� fd
d � }|� |� qW |S ) zJ
Download many large files.
@return: L{Deferred}
i�� i�� i� z%d.txt�wb� xNz RETR %d.txt)rq c s � � |t| �� d S )N)rh r� )r
�size)r r r r M s zOFTPServerPasvDataConnectionTests.test_ManyLargeDownloads.<locals>.checkDownload)
rz r �openrQ r
r� rP �writer rb )r r0 r" ZfObjr 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 )NzRETR foo)rC rD rM rl )r 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 rx rh r� rs r� r� )r Zcurrent_errors)r r r �
checkErrorh s zHFTPServerPasvDataConnectionTests.test_downloadFolder.<locals>.checkError)rT r� r� rz rb r r� )r r0 r& r' r( 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.
ztest.txtr� zNLST )rq c s. | d d� � d�}|�� � �ddg|� d S )Nr s
s foos test.txt)r r rh )r
r )r r r r � s zFFTPServerPasvDataConnectionTests.test_NLSTEmpty.<locals>.checkDownload)rz rT r� r� r� r rb )r r0 r 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.txt)rq c s � � d| � d S )Nr- )rh )r
)r r r r � s zLFTPServerPasvDataConnectionTests.test_NLSTNonexistent.<locals>.checkDownload)rz r rb )r r0 r 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.
�NLSTu
my resumér r� i� r r s my resumé
)r r 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.
r, s
my resumér r� i� r r s my resumé
)r r 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.
ztest.txtz
NLST test.txt)rq c s$ | d d� � d�}� �dg|� d S )Nr s
s test.txt)r rh )r
r )r r r r � s zMFTPServerPasvDataConnectionTests.test_NLSTOnPathToFile.<locals>.checkDownload)rz rT r� r� r rb )r r0 r r )r r �test_NLSTOnPathToFile� s
z6FTPServerPasvDataConnectionTests.test_NLSTOnPathToFile)N)N)r) r* r+ r| r r r r r r r r r% r) r* r+ r- r. r/ r r r r r� p s
-1
!
r� c @ s6 e Zd Zdd� Zd dd�Zdd� Zdd � Zd
d � ZdS )
� FTPServerPortDataConnectionTestsc C s g | _ t�| �S )N)� dataPortsr� rg )r r r r rg � 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 )Nr )r �
ServerFactoryrF r
Z callLaterr2 )r rI rd )r� r r rF � s zWFTPServerPortDataConnectionTests._makeDataConnection.<locals>.DataFactory.buildProtocolN)r) r* r+ r, r rF r )r� r r � DataFactory� s r3 r z 127.0.0.1)rA zPORT )r
r/ r r2 r
r\ r1 r# r r7 r^ r; rM rl )r rm r3 ZdataPort�cmdr )r� r r � 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
�
maybeDeferredr] )�.0r; r r r �
<listcomp>� s z=FTPServerPortDataConnectionTests.tearDown.<locals>.<listcomp>T)r� )r1 r
r5 r� �tearDownr# r� )r �lr0 r r r r8 � 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 z 127.0.0.1)rA c s � S )Nr )r� )re r r r� � r- z[FTPServerPortDataConnectionTests.test_PORTCannotConnect.<locals>.loggedIn.<locals>.<lambda>)r
r\ r r� r^ r; r] rb )rm r; r0 r )re r r � s
zIFTPServerPortDataConnectionTests.test_PORTCannotConnect.<locals>.loggedInc s � � dt�d| � dg�S )NzPORT z 127.0.0.1z 425 Can't open data connection.)ry r r7 )re )r r r �
gotPortNum� s zKFTPServerPortDataConnectionTests.test_PORTCannotConnect.<locals>.gotPortNum)rz rb )r r0 r 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.
ztest.txtzceva.txtzno.matchz
NLST *.txt)rq c s. | d d� � d�}|�� � �ddg|� d S )Nr s
s ceva.txts test.txt)r r rh )r
r )r r r r s zIFTPServerPortDataConnectionTests.test_nlstGlobbing.<locals>.checkDownload)rT r� r� rz r rb )r r0 r r )r r �test_nlstGlobbing s z2FTPServerPortDataConnectionTests.test_nlstGlobbing)N)r) r* r+ rg r r8 r; r< r r r r r0 � s
r0 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 rg 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)
rw r r� r �PortConnectionErrorrb r# �
setTimeoutr
�advancer� r� �calls)r �finishedr0 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 rF 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 rD rF r� r
rF )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.
rL N)
rw r r� r rC rD r
rE rJ rF )r r0 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.
rL N)
rw r r� r rC rb r# rD r� �clientConnectionFailedr� r
rE )r rG r0 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.
rL Nr� ) rw r r� r rC rD r
rE rO r
� TimeoutError)r r0 r r r �!test_connectionFailedAfterTimeout� s
z1DTPFactoryTests.test_connectionFailedAfterTimeoutN) r) r* r+ r| rg rH rK rM rN rP rR 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>.ProtocolInterpreterN)r) r* r+ r� r r r r r> � s r> N) r r? r
r� r@ r rA r r � StringTransportWithDisconnectionrC )r r> r r r rg � 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 contents
)r rF �makeConnectionrC �sendLiner� rh )r r� Z lineContentZdataSentr r r �test_sendLineNewline� s
z DTPTests.test_sendLineNewlineN)r) r* r+ r| rg rW r r r r rS � s rS c @ s e Zd ZdZdd� Zdd� ZdS )�
PrintLinesz3
Helper class used by FTPFileListingTests.
c C s
|| _ d S )N)�_lines)r r r r r r � s zPrintLines.__init__c C s4 x$| j D ]}| j�|�d�d � qW | j�� d S )Nzlatin-1s
)rY rC r$ r8 rD )r r$ r r r r" � s zPrintLines.connectionMadeN)r) r* r+ r| r r" r r r r rX � s rX c @ s e Zd Zdd� Zdd� ZdS )�MyFTPFileListProtocolc C s g | _ tj�| � d S )N)�otherr �FTPFileListProtocolr )r r r r r � s z MyFTPFileListProtocol.__init__c C s | j �|� d S )N)r[ r# )r r$ r r r � unknownLine� s z!MyFTPFileListProtocol.unknownLineN)r) r* r+ r r] r r r r rZ � s rZ 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 )N)�filesr[ )r� )�fileListr r r� � r- z6FTPFileListingTests.getFilesForLines.<locals>.<lambda>)rZ r �
loopbackAsyncrX rb )r r r0 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�-zmisparsed fileitem�permsz rw-r--r--zmisparsed perms�owner�rootr r[ r" i �datez Jan 29 03:26�filename�README�nlinksr� zmisparsed nlinks�
linktargetzmisparsed linktarget)r� �reprr� )Z fileOther�filer[ )r r r �check� s
z/FTPFileListingTests.test_OneLine.<locals>.check)rb rb )r r$ ro 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 linerc r0 zmisparsed fileitemre z rw-r--r--zmisparsed permsrf rg zmisparsed ownerr r[ zmisparsed groupr" i zmisparsed sizerh z Jan 9 2003zmisparsed dateri �Azmisparsed filenamerk r zmisparsed nlinksrl zmisparsed linktargetr9 r� z Jan 29 03:26�B)r� r� )r� Zfile1Zfile2r[ )r r r ro s( z4FTPFileListingTests.test_VariantLines.<locals>.check)rb rb )r �line1�line2Zline3ro 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� rm )r� r_ �others)r r r ro & s z3FTPFileListingTests.test_UnknownLine.<locals>.check�ABCz
not a file)rb rb )r ro 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 ri zmisparsed filenamezB Ar� zD C/A Brl zmisparsed linktarget)rh )r� r_ rv )r r r ro 9 s zBFTPFileListingTests.test_filenameWithUnescapedSpace.<locals>.check)rb rb )r rs rt ro 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 ) Nzunexpected others entrieszA Br ri zmisparsed filenamezB Ar� zD C/A Brl zmisparsed linktarget)rh )r� r_ rv )r r r ro P s z@FTPFileListingTests.test_filenameWithEscapedSpace.<locals>.check)rb rb )r rs rt ro 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 )N)rC r$ rD )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 r" i zmisparsed fileitemrh z Jan 29 2003ri rj )r_ r� )rm rn )r` r r r ro j s
z,FTPFileListingTests.test_Year.<locals>.check)r r\ r �Protocolr ra rb )r r| ro r0 r )r{ r` r r � test_Year\ s z FTPFileListingTests.test_YearN)
r) r* r+ rb rp ru rx ry rz 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 z 127.0.0.1)rA 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 )N)rX )rI )� responsesr r r� � r- zSFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_FailedRETR.<locals>.<lambda>r� )�passivec S s t �� }| �d|�S )Nz/file/that/doesnt/exist)r r} r� )rM rd r r r rN � s zTFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_FailedRETR.<locals>.gotClient)r r� Znoisyr
r\ rH r] r^ r; rF r_ r r� ra rb rw rx )r r! r; re r� r0 rN 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 )N)r# )r )�mr r �_eb� s
zZFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_errbacksUponDisconnect.<locals>._ebr )�CONNECTION_LOST)r r� r rT rU r r� r r� Ztwisted.internet.mainr� r4 r �Failurer� )r � ftpClient�trr0 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 r� rM r rT rC rU r )r r r r rg � 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)rM r4 r �ConnectionLost)r r r r r8 � 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.
r- 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)rh rC r� rM r% �clear)r 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
r- N� s
� és �
)rh rC r� rM rV 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 z&250 Requested File Action Completed OK)rh )r� )r r r �cbCdup� s z(FTPClientTests.test_CDUP.<locals>.cbCdups CDUP
s& 250 Requested File Action Completed OK) r� rM �cduprb rh rC r� r� r% )r r� r0 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}.
s CDUP
s! 550 ..: No such file or directory) r� rM r� rw r rx rh rC r� r� r% )r r0 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 z/bar/baz)rh r ZparsePWDResponse)r� )r r r �cbPwd s z&FTPClientTests.test_PWD.<locals>.cbPwds PWD
s 257 "/bar/baz")r� rM �pwdrb rh rC r� r% )r r� r0 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}.
s PWD
s' 550 /bar/baz: No such file or directory)
r� rM r� rw r rx rh rC r� r% )r r0 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 )Nr z&250 Requested File Action Completed OK)rh )r� )r r r �cbCwd7 s z&FTPClientTests.test_CWD.<locals>.cbCwdzbar/foos
CWD bar/foo
s& 250 Requested File Action Completed OK)r� rM �cwdrb rh rC r� r% )r r� r0 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}.
zbar/foos
CWD bar/foo
s& 550 bar/foo: No such file or directory)
r� rM r� rw r rx rh rC r� r% )r r0 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 )Ns� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)rh r. )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 )Nz 127.0.0.1i90 s4 150 File status okay; about to open data connection.s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr� )
rh rF rU r �StringTransportrM r% r1 r4 r r� r �ConnectionDone)r: r; r r� )r r r � cbConnect^ s
z2FTPClientTests.test_passiveRETR.<locals>.cbConnect�spams PASV
s RETR spam
s 226 Transfer Complete.) rM �connectFactoryr� r, r� rb rh rC r� r� r% r= )r r� r� r� r0 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 )NzPORT %sz 127.0.0.1i�& s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr� )
r r7 �textr rU r r� r1 r4 r r� r r� )�portCmdr r r � generatePort� s
z.FTPClientTests.test_RETR.<locals>.generatePortc s � � |jd� d S )Ns� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)rh r. )r� r� )r r r r� � s z(FTPClientTests.test_RETR.<locals>.cbRetrr� z PORT %s
z 127.0.0.1i�& s 200 PORT OKs RETR spam
s 226 Transfer Complete.)rM r� �generatePortCommandr� r, r� rb rh rC r� r r7 r8 r9 r� r% )r r� r� r� r0 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 )Nz 127.0.0.1i90 s4 150 File status okay; about to open data connection.r� ) rh rF rU r r� rM r% r4 r r� r r� )r: r; r r� )r r r r� � s z1FTPClientTests.test_failedRETR.<locals>.cbConnectr� s PASV
s RETR spam
s# 550 spam: No such file or directory)rM r� r� r, r� rw r rx rh rC r� r� r% r= )r r� r� r0 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 )NzPORT %sz 127.0.0.1i�& s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx) r r7 r� r rT r rU r1 r# )r� r� )r9 r r r� � s z2FTPClientTests.test_lostRETR.<locals>.generatePortr� z PORT %s
z 127.0.0.1i�& s 200 PORT OKs RETR spam
r )rM r� r� r� r, r� rh rC r� r r7 r8 r9 r� r% r� rD rw r� )r r� r� r0 r )r9 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 )Ns4 150 File status okay; about to open data connection.s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr� )
rM r% rC r$ �finishr4 r r� r r� )�sender)r r r �cbStore� s
z0FTPClientTests.test_passiveSTOR.<locals>.cbStorec s � � ��� d� d S )Ns� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)rh r� )�ign)r r� r r �cbFinish� s z1FTPClientTests.test_passiveSTOR.<locals>.cbFinishc s4 � � | d� � � |d� |�| |f�}|��� d S )Nz 127.0.0.1i90 )rh rF rU )r: r; r r� )r r� r r r� � s z2FTPClientTests.test_passiveSTOR.<locals>.cbConnectr� s PASV
s STOR spam
s 226 Transfer Complete.)r r� rM r� r� r� rb rh rC r� r� r% r= r
rc )r r� r� r� rJ rf r )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 )Ns4 150 File status okay; about to open data connection.s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr� )
rM r% rC r$ r� r4 r r� r r� )r� )r r r r� s
z/FTPClientTests.test_failedSTOR.<locals>.cbStorec s4 � � | d� � � |d� |�| |f�}|��� d S )Nz 127.0.0.1i90 )rh rF rU )r: r; r r� )r r� r r r� s z1FTPClientTests.test_failedSTOR.<locals>.cbConnectr� s PASV
s STOR spam
s. 426 Transfer aborted. Data connection closed.)r r� rM r� r� r� rb rw r rx rh rC r� r� r% r= r
rc )r r� r� rJ rf r )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 )NzPORT z 127.0.0.1i�& )r r7 r� r rU )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 )
Nz PORT %s
z 127.0.0.1i�& s 200 PORT OKs STOR spam
s4 150 File status okay; about to open data connection.s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr� s 226 Transfer Complete.)rh rC r� r r7 r8 rM r9 r� r% r$ r� r4 r r� r r� )r� )r r r r� ? s
z)FTPClientTests.test_STOR.<locals>.cbStorec s � � ��� d� d S )Ns� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx)rh r� )r� )r r� r r r� N s z*FTPClientTests.test_STOR.<locals>.cbFinishr� )
r r� rM r� r� r� r� rb r
rc )r r� r� r� rJ rf r )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 )ri r )r6 r! r r r r7 m s zCFTPClientTests.test_passiveLIST.<locals>.cbList.<locals>.<listcomp>r� r� �baz)r_ r rh )r� r` �fls�expected)r r r �cbListl s
z/FTPClientTests.test_passiveLIST.<locals>.cbListc s| � � | d� � � |d� |�| |f�}|�t�� � � j�d� dddg}x|D ]}|�|� qPW |�t �
t � d��� d S )Nz 127.0.0.1i90 s4 150 File status okay; about to open data connection.s8 -rw-r--r-- 0 spam egg 100 Oct 10 2006 foo
s8 -rw-r--r-- 3 spam egg 100 Oct 10 2006 bar
s8 -rw-r--r-- 4 spam egg 100 Oct 10 2006 baz
r� )
rh rF rU r r� rM r% r1 r4 r r� r r� )r: r; r r� �sendingr� )r r r r� s s
z2FTPClientTests.test_passiveLIST.<locals>.cbConnectzfoo/bars PASV
s LIST foo/bar
s 226 Transfer Complete.)
rM r� r� r r\ r� rb rh rC r� r� r% r= )r r� r� r` r0 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 sp dt �dd�f | _| j�t�� � � j�d� dddg}x|D ]}| j� |� q@W | j�
t � t
�d��� d S ) NzPORT %sz 127.0.0.1i�& s4 150 File status okay; about to open data connection.s8 -rw-r--r-- 0 spam egg 100 Oct 10 2006 foo
s8 -rw-r--r-- 3 spam egg 100 Oct 10 2006 bar
s8 -rw-r--r-- 4 spam egg 100 Oct 10 2006 baz
r� )r r7 r� r rU r r� rM r% r1 r4 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 )ri r )r6 r! r r r r7 � s z<FTPClientTests.test_LIST.<locals>.cbList.<locals>.<listcomp>r� r� r� )r_ r rh )r� r` r� r� )r r r r� � s
z(FTPClientTests.test_LIST.<locals>.cbListzfoo/barz PORT %s
z 127.0.0.1i�& s 200 PORT OKs LIST foo/bar
s 226 Transfer Complete.)rM r� r� r� r r\ r� rb rh rC r� r7 r8 r9 r� r% )r r� r� r` r0 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 )Nz 127.0.0.1i90 s4 150 File status okay; about to open data connection.r� ) rh rF rU r r� rM r% r4 r r� r r� )r: r; r r� )r r r r� � s z1FTPClientTests.test_failedLIST.<locals>.cbConnectzfoo/bars PASV
s LIST foo/bar
s& 550 foo/bar: No such file or directory)rM r� r� r r\ r� rw rx rh rC r� r� r% r= )r r� r` r0 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 ) NzPORT %sz 127.0.0.1i�& s4 150 File status okay; about to open data connection.s foo
s bar
s baz
r� )r r7 r� r rU r r� rM r% r1 r4 r r� r r� )r� )r r r r� � s z.FTPClientTests.test_NLST.<locals>.generatePortc s> |j �� jj��� }dddg}|�� |�� � �||� d S )Nr� r� r� )r. �decoderM r9 r r rh )r� r� r� r� )r r r r� � s
z(FTPClientTests.test_NLST.<locals>.cbListzfoo/barz PORT %s
z 127.0.0.1i�& s 200 PORT OKs NLST foo/bar
s 226 Transfer Complete.)rM r� r� r� r, �nlstrb rh rC r� r r7 r8 r9 r� r% )r r� r� �lstprotor0 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 )Ns foos bars baz)r. r r rh )r� r� r� r� )r r r r� s
z/FTPClientTests.test_passiveNLST.<locals>.cbListc sx � � | d� � � |d� |�| |f�}|�t�� � � j�d� |�d� |�d� |�d� |�t �
t � d��� d S )Nz 127.0.0.1i90 s4 150 File status okay; about to open data connection.s foo
s bar
s baz
r� )
rh rF rU r r� rM r% r1 r4 r r� r r� )r: r; r r� )r r r r� s
z2FTPClientTests.test_passiveNLST.<locals>.cbConnectzfoo/bars PASV
s NLST foo/bar
s 226 Transfer Complete.) rM r� r� r, r� rb rh rC r� r� r% r= )r r� r� r� r0 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 )Nz 127.0.0.1i90 s4 150 File status okay; about to open data connection.r� )
rh rF rU rM r% r4 r r� r r� )r: r; r r� )r r� r r r� 3 s
z1FTPClientTests.test_failedNLST.<locals>.cbConnectzfoo/bars PASV
s NLST foo/bar
s& 550 foo/bar: No such file or directory)r r� rM r� r� r, r� rw r rx rh rC r� r� r% r= )r r� r� r0 r )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.
z/spamz/hams RNFR /spam
z8350 Requested file action pending further information.
s RNTO /ham
z&250 Requested File Action Completed OK) r� rM �renamerh rC r� r� r% r8 r9 rb )r r0 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� rM r� r% rh rC 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).
z/spamz/hams RNFR /spam
s! 550 Requested file unavailable.
r- ) r� rM r� rh rC r� r� r% rw r rx )r r0 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).
z/spamz/hams RNFR /spam
s8 350 Requested file action pending further information.
s RNTO /ham
s! 550 Requested file unavailable.
) r� rM r� rh rC r� r� r% rw r rx )r r0 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.
z/spams MKD /spam
s 257 "/spam" created.z257 "/spam" created.)r� rM �
makeDirectoryrh rC r� r% rb )r r0 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� rM r� rh rC r� r% )r r0 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.
z/spams MKD /spam
s 550 PERMISSION DENIED)
r� rM r� rh rC r� r% rw r rx )r r0 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 )Nz/bar/baz)rh )r� )r r r �cbGet� s z/FTPClientTests.test_getDirectory.<locals>.cbGets PWD
s 257 "/bar/baz")r� rM � getDirectoryrb rh rC r� r% )r r� r0 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.
s PWD
s' 550 /bar/baz: No such file or directory)
r� rM r� rw r rx rh rC r� r% )r r0 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.
s PWD
s 257 /bar/baz)
r� rM r� rw r rx rh rC r� r% )r r0 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.
z /tmp/tests DELE /tmp/test
z*250 Requested file action okay, completed.)
r� rM �
removeFilerh rC r� r% r8 r9 rb )r r0 �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}.
z /tmp/tests DELE /tmp/test
z,501 Syntax error in parameters or arguments.c s �� | j� gf�S )N)rh rs )�exc)r� r r r r�
r- z6FTPClientTests.test_failedRemoveFile.<locals>.<lambda>)
r� rM r� rh rC r� r% r8 r9 rw r rx rb )r r0 r )r� r 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.
z /tmp/testz765 blah blah blahc s �� | j� gf�S )N)rh rs )r� )r� r r r r�
r- zBFTPClientTests.test_unparsableRemoveFileResponse.<locals>.<lambda>)
r� rM r� r% r8 r9 rw r � BadResponserb )r r0 r )r� r 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.
z /tmp/tests 250-perhaps a progress reports 250 okay)r� rM r� r% rb r� )r r0 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.
z /tmp/tests RMD /tmp/test
z*250 Requested file action okay, completed.)
r� rM �removeDirectoryrh rC r� r% r8 r9 rb )r r0 r� 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}.
z /tmp/tests RMD /tmp/test
z,501 Syntax error in parameters or arguments.c s �� | j� gf�S )N)rh rs )r� )r� r r r r� F
r- z;FTPClientTests.test_failedRemoveDirectory.<locals>.<lambda>)
r� rM r� rh rC r� r% r8 r9 rw r rx rb )r r0 r )r� r 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.
z /tmp/testz765 blah blah blahc s �� | j� gf�S )N)rh rs )r� )r� r r r r� U
r- zGFTPClientTests.test_unparsableRemoveDirectoryResponse.<locals>.<lambda>)
r� rM r� r% r8 r9 rw r r� rb )r r0 r )r� r 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.
z /tmp/tests 250-perhaps a progress reports 250 okay)r� rM r� r% rb r� )r r0 r r r �%test_multilineRemoveDirectoryResponseY
s
z4FTPClientTests.test_multilineRemoveDirectoryResponseN)-r) r* r+ r| rg r8 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� r� r� r� 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.
s 220 Imaginary FTP.z220 Imaginary FTP.N)r r} r% rh �greeting)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 N)r r} r% rh r� )r 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
s 220 Imaginary FTP.ZBLAHs 210-First line.s 123-Second line.s Just some text.s Hir- s 321s 210 Done.z210-First line.z123-Second line.zJust some text.ZHir� Z321z 210 Done.r N) r r} r r� rC r% rl rb r# r� r� rh )r r� r� r� r r r �test_MultilineResponse~
s8
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.
s 220 Welcome to Imaginary FTP.�bobNs
USER bob
s 200 Hello bob.r- )
r r} r r� rC r% �
queueLoginrh r� 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.
s 220 Welcome to Imaginary FTP.r� Zsecrets
USER bob
s# 230 Hello bob. No password needed.r- N)
r r} r r� rC r% r� rh r� r� )r 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: xbddgfddgfdg fddddgfddddgfddgfd dgfgD ] \}}| � t�g |�|� qDW xvdddgfd
ddgfd dgfd dgfd
dddgfddddgfdddgfdddgfgD ] \}}| � t�dg|�|� q�W xFdg fddgfddgfdddgfgD ] \}}| � t�g |�|� �qW xBdg fddgfddddgfgD ] \}}| � t�dg|�|� �qDW x�dg fdg fddgfdg fd dgfd dgfd ddgfd ddgfd dgfd!dgfd"g fd#g fg D ] \}}| � t�dg|�|� �q�W x"d$D ]}| �tjtjg |� �q�W x$d%D ] }| �tjtjdg|� �qW 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//cz..z../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/../) z..z../za/../..za/../../z/..z/../z/a/../..z /a/../../z
/a/b/../../..)z../..z../../z
../a/../..N)rh r Z
toSegmentsr� Z InvalidPath)r ZinpZoutpr r r �test_Normalizer�
sV
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 �_isGlobbingExpression)r 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� �exprz*.txtN)r� r r� )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� z*.txtz [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 � BaseFTPRealm)r 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 )N)r# )r ZavatarId)�avatarsr� r r �getHomeDirectoryW s
zKBaseFTPRealmTests.test_getHomeDirectory.<locals>.TestRealm.getHomeDirectoryN)r) r* r+ r� r )r� r� r r � TestRealmV s r� zalice@example.comN) r rS rO r r� �
requestAvatar� IFTPShellr� �FTPShellrh �filesystemRoot)r r� �realm�iface�avatar�logoutr )r� 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)
rO r r� r� r Z ANONYMOUSr� r� r� rh r� r rS )r r? 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� rO 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/homezalice@example.comN)r rU rO r� rh r rS 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.
zalice@example.comz/home/alice@example.comN)r rU rO r� rh r rS )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 �SystemFTPRealmrO r� rh r rS )r r r� r� 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 )N)�random�choice�string�
ascii_letters�digits)r6 r� r r r � <genexpr>� s z6SystemFTPRealmTests.test_noSuchUser.<locals>.<genexpr>r� N) r
r r r r� r r r rO 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�errno�ENOENTrw �FileNotFoundError)r r0 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 �EPERMrw �PermissionDeniedError)r r0 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 �EACCESrw r )r r0 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 �ENOTDIRrw �IsNotADirectoryError)r r0 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 �EEXISTrw �FileExistsError)r r0 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 ZEISDIRrw r� )r r0 r r r �test_isDirectory� s z$ErrnoToFailureTests.test_isDirectoryc C s0 y t d��W n t�dd�}| �|t �S 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 rw )r r0 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� rw r )r r� r0 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 r
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}
N)r )r 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}
N)r )r r
r r r �
fileExists. s z IFTPShellTestsMixin.fileExistsr- 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}
N)r )r r
� 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}.
zfile.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� r� rb )r r0 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� r� rw r r% )r r0 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 )Nr� )r� r, )r� )r r r r� y s z4IFTPShellTestsMixin.test_removeDirectory.<locals>.cb)r� r� r� rb )r r0 r� r )r r r� 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}.
zfile.txt)zfile.txt)r/ r� r� rw r r" )r r0 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� r� rw r r )r r0 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.
zfile.txt)zfile.txtc s � � � �d�� d S )Nzfile.txt)r� r- )r� )r r r r� � s z/IFTPShellTestsMixin.test_removeFile.<locals>.cb)r/ r� r� rb )r r0 r� r )r r r� � s
z#IFTPShellTestsMixin.test_removeFilec C s$ | � d� | j�d�}| �|tj�S )z:
removeFile should not work on directory.
�ned)r5 )r� r� r� rw r r� )r r0 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� r� rw r r )r r0 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.
r5 zfile.txt)�.c s$ | � � � �| dg fdg fg� d S )Nzfile.txtr5 )r rh )r9 )r r r r� � s z)IFTPShellTestsMixin.test_list.<locals>.cb)r� r/ r� r� rb )r r0 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.
r5 zfile.txt)r8 )r" Z permissionsc st | � � � �t| �d� � �| d d d� � �| d d d� � �t| d d �d� � �t| d d �d� d S )Nr r zfile.txtr� r5 )r rh r� )r9 )r r r r� � s z1IFTPShellTestsMixin.test_listWithStat.<locals>.cb)r� r/ r� r� rb )r r0 r� r )r r �test_listWithStat� s
z%IFTPShellTestsMixin.test_listWithStatc C s$ | � d� | j�dd�}| �|t�S )zP
Querying an invalid stat should result to a C{AttributeError}.
r5 )r8 )r" � whateverstat)r� r� r� rw �AttributeError)r r0 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.
zfile.txt)zfile.txtc s | � � � �| dg fg� d S )Nzfile.txt)r rh )r9 )r r r r� � s z-IFTPShellTestsMixin.test_listFile.<locals>.cb)r/ r� r� rb )r r0 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� rw r r )r r0 r r r � test_listNotExistingDirectory� s z1IFTPShellTestsMixin.test_listNotExistingDirectoryc C s | � d� | j�d�}|S )z+
Try to access a resource.
r5 )r5 )r� r� �access)r r0 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@ rw r r )r r0 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}.
zfile.txt)zfile.txtc s � � tj�| �� d S )N)r� r Z IReadFile�
providedBy)r� )r r r r�
s z3IFTPShellTestsMixin.test_openForReading.<locals>.cb)r/ r� r� rb )r r0 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.
)r5 )r� r� rw r r )r r0 r r r �test_openForReadingNotFound
s z/IFTPShellTestsMixin.test_openForReadingNotFoundc C s$ | � d� | j�d�}| �|tj�S )z>
openForReading should not work on directory.
r5 )r5 )r� r� r� rw r r� )r r0 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 )N)r� r �
IWriteFilerC r� rb )r� )�cb2r r r �cb1#
s z4IFTPShellTestsMixin.test_openForWriting.<locals>.cb1c s � � t�| �� d S )N)r� r rC )r� )r r r rH &
s z4IFTPShellTestsMixin.test_openForWriting.<locals>.cb2)r� r� rb )r r0 rI r )rH 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.
r5 )r5 )r� r� r� rw r r� )r r0 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.
r5 )r5 Z idonotexistr� )r� r� r� rw r r )r r0 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.
s wobble
zfile.txt)zfile.txt)r" rP c s& �� | d t� �� ��| d � d S )Nr r� )rh r� r� )r� )r. r r r r� G
s z-IFTPShellTestsMixin.test_statFile.<locals>.cb)r/ r� �statrb )r r0 r� 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.
r5 )r5 )r" rP c s � � | d � d S )Nr� )r� )r� )r r r r� T
s z2IFTPShellTestsMixin.test_statDirectory.<locals>.cb)r� r� rM rb )r r0 r� r )r r �test_statDirectoryN
s
z&IFTPShellTestsMixin.test_statDirectoryc s2 � � d� � j�dd�}� fdd�}|�|� |S )z3
Check the owner and groups stats.
r5 )r5 )rf r c s � � t| �d� d S )Nr )rh r� )r� )r r r r� `
s z3IFTPShellTestsMixin.test_statOwnerGroup.<locals>.cb)r� r� rM rb )r r0 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 )N)r r r r r �raiseNotImplementedm
s zQIFTPShellTestsMixin.test_statHardlinksNotImplemented.<locals>.raiseNotImplementedc s � | �}�|_ |S )N)ZgetNumberOfHardLinks)r
r! )�pathFuncrQ r r �notImplementedFilePathp
s zTIFTPShellTestsMixin.test_statHardlinksNotImplemented.<locals>.notImplementedFilePathr5 )r5 )Z hardlinksr N)r� �_pathr� rM rh �successResultOf)r rS r0 r )rR rQ 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 )N)r r r r r rQ �
s zRIFTPShellTestsMixin.test_statOwnerGroupNotImplemented.<locals>.raiseNotImplementedc s � | �}�|_ �|_|S )N)Z getUserIDZ
getGroupID)r
r! )rR rQ r r rS �
s zUIFTPShellTestsMixin.test_statOwnerGroupNotImplemented.<locals>.notImplementedFilePathr5 )r5 )rf r �0N)r� rT r� rM rh rU )r rS r0 r )rR rQ 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" rP )r� rM rw r r )r r0 r r r �test_statNotExisting�
s z(IFTPShellTestsMixin.test_statNotExistingc C s$ | � d� | j�dd�}| �|t�S )zP
Querying an invalid stat should result to a C{AttributeError}.
r5 )r5 )r" r; )r� r� rM rw r< )r r0 r r r �test_invalidStat�
s
z$IFTPShellTestsMixin.test_invalidStatc s. � � d� � j�dd�}� fdd�}|�|�S )z,
Try to rename a directory.
r5 )r5 )r� c s$ � � � �d�� � �� �d�� d S )Nr� r5 )r� r, r� )r� )r r r r� �
s z+IFTPShellTestsMixin.test_rename.<locals>.cb)r� r� r� rb )r r0 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� r� rw r r )r r0 r r r �test_renameNotExisting�
s z*IFTPShellTestsMixin.test_renameNotExistingN)r- )(r) r* r+ r| r, r� r- r/ r0 r1 r� r2 r� r3 r4 r� r6 r7 r9 r: r= r> r? rA rB rD rE rF rJ rK rL rN rO rP rV rX rY rZ 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 rS rO rg r� r r� r� )r r r r rg �
s
zFTPShellTests.setUpc C s | j �|��� S )z:
Test if the directory exists at C{path}.
)rg r� �isdir)r r
r r r r, �
s z FTPShellTests.directoryExistsc C s | j �|��� S )z0
Create a directory in C{path}.
)rg 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}.
)rg r� �isfile)r r
r r r r- �
s zFTPShellTests.fileExistsr- c C s | j �|��|�S )z@
Create a file named C{path} with some content.
)rg r� �
setContent)r r
r. r r r r/ �
s zFTPShellTests.createFileN)r- ) r) r* r+ r| rg 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� r. �resumeProducing)r rb 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)rb 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)r. r# rb rc )r r& r r r r$ s zTestConsumer.write)r) r* r+ r| rb rd re r$ r r r r ra �
s
ra 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 rg r� r r r r s zTestProducer.__init__c C s | j �| j� dS )z+
Send the data to consume.
N)r� r$ rg )r r r r �start# s zTestProducer.startN)r) r* r+ r| r rh r r r r rf s rf 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
N)r )r �contentr r r �
getFileReader0 s z"IReadWriteTestsMixin.getFileReaderc C s
t � �dS )zR
Return an object providing C{IWriteFile}, ready to receive data.
N)r )r r r r �
getFileWriter9 s z"IReadWriteTestsMixin.getFileWriterc C s
t � �dS )z6
Return the content of the file used.
N)r )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.
s wobble
c s | � ���� �S )N)r� rb )�reader)�cbSendr� r r r� O s z-IReadWriteTestsMixin.test_read.<locals>.cbGetc s �� d�� j��� d S )Nr- )rh r� r. )r� )r� rj r r r ro Q s z.IReadWriteTestsMixin.test_read.<locals>.cbSend)ra rk rb )r r� r )ro r� rj 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 )N)r� rb )�writer)� cbReceiver r r� ^ s z.IReadWriteTestsMixin.test_write.<locals>.cbGetc s4 t �| �}| �d d� |�� | �� |�� �� �S )NT)rf rd rh re �closerb )r� rq rb )�cbCloserj r r rr ` s
z2IReadWriteTestsMixin.test_write.<locals>.cbReceivec s �� ��� � � d S )N)rh rm )rm )rj r r r rt f s z0IReadWriteTestsMixin.test_write.<locals>.cbClose)rl rb )r r� r )rt rr rj r r �
test_writeV s
z IReadWriteTestsMixin.test_writeN) r) r* r+ r| rk rl rm rp ru r r r r ri + s ri 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.
zfile.txtN) r rS rO rg r� r r� r� ri )r r r r rg r s
zFTPReadWriteTests.setUpc C s$ | j �| j��|� | j�| jf�S )zV
Return a C{ftp._FileReader} instance with a file opened for reading.
)rg r� ri r` r� r� )r rj r r r rk | s z FTPReadWriteTests.getFileReaderc C s | j �| jf�S )zV
Return a C{ftp._FileWriter} instance with a file opened for writing.
)r� r� ri )r r r r rl � s z FTPReadWriteTests.getFileWriterc C s | j �| j��� S )z;
Return the content of the temporary file.
)rg r� ri Z
getContent)r r r r rm � s z FTPReadWriteTests.getFileContentN)r) r* r+ r| rg rk rl rm r r r r rv l s
rv 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 r. r Z FileConsumerr
rp )r Zfcr r r r� � s zCloseTestWriter.receivec C s d| _ | jS )z<
Close bytes.
@return: L{Deferred}
T)� closeStartedr0 )r r r r rs � s zCloseTestWriter.closeN)r) r* r+ r| rx r� rs r r r r rw � s rw c @ s e Zd ZdZdd� ZdS )�CloseTestShellz
Close writing shell.
c C s t �| j�S )N)r
rp rq )r Zsegsr r r r� � s z CloseTestShell.openForWritingN)r) r* r+ r| r� r r r r ry � s ry 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
rg Nr
zclose() called earlys some data herezreason is ignoredzclose() not calledzallow close() to finish)r ZFTPZworkingDirectoryry r� rw rq r
r/ r0 r[ r ZtimeOutrU r rI rA r� Zftp_STORrb r# r� rx r1 r4 r� r2 )r r! ZdiZ stor_doner0 r r r ru � s0
zFTPCloseTests.test_writeN)r) r* r+ r| ru r r r r rz � s rz c @ s e Zd ZdZdd� ZdS )�FTPResponseCodeTestsz4
Tests relating directly to response codes.
c C sv t tj�}t � }x`tt��� D ]P\}}t|t�r |�� r | �||d||f � | � ||d||f � |�
|� q W 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 r{ � s r{ )r5 r6 )Ur| rQ 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} r, r= ZTestCaser> r~ r� r� r� r0 r= rS rX r\ rZ r^ r r� r� r� r� r� r r r r) r+ r] r� ra rf ri rv rG rw ry rz r{ r r r r �<module> s�
s RU * ; =x>%7 (B /++A' *