0x1949 Team - FAZEMRX - MANAGER
Edit File: test_ftp.cpython-310.pyc
o �b� � @ s( d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z ddl m Z ddl m Z mZmZ ddlmZ ddlmZ dd lmZmZmZmZmZ dd lmZ dd l m Z m Z m Z dd l m!Z!m"Z"m#Z# dd l$m%Z% ddl&m'Z' e#j(�)� szdZ*ndZ*G dd� de j+�Z,G dd� dej-�Z.dVdd�Z/G dd� de'�Z0G dd� de0�Z1G d d � d e0�Z2G d d � d e0�Z3G d d!� d!e0�Z4G d"d#� d#e4�Z5G d$d%� d%e'�Z6G d&d'� d'e'�Z7G d(d)� d)ej-�Z8G d*d+� d+e j9�Z:G d,d-� d-e'�Z;G d.d/� d/e'�Z<G d0d1� d1e'�Z=G d2d3� d3e'�Z>G d4d5� d5e'�Z?G d6d7� d7e'�Z@G d8d9� d9e'�ZAG d:d;� d;e'�ZBG d<d=� d=e'�ZCG d>d?� d?e'�ZDG d@dA� dAe'�ZEG dBdC� dC�ZFG dDdE� dEe'eF�ZGe e�G dFdG� dG��ZHG dHdI� dI�ZIG dJdK� dK�ZJG dLdM� dMe'eJ�ZKe e jL�G dNdO� dO��ZMG dPdQ� dQ�ZNG dRdS� dSe'�ZOG dTdU� dUe'�ZPdS )Wz FTP tests. � N)�BytesIO)� implementer)� verifyClass)�checkers� credentials�portal)�UnauthorizedLogin)�IRealm)�defer�error�protocol�reactor�task)� IConsumer)�basic�ftp�loopback)�failure�filepath�runtime)� proto_helpers)�TestCasezCannot run on Windowsc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � ZdS ) �DummyNc C s g | _ g | _d S �N)�lines�rawData��self� r �7/usr/lib/python3/dist-packages/twisted/test/test_ftp.py�__init__% � zDummy.__init__c C s | j | _d S r )�factory�fr r r r �connectionMade) � zDummy.connectionMadec C � | j �|� d S r )r �append�r �liner r r � lineReceived, � zDummy.lineReceivedc C r& r )r r' �r �datar r r �rawDataReceived/ r+ zDummy.rawDataReceivedc C s d S r r r( r r r �lineLengthExceeded2 � zDummy.lineLengthExceeded) �__name__� __module__� __qualname__�lognamer r$ r* r. r/ r r r r r " s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_BufferingProtocolc C s d| _ t�� | _d S �N� )�bufferr �Deferred�dr r r r r$ 7 s z!_BufferingProtocol.connectionMadec C s | j |7 _ d S r �r8 r, r r r � dataReceived; � z _BufferingProtocol.dataReceivedc C s | j �| � d S r )r: �callback)r �reasonr r r �connectionLost> r+ z!_BufferingProtocol.connectionLostN)r1 r2 r3 r$ r< r@ r r r r r5 6 s r5 � 127.0.0.1�90 c C s dt �||�� d�}|�| 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 z227 Entering Passive Mode (z).)r �encodeHostPort�encode� _encoding)r �host�port�msgr r r �passivemode_msgB s rI 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)� anonymousRoot�userHomez test-userz test-password)r � userAnonymousr rA �� interfacec sD �j `� | �}|j�_�fdd�}��|� ��d � ��|� |S )Nc s � j jd ur� j j�� d S d S r ��serverProtocol� transport�loseConnectionr r r r � cleanupServer� s �zQFTPServerTestCase.setUp.<locals>._rememberProtocolInstance.<locals>.cleanupServer)r" � buildProtocol�wrappedProtocolrR � addCleanupr> r' )�addrr rU �rV �d1� protocolsr r r �_rememberProtocolInstance{ s z:FTPServerTestCase.setUp.<locals>._rememberProtocolInstancec s& | �_ ���j jj� � ��j � d S r )�clientrX rS rT r' )r^ )r\ r r r � gotClient� s z*FTPServerTestCase.setUp.<locals>.gotClient))�mktemp� directory�os�mkdirr �FilePath�dirPathr �Portalr �FTPRealm�registerCheckerr �AllowAnonymousAccessr � IAnonymous�'InMemoryUsernamePasswordDatabaseDontUse�username�password�addUser�IUsernamePassword� FTPFactoryrN r" r � listenTCPrG rX � stopListeningrV r r9 �getHostr � ClientCreator� clientFactory� connectTCP� addCallbackr �waitUntilAllDisconnected� gatherResults) r �p� users_checkerrG r] �portNum� clientCreator�d2r_ r rZ r �setUpY s< �� zFTPServerTestCase.setUpNc s, |du r t �d�}� ��fdd�}|�|�S )z� Asserts that a sending an FTP command receives the expected response. Returns a Deferred. Optionally accepts a deferred to chain its actions to. Nc s$ �j �� �}��fdd�}|�|�S )Nc s �� � | � d S r �� assertEqual�� responseLines)�expectedResponseLinesr r r � gotResponse� r+ zRFTPServerTestCase.assertCommandResponse.<locals>.queueCommand.<locals>.gotResponse�r^ �queueStringCommandrw )�ignoredr: r� ��commandr� r r r � queueCommand� s z=FTPServerTestCase.assertCommandResponse.<locals>.queueCommand)r �succeedrw )r r� r� � chainDeferredr� r r� r �assertCommandResponse� s z'FTPServerTestCase.assertCommandResponsec sP |d u r t �d �}� �fdd�}|�|� ��|tj� ��fdd�}|�|�S )Nc s �j �� �S r �r^ r� �r� �r� r r r r� � r% z;FTPServerTestCase.assertCommandFailed.<locals>.queueCommandc s" � d ur�� � | jd � d S d S �Nr �r� �args)� exception)�expectedResponser r r �failed� s �z5FTPServerTestCase.assertCommandFailed.<locals>.failed)r r� rw � assertFailurer � CommandFailed)r r� r� r� r� r� r )r� r� r r �assertCommandFailed� s z%FTPServerTestCase.assertCommandFailedc C � | � ddg�}| j ddg|d�S )N�USER anonymous�8331 Guest login ok, type your email address as password.�PASS test@twistedmatrix.com�2230 Anonymous login ok, access restrictions apply.�r� �r� �r r: r r r �_anonymousLogin� s ��z!FTPServerTestCase._anonymousLoginc C s2 | � d| j d| j g�}| j d| j dg|d�S )zx Authenticates the FTP client using the test account. @return: L{Deferred} of command response zUSER %sz 331 Password required for %s.zPASS %sz230 User logged in, proceedr� )r� rl rm r� r r r � _userLogin� s ��z FTPServerTestCase._userLoginr )NN) r1 r2 r3 �__doc__r �FTPClientBasicru rN r r� r� r� r� r r r r rJ O s C rJ 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" �guestc C r� )a Tests whether the changing of the anonymous username is working or not. The FTP server should not comply about the need of password for the username 'guest', letting it login as anonymous asking just an email address as password. z USER guestr� r� r� r� r� r� r r r �test_anonymousLogin� s ��z%FTPAnonymousTests.test_anonymousLoginN)r1 r2 r3 r� rN 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 � rA c S s | j S r )r: ��protor r r r_ 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. �r� r8 r� r r r �onConnectionLost s �zEBasicFTPServerTests.test_tooManyConnections.<locals>.onConnectionLost) r" �connectionLimitr rt r r5 rv rG rs rw )r �ccr: r_ r� r r r �test_tooManyConnections� s z+BasicFTPServerTests.test_tooManyConnectionsc s� g d�}dg}� fdd�}� fdd�}g }|D ]}� j �|�}� �|tj� |�||� |�|� q|D ]}� j �|�}|�||� |�|� q4tj|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. ) �CDUP�CWD�LIST�MODE�PASV�PWD�RETR�STRU�SYST�TYPE�FEATc s0 | j d }� �|d �d�d||d f � d S )Nr ����530z'%s - Response didn't start with 530: %r)r� � assertTrue� startswith)r� r� �failureResponseLinesr 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� )�resultr� r r r �checkPassResponse0 s ���zDBasicFTPServerTests.test_NotLoggedInReply.<locals>.checkPassResponseT��fireOnOneErrback) r^ r� r� r r� rw r' r � DeferredList)r �loginRequiredCommandList�loginNotRequiredCommandListr� r� � deferredsr� �deferredr r r �test_NotLoggedInReply s z)BasicFTPServerTests.test_NotLoggedInReplyc C � | � ddg�S )z@ Issuing PASS before USER should give an error. zPASS fooz=503 Incorrect sequence of commands: USER required before PASS�r� r r r r �test_PASSBeforeUSERI s �z'BasicFTPServerTests.test_PASSBeforeUSERc C r� )zD Issuing USER without a username is a syntax error. �USERz,500 Syntax error: USER requires an argument.r� r r r r �test_NoParamsForUSERR s �z(BasicFTPServerTests.test_NoParamsForUSERc C s | j �d�}| jddg|d�S )zD Issuing PASS without a password is a syntax error. �USER foo�PASSz,500 Syntax error: PASS requires an argument.r� )r^ r� r� r� r r r �test_NoParamsForPASSZ s �z(BasicFTPServerTests.test_NoParamsForPASSc sF dd� }|� j j_� j�d�}� jddg|d� |j� fdd��}|S ) zI Unexpected exceptions from the login handler are caught c _ s t �td��S )Nztest exception�r �fail�AssertionError)r� �kwargsr r r �_fake_loginhandlerh � z?BasicFTPServerTests.test_loginError.<locals>._fake_loginhandlerr� zPASS barz5550 Requested action not taken: internal server errorr� c s. � � � }� �dt|�� � �|d jt� d S )Nr� r ��flushLoggedErrorsr� �len�assertIsInstance�valuer� �r� �logsr r r � checkLogss s z6BasicFTPServerTests.test_loginError.<locals>.checkLogs)rR r �loginr^ r� r� rw )r r� r: r� r r r �test_loginErrorc s �z#BasicFTPServerTests.test_loginErrorc C s | � � S )z/ Login with userid 'anonymous' )r� r r r r �test_AnonymousLogin{ s z'BasicFTPServerTests.test_AnonymousLoginc C � | � � }| jddg|d�S )zm Issuing QUIT should return a 221 message. @return: L{Deferred} of command response �QUITz 221 Goodbye.r� �r� r� r� r r r � test_Quit� s z BasicFTPServerTests.test_Quitc C sX d| j _t�� }| j j�|tj� | �ddg�}| j ddg|d�}| j ddg|d�}|S ) z� Reconfigure the server to disallow anonymous access, and to have an IUsernamePassword checker that always rejects. @return: L{Deferred} of command response Fr� z$331 Password required for anonymous.r� z!530 Sorry, Authentication failed.r� r� z$530 Please login with USER and PASS.) r" �allowAnonymousr rk r rh r ro r� r� )r �denyAlwaysCheckerr: r r r �test_AnonymousLoginDenied� s" ����z-BasicFTPServerTests.test_AnonymousLoginDeniedc C r� )z� When an anonymous user attempts to edit the server-side filesystem, they will receive a 550 error with a descriptive message. @return: L{Deferred} of command response z MKD newdirz:550 Anonymous users are forbidden to change the filesystemr� �r� r� r� r r r �test_anonymousWriteDenied� s �z-BasicFTPServerTests.test_anonymousWriteDeniedc C r� )z\ Send an invalid command. @return: L{Deferred} of command response � GIBBERISHz'502 Command 'GIBBERISH' not implementedr� r� r� r r r �test_UnknownCommand� s �z'BasicFTPServerTests.test_UnknownCommandc C r� )zb Send RETR before sending PORT. @return: L{Deferred} of command response �RETR foozE503 Incorrect sequence of commands: PORT or PASV required before RETRr� r� r� r r r �test_RETRBeforePORT� � ��z'BasicFTPServerTests.test_RETRBeforePORTc C r� )zb Send STOR before sending PORT. @return: L{Deferred} of command response zSTOR foozE503 Incorrect sequence of commands: PORT or PASV required before STORr� r� r� r r r �test_STORBeforePORT� r� z'BasicFTPServerTests.test_STORBeforePORTc C s0 | � � }| jddg|d� | jddg|d� |S )zd Send command with bad arguments. @return: L{Deferred} of command response zMODE zz&504 Not implemented for parameter 'z'.r� zSTRU Iz&504 Not implemented for parameter 'I'.r� r� r r r �test_BadCommandArgs� s ��z'BasicFTPServerTests.test_BadCommandArgsc C s\ | � t�d�d� td�}td�D ]}t|�}d||< d�tt|��}| �t tj|� qdS )z% Decode a host port. z25,234,129,22,100,23)z 25.234.129.22id � � �,N) r� r �decodeHostPort�range�list�join�map�str� assertRaises� ValueError)r �nums�i�badValue�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 � � j �d�S )Nr� r� ��_r r r �<lambda> � 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� r� rR �dtpPortrs rG )r� rF rG r r r �cb s z)BasicFTPServerTests.test_PASV.<locals>.cbc s � j j�� S r rQ r r r r r r �r� rw �r r: r r r r � test_PASV� s z BasicFTPServerTests.test_PASVc C � | � � }| jddg|d� |S )z? SYST command will always return UNIX Type: L8 r� z215 UNIX Type: L8r� r� r� r r r � test_SYST s z BasicFTPServerTests.test_SYSTc sp � j �� j��� � j �� j��d��� � �� }� jddg|d� � jddg|d� � fdd�}|�|� |S ) z� Sending the RNFR command followed by RNTO, with valid filenames, will perform a successful rename operation. �foo�RNFR foo�6350 Requested file action pending further information.r� zRNTO bar�&250 Requested File Action Completed OKc s" � � � j�� j��d��� � | S �N�bar)r� re �childrl �exists�r� r r r � check_rename1 s z:BasicFTPServerTests.test_RNFRandRNTO.<locals>.check_rename)re r rl �createDirectory�touchr� r� rw )r r: r r r r �test_RNFRandRNTO s �� z$BasicFTPServerTests.test_RNFRandRNTOc C s0 | � � }| jddg|d� | jddg|d� |S )z� Sending the RNFR command followed by any command other than RNTO should return an error informing users that RNFR should be followed by RNTO. r r r� zOTHER don-tcarez<503 Incorrect sequence of commands: RNTO required after RNFR)r� r� r� r� r r r �test_RNFRwithoutRNTO8 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 � t � �r )� RuntimeError�� portNumberr" r r r � listenFactoryQ s zEBasicFTPServerTests.test_portRangeForwardError.<locals>.listenFactoryN)rR r' r r$ � getDTPPortr �Factory)r r' r r r �test_portRangeForwardErrorK 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 | dv r t �d| d��| S )N)�V iV iV � localhostr )r �CannotListenErrorr% r r r r' b s z9BasicFTPServerTests.test_portRange.<locals>.listenFactoryr r+ i iV N) rR r' r( r r) r� r� �passivePortRanger r r- )r r' rG r r r �test_portRangeZ 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. rV r� rW )r � portRanger r r r �"test_portRangeInheritedFromFactoryu s z6BasicFTPServerTests.test_portRangeInheritedFromFactoryc �" � 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� )r� �assertInr� r r r r� � s z2BasicFTPServerTests.test_FEAT.<locals>.gotResponser� )r r: r� r r r � test_FEAT� s z BasicFTPServerTests.test_FEATc C r )z� When the server receives 'OPTS something', it should report that the FTP server does not support the option called 'something'. zOPTS somethingz'502 Option 'something' not implemented.r� r� r� r r r � test_OPTS� s �z BasicFTPServerTests.test_OPTSc st � j �� j��� � j �� j��d��� � �� }� fdd�}� fdd�}|�|� |�|� � jddg|d� |S ) zl Any FTP error raised inside STOR while opening the file is returned to the client. �folderc r )zE Send the PASV command required before port. r� r� r r r r �sendPASV� s zCBasicFTPServerTests.test_STORreturnsErrorFromOpen.<locals>.sendPASVc s � j jj�d� t� � j _| S )z� Fake an incoming connection and create a mock DTPInstance so that PORT command will start processing the request. N)rR � dtpFactoryr� r> �object� dtpInstancer r r r �mockDTPInstance� s zJBasicFTPServerTests.test_STORreturnsErrorFromOpen.<locals>.mockDTPInstancez STOR folderz550 folder: is a directoryr� )re r rl r r� rw r� )r r: r7 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 � t �t� �S )z� Override openForWriting. @param ignore: ignored, used for callback @return: an error r� )�ignorer r r �failingOpenForWriting� s z[BasicFTPServerTests.test_STORunknownErrorBecomesFileNotFound.<locals>.failingOpenForWritingc r )z Send the PASV command required before port. @param result: parameter used in L{Deferred} r� r� r r r r r7 � 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)rR r8 r� r> r9 r: �shell�openForWritingr �r? r r r r; � s zUBasicFTPServerTests.test_STORunknownErrorBecomesFileNotFound.<locals>.mockDTPInstancec s. � � � }� �dt|�� � �|d jt� dS )zy Check that unknown errors are logged. @param result: parameter used in L{Deferred} r� r Nr� r� r r r r� � s zOBasicFTPServerTests.test_STORunknownErrorBecomesFileNotFound.<locals>.checkLogszSTOR somethingz)550 something: No such file or directory.r� )r� rw r� )r r: r7 r; r� r rB r �(test_STORunknownErrorBecomesFileNotFound� s � z<BasicFTPServerTests.test_STORunknownErrorBecomesFileNotFoundN) r1 r2 r3 r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r! r" r* r/ r1 r4 r5 r< rC r r r r r� � s8 8 $r� c @ s6 e Zd ZdZejZdd� Zdd� Zdd� Z dd � Z d S ) � FTPServerAdvancedClientTestsz: Test FTP server with the L{ftp.FTPClient} class. c s4 � fdd�}� j �d�\}}|�|� t�||g�S )zl Try to make an STOR as anonymous, and check that we got a permission denied error. c s( | � tj� � �| jjd d d� d S )Nr z550 foo: Permission denied.)�trapr r� r� r� r� ��resr r r �eb s z;FTPServerAdvancedClientTests.test_anonymousSTOR.<locals>.ebr )r^ � storeFile� addErrbackr ry )r rH r[ r~ r r r �test_anonymousSTOR s z/FTPServerAdvancedClientTests.test_anonymousSTORc �b 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 @ � e Zd Zdd� ZdS )zXFTPServerAdvancedClientTests.test_STORtransferErrorIsReturned.<locals>.FailingFileWriterc S � t �t�d��S )N� failing_file�r r� r �IsADirectoryErrorr r r r �receive r+ z`FTPServerAdvancedClientTests.test_STORtransferErrorIsReturned.<locals>.FailingFileWriter.receiveN�r1 r2 r3 rR r r r r �FailingFileWriter � rT c � t �� d ��S r �r r� ��a�b�rT r r � failingSTOR r� zRFTPServerAdvancedClientTests.test_STORtransferErrorIsReturned.<locals>.failingSTORrA c sT | � tj� � �� }� �dt|�� � �|d jtj� � �| jj d d d� d S )Nr� r z 550 failing_file: is a directory) rE r r� r� r� r� r� r� rQ r� �rG r� r r r rH & s zIFTPServerAdvancedClientTests.test_STORtransferErrorIsReturned.<locals>.ebrO � r � _FileWriter�patch�FTPAnonymousShellr^ rI rJ r ry �r r\ rH r[ r~ r �rT r r � test_STORtransferErrorIsReturned s z=FTPServerAdvancedClientTests.test_STORtransferErrorIsReturnedc rL ) 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 @ rM )zaFTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.FailingFileWriterc S r= r r� r r r r rR : r% ziFTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.FailingFileWriter.receiveNrS r r r r rT 9 rU rT c rV r rW rX r[ r r r\ = r� z[FTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.failingSTORrA c sR | � tj� � �� }� �dt|�� � �|d jt� � �| jj d d d� d S )Nr� r z.426 Transfer aborted. Data connection closed.) rE r r� r� r� r� r� r� r� r� r] r r r rH D s �zRFTPServerAdvancedClientTests.test_STORunknownTransferErrorBecomesAbort.<locals>.ebrO r^ rb r rc r �)test_STORunknownTransferErrorBecomesAbort1 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 @ rM )zJFTPServerAdvancedClientTests.test_RETRreadError.<locals>.FailingFileReaderc S rN )N�blahrP )r �consumerr r r �sendX r+ zOFTPServerAdvancedClientTests.test_RETRreadError.<locals>.FailingFileReader.sendN)r1 r2 r3 rh r r r r �FailingFileReaderW rU ri c rV r rW rX )ri r r � failingRETR[ r� 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� rE r r� r� r� r� �r r r r �check_responseb s � zGFTPServerAdvancedClientTests.test_RETRreadError.<locals>.check_responserO )r � _FileReaderr` ra r5 r^ � retrieveFilerJ )r rj rm r� r: r )ri r r �test_RETRreadErrorQ s z/FTPServerAdvancedClientTests.test_RETRreadErrorN) r1 r2 r3 r� r � FTPClientru rK rd re rp r r r r rD s rD 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 rA )r r� r rt r r5 rv )r� rF rG r� r r r �gotPASV� s zEFTPServerPasvDataConnectionTests._makeDataConnection.<locals>.gotPASVr� )r r� r: rs r r r �_makeDataConnectionv s z4FTPServerPasvDataConnectionTests._makeDataConnectionc sH |du r t �d�}|��j� � �fdd�}|�|� dd� }|�|�S )z� Download file. @param command: command to run @param chainDeferred: L{Deferred} used to queue commands. @return: L{Deferred} of command response Nc s �j �� �}| j}t�||g�S r )r^ r� r: r ry )� downloaderr[ r~ r� r r r� � s z@FTPServerPasvDataConnectionTests._download.<locals>.queueCommandc S s | \}}|j S r r; )r� r� ru r r r � downloadDone� s z@FTPServerPasvDataConnectionTests._download.<locals>.downloadDone)r r� rw rt )r r� r� r� rv r r� r � _download� s z*FTPServerPasvDataConnectionTests._downloadc �, � � � }� jd|d� � fdd�}|�|�S )zM When listing empty folders, LIST returns an empty response. r� r� c � � � d| � d S r6 r� r r r r � checkEmpty� r+ zCFTPServerPasvDataConnectionTests.test_LISTEmpty.<locals>.checkEmpty�r� rw rw )r r: rz r r r �test_LISTEmpty� s z/FTPServerPasvDataConnectionTests.test_LISTEmptyc sX t �t j�� jd�� t �t j�� jd�� � �� }� jd|d� � fdd�}|�|�S )z} LIST ignores requests for folder with names like '-al' and will list the content of current folder. r r zLIST -aLr� c sR g }| � � D ] }|�|�d�d � q� �dt|�� � �d|� � �d|� d S )N� r� � � foo� bar)� splitlinesr' �splitr� r� r3 )�download�namesr) r r r � checkDownload� s zOFTPServerPasvDataConnectionTests.test_LISTWithBinLsFlags.<locals>.checkDownload�rb rc �pathr� ra r� rw rw �r r: 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� r� c s" � � dt| d d� �d��� d S )Nr~ ���� )r� r� r� �r� r r r r� � s "zLFTPServerPasvDataConnectionTests.test_LISTWithContent.<locals>.checkDownload�NLST c �. | d d� � d�}|�� � �ddg|� d S )Nr� r� r� r �r� �sortr� �r� � filenamesr r r r� � � zLIST fooc ry r6 r� r� r r r r� � r+ c r )NzCWD foor� r� r r r �chdir� r% zDFTPServerPasvDataConnectionTests.test_LISTWithContent.<locals>.chdirc ry r6 r� r� r r r r� � r+ r� )r r: r� r� r r r �test_LISTWithContent� s" z5FTPServerPasvDataConnectionTests.test_LISTWithContentc s\ �� � }d �fdd� ���fdd�}|�|� �j|� d�|d� � ��fdd �}|�|�S ) a5 Exercise handling by the implementation of I{LIST} or I{NLST} of certain return values and types from an L{IFTPShell.list} implementation. This will issue C{command} and assert that if the L{IFTPShell.list} implementation includes C{listOutput} as one of the file entries then the result given to the client is matches C{expectedOutput}. @param command: Either C{b"LIST"} or C{b"NLST"} @type command: L{bytes} @param listOutput: A value suitable to be used as an element of the list returned by L{IFTPShell.list}. Vary the values and types of the contents to exercise different code paths in the server's handling of this result. @param expectedOutput: A line of output to expect as a result of C{listOutput} being transformed into a response to the command issued. @type expectedOutput: L{bytes} @return: A L{Deferred} which fires when the test is done, either with an L{Failure} if the test failed or with a function object if it succeeds. The function object is the function which implements L{IFTPShell.list} (and is useful to make assertions about what warnings might have been emitted). @rtype: L{Deferred} r c s t �� g�S r rW )�segments�keys)� listOutputr r � patchedList r% zEFTPServerPasvDataConnectionTests._listTestHelper.<locals>.patchedListc s � �j j_| S r )rR r@ r� r )r� r r r �loggedIn � zBFTPServerPasvDataConnectionTests._listTestHelper.<locals>.loggedInz somethingr� c s �� � | � �S r r� r� )�expectedOutputr� r r r r� ( s zGFTPServerPasvDataConnectionTests._listTestHelper.<locals>.checkDownloadN)r )r� rw rw )r r� r� r� r: r� r� r )r� r� r� r r �_listTestHelper� s z0FTPServerPasvDataConnectionTests._listTestHelperc C �& | � ddddt�d�ddddffd�S ) z� Unicode filenames returned from L{IFTPShell.list} are encoded using UTF-8 before being sent with the response. r� � my resumér r� � �user�group�L drwxrwxrwx 0 user group 0 Jan 01 1970 my resumé �r� r � Permissionsr r r r �test_LISTUnicode. � ��z1FTPServerPasvDataConnectionTests.test_LISTUnicodec C r� ) z� When LIST receive a filename as byte string from L{IFTPShell.list} it will just pass the data to lower level without any change. @return: L{_listTestHelper} r� � my resumér r� r� r� r� r� r� r r r r �test_LISTNonASCIIBytes= s ��z7FTPServerPasvDataConnectionTests.test_LISTNonASCIIBytesc s� � � � }tddd�D ]>}ttj�� jd|f �d��}|�d| � W d � n1 s,w Y � jd|f |d � |f� fd d � }|� |� q |S ) zJ Download many large files. @return: L{Deferred} i�� i�� i� z%d.txt�wb� xNz RETR %d.txtr� c s � � |t| �� d S r �r� r� )r� �sizer r r r� ^ � zOFTPServerPasvDataConnectionTests.test_ManyLargeDownloads.<locals>.checkDownload) r� r� �openrb r� r� ra �writerw rw )r r: r� �fObjr� r r r �test_ManyLargeDownloadsN s � z8FTPServerPasvDataConnectionTests.test_ManyLargeDownloadsc sf � j �d��� � �� }|�� j� � fdd�}|�|� dd� }|�|� � fdd�}|�|� |S )zo When RETR is called for a folder, it will fail complaining that the path is a folder. r c s | j �� � j�d�}|S )Nr� )rS rT r^ r� )ru r� r r r � retrFoldero s zHFTPServerPasvDataConnectionTests.test_downloadFolder.<locals>.retrFolderc S s t d��)Nz(Downloading a folder should not succeed.)r� r r r r � failOnSuccessv 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.)rE r r� r� r� r� r� r� )r �current_errorsr r r � checkError{ s �zHFTPServerPasvDataConnectionTests.test_downloadFolder.<locals>.checkError)re r r r� rw rt rJ )r r: r� r� r� r r r �test_downloadFolderd s z4FTPServerPasvDataConnectionTests.test_downloadFolderc sL � � � }� j�d��� � j�d��� � jd|d� � fdd�}|�|�S )zp NLST with no argument returns the directory listing for the current working directory. �test.txtr r� r� c r� )Nr� r� r � test.txtr� r� r r r r� � r� zFFTPServerPasvDataConnectionTests.test_NLSTEmpty.<locals>.checkDownload)r� re r r r rw rw r� r r r �test_NLSTEmpty� s z/FTPServerPasvDataConnectionTests.test_NLSTEmptyc rx )zH NLST on a non-existent file/directory returns nothing. zNLST nonexistent.txtr� c ry r6 r� r� r r r r� � r+ zLFTPServerPasvDataConnectionTests.test_NLSTNonexistent.<locals>.checkDownloadr{ r� r r r �test_NLSTNonexistent� s z5FTPServerPasvDataConnectionTests.test_NLSTNonexistentc C r� ) zs NLST will receive Unicode filenames for IFTPShell.list, and will encode them using UTF-8. �NLSTr� r r� r� r� r� � my resumé r� r r r r �test_NLSTUnicode� r� z1FTPServerPasvDataConnectionTests.test_NLSTUnicodec C r� ) zJ NLST will just pass the non-Unicode data to lower level. r� r� r r� r� r� 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. r� z NLST test.txtr� c s$ | d d� � d�}� �dg|� d S )Nr� r� r� )r� r� r� r r r r� � s zMFTPServerPasvDataConnectionTests.test_NLSTOnPathToFile.<locals>.checkDownload)r� re r r rw rw r� r r r �test_NLSTOnPathToFile� s z6FTPServerPasvDataConnectionTests.test_NLSTOnPathToFiler )r1 r2 r3 r� rt rw r| r� r� r� r� r� r� r� r� r� r� r� r� r r r r rr q s" 51$ rr c @ s6 e Zd Zdd� Zd dd�Zdd� Zdd � Zd d � ZdS ) � FTPServerPortDataConnectionTestsc C s g | _ t�| �S r )� dataPortsrr r r r r r r � r! z&FTPServerPortDataConnectionTests.setUpNc sb t �� � G � fdd�dtj�}tjd|� dd�}| j�|� dt� d|� � j � }| j � |� � S )Nc s e Zd ZeZ� fdd�ZdS )zIFTPServerPortDataConnectionTests._makeDataConnection.<locals>.DataFactoryc s" t j�| |�}t�d� j|� |S r� )r � ServerFactoryrV r � callLaterr> )r rY rz �r� r r rV � s zWFTPServerPortDataConnectionTests._makeDataConnection.<locals>.DataFactory.buildProtocolN)r1 r2 r3 r5 r rV r r� r r � DataFactory� s r� r rA rO �PORT )r r9 r r� r rq r� r' r rC rs rG r^ r� )r r� r� �dataPort�cmdr r� r rt � 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 � maybeDeferredrr )�.0rG r r r � <listcomp>� s z=FTPServerPortDataConnectionTests.tearDown.<locals>.<listcomp>Tr� )r� r r� rr �tearDownr' r� )r �lr: r r r r� � s z)FTPServerPortDataConnectionTests.tearDownc s0 � � � }dd� }|�|� � fdd�}|�|�S )z� Listen on a port, and immediately stop listening as a way to find a port number that is definitely closed. c s< t jdt�� dd�}|�� j� |�� }|�� fdd�� |S )Nr rA rO c s � S r r r �r| r r r s z[FTPServerPortDataConnectionTests.test_PORTCannotConnect.<locals>.loggedIn.<locals>.<lambda>)r rq r r) rs rG rr rw )r� rG r: r r� r r� s zIFTPServerPortDataConnectionTests.test_PORTCannotConnect.<locals>.loggedInc s � � dt�d| � dg�S )Nr� rA z 425 Can't open data connection.)r� r rC r� r r r � gotPortNum s �zKFTPServerPortDataConnectionTests.test_PORTCannotConnect.<locals>.gotPortNumr )r r: 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. r� zceva.txtzno.matchz NLST *.txtr� c r� )Nr� r� s ceva.txtr� r� r� r r r r� $ r� zIFTPServerPortDataConnectionTests.test_nlstGlobbing.<locals>.checkDownload)re r r r� rw rw r� r r r �test_nlstGlobbing s z2FTPServerPortDataConnectionTests.test_nlstGlobbingr )r1 r2 r3 r rt r� r� r� r r r r r� � s r� c @ �H e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�DTPFactoryTestsz& Tests for L{ftp.DTPFactory}. c C s8 t �� | _G dd� d�}|� | _t�| jd| j�| _dS )zf Create a fake protocol interpreter and a L{ftp.DTPFactory} instance to test. c @ � e Zd ZdZdS )z2DTPFactoryTests.setUp.<locals>.ProtocolInterpreterN�r1 r2 r3 r: r r r r �ProtocolInterpreter8 � r� N)r �Clockr �protocolInterpreterr � DTPFactoryr" �r r� r r r r 1 s zDTPFactoryTests.setUpc C sl g }| � | jjtj�}|�|j� | j�d� | j� d� | � |� | j� d� | � |� | � | jj � dS )z� L{ftp.DTPFactory.setTimeout} uses the reactor passed to its initializer to set up a timed event to time out the DTP setup after the specified number of seconds. r� � r� N) r� r" r� r �PortConnectionErrorrw r' � setTimeoutr �advancer� r� �calls�r �finishedr: 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" rV r� r �DTP� assertIsNone)r r r r r �test_buildProtocolOnce[ s z&DTPFactoryTests.test_buildProtocolOncec C s* | j �d� | j �d� | �| jj� dS )z� If a timeout has been set up using L{ftp.DTPFactory.setTimeout}, it is cancelled by L{ftp.DTPFactory.buildProtocol}. � N)r" r� rV r� r r� r r r r �test_timeoutAfterConnectionf s z+DTPFactoryTests.test_timeoutAfterConnectionc C s@ | � | jjtj�}| j�d� | j�d� | �| j� d�� |S )z� If L{ftp.DTPFactory.buildProtocol} is called after the timeout specified by L{ftp.DTPFactory.setTimeout} has elapsed, L{None} is returned. r� N) r� r" r� r r� r� r r� r� rV r� r r r �test_connectionAfterTimeoutp s z+DTPFactoryTests.test_connectionAfterTimeoutc C s` g }| � | jjtj�}|�|j� | j�d� | �|� | j� dd� | � |� | j � d� |S )a L{ftp.DTPFactory.deferred} fails with L{PortConnectionError} when L{ftp.DTPFactory.clientConnectionFailed} is called. If the timeout specified with L{ftp.DTPFactory.setTimeout} expires after that, nothing additional happens. r� N) r� r" r� r r� rw r' r� r� �clientConnectionFailedr� r r� r� r r r �!test_timeoutAfterConnectionFailed� s z1DTPFactoryTests.test_timeoutAfterConnectionFailedc C sB | � | jjtj�}| j�d� | j�d� | j�dt � d�� |S )z� If L{ftp.DTPFactory.clientConnectionFailed} is called after the timeout specified by L{ftp.DTPFactory.setTimeout} has elapsed, nothing beyond the normal timeout before happens. r� Nr ) r� r" r� r r� r� r r� r� r � TimeoutErrorr� r r r �!test_connectionFailedAfterTimeout� s z1DTPFactoryTests.test_connectionFailedAfterTimeoutN) r1 r2 r3 r� r r� r� r� r� r� r� r r r r r� , s r� c @ � e Zd ZdZdd� Zdd� ZdS )�DTPTestszx Tests for L{ftp.DTP}. The DTP instances in these tests are generated using DTPFactory.buildProtocol() c C sB t �� | _G dd� d�}|� | _t�| jd| j�| _t�� | _ dS )z� Create a fake protocol interpreter, a L{ftp.DTPFactory} instance, and dummy transport to help with tests. c @ r� )z+DTPTests.setUp.<locals>.ProtocolInterpreterNr� r r r r r� � r� r� N) r r� r r� r r� r" r � StringTransportWithDisconnectionrS r� r r r r � 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 contentr� )r" rV �makeConnectionrS �sendLiner� r� )r r: � lineContent�dataSentr r r �test_sendLineNewline� s z DTPTests.test_sendLineNewlineN)r1 r2 r3 r� r r� r r r r r� � s r� c @ r� )� PrintLinesz3 Helper class used by FTPFileListingTests. c C s || _ d S r )�_lines)r r r r r r � s zPrintLines.__init__c C s0 | j D ] }| j�|�d�d � q| j�� d S )Nzlatin-1r� )r� rS r� rD rT r( r r r r$ � s zPrintLines.connectionMadeN)r1 r2 r3 r� r r$ r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�MyFTPFileListProtocolc C s g | _ tj�| � d S r )�otherr �FTPFileListProtocolr r r r r r � s z MyFTPFileListProtocol.__init__c C r& r )r� r' r( r r r � unknownLine� r+ z!MyFTPFileListProtocol.unknownLineN)r1 r2 r3 r r� r r r r r� � s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�FTPFileListingTestsc s, t � � t�t|�� �}|�� fdd�� |S )Nc s � j � jfS r )�filesr� r ��fileListr r r � r z6FTPFileListingTests.getFilesForLines.<locals>.<lambda>)r� r � loopbackAsyncr� rw )r r r: 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|�� �� � �|d dkd� � �|d dkd� � �|d d kd� � �|d d kd� � �|d d kd� � �|d dkd� � �|d dkd� � �|d dkd� � � |d d� d S )Nzunexpect unparsable lines: �filetype�-�misparsed fileitem�perms� rw-r--r--�misparsed perms�owner�rootr� r� r� � �date� Jan 29 03:26�filename�README�nlinksr� �misparsed nlinks� linktarget�misparsed linktarget)r� �reprr� )� fileOther�filer� r r r �check� s z/FTPFileListingTests.test_OneLine.<locals>.check�r rw )r r) r r r r � test_OneLine� s z FTPFileListingTests.test_OneLinec s. d}d}d}� fdd�}� � |||g��|�S )z Variant lines. z7drw-r--r-- 2 root other 531 Jan 9 2003 Az<lrw-r--r-- 1 root other 1 Jan 29 03:26 B -> Azwoohoo! c s� | \\}}\}� � |dkd� � � |d dkd� � � |d dkd� � � |d d kd � � � |d d kd� � � |d dkd� � � |d dkd� � � |d dkd� � � |d dkd� � �|d d � � � |d d kd� � � |d dkd� � � |d d kd � � � |d d kd� � � |d d kd� � � |d d kd� � � |d d kd� � � |d d kd� � � |d dkd � d S )!Nz woohoo! zincorrect other liner r: r r r r r r zmisparsed ownerr� r� zmisparsed groupr� r zmisparsed sizer z Jan 9 2003zmisparsed dater �A�misparsed filenamer r~ r r r r� r� r �B)r� r� )r� �file1�file2r� r r r r s( z4FTPFileListingTests.test_VariantLines.<locals>.checkr )r �line1�line2�line3r r r r �test_VariantLines s z%FTPFileListingTests.test_VariantLinesc s � fdd�}� � ddg��|�S )z Unknown lines. c s4 | \}}� � |d� � �|ddgkdt|� � d S )Nzunexpected file entrieszABC z not a file z incorrect unparsable lines: %s)r� r� r �r� r� �othersr r r r 1 s �z3FTPFileListingTests.test_UnknownLine.<locals>.check�ABCz not a filer )r r r r r �test_UnknownLine, s z$FTPFileListingTests.test_UnknownLinec �( 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 �\ | \}}� � g |d� � � d|d d d� � � d|d d d� � � d|d d d � d S � Nzunexpected others entrieszA Br r r zB Ar� zD C/A Br r r� r# r r r r E � zBFTPFileListingTests.test_filenameWithUnescapedSpace.<locals>.checkr �r r r r r r r � test_filenameWithUnescapedSpace; � � z3FTPFileListingTests.test_filenameWithUnescapedSpacec r' )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 r( r) r� r# r r r r X r* z@FTPFileListingTests.test_filenameWithEscapedSpace.<locals>.checkr r+ r r r � test_filenameWithEscapedSpaceN r- z1FTPFileListingTests.test_filenameWithEscapedSpacec sH t �� �d� G � fdd�dtj�}��fdd�}t�|� ��}|�|�S )z{ This example derived from bug description in issue 514. @return: L{Deferred} of command response s< -rw-r--r-- 1 root other 531 Jan 29 2003 README c s e Zd Z� fdd�ZdS )z0FTPFileListingTests.test_Year.<locals>.PrintLinec s | j �� � | j �� d S r )rS r� rT r �� exampleLiner r r$ k s z?FTPFileListingTests.test_Year.<locals>.PrintLine.connectionMadeN)r1 r2 r3 r$ r r/ r r � PrintLinej s r1 c sJ � j d }��|d dkd� ��|d dkd� ��|d dkd� d S ) Nr r� r r r z Jan 29 2003r r )r� r� )r� r )r r r r r o s z,FTPFileListingTests.test_Year.<locals>.check)r r� r �Protocolr r rw )r r1 r r: r )r0 r r r � test_Yeara s z FTPFileListingTests.test_YearN) r1 r2 r3 r r r" r& r, r. r3 r r r r r� � s " r� c @ r� )�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 rA rO z1220 ready, dude (vsFTPd 1.0.0: beat me, break me)z 331 Please specify the password.z 230 Login successful. Have fun.z200 Binary it is, then.z+227 Entering Passive Mode (127,0,0,1,%d,%d)� � z550 Failed to open file.c s t � �S r )r� )rY �� responsesr r r � s zSFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_FailedRETR.<locals>.<lambda>r� )�passivec S s t �� }| �d|�S )Nz/file/that/doesnt/exist)r r2 ro )r^ rz r r r r_ � s zTFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_FailedRETR.<locals>.gotClient)r r) �noisyr rq rX rr rs rG rV rt r rq rv rw r� r� )r r# rG r| r� r: r_ r r7 r �test_FailedRETR~ s( �� zAFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_FailedRETRc st t �� }t�� }|�|� ||_|�dt� �}g � � fdd�}|�|� ddl m } |� t � |�� | �� � � |S )zl Test the ftp command errbacks when a connection lost happens during the operation. z some pathc s � � | � d S r �r' rl ��mr r �_eb� r� zZFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_errbacksUponDisconnect.<locals>._ebr )�CONNECTION_LOST)r rq r r� r� r r� r rJ �twisted.internet.mainr@ r@ r �Failurer� )r � ftpClient�trr: r? r@ r r= r �test_errbacksUponDisconnect� s zMFTPClientFailedRETRAndErrbacksUponDisconnectTests.test_errbacksUponDisconnectN)r1 r2 r3 r� r; rE r r r r r4 y s $r4 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 rq r^ r r� rS r� r r r r r r � 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)r^ r@ r �ConnectionLostr r r r r� � s zFTPClientTests.tearDownc C sr | � | j�� d� | j�d� | � | j�� d� | j�� | j�d� | � | j�� d� | j�� | j�d� dS )z& Test the login part. r7 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)r� rS r� r^ r* �clearr r r r � _testLogin� s � zFTPClientTests._testLoginc C sz | � | j�� d� | j�d� | � | j�� d� | j�d� | � | j�� d� | j�� | j�d� | � | j�� d� dS )z5 Test encoding behaviour of sendLine r7 N� r� � és � )r� rS r� r^ r� rH 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 � � � | d d� d S �Nr r r� rF r r r �cbCdup� r� z(FTPClientTests.test_CDUP.<locals>.cbCdup� CDUP �& 250 Requested File Action Completed OK) rI r^ �cduprw r� rS r� rH r* )r rO r: 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}. rP s! 550 ..: No such file or directory) rI r^ rR r� r r� r� rS r� rH r* r� r r r �test_failedCDUP s z FTPClientTests.test_failedCDUPc �F � fdd�}� � � � j�� �|�}� �� j�� d� � j�d� |S )a Test the PWD command. L{ftp.FTPClient.pwd} should return a Deferred which fires with a sequence of one element which is a string representing the current working directory on the server. (XXX - This is a bad API) c s � � t�| d �d� d S )Nr �/bar/baz)r� r �parsePWDResponserF r r r �cbPwd s z&FTPClientTests.test_PWD.<locals>.cbPwd� PWD � 257 "/bar/baz")rI r^ �pwdrw r� rS r� r* )r rX r: r r r �test_PWD s zFTPClientTests.test_PWDc C �B | � � | 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}. rY �' 550 /bar/baz: No such file or directory) rI r^ r[ r� r r� r� rS r� r* r� r r r �test_failedPWD# � 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 rM rN r� rF r r r �cbCwd< r� z&FTPClientTests.test_CWD.<locals>.cbCwd�bar/foo� CWD bar/foo rQ )rI r^ �cwdrw r� rS r� r* )r ra r: r r r �test_CWD1 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}. rb rc s& 550 bar/foo: No such file or directory) rI r^ rd r� r r� r� rS r� r* r� r r r �test_failedCWDE 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 � � � |jd� d S �N�� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr� �rG r� r r r �cbRetr_ r= z/FTPClientTests.test_passiveRETR.<locals>.cbRetrc sd � � | d� � � |d� |�| |f�}|�t�� � � j�d� |�d� |�t � t � d��� d S )NrA rB �4 150 File status okay; about to open data connection.ri rJ � r� rV r� r �StringTransportr^ r* r< r@ r rB r �ConnectionDone�rF rG r" r� r r r � cbConnectb s � z2FTPClientTests.test_passiveRETR.<locals>.cbConnect�spam� PASV � RETR spam � 226 Transfer Complete.) r^ �connectFactoryrI r5 ro rw r� rS r� rH r* rI )r rk rq r� r: r r r �test_passiveRETRS s z FTPClientTests.test_passiveRETRc s� d� j _dd� }� fdd�}|� j _� �� t� }� j �d|�}|�||� � �� j� � d� t � dd ��� � 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 d� t�dd��| _| j�t�� � | j�d� | j� t � t � d��� d S )N�PORT {}rA �& ri rJ )�formatr rC �textr r� r rn r< r@ r rB r ro ��portCmdr r r � generatePort� s z.FTPClientTests.test_RETR.<locals>.generatePortc rg rh r� rj r r r rk � r= z(FTPClientTests.test_RETR.<locals>.cbRetrrr � PORT {} rA ry � 200 PORT OKrt ru )r^ r9 �generatePortCommandrI r5 ro rw r� rS r� rz r rC rD rE rH r* )r r~ rk r� r: r r r � test_RETRz 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 �Z � � | d� � � |d� |�| |f�}|�t�� � � j�d� |�t� t � d��� d S �NrA rB rl rJ � r� rV r� r rn r^ r* r@ r rB r ro rp r r r rq � � �z1FTPClientTests.test_failedRETR.<locals>.cbConnectrr rs rt s# 550 spam: No such file or directory)r^ rv rI r5 ro r� r r� r� rS r� rH r* rI )r rq r� r: r r r �test_failedRETR� s z FTPClientTests.test_failedRETRc s� d| j _g � � fdd�}|| j _| �� t� }| j �d|�}| �| j�� d� t � dd��� | 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 d� t�dd��| _t�� }| j�|� | j|_| j�d� � � |� d S )Nrx rA ry s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx) rz r rC r{ r r� r r� r< r' )r} rD �r� r r r~ � s z2FTPClientTests.test_lostRETR.<locals>.generatePortrr r rA ry r� rt r )r^ r9 r� rI r5 ro r� rS r� rz r rC rD rE rH r* r� rT r� rG )r r~ r� r: r r� r � test_lostRETR� s* �� z FTPClientTests.test_lostRETRc s� t �� �� fdd�}� �fdd�}� �fdd�}|� j_� �� � j�d�\}}|�|� |�|� � �� j� � d� � j� � � j� t � j�� � �� j� � d � � j� � � j� d � t �||g�S ) a� Test the STOR command: send a file and verify its content. L{ftp.FTPClient.storeFile} should return a two-tuple of Deferreds. The first of which should fire with a protocol instance when the data connection has been established and is responsible for sending the contents of the file. The second of which should fire when the upload has completed, the data connection has been closed, and the server has acknowledged receipt of the file. (XXX - storeFile should take a producer as an argument, instead, and only return a Deferred which fires when the upload has succeeded or failed). c �: � j �d� | j�d� | �� | �t�t� d��� d S �Nrl ri rJ � r^ r* rS r� �finishr@ r rB r ro ��senderr r r �cbStore� � � z0FTPClientTests.test_passiveSTOR.<locals>.cbStorec � � � ��� d� d S rh �r� r� ��ign�r rD r r �cbFinish� r� z1FTPClientTests.test_passiveSTOR.<locals>.cbFinishc �4 � � | d� � � |d� |�| |f�}|��� d S �NrA rB �r� rV r� rp r� r r rq � z2FTPClientTests.test_passiveSTOR.<locals>.cbConnectrr rs � STOR spam ru )r rn r^ rv rI rI rw r� rS r� rH r* rI r ry )r r� r� rq r[ 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 r� r� r� r� r r r r� r� z/FTPClientTests.test_failedSTOR.<locals>.cbStorec r� r� r� rp r� r r rq & r� z1FTPClientTests.test_failedSTOR.<locals>.cbConnectrr rs r� s. 426 Transfer aborted. Data connection closed.)r rn r^ rv rI rI rw r� r r� r� rS r� rH r* rI r ry )r r� rq r[ 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 )Nr� rA ry )r rC r{ r r� r| )rD r r r~ D s z.FTPClientTests.test_STOR.<locals>.generatePortc s� � � � j�� d�t�dd���� jj�� � j� � � j� d� � � � j�� d� � j� � � j� d� | j� d� | � � | � t�t�d��� � j� d � d S ) Nr rA ry r� r� rl ri rJ ru )r� rS r� rz r rC rD r^ rE rH r* r� r� r@ r rB r ro r� r r r r� H s" �� � z)FTPClientTests.test_STOR.<locals>.cbStorec r� rh r� r� r� r r r� [ r� z*FTPClientTests.test_STOR.<locals>.cbFinishrr ) r rn r^ r9 r� rI rI rw r ry )r r~ r� r� r[ r~ r r� r � test_STOR9 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 �8 dd� |j D �}g d�}|�� |�� � �||� d S )Nc S � g | ]}|d �qS �r r �r� r# r r r r� z � zCFTPClientTests.test_passiveLIST.<locals>.cbList.<locals>.<listcomp>�r r �baz�r� r� r� �rG r �fls�expectedr r r �cbListy � z/FTPClientTests.test_passiveLIST.<locals>.cbListc sv � � | d� � � |d� |�| |f�}|�t�� � � j�d� g d�}|D ]}|�|� q&|�t � t � d��� d S )NrA rB rl �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 rJ rm )rF rG r" r� �sendingr r r r rq � s � z2FTPClientTests.test_passiveLIST.<locals>.cbConnect�foo/barrs � LIST foo/bar ru ) r^ rv rI r r� r� rw r� rS r� rH r* rI )r r� rq r r: r r r �test_passiveLISTe s z FTPClientTests.test_passiveLISTc s� d� j _� fdd�}� fdd�}|� j _� �� t�� }� j �d|��||�}� �� j � � d� t� dd ��� � 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 sj d� t�dd��| _| j�t�� � � j� d� g d�}|D ]}| j� |� q | j� t � t�d��� d S )Nrx rA ry rl r� rJ �rz r rC r{ r r� r rn r^ r* r< r@ r rB r ro )r} r� r r r r r~ � s � z.FTPClientTests.test_LIST.<locals>.generatePortc r� )Nc S r� r� r r� r r r r� � r� z<FTPClientTests.test_LIST.<locals>.cbList.<locals>.<listcomp>r� r� r� r r r r� � r� z(FTPClientTests.test_LIST.<locals>.cbListr� r rA ry r� r� ru )r^ r9 r� rI r r� r� rw r� rS r� rz rC rD rE rH r* )r r~ r� r r: 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 r� r� r� rp r r r rq � r� z1FTPClientTests.test_failedLIST.<locals>.cbConnectr� rs r� �& 550 foo/bar: No such file or directory)r^ rv rI r r� r� r� r� r� rS r� rH r* rI )r rq r r: r r r �test_failedLIST� s z FTPClientTests.test_failedLISTc s� d� j _� fdd�}� fdd�}|� j _� �� t� }� j �d|��||�}� �� j� � d� t � dd ��� � 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 d� t�dd��| _| j�t�� � � j� d� | j� d� | j� d� | j� d� | j� t � t�d��� d S ) Nrx rA ry rl � foo � bar � baz rJ r� r| r r r r~ � s � z.FTPClientTests.test_NLST.<locals>.generatePortc s< |j �� jj��� }g d�}|�� |�� � �||� d S )Nr� )r8 �decoder^ rE r� r� r� �rG r� r� r� r r r r� s z(FTPClientTests.test_NLST.<locals>.cbListr� r rA ry r� � NLST foo/bar ru )r^ r9 r� rI r5 �nlstrw r� rS r� rz r rC rD rE rH r* )r r~ r� �lstprotor: 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 s2 |j �� }g d�}|�� |�� � �||� d S )N)r r� s baz)r8 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 )NrA rB rl r� r� r� rJ rm rp r r r rq ' s � z2FTPClientTests.test_passiveNLST.<locals>.cbConnectr� rs r� ru ) r^ rv rI r5 r� rw r� rS r� rH r* rI )r r� rq r� r: r r r �test_passiveNLST s z FTPClientTests.test_passiveNLSTc s� t �� �� �fdd�}|� j_� �� t� }� j�d|�}� �|tj � � � � j � � d� � j � � � j�t� j�� � � � j � � d� � j�d� |S )z� Test a failure in NLST command. L{ftp.FTPClient.nlst} should return a Deferred which fails with L{ftp.CommandFailed} if the server indicates the indicated path is invalid for some reason. c sV � � | d� � � |d� |�| |f�}|��� � j�d� |�t�t� d��� d S r� ) r� rV r� r^ r* r@ r rB r ro rp r� r r rq I s �z1FTPClientTests.test_failedNLST.<locals>.cbConnectr� rs r� r� )r rn r^ rv rI r5 r� r� r r� r� rS r� rH r* rI )r rq r� 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. �/spam�/ham� RNFR /spam z8350 Requested file action pending further information. � RNTO /ham r ) rI r^ �renamer� rS r� rH r* rD rE rw )r r: � fromResponse� toResponser r r �test_renameFromTo_ 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)rI r^ r� r* r� rS r� )r �fromFile�toFiler r r � test_renameFromToEscapesPathss 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). r� r� r� �! 550 Requested file unavailable. r7 � rI r^ r� r� rS r� rH r* r� r r� r� r r r �$test_renameFromToFailingOnFirstError� 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). r� r� r� s8 350 Requested file action pending further information. r� r� r� r� r r r �"test_renameFromToFailingOnRenameTo� s � z1FTPClientTests.test_renameFromToFailingOnRenameToc C sB | � � | j�d�}| �| j�� d� | j�d� |�| jdg�S )z� L{ftp.FTPClient.makeDirectory} issues a I{MKD} command and returns a L{Deferred} which is called back with the server's response if the directory is created. r� � MKD /spam s 257 "/spam" created.z257 "/spam" created.)rI r^ � makeDirectoryr� rS r� r* rw r� r r r �test_makeDirectory� s z!FTPClientTests.test_makeDirectoryc C s6 | � � | j�d�}| �| j�� d� | j�d� |S )z� L{ftp.FTPClient.makeDirectory} escapes the path name it sends according to U{http://cr.yp.to/ftp/filesystem.html}. z/sp ams MKD /sp am s 257 win)rI r^ r� r� rS r� r* r� r r r � test_makeDirectoryPathEscape� s z+FTPClientTests.test_makeDirectoryPathEscapec C s@ | � � | j�d�}| �| j�� d� | j�d� | �|tj �S )z� L{ftp.FTPClient.makeDirectory} returns a L{Deferred} which is errbacked with L{CommandFailed} if the server returns an error response code. r� r� s 550 PERMISSION DENIED) rI r^ r� r� rS r� r* r� r r� r� r r r �test_failedMakeDirectory� s z'FTPClientTests.test_failedMakeDirectoryc rU )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 )NrV r� rF r r r �cbGet� r+ z/FTPClientTests.test_getDirectory.<locals>.cbGetrY rZ )rI r^ � getDirectoryrw r� rS r� r* )r r� r: r r r �test_getDirectory� s z FTPClientTests.test_getDirectoryc C r] )zj Test a failure in getDirectory method. The behaviour should be the same as PWD. rY r^ � rI r^ r� r� r r� r� rS r� r* r� r r r �test_failedGetDirectory� s z&FTPClientTests.test_failedGetDirectoryc C r] )z� Test a different failure in getDirectory method. The response should be quoted to be parsed, so it returns an error otherwise. rY s 257 /bar/bazr� r� r r r � test_anotherFailedGetDirectory� r` z-FTPClientTests.test_anotherFailedGetDirectoryc C �P | � � | j�d�}| �| j�� d� d}| j�|�| jj�� |� | j|g�S )z� L{ftp.FTPClient.removeFile} sends a I{DELE} command to the server for the indicated file and returns a Deferred which fires after the server sends a 250 response code. � /tmp/test� DELE /tmp/test �*250 Requested file action okay, completed.) rI r^ � removeFiler� rS r� r* rD rE rw �r r: �responser r r �test_removeFile� � z FTPClientTests.test_removeFilec �f �� � �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}. r� r� �,501 Syntax error in parameters or arguments.c � �� | j� gf�S r r� ��exc�r� r r r r � z6FTPClientTests.test_failedRemoveFile.<locals>.<lambda>) rI r^ r� r� rS r� r* rD rE r� r r� rw r� r r� r �test_failedRemoveFile � z$FTPClientTests.test_failedRemoveFilec �T �� � �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. r� �765 blah blah blahc r� r r� r� r� r r r # r� zBFTPClientTests.test_unparsableRemoveFileResponse.<locals>.<lambda>) rI r^ r� r* rD rE r� r � BadResponserw r� r r� r �!test_unparsableRemoveFileResponse � z0FTPClientTests.test_unparsableRemoveFileResponsec C �8 | � � | 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. r� � 250-perhaps a progress report� 250 okay)rI r^ r� r* rw r� r� r r r � test_multilineRemoveFileResponse& � z/FTPClientTests.test_multilineRemoveFileResponsec C r� )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. r� � RMD /tmp/test r� ) rI r^ �removeDirectoryr� rS r� r* rD rE rw r� r r r �test_removeDirectory2 r� z#FTPClientTests.test_removeDirectoryc r� )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}. r� r� r� c r� r r� r� r� r r r L r� z;FTPClientTests.test_failedRemoveDirectory.<locals>.<lambda>) rI r^ r� r� rS r� r* rD rE r� r r� rw r� r r� r �test_failedRemoveDirectory? r� z)FTPClientTests.test_failedRemoveDirectoryc r� )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. r� r� c r� r r� r� r� r r r Z r� zGFTPClientTests.test_unparsableRemoveDirectoryResponse.<locals>.<lambda>) rI r^ r� r* rD rE r� r r� rw r� r r� r �&test_unparsableRemoveDirectoryResponseO r� z5FTPClientTests.test_unparsableRemoveDirectoryResponsec C r� )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. r� r� r� )rI r^ r� r* rw r� r� r r r �%test_multilineRemoveDirectoryResponse] r� z4FTPClientTests.test_multilineRemoveDirectoryResponseN)-r1 r2 r3 r� r r� rI rL rS rT r\ r_ re rf rw 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 rF � sV '% &/%,71 +( rF c @ �8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �FTPClientBasicTestsz FTP client c C �& t �� }|�d� | �dg|j� dS )z? The first response is captured as a greeting. � 220 Imaginary FTP.z220 Imaginary FTP.N�r r� r* r� �greeting�r rC r r r � test_greetingo s z!FTPClientBasicTests.test_greetingc C r� )z� Responses with no message are still valid, i.e. three digits followed by a space is complete response. s 220 z220 Nr r r r r �test_responseWithNoMessagew 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 � | � g d �|d � d S ) z$ Multiline response r �BLAHs 210-First line.s 123-Second line.s Just some text.s Hir7 s 321s 210 Done.)z210-First line.z123-Second line.zJust some text.�HirJ �321z 210 Done.r N) r r� r rn rS r* r� rw r' rJ r� r� )r rC r� r� r r r �test_MultilineResponse� s0 �z*FTPClientBasicTests.test_MultilineResponsec C sd t �� }t�� |_|�d� |�dd� | �d|j�� � |j� � |�d� | �d|j�� � dS )zO Passing None as the password avoids sending the PASS command. � 220 Welcome to Imaginary FTP.�bobN� USER bob s 200 Hello bob.r7 � r r� r rn rS r* � queueLoginr� r� rH r r r r �test_noPasswordGiven� s z(FTPClientBasicTests.test_noPasswordGivenc C sd t �� }t�� |_|�d� |�dd� | �d|j�� � |j� � |�d� | �d|j�� � dS )zQ Receiving a 230 response to USER prevents PASS from being sent. r r �secretr s# 230 Hello bob. No password needed.r7 Nr r r r r �test_noPasswordNeeded� s z)FTPClientBasicTests.test_noPasswordNeededN) r1 r2 r3 r� r r r r r r r r r r� j s 6 r� c @ � e Zd ZdZdd� ZdS )�PathHandlingTestsz Handling paths. c C s ddgfddgfdg fdg d�fdg d�fddgfddgffD ]\}}| � t�g |�|� q d dd gfd dd gfd d gfd d gfd g d�fdg d�fdd dgfdd dgffD ]\}}| � t�dg|�|� qVdg fddgfddgfddd gffD ]\}}| � t�g |�|� qxdg fdd gfdg d�ffD ]\}}| � t�dg|�|� q�dg fdg fddgfd g fd dgfd dgfd ddgfd ddgfd!dgfd"dgfd#g fd$g ff D ]\}}| � t�dg|�|� q�d%D ] }| �tjtjg |� q�d&D ] }| �tjtjdg|� q�d'S )(z" Normalize paths. rY z/a�/za/b/c)rY rZ �cz/a/b/cz/a/za/rZ zb/z/bz/b/zb/czb/c/z/b/cr z/b/c/z//z//aza//za//bz//bzb//c�..�../za/..�xz/a/..z/a/b/..z/a/b/../z /a/b/../cz /a/b/../c/z /a/b/../../cz /a/b/../../c/z/a/b/../../c/..z/a/b/../../c/../) r r za/../..za/../../z/..z/../z/a/../..z /a/../../z /a/b/../../..)z../..z../../z ../a/../..N)r� r � toSegmentsr � InvalidPath)r �inp�outpr r r �test_Normalizer� s\ � � �$ � �z!PathHandlingTests.test_NormalizerN)r1 r2 r3 r� r r r r r r � � r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �IsGlobbingExpressionTestsz; Tests for _isGlobbingExpression utility function. c C s2 | � t�� � | � t�g �� | � t�d�� dS )z^ _isGlobbingExpression will return False for None, or empty segments. N�r� r �_isGlobbingExpressionr r r r �&test_isGlobbingExpressionEmptySegments3 s z@IsGlobbingExpressionTests.test_isGlobbingExpressionEmptySegmentsc C s, | � t�ddg�� | � t�ddg�� dS )z� _isGlobbingExpression will return False for plain segments. Also, it only checks the last segment part (filename) and will not check the path name. r> �expr�*.txtNr r r r r � test_isGlobbingExpressionNoGlob< s z9IsGlobbingExpressionTests.test_isGlobbingExpressionNoGlobc C s@ | � t�ddg�� | � t�ddg�� | � t�ddg�� dS )z� _isGlobbingExpression will return True for segments which contains globbing characters in the last segment part (filename). r> r$ z [a-b].txtzfil?.txtN)r� r r! r r r r � test_isGlobbingExpressionGlobF s z7IsGlobbingExpressionTests.test_isGlobbingExpressionGlobN)r1 r2 r3 r� r" r% r& r r r r r . s r c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �BaseFTPRealmTestsz� Tests for L{ftp.BaseFTPRealm}, a base class to help define L{IFTPShell} realms with different user home directory policies. c C s | � tttj�� dS )z; L{ftp.BaseFTPRealm} implements L{IRealm}. N)r� r r r � BaseFTPRealmr r r r �test_interfaceV s z BaseFTPRealmTests.test_interfacec sl t �| �� ��g � G � �fdd�dtj�}|| �� �}|�ddtj�\}}}| �|tj� | � |j �� dS )z� L{ftp.BaseFTPRealm} calls its C{getHomeDirectory} method with the avatarId being requested to determine the home directory for that avatar. c s e Zd Z� �fdd�ZdS )z:BaseFTPRealmTests.test_getHomeDirectory.<locals>.TestRealmc s � � |� �S r r< )r �avatarId��avatarsr� r r �getHomeDirectoryf r� zKBaseFTPRealmTests.test_getHomeDirectory.<locals>.TestRealm.getHomeDirectoryN)r1 r2 r3 r. r r, r r � TestRealme s r/ �alice@example.comN) r rd r` r r) � requestAvatar� IFTPShellr� �FTPShellr� �filesystemRoot)r r/ �realm�iface�avatar�logoutr r, r �test_getHomeDirectory\ 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) r` r r) r1 r � ANONYMOUSr2 r� ra r� r4 r rd )r rK r5 r6 r7 r8 r r r �test_anonymousq 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) r` r �NotImplementedErrorr. r9 )r r5 r r r �test_notImplemented~ s z%BaseFTPRealmTests.test_notImplementedN)r1 r2 r3 r� r* r9 r; r= r r r r r( P s r( c @ r� )� 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/homer0 N)r rg r` r. r� r rd r )r �baser5 �homer r r r9 � 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. r0 z/home/alice@example.comN)r rg r` r. r� r rd )r r5 r@ r r r �test_defaultHomeDirectory� s z'FTPRealmTests.test_defaultHomeDirectoryN)r1 r2 r3 r� r9 rA r r r r r> � 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) �getpass�getuserr[ �getpwnam�pw_dirr �SystemFTPRealmr` r. r� r rd )r r� r[ r� r5 r@ r r r r9 � 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. rJ c s s" � | ] }t �tjtj �V qd S r )�random�choice�string� ascii_letters�digits)r� r r r r � <genexpr>� s � �z6SystemFTPRealmTests.test_noSuchUser.<locals>.<genexpr>r N) rH rI rJ rK r� r� r rG r` r r r. )r r� r5 r r r �test_noSuchUser� s �z#SystemFTPRealmTests.test_noSuchUserN)r1 r2 r3 r� � nonPOSIXSkip�skipr9 rN r r r r rB � s rB c @ r� )�ErrnoToFailureTestsz9 Tests for L{ftp.errnoToFailure} errno checking. c C � t �tjd�}| �|t j�S )zS C{errno.ENOENT} should be translated to L{ftp.FileNotFoundError}. r )r �errnoToFailure�errno�ENOENTr� �FileNotFoundErrorr� r r r � test_notFound� � z!ErrnoToFailureTests.test_notFoundc C rR )zV C{errno.EPERM} should be translated to L{ftp.PermissionDeniedError}. r )r rS rT �EPERMr� �PermissionDeniedErrorr� r r r �test_permissionDenied� rX z)ErrnoToFailureTests.test_permissionDeniedc C rR )zW C{errno.EACCES} should be translated to L{ftp.PermissionDeniedError}. r )r rS rT �EACCESr� rZ r� r r r �test_accessDenied� rX z%ErrnoToFailureTests.test_accessDeniedc C rR )zW C{errno.ENOTDIR} should be translated to L{ftp.IsNotADirectoryError}. r )r rS rT �ENOTDIRr� �IsNotADirectoryErrorr� r r r �test_notDirectory� rX z%ErrnoToFailureTests.test_notDirectoryc C rR )zQ C{errno.EEXIST} should be translated to L{ftp.FileExistsError}. r )r rS rT �EEXISTr� �FileExistsErrorr� r r r �test_fileExists� rX z#ErrnoToFailureTests.test_fileExistsc C rR )zS C{errno.EISDIR} should be translated to L{ftp.IsADirectoryError}. r )r rS rT �EISDIRr� rQ r� r r r �test_isDirectory� rX z$ErrnoToFailureTests.test_isDirectoryc C s4 zt d�� ty t�dd�}| �|t � Y S w )z� If an unknown errno is passed to L{ftp.errnoToFailure}, it should let the originating exception pass through. r r� r )r$ � BaseExceptionr rS r� r� r r r �test_passThrough� s �z$ErrnoToFailureTests.test_passThroughN) r1 r2 r3 r� rW r[ r] r` rc re rg r r r r rQ � s rQ c @ r )�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. rJ �r )r ra rA r� rZ )r r@ r: r r r �test_anonymousWrite s z*AnonymousFTPShellTests.test_anonymousWriteN)r1 r2 r3 r� rj r r r r rh r rh 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 r# )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} �r< �r r� r r r �directoryExists � z#IFTPShellTestsMixin.directoryExistsc C r# )z� Create a directory in C{path}. @param path: the relative path of the directory to create, with one segment. @type path: C{str} rl rm r r r r ' s z#IFTPShellTestsMixin.createDirectoryc C r# )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} rl rm r r r � fileExists1 ro z IFTPShellTestsMixin.fileExistsr7 c C r# )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} rl �r r� � fileContentr r r � createFile> ro z IFTPShellTestsMixin.createFilec C �. | � | �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� rn r r� r r r r �test_createDirectoryK s z(IFTPShellTestsMixin.test_createDirectoryc C rt )z� C{fileExists} should report correctly about file existence, and C{createFile} should create a file detectable by C{fileExists}. �file.txtN)r� rp rs r� r r r r �test_createFileU s z#IFTPShellTestsMixin.test_createFilec r2 )zI Create a directory and check it ends in the filesystem. ri c � � � � �d�� d S )Nr )r� rn r r r r r d r� z2IFTPShellTestsMixin.test_makeDirectory.<locals>.cb)r@ r� rw r r r r r� ^ s z&IFTPShellTestsMixin.test_makeDirectoryc C �$ | � d� | j�d�}| �|tj�S )zm Creating a directory that already exists should fail with a C{ftp.FileExistsError}. r ri )r r@ r� r� r rb r� r r r �test_makeDirectoryErrori � z+IFTPShellTestsMixin.test_makeDirectoryErrorc �, � � d� � j�d�}� fdd�}|�|�S )zW Try to remove a directory and check it's removed from the filesystem. r �r c rx r )r� rn r r r r r y r� z4IFTPShellTestsMixin.test_removeDirectory.<locals>.cb)r r@ r� rw r r r r r� r s z(IFTPShellTestsMixin.test_removeDirectoryc C ry )zn removeDirectory should not work in file and fail with a C{ftp.IsNotADirectoryError}. rv �rv )rs r@ r� r� r r_ r� r r r �test_removeDirectoryOnFile~ r{ z.IFTPShellTestsMixin.test_removeDirectoryOnFilec C � | j �d�}| �|tj�S )zl Removing directory that doesn't exist should fail with a C{ftp.FileNotFoundError}. r} )r@ r� r� r rV r� r r r � test_removeNotExistingDirectory� � z3IFTPShellTestsMixin.test_removeNotExistingDirectoryc �0 � � d� � j�d�}� fdd�}|�|� |S )zR Try to remove a file and check it's removed from the filesystem. rv r~ c rx �Nrv )r� rp rF r r r r � r� z/IFTPShellTestsMixin.test_removeFile.<locals>.cb)rs r@ r� rw r r r r r� � � z#IFTPShellTestsMixin.test_removeFilec C ry )z: removeFile should not work on directory. �ned�r� )r r@ r� r� r rQ r� r r r �test_removeFileOnDirectory� � z.IFTPShellTestsMixin.test_removeFileOnDirectoryc C r� )zl Try to remove a non existent file, and check it raises a L{ftp.FileNotFoundError}. ri )r@ r� r� r rV r� r r r �test_removeNotExistingFile� r� z.IFTPShellTestsMixin.test_removeNotExistingFilec s6 � � d� � �d� � j�d�}� fdd�}|�|�S )z6 Check the output of the list method. r� rv ��.c s$ | � � � �| dg fdg fg� d S )Nrv r� �r� r� r� r r r r � s z)IFTPShellTestsMixin.test_list.<locals>.cb�r rs r@ r� rw r r r r � test_list� s z IFTPShellTestsMixin.test_listc s8 � � d� � �d� � j�dd�}� fdd�}|�|�S )z< Check the output of list with asked stats. r� rv r� )r� � permissionsc st | � � � �t| �d� � �| d d d� � �| d d d� � �t| d d �d� � �t| d d �d� d S )Nr~ r rv r� r� )r� r� r� r� r r r r � s z1IFTPShellTestsMixin.test_listWithStat.<locals>.cbr� r r r r �test_listWithStat� s � z%IFTPShellTestsMixin.test_listWithStatc C �$ | � d� | j�dd�}| �|t�S )�P Querying an invalid stat should result to a C{AttributeError}. r� r� �r� � whateverstat)r r@ r� r� �AttributeErrorr� r r r �test_listWithInvalidStat� s � z,IFTPShellTestsMixin.test_listWithInvalidStatc r| )z@ Check the output of the list method on a file. rv r~ c s | � � � �| dg fg� d S r� r� r� r r r r � s z-IFTPShellTestsMixin.test_listFile.<locals>.cb)rs r@ r� rw r r r r � test_listFile� s z!IFTPShellTestsMixin.test_listFilec C r� )zm list on a directory that doesn't exist should fail with a L{ftp.FileNotFoundError}. ri )r@ r� r� r rV r� r r r � test_listNotExistingDirectory� r� z1IFTPShellTestsMixin.test_listNotExistingDirectoryc C s | � d� | j�d�}|S )z+ Try to access a resource. r� r� )r r@ �accessr� r r r � test_access� s z IFTPShellTestsMixin.test_accessc C r� )zF access should fail on a resource that doesn't exist. ri )r@ r� r� r rV r� r r r �test_accessNotFound� s z'IFTPShellTestsMixin.test_accessNotFoundc r� )zY Check that openForReading returns an object providing C{ftp.IReadFile}. rv r~ c s � � tj�| �� d S r )r� r � IReadFile� providedByrF r r r r s z3IFTPShellTestsMixin.test_openForReading.<locals>.cb)rs r@ rk rw r r r r �test_openForReading r� z'IFTPShellTestsMixin.test_openForReadingc C r� )zr openForReading should fail with a C{ftp.FileNotFoundError} on a file that doesn't exist. r� )r@ rk r� r rV r� r r r �test_openForReadingNotFound r� z/IFTPShellTestsMixin.test_openForReadingNotFoundc C ry )z> openForReading should not work on directory. r� r� )r r@ rk r� r rQ r� r r r � test_openForReadingOnDirectory r� z2IFTPShellTestsMixin.test_openForReadingOnDirectoryc s4 �j �d�}� �fdd�}�fdd�� |�|� |S )zZ Check that openForWriting returns an object providing C{ftp.IWriteFile}. ri c s �� tj�| �� | �� �� �S r )r� r � IWriteFiler� rR rw rF ��cb2r r r �cb1) s z4IFTPShellTestsMixin.test_openForWriting.<locals>.cb1c s � � t�| �� d S r )r� r r� rF r r r r� - r� z4IFTPShellTestsMixin.test_openForWriting.<locals>.cb2)r@ rA rw )r r: r� r r� r �test_openForWriting# s z'IFTPShellTestsMixin.test_openForWritingc C ry )zd openForWriting should not be able to open a directory that already exists. r� r� )r r@ rA r� r rQ r� r r r �$test_openForWritingExistingDirectory3 r{ z8IFTPShellTestsMixin.test_openForWritingExistingDirectoryc C ry )z� openForWring should fail with a L{ftp.FileNotFoundError} if you specify a file in a directory that doesn't exist. r� )r� � idonotexistr )r r@ rA r� r rV r� r r r �)test_openForWritingInNotExistingDirectory< r{ z=IFTPShellTestsMixin.test_openForWritingInNotExistingDirectoryc s: d� �� d� � �j�dd�}� �fdd�}|�|� |S )z@ Check the output of the stat method on a file. � wobble rv r~ �r� ra c s& �� | d t� �� ��| d � d S )Nr r� )r� r� r� rF �rr r r r r M s z-IFTPShellTestsMixin.test_statFile.<locals>.cb)rs r@ �statrw r r r� r � test_statFileE s z!IFTPShellTestsMixin.test_statFilec �2 � � d� � j�dd�}� fdd�}|�|� |S )zE Check the output of the stat method on a directory. r� r� r� c s � � | d � d S )Nr� )r� rF r r r r [ r= z2IFTPShellTestsMixin.test_statDirectory.<locals>.cb�r r@ r� rw r r r r �test_statDirectoryT � z&IFTPShellTestsMixin.test_statDirectoryc r� )z3 Check the owner and groups stats. r� r� �r r� c s � � t| �d� d S )Nr~ r� rF r r r r h r� z3IFTPShellTestsMixin.test_statOwnerGroup.<locals>.cbr� r r r r �test_statOwnerGroupa r� 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 � t �r rl r r r r �raiseNotImplementedu r0 zQIFTPShellTestsMixin.test_statHardlinksNotImplemented.<locals>.raiseNotImplementedc s � | �}�|_ |S r )�getNumberOfHardLinks�r� r# ��pathFuncr� r r �notImplementedFilePathx s zTIFTPShellTestsMixin.test_statHardlinksNotImplemented.<locals>.notImplementedFilePathr� r� )� hardlinksr N�r@ �_pathr r� r� �successResultOf�r r� r: r r� r � test_statHardlinksNotImplementedn 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 r� r rl r r r r r� � r0 zRIFTPShellTestsMixin.test_statOwnerGroupNotImplemented.<locals>.raiseNotImplementedc s � | �}�|_ �|_|S r )� getUserID� getGroupIDr� r� r r r� � s zUIFTPShellTestsMixin.test_statOwnerGroupNotImplemented.<locals>.notImplementedFilePathr� r� r� �0Nr� r� r r� r �!test_statOwnerGroupNotImplemented� s z5IFTPShellTestsMixin.test_statOwnerGroupNotImplementedc C � | j �dd�}| �|tj�S )zf stat should fail with L{ftp.FileNotFoundError} on a file that doesn't exist. ri r� )r@ r� r� r rV r� r r r �test_statNotExisting� � z(IFTPShellTestsMixin.test_statNotExistingc C r� )r� r� r� r� )r r@ r� r� r� r� r r r �test_invalidStat� s z$IFTPShellTestsMixin.test_invalidStatc s. � � d� � j�dd�}� fdd�}|�|�S )z, Try to rename a directory. r� r� ri c s$ � � � �d�� � �� �d�� d S )Nr r� )r� rn r� rF r r r r � s z+IFTPShellTestsMixin.test_rename.<locals>.cb)r r@ r� rw r r r r � test_rename� s z IFTPShellTestsMixin.test_renamec C r� )zl Renaming a directory that doesn't exist should fail with L{ftp.FileNotFoundError}. ri r} )r@ r� r� r rV r� r r r �test_renameNotExisting� r� z*IFTPShellTestsMixin.test_renameNotExistingN�r7 )(r1 r2 r3 r� rn r rp rs ru rw r� rz 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 rk sL rk 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 rd r` r r r r3 r@ r r r r r � s zFTPShellTests.setUpc C � | j �|��� S )z: Test if the directory exists at C{path}. )r r �isdirrm r r r rn � � z FTPShellTests.directoryExistsc C r� )z0 Create a directory in C{path}. )r r r rm r r r r � r� z FTPShellTests.createDirectoryc C r� )z5 Test if the file exists at C{path}. )r r �isfilerm r r r rp � r� zFTPShellTests.fileExistsr7 c C s | j �|��|�S )z@ Create a file named C{path} with some content. )r r � setContentrq r r r rs � � zFTPShellTests.createFileNr� ) r1 r2 r3 r� r rn r rp rs 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 du sJ �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)�producerr8 �resumeProducing)r r� � streamingr r r �registerProducer� s z TestConsumer.registerProducerc C s | j dusJ �d| _ dS )zN Unregister the producer, it should be done after a register. N)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)r8 r' r� r� r, r r r r� s zTestConsumer.write)r1 r2 r3 r� r� r� r� r� r r r r r� � s r� c @ r� )� 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)� toProducerg )r r� rg r r r r s zTestProducer.__init__c C s | j �| j� dS )z+ Send the data to consume. N)rg r� r� r r r r �start r� zTestProducer.startN)r1 r2 r3 r� r r� r r r r r� s r� c @ r� ) �IReadWriteTestsMixinzJ Generic tests for the C{IReadFile} and C{IWriteFile} interfaces. c C r# )z Return an object providing C{IReadFile}, ready to send data C{content}. @param content: data to send rl �r �contentr r r � getFileReader* s z"IReadWriteTestsMixin.getFileReaderc C r# )zR Return an object providing C{IWriteFile}, ready to receive data. rl r r r r � getFileWriter2 � z"IReadWriteTestsMixin.getFileWriterc C r# )z6 Return the content of the file used. rl r r r r �getFileContent8 r� z#IReadWriteTestsMixin.getFileContentc s8 d�t � �� �fdd�}���fdd�� �����|�S )z� Test L{ftp.IReadFile}: the implementation should have a send method returning a C{Deferred} which fires when all the data has been sent to the consumer, and the data should be correctly send to the consumer. r� c s | � ���� �S r )rh rw )�reader)�cbSendrg r r r� G r+ z-IReadWriteTestsMixin.test_read.<locals>.cbGetc s �� d�� j��� d S r6 )r� r� r8 rF )rg r� r r r r� J s z.IReadWriteTestsMixin.test_read.<locals>.cbSend)r� r� rw �r r� r )r� rg r� r r � test_read> s z IReadWriteTestsMixin.test_readc s: d��fdd�}� �fdd����fdd�� �� � �|�S )a Test L{ftp.IWriteFile}: the implementation should have a receive method returning a C{Deferred} which fires with a consumer ready to receive data to be written. It should also have a close() method that returns a Deferred. s elbbow c s | � � �� | �S r )rR rw )�writer)� cbReceiver r r� X r+ z.IReadWriteTestsMixin.test_write.<locals>.cbGetc s4 t �| �}| �d d� |�� | �� |�� �� �S )NT)r� r� r� r� �closerw )rg r� r� )�cbCloser� r r r� [ s z2IReadWriteTestsMixin.test_write.<locals>.cbReceivec s �� ��� � � d S r )r� r� r� )r� r r r r� b r� z0IReadWriteTestsMixin.test_write.<locals>.cbClose)r� rw r� r )r� r� r� r r � test_writeO s z IReadWriteTestsMixin.test_writeN) r1 r2 r3 r� r� r� r� r� r� r r r r r� % s r� c @ r' ) �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. rv N) r rd r` r r r r3 r@ r r r r r r n s zFTPReadWriteTests.setUpc C s$ | j �| j��|� | j�| jf�S )zV Return a C{ftp._FileReader} instance with a file opened for reading. )r r r r� r@ rk r� r r r r� w s z FTPReadWriteTests.getFileReaderc C s | j �| jf�S )zV Return a C{ftp._FileWriter} instance with a file opened for writing. )r@ rA r r r r r r� ~ r� z FTPReadWriteTests.getFileWriterc C s | j �| j��� S )z; Return the content of the temporary file. )r r r � getContentr r r r r� � r� z FTPReadWriteTests.getFileContentN)r1 r2 r3 r� r r� r� r� r r r r r� h s r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�CloseTestWriterz" Close writing to a file. Fc C s t � | _t�| j�}t�|�S )z> Receive bytes. @return: L{Deferred} )r r8 r � FileConsumerr r� )r �fcr r r rR � s zCloseTestWriter.receivec C s d| _ | jS )z< Close bytes. @return: L{Deferred} T)� closeStartedr: r r r r r� � s zCloseTestWriter.closeN)r1 r2 r3 r� r� rR r� r r r r r� � s r� c @ r )�CloseTestShellz Close writing shell. c C s t �| j�S r )r r� r� )r �segsr r r rA � r% z CloseTestShell.openForWritingN)r1 r2 r3 r� rA r r r r r� � r r� c @ r )� FTPCloseTestsz8 Tests that the server invokes IWriteFile.close c C s� t �� }dg|_t� |_t� |j_t�� |jj_ t � � |_ d|j _ |� t� � t �� }t �|�|_ ||_|� d� g }|�d�}|�|j� | �|jjjd� |�d� | �|jjjd� |�d� | �|jjjd� | �|� |jjj �d� | �|� |S ) z� Confirm that FTP uploads (i.e. ftp_STOR) correctly call and wait upon the IWriteFile object's close() method r Nr� zclose() called earlys some data herezreason is ignoredzclose() not calledzallow close() to finish)r �FTP�workingDirectoryr� r@ r� r� r r9 r: rp r" �timeOutr� r r� r� r: �ftp_STORrw r' r� r� r< r@ r� r> )r r# �di� stor_doner: r r r r� � s0 zFTPCloseTests.test_writeN)r1 r2 r3 r� r� r r r r r� � r r� c @ r )�FTPResponseCodeTestsz4 Tests relating directly to response codes. c C sv t tj�}t � }tt��� D ]*\}}t|t�r8|�� r8| �||d� ||�� | � ||d|�d|��� |� |� qdS )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. z4Code {!r} with value {!r} missing from RESPONSE dictzDuplicate code z with value N) �setr �RESPONSE�vars�items� isinstancer �isupperr3 rz � assertNotIn�add)r � allValues� seenValues�keyr� r r r � test_unique� s& ��� ��z FTPResponseCodeTests.test_uniqueN)r1 r2 r3 r� r r r r r r� � r r� )rA rB )Qr� rT rC rb rH rJ �ior �zope.interfacer �zope.interface.verifyr � twisted.credr r r �twisted.cred.errorr �twisted.cred.portalr �twisted.internetr r r r r �twisted.internet.interfacesr �twisted.protocolsr r r �twisted.pythonr r r � twisted.testr �twisted.trial.unittestr �platform� isWindowsrO � LineReceiverr r2 r5 rI rJ r� r� rD rr r� r� r� r� r� r� r� r4 rF r� r r r( r> rB rQ rh rk r� r� r� r� r� r� r� r� r� r� r r r r �<module> s� q kR ( A 7xL"7); .&(C# )