o pjG@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZdd lmZdd lmZmZmZmZmZdd lmZdd lm Z m!Z!dd l"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/e+dZ0e+dZ1e+dZ2e1re0rz(ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:m;Z;ddlZ>Wn e?yYnwdZ@de2e1e0eAedfvrdZ@e e@dGd d!d!e/ZBGd"d#d#e/ZCGd$d%d%e-ZDGd&d'd'ZEGd(d)d)ZFe eGd*d+d+e ZGe e@dGd,d-d-e/ZHGd.d/d/ZIGd0d1d1ejJZKGd2d3d3eZLe e@dGd4d5d5eLZMe e@dGd6d7d7eLZNe e@de e)d8 d9Gd:d;d;eLZOdS)/usr/lib/python3/dist-packages/twisted/conch/test/test_cftp.pytest_eofReceived>s z SSHSessionTests.test_eofReceivedN)__name__ __module__ __qualname____doc__r+r)r)r)r*r"8s r"c@seZdZdZeedddurdZddZddZd d Z d d Z d dZ e Zzz e e jdWn e jy>dZYnwe e jeWnYeedddZddZdS) ListingTestsz Tests for L{lsLine}, the function which generates an entry for a file or directory in an SFTP I{ls} command's output. tzsetNz8Cannot test timestamp formatting code without time.tzsetcsjd_fdd}td|dtjvr*tjtjdtjdtj dSdd}|dS) zo Patch the L{ls} module's time function so the results of L{lsLine} are deterministic. i[csjSN)nowr)r'r)r*fakeTimeZsz$ListingTests.setUp..fakeTimetimeTZcSs,ztjd=Wn tyYnwtdS)Nr7)osenvironKeyErrorr6r1r)r)r)r*cleanupfs    z#ListingTests.setUp..cleanupN) r3patchrr8r9 addCleanupoperatorsetitemr6r1)r'r5r;r)r4r*setUpSs  zListingTests.setUpcCs|tjd<ttd|S)zl Call L{ls.lsLine} after setting the timezone to C{timezone} and return the result. r7foo)r8r9r6r1rlsLine)r'timezonestatr)r)r* _lsInTimezoneps  zListingTests._lsInTimezonec CT|jd}tdddddddd|df }||d|d||d|ddS)z A file with an mtime six months (approximately) or more in the past has a listing including a low-resolution timestamp. rAmerica/New_Yorkz;!--------- 0 0 0 0 Apr 26 1973 fooPacific/Aucklandz;!--------- 0 0 0 0 Apr 27 1973 fooNr3r8 stat_result assertEqualrEr'thenrDr)r)r* test_oldFile{   zListingTests.test_oldFilec CX|jdd}tdddddddd|df }||d|d||d|ddS) A file with a high-resolution timestamp which falls on a day of the month which can be represented by one decimal digit is formatted with one padding 0 to preserve the columns which come after it. rGirrHz;!--------- 0 0 0 0 May 01 1973 foorIz;!--------- 0 0 0 0 May 02 1973 fooNrJrMr)r)r*test_oldSingleDigitDayOfMonth  z*ListingTests.test_oldSingleDigitDayOfMonthc CrF)z A file with an mtime fewer than six months (approximately) in the past has a listing including a high-resolution timestamp excluding the year. 逛zrrH;!--------- 0 0 0 0 Aug 28 17:33 foorI;!--------- 0 0 0 0 Aug 29 09:33 fooNrJrMr)r)r* test_newFilerPzListingTests.test_newFile es_AR.UTF8Tz'The es_AR.UTF8 locale is not installed.c Cs||jd}tdddddddd|df }t}ttjd|tjtj||| d|d|| d|ddS) zC The month name in the date is locale independent. rUrrYrHrVrIrWN) r3r8rKlocale getlocale setlocaleLC_ALLr=rLrE)r'rNrD currentLocaler)r)r*test_localeIndependents   z#ListingTests.test_localeIndependentc CrQ) rRrUiFrrHz;!--------- 0 0 0 0 Sep 01 17:33 foorIz;!--------- 0 0 0 0 Sep 02 09:33 fooNrJrMr)r)r*test_newSingleDigitDayOfMonthrTz*ListingTests.test_newSingleDigitDayOfMonth)r,r-r.r/getattrr6skipr@rErOrSrXrZr[r^r\r]Error localeSkiprr_r`r)r)r)r*r0Js.   r0cs eZdZdZfddZZS)InMemorySSHChannelzp Minimal implementation of a L{SSHChannel} like class which only reads and writes data from memory. cs||_d|_tdS)zt @param conn: The SSH connection associated with this channel. @type conn: L{SSHConnection} rN)conn localClosedsuper__init__)r'rf __class__r)r*riszInMemorySSHChannel.__init__)r,r-r.r/ri __classcell__r)r)rjr*resrec@(eZdZdZddZddZddZdS) FilesystemAccessExpectationszC A test helper used to support expected filesystem access. cCs i|_dSr2_cacher4r)r)r*ri z%FilesystemAccessExpectations.__init__cCs||j||f<dS)z @param path: Path at which the stream is requested. @type path: L{str} @param path: Flags with which the stream is requested. @type path: L{str} @param stream: A stream. @type stream: C{File} Nro)r'pathflagsstreamr)r)r*puts z FilesystemAccessExpectations.putcCs|j||fS)a Remove a stream from the memory. @param path: Path at which the stream is requested. @type path: L{str} @param path: Flags with which the stream is requested. @type path: L{str} @return: A stream. @rtype: C{File} )rppop)r'rrrsr)r)r*rvs z FilesystemAccessExpectations.popN)r,r-r.r/rirurvr)r)r)r*rns  rnc@s eZdZdZddZddZdS)InMemorySFTPClienta' A L{filetransfer.FileTransferClient} which does filesystem operations in memory, without touching the local disc or the network interface. @ivar _availableFiles: File like objects which are available to the SFTP client. @type _availableFiles: L{FilesystemRegister} cCs t||_ddd|_||_dS)N )requests buffersize)re transportoptions_availableFiles)r'availableFilesr)r)r*ri,s  zInMemorySFTPClient.__init__cCs|j||S)z} @see: L{filetransfer.FileTransferClient.openFile}. Retrieve and remove cached file based on flags. )r~rv)r'filenamersattrsr)r)r*openFile4szInMemorySFTPClient.openFileN)r,r-r.r/rirr)r)r)r*rw"s rwc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)InMemoryRemoteFilez; An L{ISFTPFile} which handles all data in memory. cCs||_t|dS)zL @param name: Name of this file. @type name: L{str} N)namerri)r'rr)r)r*riCszInMemoryRemoteFile.__init__cCs||||t|S)z/ @see: L{ISFTPFile.writeChunk} )seekwriter succeed)r'startdatar)r)r* writeChunkKs   zInMemoryRemoteFile.writeChunkcCs d|_dS)zo @see: L{ISFTPFile.writeChunk} Keeps data after file was closed to help with testing. TN)_closedr4r)r)r*closeS zInMemoryRemoteFile.closecCdSr2r)r4r)r)r*getAttrs[zInMemoryRemoteFile.getAttrscCrr2r))r'offsetlengthr)r)r* readChunk_rzInMemoryRemoteFile.readChunkcCrr2r))r'rr)r)r*setAttrscrzInMemoryRemoteFile.setAttrscCs t|S)zb Get current data of file. Allow reading data event when file is closed. )rgetvaluer4r)r)r*rgrzInMemoryRemoteFile.getvalueN) r,r-r.r/rirrrrrrr)r)r)r*r=s rc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZd)ddZd*dd Zd!d"Zd#d$Zd%d&Zd'd(ZdS)+StdioClientTestsz( Tests for L{cftp.StdioClient}. cCsTt|_t|j}t||_d|j_t|_|j_ | dd|jjj |j_ dS)zm Create a L{cftp.StdioClient} hooked up to dummy transport and a fake user database. /iN) rnfakeFilesystemrwr StdioClientclientcurrentDirectoryrdatabase_pwdsetKnownConsoleSizer|)r' sftpClientr)r)r*r@vs   zStdioClientTests.setUpc Cs@|jtdtdddtj|j d}| |j d|S)zs The I{exec} command runs its arguments locally in a child process using the user's shell. secretrAbarzexec print(1 + 2)s3 ) raddUsergetpassgetuserr8getuidsys executabler_dispatchCommand addCallbackrLr'dr)r)r* test_execs  zStdioClientTests.test_execc C>|jtdtdddd|jd}||j d|S)zr If the local user has no shell, the I{exec} command runs its arguments using I{/bin/sh}. rrrArzexec echo hellohello rrrrr8rrrrrLrr)r)r*test_execWithoutShells  z&StdioClientTests.test_execWithoutShellc Cr)zO The I{exec} command is run for lines which start with C{"!"}. rrrArz/bin/shz !echo hellorrrr)r)r* test_bangs  zStdioClientTests.test_bangcs2ddlGfddd}|td|dS)a For the duration of this test, patch C{cftp}'s C{fcntl} module to return a fixed width and height. @param width: the width in characters @type width: L{int} @param height: the height in characters @type height: L{int} rNcseZdZfddZdS)z7StdioClientTests.setKnownConsoleSize..FakeFcntlcs&|jkr |dtdddS)Nz#Only window-size queries supported.4Hr) TIOCGWINSZfailstructpack)r'fdoptmutateheightttywidthr)r*ioctls  z=StdioClientTests.setKnownConsoleSize..FakeFcntl.ioctlN)r,r-r.rr)rr)r* FakeFcntlsrfcntl)rr<r)r'rrrr)rr*rs z$StdioClientTests.setKnownConsoleSizecCs|ddt}|j_td}d|_t|}d|_| }| d|j d7_ |j ||d}| |jj|d S) a( L{StdioClient._printProgressBar} prints a progress description, including percent done, amount transferred, transfer rate, and time remaining, all based the given start time, the given L{FileWrapper}'s progress information and the reactor's current time. ry"xsamplei(g@is# b'sample' 40% 4.0kB 2.0kBps 00:03 N)rrrrrrr FileWrappersizesecondsadvancetotal_printProgressBarrLr|valuer'clockwrappedwrapper startTimeresultr)r)r*test_printProgressBarReportings   z/StdioClientTests.test_printProgressBarReportingcCsd|ddt}|j_td}d|_t|}|}|j ||d}| |jj |dS)z L{StdioClient._printProgressBar} prints a progress description that indicates 0 bytes transferred if no bytes have been transferred and no time has passed. ryrrrs! b'sample' 0% 0.0B 0.0Bps 00:00 N) rrrrrrrrrrrLr|rrr)r)r*test_printProgressBarNoProgresss  z0StdioClientTests.test_printProgressBarNoProgresscCsL|ddt}d|_t|}|j|dd}|||jj dS)z5 Print the progress for empty files. ryrs empty-filers% b'empty-file'100% 0.0B 0.0Bps 00:00 N) rrrrrrrrLr|r)r'rrrr)r)r*test_printProgressBarEmptyFiles  z/StdioClientTests.test_printProgressBarEmptyFilecC|jd}|d|dS)zK Returns empty value for both filename and remaining data. z )rrNr _getFilenamerLr'rr)r)r*test_getFilenameEmptys z&StdioClientTests.test_getFilenameEmptycCr)zd Returns empty value for remaining data when line contains only a filename. only-local)rrNrrr)r)r*test_getFilenameOnlyLocal z*StdioClientTests.test_getFilenameOnlyLocalcCr)ze Returns filename and remaining data striped of leading and trailing spaces. z local remote file )localz remote fileNrrr)r)r*test_getFilenameNotQuoted rz*StdioClientTests.test_getFilenameNotQuotedcCr)z Returns filename and remaining data not striped of leading and trailing spaces when quoted paths are requested. z" " local file " " remote file " )z local file z" remote file "Nrrr)r)r*test_getFilenameQuotedrz'StdioClientTests.test_getFilenameQuotedNcCsJ|dur|}t|d}||Wd|S1swY|S)ab Create a local file and return its path. When `path` is L{None}, it will create a new temporary file. @param path: Optional path for the new file. @type path: L{str} @param content: Content to be written in the new file. @type content: L{bytes} @return: Path to the newly create file. Nwb)mktempopenr)r'rrcontentfiler)r)r*makeFiles   zStdioClientTests.makeFileFcCsh|jj}|d}|d}g}g}|D]\}}}g} |D] } | |d| q| d|d||| |ddd} | dd } | d d d } | | g}| dd D]$} |  dd d} | dd } || dd| d qd|| d ||q|r| t |t |n| ||| dt |d dS)a Check output of cftp client for a put request. @param transfers: List with tuple of (local, remote, progress). @param randomOrder: When set to C{True}, it will ignore the order in which put reposes are received utf-8z  z Transferred z to r N rxz5There are still put responses which were not checked.) rr|rdecodesplitappendrvstripextendrsplitrLsortedlen)r' transfers randomOrderoutputexpectedOutput actualOutputrremoteexpectedexpectedTransferline progressPartsactuallastactualTransferr)r)r*checkPutMessage3s<     "  z StdioClientTests.checkPutMessagecCsd}|j|d}tjtjBtjB}tjdtj|}t |}|j ||t |d|jjjd<|j|}|||||||j|||gdfgdS)z A name based on local path is used when remote path is not provided. The progress is updated while chunks are transferred. s Test Content)rrryr{)z 76% 10.0B 100% 13.0BrN)rr FXF_WRITE FXF_CREAT FXF_TRUNCr8rrjoinbasenamerrrur rrr}cmd_PUTsuccessResultOfrLrr%rr)r'r localPathrs remoteName remoteFiledeferredr)r)r*test_cmd_PUTSingleNoRemotePathps    z/StdioClientTests.test_cmd_PUTSingleNoRemotePathcCs|}tjtjBtjB}d}t|}|j||t ||j |d|d}| || ||dgfg||j|d|dS)z Remote path is extracted from first filename after local file. Any other data in the line is ignored. z /remote-pathrz ignored 100% 0.0BrN)rrrrrrrrur rrr r rr%rrLr)r'r rsrrrr)r)r*test_cmd_PUTSingleRemotePaths  z-StdioClientTests.test_cmd_PUTSingleRemotePathc Cs|}tj|}d}tj|}|jtj||d}tjtjBtj B}d|}d|}t |} t |} |j ||t | |j ||t | |jtj|d} || || j|d| || j|d| |j||dgf||dgfgddd S) z When a gobbing expression is used local files are transferred with remote file names based on local names. second-namerrr*rrTrN)rr8rrr dirnamer rrrrrrrur rrr r r%rrLrr) r'first firstName secondNameparentsecondrsfirstRemotePathsecondRemotePathfirstRemoteFilesecondRemoteFilerr)r)r* test_cmd_PUTMultipleNoRemotePaths4           z1StdioClientTests.test_cmd_PUTMultipleNoRemotePathc Cs |}tj|}d}tj|}|jtj||d}tjtjBtj B}t |}t |}d|} d|} |j | |t ||j | |t ||jdtj|d} || ||j|d|||j|d||j||dgf||dgfgdd d S) z When a gobbing expression is used local files are transferred with remote file names based on local names. when a remote folder is requested remote paths are composed from remote path and local filename. rrz/remote/z {} remoterrrTrN)rr8rrr rr rrrrrrrur rrr formatr r%rrLrr) r'rrrrrrsr r!rrrr)r)r*"test_cmd_PUTMultipleWithRemotePaths4           z3StdioClientTests.test_cmd_PUTMultipleWithRemotePath)Nr)F)r,r-r.r/r@rrrrrrrrrrrrrrrr"r$r)r)r)r*rps(     = #rc@seZdZddZddZdS)FileTransferTestRealmcCs ||_dSr2)testDir)r'r&r)r)r*rirqzFileTransferTestRealm.__init__cGst|j}|d|ddfS)NrcSrr2r)r)r)r)r*sz5FileTransferTestRealm.requestAvatar..)rr&)r'avatarIDmindr ar)r)r* requestAvatars z#FileTransferTestRealm.requestAvatarN)r,r-r.rir+r)r)r)r*r%s r%c@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)SFTPTestProcessz Protocol for testing cftp. Provides an interface between Python (where all the tests are) and the cftp client process (which does the work that is being tested). cCs$|||_d|_d|_d|_dS)zr @param onOutReceived: A L{Deferred} to be fired as soon as data is received from stdout. NF) clearBuffer onOutReceived onProcessEnd_expectingCommand _processEnded)r'r.r)r)r*ris  zSFTPTestProcess.__init__cCsd|_g|_d|_dS)zR Clear any buffered data received from stdout. Should be private. rN)buffer_linesReceived _lineBufferr4r)r)r*r-s zSFTPTestProcess.clearBuffercCsrtd||j|d}|d|_|j||jdur,|jd}|_|||j |7_ | dS)zO Called by Twisted when the cftp client prints data to stdout. zgot %r rN) rmsgr4rrvr3rr.callbackr2_checkForCommand)r'rlinesrr)r)r* outReceiveds     zSFTPTestProcess.outReceivedcCsfd}|jr/|j|kr1d|j}||r|t|d}||jd}|_||dSdSdS)Nscftp> r5)r0r4r r3 startswithrr-r7)r'promptbufrr)r)r*r8s  z SFTPTestProcess._checkForCommandcCstd|dS)zO Called by Twisted when the cftp client prints data to stderr. zerr: %sN)rr6)r'rr)r)r* errReceived"szSFTPTestProcess.errReceivedcCs|jS)zQ Return the contents of the buffer of data received from stdout. )r2r4r)r)r* getBuffer(szSFTPTestProcess.getBuffercCs<t|_|t|tr|d}|j|d|jS)a Issue the given command via the cftp client. Return a C{Deferred} that fires when the server returns a result. Note that the C{Deferred} will callback even if the server returns some kind of error. @param command: A string containing an sftp command. @return: A C{Deferred} that fires when the sftp server returns a result. The payload is the server's response string. rr5) r Deferredr0r- isinstancestrencoder|rr'commandr)r)r* runCommand.s   zSFTPTestProcess.runCommandcs(tdfdd|D}t|S)ax Run each command in sequence and return a Deferred that fires when all commands are completed. @param commands: A list of strings containing sftp commands. @return: A C{Deferred} that fires when all commands are completed. The payload is a list of response strings from the server, in the same order as the commands. rxcsg|] }j|qSr))runrF).0rEr'semr)r* Lsz-SFTPTestProcess.runScript..)r DeferredSemaphore gatherResults)r'commandsdlr)rIr* runScript@s  zSFTPTestProcess.runScriptcCs,|jrtdSt|_|jd|jS)z Kill the process if it is still running. If the process is still running, sends a KILL signal to the transport and returns a C{Deferred} which fires when L{processEnded} is called. @return: a C{Deferred}. NKILL)r1r rr@r/r| signalProcessr4r)r)r* killProcessOs    zSFTPTestProcess.killProcesscCs,d|_|jr|jd}|_|ddSdS)zF Called by Twisted when the cftp client process ends. TN)r1r/r7)r'reasonrr)r)r* processEnded^s zSFTPTestProcess.processEndedN)r,r-r.r/rir-r:r8r>r?rFrPrSrUr)r)r)r*r,s   r,c@s4eZdZddZddZddZddZd d Zd S) CFTPClientTestBasecCstdd}|tjWdn1swYtdd}|tjWdn1s1wYtddtdd}|dtjWdn1sTwYt |S)N dsa_test.pubrdsa_testikh_tests 127.0.0.1 ) rrrpublicDSA_opensshprivateDSA_opensshr8chmodpublicRSA_opensshrr@r'fr)r)r*r@is     zCFTPClientTestBase.setUpcCsFt|j}t|}|tt}||_tj d|dd|_ dS)Nrz 127.0.0.1) interface) r%r&r PortalregisterCheckerrconchTestPublicKeyCheckerConchTestServerFactoryr listenTCPserver)r'realmpfacr)r)r* startServerss  zCFTPClientTestBase.startServercCsHt|jjds |dSd|jjj_t|jjjjj }| |j|S)Nprotorx) hasattrrffactory _cbStopServerrkexpectedLoseConnectionr maybeDeferredr|loseConnectionrrr)r)r* stopServer{s    zCFTPClientTestBase.stopServercCst|jjSr2)r rprf stopListeningr'ignoredr)r)r*rnsz CFTPClientTestBase._cbStopServerc Cs4dD]}zt|WqtyYqwt|S)N)rWrXrY)r8remove BaseExceptionrtearDownr^r)r)r*rxs  zCFTPClientTestBase.tearDownN)r,r-r.r@rjrrrnrxr)r)r)r*rVhs   rVc@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ d-ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,ZdS).OurServerCmdLineClientTestsz Functional tests which launch a SFTP server over TCP on localhost and check cftp command line interface using a spawned process. Due to the spawned process you can not add a debugger breakpoint for the client code. c s0td}jj}tj||dd}t dt j d|t }t|_|fddtj}tjt j|d<g}i}|D]}t|trZ|d }||qN|D]}||} t|trr|d }t| tr|| d } | ||<qbt |t |tjjt j ||d |S) Nz-p %i -l testuser --known-hosts kh_test --user-authentications publickey --host-key-algorithms ssh-rsa -i dsa_test -a -v 127.0.0.1rmodrunning rc jSr2)processProtocolr-_r4r)r*r' z3OurServerCmdLineClientTests.setUp.. PYTHONPATHrenv)rVr@rjrfgetHostportr _makeArgsrrr6rrr r@r,r~rr8r9copypathsepr rrrArBrCrr spawnProcess) r'cmdsrrr encodedCmds encodedEnvcmdvarvalr)r4r*r@s>               z!OurServerCmdLineClientTests.setUpcs}|fdd|S)Ncr}r2)r~rSrr4r)r*r'rz6OurServerCmdLineClientTests.tearDown..)rrrrr)r4r*rxsz$OurServerCmdLineClientTests.tearDowncCs,z |jjdWdStjyYdSw)NrQ)r~r|rRr ProcessExitedAlreadyrtr)r)r* _killProcesss z(OurServerCmdLineClientTests._killProcesscC |j|S)z Run the given command with the cftp client. Return a C{Deferred} that fires when the command is complete. Payload is the server's output for that command. )r~rFrDr)r)r*rF z&OurServerCmdLineClientTests.runCommandcGr)z Run the given commands with the cftp client. Returns a C{Deferred} that fires when the commands are all complete. The C{Deferred}'s payload is a list of output for each command. )r~rP)r'rNr)r)r*rPrz%OurServerCmdLineClientTests.runScriptcCsL|j}|ddddd}dd}||||j|jtd|jg|S)z Test that 'pwd' reports the current remote directory, that 'lpwd' reports the current local directory, and that changing to a subdirectory then changing to its parent leaves you in the original remote directory. pwdlpwdzcd testDirectorycd ..cSsDg}|D]}t|tr|d}||q|dd|ddS)zH Callback function for handling command output. rN)rAbytesrr)rrrr)r)r* cmdOutputs    z8OurServerCmdLineClientTests.testCdPwd..cmdOutputr)r&rPrrLrrr8getcwd)r'homeDirrrr)r)r* testCdPwds  z%OurServerCmdLineClientTests.testCdPwdcs&fdd}dddd}||S)z Check that 'ls -l' output includes the access permissions and that this output changes appropriately with 'chmod'. csZ|dd|dd|dd|d|dddS)Nrs -rw-r--r--rxrrs ----------r)flushLoggedErrorsr%r;rLresultsr4r)r*_checks z7OurServerCmdLineClientTests.testChAttrs.._checkzls -l testfile1zchmod 0 testfile1zchmod 644 testfile1rPrr'rrr)r4r* testChAttrss  z'OurServerCmdLineClientTests.testChAttrscs@fdd}ddjddd}|dd ||S) z Check 'ls' works as expected. Checks for wildcards, hidden files, listing directories and listing empty directories. csf|dgd|dgd|dddg|dgd|d d gdS) Nrs testDirectorytestRemoveFiletestRenameFiles testfile1rxrrrr)s.testHiddenFilerrrrrLrr4r)r*rs z4OurServerCmdLineClientTests.testList.._checkrzls ../zls *Filez ls -a *Filezls -l testDirectorycSsdd|DS)NcSsg|]}|dqS)r5)r)rHxr)r)r*rK*zJOurServerCmdLineClientTests.testList....r))xsr)r)r*r'*sz6OurServerCmdLineClientTests.testList..)rPr&r rrr)r4r*testLists   z$OurServerCmdLineClientTests.testListcCsD|d}tdd}t|tr|d}||j ||S)zB Check that running the '?' command returns help. ?Nrr) rFrrcmd_HELPrrArBrCrrL)r'rhelpTextr)r)r*testHelp-s   z$OurServerCmdLineClientTests.testHelpNcCs||||dS)zg Assert that the files at C{name1} and C{name2} contain exactly the same data. N)rL getContent)r'name1name2r6r)r)r*assertFilesEqual9sz,OurServerCmdLineClientTests.assertFilesEqualcsldjjjjttrdfdd}djjd}|||fdd|S) z Test that 'get' saves the remote file to the correct local location, that the output of 'get' is correct and that 'rm' actually removes the file. z)Transferred {}/testfile1 to {}/test file2rcs8|jdjdddS)N testfile1 test file2z get failedzrm "test file2")r%endswithrr&childrFrrr'r)r* _checkGetNs   z6OurServerCmdLineClientTests.testGet.._checkGetzget testfile1 "z /test file2"cjdS)Nr assertFalser&rexistsrr4r)r*r'Zrz5OurServerCmdLineClientTests.testGet..)r#r&rrrArBrCrFr)r'rrr)rr*testGet@s    z#OurServerCmdLineClientTests.testGetcs fdd}d}||S)zQ Test that 'get' works correctly when given wildcard parameters. cs8jdtddjdtdddS)NtestRemoveFiletestRemoveFile get failedtestRenameFiletestRenameFile get failed)rr&rr)rur4r)r*rcs  z;OurServerCmdLineClientTests.testWildcardGet.._checkz get testR*)rFrrr)r4r*testWildcardGet^s  z+OurServerCmdLineClientTests.testWildcardGetcshdjjdjjdfdd}djjd}|||fdd |S) z Check that 'put' uploads files correctly and that they can be successfully removed. Also check the output of the put command. s Transferred s/testfile1 to s /test"file2cs6jdjd|dS)Nr test"file2zrm "test\"file2")rr&rr%rrFrrr)r* _checkPuts  z6OurServerCmdLineClientTests.testPut.._checkPutput z/testfile1 "test\"file2"cr)Nrrrr4r)r*r'rz5OurServerCmdLineClientTests.testPut..)r& asBytesModerrrFr)r'rrr)rr*testPutrs"    z#OurServerCmdLineClientTests.testPutcsjdjdd }|dWdn1swYjdjdd }|dWdn1s9wYfdd }d jjd }|||S) zb Check that 'put' uploads files correctly when overwriting a longer file. shorterFilewmodeaN longerFilebbcs jdjddS)Nrr)rr&rrr4r)r*rszEOurServerCmdLineClientTests.test_putOverLongerFile.._checkPutrz/shorterFile longerFile)r&rrrrFrrrr'r_rrr)r4r*test_putOverLongerFiles    z2OurServerCmdLineClientTests.test_putOverLongerFilecsjddjdd }|dWdn1s#wYjdjdd }|dWdn1sBwYfdd }d jjd }|||S) z Check that 'put' uploads files correctly when overwriting a longer file and you use a wildcard to specify the files to upload. dirrrrrNrcsdjddS)Nr)rrr&rr'someDirr)r*rszMOurServerCmdLineClientTests.test_putMultipleOverLongerFile.._checkPutrz/dir/*)r&rcreateDirectoryrrrFrrrrr)rr*test_putMultipleOverLongerFiles    z:OurServerCmdLineClientTests.test_putMultipleOverLongerFilecs>fdd}ddjjddj}|||S)z What happens if you issue a 'put' command and include a wildcard (i.e. '*') in parameter? Check that all files matching the wildcard are uploaded to the correct directory. csh|dd|ddjdjddjdjdddS)Nrrrrrrr)rLrr&rrrr4r)r*checks  z:OurServerCmdLineClientTests.testWildcardPut..checkrrz/testR*zcd %s)rPr&rrr r)r'rrr)r4r*testWildcardPuts   z+OurServerCmdLineClientTests.testWildcardPutc4fdd}dd}|||jd|S)z Test that 'ln' creates a file which appears as a link in the output of 'ls'. Check that removing the new file succeeds without output. cs8|dd|ddddS)Nrrrxlz link failedz rm testLink)rrLr%r;rFrr4r)r*rs z4OurServerCmdLineClientTests.testLink.._checkzln testLink testfile1zls -l testLinkrrPrrLrr)r4r*testLinks   z$OurServerCmdLineClientTests.testLinkcr)zV Test that we can create and remove directories with the cftp client. cs.|dd|dddS)Nrrrxdzrmdir testMakeDirectory)rLr%r;rFrr4r)r*rs z?OurServerCmdLineClientTests.testRemoteDirectory.._checkmkdir testMakeDirectoryzls -l testMakeDirector?rrrr)r4r*testRemoteDirectory   z/OurServerCmdLineClientTests.testRemoteDirectorycs&fdd}dd}|||S)z Test that a C{mkdir} on an existing directory fails with the appropriate error, and doesn't log an useless error server side. cs$|dd|dddS)Nrrrxsremote error 11: mkdir failedrrr4r)r*rszHOurServerCmdLineClientTests.test_existingRemoteDirectory.._checkrrrr)r4r*test_existingRemoteDirectorys   z8OurServerCmdLineClientTests.test_existingRemoteDirectorycsHdjjd}|jd|fdd|jd|S)z Test that we can create a directory locally and remove it with the cftp client. This test works because the 'remote' server is running out of a local directory. zlmkdir z/testLocalDirectoryrcs dS)Nzrmdir testLocalDirectory)rFrr4r)r*r' rz@OurServerCmdLineClientTests.testLocalDirectory..)rFr&rrrrLrr)r4r*testLocalDirectorys z.OurServerCmdLineClientTests.testLocalDirectorycr)z1 Test that we can rename a file. cs*|dd|dddS)Nrrrxs testfile2zrename testfile2 testfile1)rLrFrr4r)r*rs z6OurServerCmdLineClientTests.testRename.._checkzrename testfile1 testfile2z ls testfile?rrrr)r4r* testRenamerz&OurServerCmdLineClientTests.testRenamer2)r,r-r.r/r@rxrrFrPrrrrrrrrrrrrrrrrr)r)r)r*rys.*  ryc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)OurServerBatchFileTestszm Functional tests which launch a SFTP server over localhost and checks csftp in batch interface. cCst||dSr2rVr@rjr4r)r)r*r@%s  zOurServerBatchFileTests.setUpcCt||Sr2)rVrxrrr4r)r)r*rx) z OurServerBatchFileTests.tearDowncs|td }||Wdn1swY|jj}d|f}tj|dddd}t dt j d|t j}t jt j|d<d|jj_tt j ||d }fd d }|d d |||S)Nrz-p %i -l testuser --known-hosts kh_test --user-authentications publickey --host-key-algorithms ssh-rsa -i dsa_test -a -v -b %s 127.0.0.1rrzrxr|rrrcst|Sr2)r8rvresfnr)r*_cleanupDs z9OurServerBatchFileTests._getBatchOutput.._cleanupcSs|dS)Nrr)rr)r)r*r'Hsz9OurServerBatchFileTests._getBatchOutput..)rrrrfrrrrrrr6rrr8r9rrr rrrmrorraddBoth)r'r_fprrrrrr)rr*_getBatchOutput-s$        z'OurServerBatchFileTests._getBatchOutputc(d}fdd}|}|||S)z Test whether batch file function of cftp ('cftp -b batchfile'). This works by treating the file as a list of commands to be run. z pwd ls exit csP|d}tdt|jj|d|ddgddS)Nr5zRES %srxrr) rrr6reprassertInr&rrrrLrr4r)r*_cbCheckResultWs  z=OurServerBatchFileTests.testBatchFile.._cbCheckResultrrr'rrrr)r4r* testBatchFileMs   z%OurServerBatchFileTests.testBatchFilecr)zO Test that an error in the batch file stops running the batch. zchown 0 missingFile pwd exit cjj|dSr2) assertNotInr&rrrrr4r)r*rmz9OurServerBatchFileTests.testError.._cbCheckResultrrr)r4r* testErrords    z!OurServerBatchFileTests.testErrorcr)z_ Test that a minus sign '-' at the front of a line ignores any errors. z-chown 0 missingFile pwd exit crr2)rr&rrrrr4r)r*r~rz@OurServerBatchFileTests.testIgnoredError.._cbCheckResultrrr)r4r*testIgnoredErrorts    z(OurServerBatchFileTests.testIgnoredErrorN) r,r-r.r/r@rxrrrrr)r)r)r*rs  rsftpz%no sftp command-line client availablec@rm) OurServerSftpClientTestsz@ Test the sftp server against sftp command line client. cCrr2rr4r)r)r*r@rzOurServerSftpClientTests.setUpcCs|Sr2)rrr4r)r)r*rxsz!OurServerSftpClientTests.tearDowncstd }|dWdn1swYjjtjfdd}td|djj _ t dd }fd d }fd d }| || dd| |S)a Test the return of extended attributes by the server: the sftp client should ignore them, but still be able to parse the response correctly. This test is mainly here to check that L{filetransfer.FILEXFER_ATTR_EXTENDED} has the correct value. rz ls . exitNcs||}d|d<|S)Nrext_foor))r'sr) oldGetAttrr)r* _getAttrss zCOurServerSftpClientTests.test_extendedAttributes.._getAttrsrTssh)-oPubkeyAcceptedKeyTypes=ssh-dssz-Vcs>|dkrd}nd}|dddddddd dd fd d f 7}|S) Nr)rrr)z-Fz /dev/nullrzIdentityFile=dsa_testzUserKnownHostsFile=kh_testzHostKeyAlgorithms=ssh-rsazPort=%iz-bztestuser@127.0.0.1r))statusargs)rrr)r*hasPAKTs&zAOurServerSftpClientTests.test_extendedAttributes..hasPAKTcs.|dddD] }||dq dS)Nrrr)rLr)rir4r)r*rsz?OurServerSftpClientTests.test_extendedAttributes..checkcSs td|S)Nr)r)r r)r)r*r'rzBOurServerSftpClientTests.test_extendedAttributes..) rrrrfrrr!rr<rmrorr)r'r_rrr rr))rrrr'r*test_extendedAttributess         z0OurServerSftpClientTests.test_extendedAttributesN)r,r-r.r/r@rxr r)r)r)r*rs  r)Pr/rrZr>r8rrr6iorunittestrzope.interfacer twisted.conchrtwisted.conch.interfacesr$twisted.conch.test.test_filetransferrr twisted.credr twisted.internetr r r r rtwisted.internet.taskrtwisted.internet.utilsrrtwisted.pythonrtwisted.python.fakepwdrtwisted.python.filepathrtwisted.python.procutilsrtwisted.python.reflectrtwisted.test.proto_helpersrtwisted.trial.unittestrrrunixtwisted.conch.scriptsrtwisted.conch.scripts.cftprtwisted.conch.sshrtwisted.conch.testrrtwisted.conch.test.test_conchr r! ImportError skipTestsIReactorProcessr"r0rernrwrrr%ProcessProtocolr,rVryrrr)r)r)r*s                   #&2u {'g