← Back
Editing: test_filetransfer.cpython-38.pyc
U \�[7r � @ s� d Z ddlmZmZ ddlZddlZddlZddlmZ ddl m Z ed�Z ed�Z e r~ddl mZ dd lmZmZmZmZ nG d d � d �Zdd lmZ dd lmZ ddlmZ ddlmZm Z ddl m Z G dd� dej �Z G dd� de �Z!G dd� d�Z"e �rNe#e d��s,ddl$Z$e$�%de f � dZ n"G dd� de j&�Z'e�(e'e ej)� G dd � d e j*�Z+G d d � d e+�Z,G d d � d �Z-G d!d"� d"e j*�Z.G d#d$� d$e j*�Z/G d%d&� d&e j*�Z0dS )'z. Tests for L{twisted.conch.ssh.filetransfer}. � )�division�absolute_importN)� requireModule)�unittest� cryptographyztwisted.conch.unix��avatar)�common� connection� filetransfer�sessionc @ s e Zd ZG dd� d�ZdS )r c @ s e Zd ZdS )zavatar.ConchUserN)�__name__� __module__� __qualname__� r r �F/usr/lib/python3/dist-packages/twisted/conch/test/test_filetransfer.py� ConchUser s r N)r r r r r r r r r s r )�defer)�loopback)� components)�long� _PY37PLUS)�FilePathc @ s e Zd Zdd� Zdd� ZdS )� TestAvatarc C s( t j�| � tj| jd< tj| jd< d S )N� session� sftp) r r �__init__r � SSHSessionZ channelLookupr �FileTransferServerZsubsystemLookup��selfr r r r $ s zTestAvatar.__init__c O s~ z t |�}W n tk r, |||fg}Y nX |D ]F}|d }t|�dkrR|d pTd}t|�dkrj|d pli }|||�}q2|S )Nr � r � )�iter� TypeError�len)r �f�args�kw�i�func�rr r r � _runAsUser) s zTestAvatar._runAsUserN)r r r r r, r r r r r # s r c @ s e Zd Zdd� Zdd� ZdS )�FileTransferTestAvatarc C s t �| � || _d S �N)r r �homeDir)r r/ r r r r 8 s z FileTransferTestAvatar.__init__c C s t t�� ��| jj�S r. )r �os�getcwd� preauthChildr/ �pathr r r r � getHomeDir< s z!FileTransferTestAvatar.getHomeDirN)r r r r r4 r r r r r- 6 s r- c @ s e Zd Zdd� ZdS )�ConchSessionForTestAvatarc C s || _ d S r. r )r r r r r r B s z"ConchSessionForTestAvatar.__init__N)r r r r r r r r r5 @ s r5 �SFTPServerForUnixConchUserzNtwisted.conch.unix imported %r, but doesn't define SFTPServerForUnixConchUser'c @ s e Zd Zdd� Zdd� ZdS )�FileTransferForTestAvatarc C s ddiS )N� conchTest� ext datar )r �versionZotherExtr r r � gotVersionQ s z$FileTransferForTestAvatar.gotVersionc C s |dkr dS t �d S )N� testExtendedRequest� bar)�NotImplementedError)r ZextName�extDatar r r �extendedRequestT s z)FileTransferForTestAvatar.extendedRequestN)r r r r; r@ r r r r r7 O s r7 c @ s e Zd Zdd� ZdS )� SFTPTestBasec C s t | �� �| _| j�d�| _| j�d��d� | j�d�jdd��6}|�d� tdd ��}|�|�d �� W 5 Q R X W 5 Q R X | j�d��d � | j�d �jdd��}|�d � W 5 Q R X | j�d�jdd��}|�d � W 5 Q R X | j�d�jdd��}|�d � W 5 Q R X d S )NZextraZ testDirectoryT� testfile1�wb)�mode� aaaaaaaaaabbbbbbbbbbz /dev/urandom�rbi i� ZtestRemoveFile� aZtestRenameFilez.testHiddenFile) r Zmktemp�testDir�child�makedirs�open�write�read�chmod)r r& �f2r r r �setUp_ s $zSFTPTestBase.setUpN)r r r rP r r r r rA ] s rA c @ s� e Zd ZesdZdd� Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zej d.d/� �Z e r�d0e _ej d1d2� �Z d3S )4�OurServerOurClientTests� can't run on non-posix computersc s� t �� � t� j�� _tj� jd�� _t� � j�}t� � � _ d � _ d � _ � fdd�}|� j _t� � j �}� j �|� � j�|� |� _|� _� �� d S )Nr c s | � _ |� _d S r. )�_serverVersion�_extData)Z serverVersionr? r r r �_� s z(OurServerOurClientTests.setUp.<locals>._)rA rP r- rH r r r �serverr Z LoopbackRelay�FileTransferClient�clientrS rT ZgotServerVersionZmakeConnection�clientTransport�serverTransport� _emptyBuffers)r rY rU rZ r r r rP x s z OurServerOurClientTests.setUpc C s* | j js| jjr&| j �� | j�� q d S r. )rZ �bufferrY � clearBufferr r r r r[ � s z%OurServerOurClientTests._emptyBuffersc C s, | j �� | j�� | j �� | j�� d S r. )rZ �loseConnectionrY r] r r r r �tearDown� s z OurServerOurClientTests.tearDownc C s$ | � | jd� | � | jddi� d S )N� r8 r9 )� assertEqualrS rT r r r r �test_serverVersion� s z*OurServerOurClientTests.test_serverVersionc C s& | � tj�| jj�d| jjf � dS )z: It implements the ISFTPServer interface. z ISFTPServer not provided by %rN)� assertTruer � ISFTPServerZ providedByrV rX r r r r � test_interface_implementation� s �z5OurServerOurClientTests.test_interface_implementationc sb �j �dtjtjB i �}��� tj�g � � �fdd�}��td|� � �fdd�}|� |� |S )zV A file opened with C{openFile} is close when the connection is lost. � testfile1c s � � | � �| � d S r. ��append)�fd)�closed�oldCloser r �close� s zJOurServerOurClientTests.test_openedFileClosedWithConnection.<locals>.closerl c s` �j j| jdd � j}�j�� �j�� �j�� �j�� ���j ji � �� |� � d S )N� ) rV Z openFilesZhandleri rZ r^ rY r] ra ZassertIn)�openFileri )rj r r r � _fileOpened� s zPOurServerOurClientTests.test_openedFileClosedWithConnection.<locals>._fileOpened) rX rn r �FXF_READ� FXF_WRITEr[ r0 rl Zpatch� addCallback)r �drl ro r )rj rk r r �#test_openedFileClosedWithConnection� s �� z;OurServerOurClientTests.test_openedFileClosedWithConnectionc s. � j �d�}� �� � fdd�}|�|� |S )zh A directory opened with C{openDirectory} is close when the connection is lost. � c s< � j �� � j�� � j �� � j�� � �� jji � d S r. )rZ r^ rY r] ra rV ZopenDirs��openDirr r r � _getFiles� s zSOurServerOurClientTests.test_openedDirectoryClosedWithConnection.<locals>._getFiles�rX � openDirectoryr[ rr )r rs rx r r r �(test_openedDirectoryClosedWithConnection� s z@OurServerOurClientTests.test_openedDirectoryClosedWithConnectionc sd �j �dtjtjB i �}��� � ��fdd�}�fdd�� ��fdd���fdd ��|�|� |S ) Nrf c s* �� | t�| �� � | �}|��| � |S r. )ra r Z ISFTPFilerr �rn rs )� _readChunk� _writeChunkr r r ro � s z<OurServerOurClientTests.test_openFileIO.<locals>._fileOpenedc s& | � dd�}� �� |�� jd� |S )Nr � rE �Z readChunkr[ rr ra r| r r r r} � s z;OurServerOurClientTests.test_openFileIO.<locals>._readChunkc s$ |� dd�}��� |�� |� |S )Nr s cccccccccc)Z writeChunkr[ rr �rU rn rs )� _readChunk2r r r r~ � s z<OurServerOurClientTests.test_openFileIO.<locals>._writeChunkc s& |� dd�}� �� |�� jd� |S )Nr � s aaaaaaaaaabbbbbbbbbbccccccccccr� r� r r r r� � s z<OurServerOurClientTests.test_openFileIO.<locals>._readChunk2�rX rn r rp rq r[ rr )r rs ro r )r} r� r~ r r �test_openFileIO� s �� z'OurServerOurClientTests.test_openFileIOc sV �j �dtjtjB i �}��� �fdd���fdd�� � ��fdd�}|�|� |S )Nrf c s |� � }� �� |S r. )�getAttrsr[ r� r r r � _getAttrs s zBOurServerOurClientTests.test_closedFileGetAttrs.<locals>._getAttrsc s � � � | S r. )ZflushLoggedErrors�r& r r r �_err s z=OurServerOurClientTests.test_closedFileGetAttrs.<locals>._errc s4 | � � }��� |��| � |�� � ��|tj�S r. )rl r[ rr � addErrback� assertFailurer � SFTPErrorr| �r� r� r r r �_close s z?OurServerOurClientTests.test_closedFileGetAttrs.<locals>._closer� )r rs r� r r� r �test_closedFileGetAttrs� s �� z/OurServerOurClientTests.test_closedFileGetAttrsc sD �j �dtjtjB i �}��� � �fdd�}�fdd�� |�|�S )Nrf c s | � � }��� |�� � |S r. �r� r[ rr r| �� _getAttrs2r r r r� s zBOurServerOurClientTests.test_openFileAttributes.<locals>._getAttrsc s& � j �d�}� �� |�� j| � |S �Nrf �rX r� r[ rr ra )Zattrs1rs r r r r� s zCOurServerOurClientTests.test_openFileAttributes.<locals>._getAttrs2r� �r rs r� r r� r �test_openFileAttributes s �� z/OurServerOurClientTests.test_openFileAttributesc sV �j �dtjtjB i �}��� ��fdd�}� �fdd���fdd�� |�|� |S )Nrf c s | � � }��� |�� � |S r. r� r| )� _setAttrsr r r r� 0 s z@OurServerOurClientTests.test_openFileSetAttrs.<locals>._getAttrsc s: d| d< �j �d| �}��� |�� � |��j| � |S )Nr Zatimerf )rX ZsetAttrsr[ rr ra ��attrsrs r� r r r� 6 s z@OurServerOurClientTests.test_openFileSetAttrs.<locals>._setAttrsc s � j �d�}� �� |S r� �rX r� r[ �rU rs r r r r� > s zAOurServerOurClientTests.test_openFileSetAttrs.<locals>._getAttrs2r� r� r )r� r� r r �test_openFileSetAttrs) s �� z-OurServerOurClientTests.test_openFileSetAttrsc sb i ��j jj� � �fdd�}|�j j_�j�dtjtjB ddi�}��� ��fdd�}|�|�S )z� Check that L{filetransfer.FileTransferClient.openFile} can send extended attributes, that should be extracted server side. By default, they are ignored, so we just verify they are correctly parsed. c s �� |� � | ||�S r. )�update)�filename�flagsr� )� oldOpenFile�savedAttributesr r rn O s zIOurServerOurClientTests.test_openFileExtendedAttributes.<locals>.openFilerf �ext_foor= c s �� � ddi� d S )Nr� r= �ra )Zign)r� r r r �checkX s zFOurServerOurClientTests.test_openFileExtendedAttributes.<locals>.check)rV rX rn r rp rq r[ rr )r rn rs r� r )r� r� r r � test_openFileExtendedAttributesG s ��z7OurServerOurClientTests.test_openFileExtendedAttributesc sB � j �d�}� �� � fdd�}|�|� |�|� � �|tj�S )N� testRemoveFilec s � j �d�}� �� |S )Nr� )rX Z removeFiler[ )�ignoredrs r r r � _removeFileb s z<OurServerOurClientTests.test_removeFile.<locals>._removeFile)rX r� r[ rr r� r r� )r rs r� r r r �test_removeFile^ s z'OurServerOurClientTests.test_removeFilec s8 �j �d�}��� � �fdd�}�fdd�� |�|�S )N� testRenameFilec s& �j �dd�}��� |�� | � |S )Nr� � testRenamedFile)rX Z renameFiler[ rr r� �� _testRenamedr r r �_renamep s z8OurServerOurClientTests.test_renameFile.<locals>._renamec s& � j �d�}� �� |�� j|� d S )Nr� r� )rU r� rs r r r r� v s z=OurServerOurClientTests.test_renameFile.<locals>._testRenamed)rX r� r[ rr )r rs r� r r� r �test_renameFilel s z'OurServerOurClientTests.test_renameFilec C s" | j �d�}| �� | �|tj�S �N� testMakeDirectory�rX r� r[ r� r r� �r rs r r r �test_directoryBad~ s z)OurServerOurClientTests.test_directoryBadc sZ � j �di �}� �� � fdd�}� fdd�}|�|� |�|� |�|� � �|tj�S )Nr� c s � j �d�}� �� |S r� r� r� r r r r� � s zAOurServerOurClientTests.test_directoryCreation.<locals>._getAttrsc s � j �d�}� �� |S r� )rX ZremoveDirectoryr[ r� r r r �_removeDirectory� s zHOurServerOurClientTests.test_directoryCreation.<locals>._removeDirectory)rX Z makeDirectoryr[ rr r� r r� )r rs r� r� r r r �test_directoryCreation� s z.OurServerOurClientTests.test_directoryCreationc s\ �j �d�}��� g �� ���fdd����fdd�}�fdd�� |��� |�|� |S )N� c sF �� fdd�}t �� j�}��� |�|� |��� |��� � |S )Nc s � � | � �S r. rg r� )�filesrw r r rh � s zMOurServerOurClientTests.test_openDirectory.<locals>._getFiles.<locals>.append)r Z maybeDeferred�nextr[ rr r� )rw rh rs �r� rx r� r rv r rx � s z=OurServerOurClientTests.test_openDirectory.<locals>._getFilesc s6 t t t� � �d �}|�� ��|dddddg� d S )Nr � .testHiddenFile� testDirectoryr� r� rf )�list�zip�sortra )r� Zfs)r� r r r � _checkFiles� s ��z?OurServerOurClientTests.test_openDirectory.<locals>._checkFilesc s |� � }� �� |S r. )rl r[ )rU rw rs r r r r� � s z:OurServerOurClientTests.test_openDirectory.<locals>._closery )r rs r� r r� r �test_openDirectory� s z*OurServerOurClientTests.test_openDirectoryc C s" | j �d�}| �� | �|tj�S )N� testLinkr� r� r r r �test_linkDoesntExist� s z,OurServerOurClientTests.test_linkDoesntExistc sB � j �dd�}� �� � fdd�}� fdd�}|�|� |�|�S )Nr� rf c s � j �dd�}� �� |S )Nr� r! r� r� r r r �_getFirstAttrs� s zDOurServerOurClientTests.test_linkSharesAttrs.<locals>._getFirstAttrsc s& � j �d�}� �� |�� j| � |S r� r� )Z firstAttrsrs r r r �_getSecondAttrs� s zEOurServerOurClientTests.test_linkSharesAttrs.<locals>._getSecondAttrs�rX ZmakeLinkr[ rr )r rs r� r� r r r �test_linkSharesAttrs� s z,OurServerOurClientTests.test_linkSharesAttrsc sF � j �dd�}� �� � fdd�}� fdd�}|�|� |�|� |S )Nr� rf c sH � j �d�}� �� tt�� ��� jj�}|� d�}|� � j |j� |S �Nr� rB ) rX ZreadLinkr[ r r0 r1 r2 rH r3 rI rr ra )rU rs ZtestFiler r r � _readLink� s �z8OurServerOurClientTests.test_linkPath.<locals>._readLinkc sH � j �d�}� �� tt�� ��� jj�}|� d�}|� � j |j� |S r� ) rX ZrealPathr[ r r0 r1 r2 rH r3 rI rr ra )rU rs ZtestLinkr r r � _realPath� s �z8OurServerOurClientTests.test_linkPath.<locals>._realPathr� )r rs r� r� r r r � test_linkPath� s z%OurServerOurClientTests.test_linkPathc C s4 | j �dd�}| �� |�| jd� |�| j� |S )Nr< s foor= )rX r@ r[ rr ra �_cbTestExtendedRequestr� r r r �test_extendedRequest� s z,OurServerOurClientTests.test_extendedRequestc C s" | j �dd�}| �� | �|t�S )Ns testBadRequestr� )rX r@ r[ r� r> )r r� rs r r r r� � s z.OurServerOurClientTests._cbTestExtendedRequestc c s� | j �d�}| �� |V }t� }z,|D ]"}| �� |V \}}}|�|� q&W 5 |�� }| �� |V X | �� | �|tdddddg�� dS )z� Check that the object returned by L{filetransfer.FileTransferClient.openDirectory} can be used as an iterator. r� r� r� r� r� rf N)rX rz r[ �setrl �addra )r rs rw � filenamesr& r� rU Z fileattrsr r r �test_openDirectoryIterator s( ��z2OurServerOurClientTests.test_openDirectoryIteratorz!Broken by PEP 479 and deprecated.c c sj | j �d�}| �� |V }|�� | �� }d}| �dt|�� | �t|d d � | �||d d � dS )zJ Using client.openDirectory as an iterator is deprecated. r� zeUsing twisted.conch.ssh.filetransfer.ClientDirectory as an iterator was deprecated in Twisted 18.9.0.r! r �category�messageN)rX rz r[ r� Z flushWarningsra r% �DeprecationWarning)r rs rw �warningsr� r r r �$test_openDirectoryIteratorDeprecated+ s �z<OurServerOurClientTests.test_openDirectoryIteratorDeprecatedN) r r r �unix�skiprP r[ r_ rb re rt r{ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r ZinlineCallbacksr� r r� r r r r rQ s s> ! " #�rQ c @ s e Zd Zdd� ZdS )�FakeConnc C s d S r. r )r Zchannelr r r � sendCloseA s zFakeConn.sendCloseN)r r r r� r r r r r� @ s r� c @ sL e Zd ZesdZdd� Zdd� Zdd� Zdd � Zd d � Z d d � Z dd� Z dS )�FileTransferCloseTestsrR c C s t � | _d S r. )r r r r r r rP J s z FileTransferCloseTests.setUpc C s, t �� }G dd� d�}|� |_| j|j_|S )Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zDFileTransferCloseTests.buildServerConnection.<locals>.DummyTransportc S s | | _ d S r. )� transportr r r r r R s zMFileTransferCloseTests.buildServerConnection.<locals>.DummyTransport.__init__c S s d S r. r )r Zkind�datar r r � sendPacketT s zOFileTransferCloseTests.buildServerConnection.<locals>.DummyTransport.sendPacketc S s dS )Nzdummy transportr r r r r � logPrefixV s zNFileTransferCloseTests.buildServerConnection.<locals>.DummyTransport.logPrefixN)r r r r r� r� r r r r �DummyTransportQ s r� )r Z SSHConnectionr� r )r �connr� r r r �buildServerConnectionM s z,FileTransferCloseTests.buildServerConnectionc s$ d�_ |j� � �fdd�}||_d S )NFc s d�_ � | � d S )NT)�connectionLostFired)�reason�ZorigConnectionLostr r r �connectionLost` s zFFileTransferCloseTests.interceptConnectionLost.<locals>.connectionLost)r� r� )r � sftpServerr� r r� r �interceptConnectionLost] s z.FileTransferCloseTests.interceptConnectionLostc C s | � | jd� d S )Nz*sftpServer's connectionLost was not called)rc r� r r r r �assertSFTPConnectionLostf s �z/FileTransferCloseTests.assertSFTPConnectionLostc C sJ t jt� | jd�}|�t�d�� |jjj }| � |� |� � | � � dS )ze Closing a session should notify an SFTP subsystem launched by that session. )r� r r N) r r r� r �request_subsystemr �NSrX r� �protor� Z closeReceivedr� )r Z testSessionr� r r r �test_sessionClosek s z(FileTransferCloseTests.test_sessionClosec C s� | � � }t�d�t�dd�d }|�|� |jd }|�t�d�� |jj j }| � |� | � |� |� t�dd�� | � � dS )zy A client sending CHANNEL_CLOSE should trigger closeReceived on the associated channel instance. r �>Lr r` r N)r� r r� �struct�pack�ssh_CHANNEL_OPEN�channelsr� rX r� r� r� Zssh_CHANNEL_CLOSEr� �r r� ZpacketZsessionChannelr� r r r �%test_clientClosesChannelOnConnnection� s z<FileTransferCloseTests.test_clientClosesChannelOnConnnectionc C sn | � � }t�d�t�dd�d }|�|� |jd }|�t�d�� |jj j }| � |� |� � | � � dS )zP Closing an SSH connection should close all sessions within it. r r� r r` r N)r� r r� r� r� r� r� r� rX r� r� r� ZserviceStoppedr� r� r r r �'test_stopConnectionServiceClosesChannel� s z>FileTransferCloseTests.test_stopConnectionServiceClosesChannelN) r r r r� r� rP r� r� r� r� r� r� r r r r r� E s r� c @ s, e Zd ZdZesdZddddgZdd� Zd S ) �ConstantsTestsag Tests for the constants used by the SFTP protocol implementation. @ivar filexferSpecExcerpts: Excerpts from the draft-ietf-secsh-filexfer-02.txt (draft) specification of the SFTP protocol. There are more recent drafts of the specification, but this one describes version 3, which is what conch (and OpenSSH) implements. � Cannot run without cryptographya� The following values are defined for packet types. #define SSH_FXP_INIT 1 #define SSH_FXP_VERSION 2 #define SSH_FXP_OPEN 3 #define SSH_FXP_CLOSE 4 #define SSH_FXP_READ 5 #define SSH_FXP_WRITE 6 #define SSH_FXP_LSTAT 7 #define SSH_FXP_FSTAT 8 #define SSH_FXP_SETSTAT 9 #define SSH_FXP_FSETSTAT 10 #define SSH_FXP_OPENDIR 11 #define SSH_FXP_READDIR 12 #define SSH_FXP_REMOVE 13 #define SSH_FXP_MKDIR 14 #define SSH_FXP_RMDIR 15 #define SSH_FXP_REALPATH 16 #define SSH_FXP_STAT 17 #define SSH_FXP_RENAME 18 #define SSH_FXP_READLINK 19 #define SSH_FXP_SYMLINK 20 #define SSH_FXP_STATUS 101 #define SSH_FXP_HANDLE 102 #define SSH_FXP_DATA 103 #define SSH_FXP_NAME 104 #define SSH_FXP_ATTRS 105 #define SSH_FXP_EXTENDED 200 #define SSH_FXP_EXTENDED_REPLY 201 Additional packet types should only be defined if the protocol version number (see Section ``Protocol Initialization'') is incremented, and their use MUST be negotiated using the version number. However, the SSH_FXP_EXTENDED and SSH_FXP_EXTENDED_REPLY packets can be used to implement vendor-specific extensions. See Section ``Vendor-Specific-Extensions'' for more details. a� The flags bits are defined to have the following values: #define SSH_FILEXFER_ATTR_SIZE 0x00000001 #define SSH_FILEXFER_ATTR_UIDGID 0x00000002 #define SSH_FILEXFER_ATTR_PERMISSIONS 0x00000004 #define SSH_FILEXFER_ATTR_ACMODTIME 0x00000008 #define SSH_FILEXFER_ATTR_EXTENDED 0x80000000 a� The `pflags' field is a bitmask. The following bits have been defined. #define SSH_FXF_READ 0x00000001 #define SSH_FXF_WRITE 0x00000002 #define SSH_FXF_APPEND 0x00000004 #define SSH_FXF_CREAT 0x00000008 #define SSH_FXF_TRUNC 0x00000010 #define SSH_FXF_EXCL 0x00000020 a� Currently, the following values are defined (other values may be defined by future versions of this protocol): #define SSH_FX_OK 0 #define SSH_FX_EOF 1 #define SSH_FX_NO_SUCH_FILE 2 #define SSH_FX_PERMISSION_DENIED 3 #define SSH_FX_FAILURE 4 #define SSH_FX_BAD_MESSAGE 5 #define SSH_FX_NO_CONNECTION 6 #define SSH_FX_CONNECTION_LOST 7 #define SSH_FX_OP_UNSUPPORTED 8 c C s� i }| j D ]<}|�� D ].}t�d|�}|rt|�d�d�||�d�< qq | �t|�dkd� |�� D ]\}}| � |t t |�� qddS )z� The constants used by the SFTP protocol implementation match those found by searching through the spec. z)^\s*#define SSH_([A-Z_]+)\s+([0-9x]*)\s*$r" r r! z,No constants found (the test must be buggy).N) �filexferSpecExcerpts� splitlines�re�matchr �grouprc r% �itemsra �getattrr )r Z constantsZexcerpt�line�m�k�vr r r �test_constantsAgainstSpec s �z(ConstantsTests.test_constantsAgainstSpecN)r r r �__doc__r r� r� r r r r r r� � s & �Lr� c @ sP e Zd ZdZesdZdd� Zdd� Zdd� Zd d � Z d d � Z d d� Z dd� Z dS )�RawPacketDataTestsz� Tests for L{filetransfer.FileTransferClient} which explicitly craft certain less common protocol messages to exercise their handling. r� c C s t �� | _d S r. )r rW �ftcr r r r rP " s zRawPacketDataTests.setUpc C sT t �� }|�| j� || jjd< t�ddtj �t � d� t � d� }| j� |� |S )a/ A STATUS packet containing a result code, a message, and a language is parsed to produce the result of an outstanding request L{Deferred}. @see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>} of the SFTP Internet-Draft. r! �!LL� msg� lang) r �Deferredrr �_cbTestPacketSTATUSr � openRequestsr� r� r �FX_OKr r� � packet_STATUS�r rs r� r r r �test_packetSTATUS& s $ z$RawPacketDataTests.test_packetSTATUSc C s$ | � |d d� | � |d d� dS )z{ Assert that the result is a two-tuple containing the message and language from the STATUS packet. r r r! r Nr� �r �resultr r r r 6 s z&RawPacketDataTests._cbTestPacketSTATUSc C s@ t �� }|�| j� || jjd< t�ddtj �}| j� |� |S )az A STATUS packet containing only a result code can also be parsed to produce the result of an outstanding request L{Deferred}. Such packets are sent by some SFTP implementations, though not strictly legal. @see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>} of the SFTP Internet-Draft. r! r ) r r rr �_cbTestPacketSTATUSShortr r r� r� r r r r r r r �test_packetSTATUSShort? s z)RawPacketDataTests.test_packetSTATUSShortc C s$ | � |d d� | � |d d� dS )z� Assert that the result is a two-tuple containing empty strings, since the STATUS packet had neither a message nor a language. r r� r! Nr� r r r r r P s z+RawPacketDataTests._cbTestPacketSTATUSShortc C sJ t �� }|�| j� || jjd< t�ddtj �t � d� }| j� |� |S )a� A STATUS packet containing a result code and a message but no language can also be parsed to produce the result of an outstanding request L{Deferred}. Such packets are sent by some SFTP implementations, though not strictly legal. @see: U{section 9.1<http://tools.ietf.org/html/draft-ietf-secsh-filexfer-13#section-9.1>} of the SFTP Internet-Draft. r! r r ) r r rr � _cbTestPacketSTATUSWithoutLangr r r� r� r r r r� r r r r r � test_packetSTATUSWithoutLangY s z/RawPacketDataTests.test_packetSTATUSWithoutLangc C s$ | � |d d� | � |d d� dS )z� Assert that the result is a two-tuple containing the message from the STATUS packet and an empty string, since the language was missing. r r r! r� Nr� r r r r r k s z1RawPacketDataTests._cbTestPacketSTATUSWithoutLangN) r r r r r r� rP r r r r r r r r r r r s r )1r Z __future__r r r0 r� r� Ztwisted.python.reflectr Z twisted.trialr r r� Z twisted.conchr Ztwisted.conch.sshr r r r Ztwisted.internetr Ztwisted.protocolsr Ztwisted.pythonr Ztwisted.python.compatr r Ztwisted.python.filepathr r r r- r5 �hasattrr� �warnr6 r7 ZregisterAdapterrd ZTestCaserA rQ r� r� r� r r r r r �<module> sR � � Pjj
Save File
Cancel