o >h@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZddlmZddlmZddlmZmZdd lmZdd lmZmZmZmZmZdd lmZdd l 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-dZ1e0rz.ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;mZ>ddlm?Z?Wn e@yYnwdZAde1e0eBedfvrdZAe eAdGd d!d!e/ZCGd"d#d#e/ZDGd$d%d%e!ZEGd&d'd'ZFGd(d)d)ZGeeGd*d+d+e ZHe eAdGd,d-d-e/ZIGd.d/d/ZJGd0d1d1ejKZLGd2d3d3eZMe eAdGd4d5d5eMZNe eAdGd6d7d7eMZOe eAde e+d8 d9e e+d: d;Gdz* Tests for L{twisted.conch.scripts.cftp}. N)BytesIO TextIOWrapper)skipIf) implementer)ls) ISFTPFile)FileTransferTestAvatar SFTPTestBase)portal)defererror interfacesprotocolreactor)Clock)StringTransport)getProcessOutputAndValue)log) UserDatabase)FilePath)which) requireModule)TestCase cryptographyztwisted.conch.unix)cftp) SSHSession) filetransfer)EXTENDED_DATA_STDERR) test_conchtest_ssh) FakeStdio)FileTransferForTestAvatarFTz*don't run w/o spawnProcess or cryptographyc@s.eZdZdZd ddZd ddZd dd ZdS) SSHSessionTestsz= Tests for L{twisted.conch.scripts.cftp.SSHSession}. returnNcCs<t|_t|_|j|j_t|_t|j|_|j|j_dSN)r stdiorchannelr stderrBufferrstderrselfr+|/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/conch/test/test_cftp.pysetUp>s   zSSHSessionTests.setUpcCs|j||jjdS)z| L{twisted.conch.scripts.cftp.SSHSession.eofReceived} loses the write half of its stdio connection. N)r& eofReceived assertTruer% writeConnLostr)r+r+r,test_eofReceivedFs z SSHSessionTests.test_eofReceivedcCs:d}|d}|jt|d||j|ddS)z L{twisted.conch.scripts.cftp.SSHSession.extReceived} decodes stderr data using UTF-8 with the "backslashescape" error handling and writes the result to its own stderr. u☃utf-8s\xffN)encoder& extReceivedr assertEqualr'getvalue)r* errorText errorBytesr+r+r,test_extReceivedStderrNs z&SSHSessionTests.test_extReceivedStderr)r#N)__name__ __module__ __qualname____doc__r-r1r:r+r+r+r,r"8s   r"c @seZdZdZeedddurdZddZddZd d Z d d Z d dZ e Zz!z e e jdWn e jy>dZYnwdZWe e jene e jeweedddZddZdS) 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[csjSr$)nowr+r)r+r,fakeTimepsz$ListingTests.setUp..fakeTimetimeTZcSs,ztjd=Wn tyYnwtdS)NrD)osenvironKeyErrorrCr@r+r+r+r,cleanup|s    z#ListingTests.setUp..cleanupN) rApatchrrErF addCleanupoperatorsetitemrCr@)r*rBrHr+r)r,r-is  zListingTests.setUpcCs|tjd<ttd|S)zl Call L{ls.lsLine} after setting the timezone to C{timezone} and return the result. rDfoo)rErFrCr@rlsLine)r*timezonestatr+r+r, _lsInTimezones  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 fooNrArE stat_resultr6rQr*thenrPr+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. rSirrTz;!--------- 0 0 0 0 May 01 1973 foorUz;!--------- 0 0 0 0 May 02 1973 fooNrVrXr+r+r,test_oldSingleDigitDayOfMonth  z*ListingTests.test_oldSingleDigitDayOfMonthc CrR)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. 逛zrrT;!--------- 0 0 0 0 Aug 28 17:33 foorU;!--------- 0 0 0 0 Aug 29 09:33 fooNrVrXr+r+r, test_newFiler[zListingTests.test_newFile es_AR.UTF8TFz'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. r`rrdrTrarUrbN) rArErWlocale getlocale setlocaleLC_ALLrJr6rQ)r*rYrP currentLocaler+r+r,test_localeIndependents   z#ListingTests.test_localeIndependentc Cr\) r]r`iFrrTz;!--------- 0 0 0 0 Sep 01 17:33 foorUz;!--------- 0 0 0 0 Sep 02 09:33 fooNrVrXr+r+r,test_newSingleDigitDayOfMonthr_z*ListingTests.test_newSingleDigitDayOfMonth)r;r<r=r>getattrrCskipr-rQrZr^rcrerfrirgrhError localeSkiprrjrkr+r+r+r,r?`s,    r?cs 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*rq __class__r+r,rtszInMemorySSHChannel.__init__)r;r<r=r>rt __classcell__r+r+rur,rpsrpc@(eZdZdZddZddZddZdS) FilesystemAccessExpectationszC A test helper used to support expected filesystem access. cCs i|_dSr$_cacher)r+r+r,rt 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} Nrz)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} )r{pop)r*r}r~r+r+r,r(s z FilesystemAccessExpectations.popN)r;r<r=r>rtrrr+r+r+r,rys  ryc@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)rp transportoptions_availableFiles)r*availableFilesr+r+r,rtBs  zInMemorySFTPClient.__init__cCs|j||S)z} @see: L{filetransfer.FileTransferClient.openFile}. Retrieve and remove cached file based on flags. )rr)r*filenamer~attrsr+r+r,openFileJszInMemorySFTPClient.openFileN)r;r<r=r>rtrr+r+r+r,r8s rc@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)namerrt)r*rr+r+r,rtYszInMemoryRemoteFile.__init__cCs||||t|S)z/ @see: L{ISFTPFile.writeChunk} )seekwriter succeed)r*startdatar+r+r, writeChunkas   zInMemoryRemoteFile.writeChunkcCs d|_dS)zo @see: L{ISFTPFile.writeChunk} Keeps data after file was closed to help with testing. TN)_closedr)r+r+r,closei zInMemoryRemoteFile.closecCdSr$r+r)r+r+r,getAttrsqzInMemoryRemoteFile.getAttrscCrr$r+)r*offsetlengthr+r+r, readChunkurzInMemoryRemoteFile.readChunkcCrr$r+)r*rr+r+r,setAttrsyrzInMemoryRemoteFile.setAttrscCs t|S)zb Get current data of file. Allow reading data event when file is closed. )rr7r)r+r+r,r7}rzInMemoryRemoteFile.getvalueN) r;r<r=r>rtrrrrrr7r+r+r+r,rSs 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) ryfakeFilesystemrr StdioClientclientcurrentDirectoryrdatabase_pwdsetKnownConsoleSizer)r* sftpClientr+r+r,r-s   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. secretrMbarzexec print(1 + 2)s3 ) raddUsergetpassgetuserrEgetuidsys executabler_dispatchCommand addCallbackr6r*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}. rrrMrzexec echo hellohello rrrrrErrrrr6rr+r+r,test_execWithoutShells  z&StdioClientTests.test_execWithoutShellc Cr)zO The I{exec} command is run for lines which start with C{"!"}. rrrMrz/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)rrIr)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. r"xsamplei(g@is# b'sample' 40% 4.0kB 2.0kBps 00:03 N)rrrrrrr FileWrappersizesecondsadvancetotal_printProgressBarr6rvaluer*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. rrrrs! b'sample' 0% 0.0B 0.0Bps 00:00 N) rrrrrrrrrrr6rrrr+r+r,test_printProgressBarNoProgresss  z0StdioClientTests.test_printProgressBarNoProgresscCsL|ddt}d|_t|}|j|dd}|||jj dS)z5 Print the progress for empty files. rrs empty-filers% b'empty-file'100% 0.0B 0.0Bps 00:00 N) rrrrrrrr6rr)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 _getFilenamer6r*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_getFilenameQuoted,rz'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*r}contentfiler+r+r,makeFile5s   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 r2z  z Transferred z to r N rz5There are still put responses which were not checked.) rrrdecodesplitappendrstripextendrsplitr6sortedlen)r* transfers randomOrderoutputexpectedOutput actualOutputrremoteexpectedexpectedTransferline progressPartsactuallastactualTransferr+r+r,checkPutMessageIs<     "  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)rrrr)z 76% 10.0B 100% 13.0BrN)rr FXF_WRITE FXF_CREAT FXF_TRUNCrEr}joinbasenamerrrr rrrcmd_PUTsuccessResultOfr6r7r/rr )r*r localPathr~ remoteName remoteFiledeferredr+r+r,test_cmd_PUTSingleNoRemotePaths    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)rrrrrrrrr rrrrr r/rr6r7)r*rr~rrrr+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-namer}r*rrTrN)rrEr}rdirnamerrrrrrrrr rrrrr/rr6r7r ) r*first firstName secondNameparentsecondr~firstRemotePathsecondRemotePathfirstRemoteFilesecondRemoteFilerr+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 {} remoterrrTr N)rrEr}rr!rrrrrrrrr rrrformatrr/rr6r7r ) r*r"r#r$r%r&r~r)r*r'r(rr+r+r,"test_cmd_PUTMultipleWithRemotePaths4           z3StdioClientTests.test_cmd_PUTMultipleWithRemotePath)Nr)F)r;r<r=r>r-rrrrrrrrrrrrr rrr+r-r+r+r+r,rs(     = #rc@seZdZddZddZdS)FileTransferTestRealmcCs ||_dSr$)testDir)r*r/r+r+r,rtr|zFileTransferTestRealm.__init__cGst|j}|d|ddfS)NrcSrr$r+r+r+r+r,sz5FileTransferTestRealm.requestAvatar..)rr/)r*avatarIDmindr ar+r+r, requestAvatars z#FileTransferTestRealm.requestAvatarN)r;r<r=rtr4r+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*r7r+r+r,rt s  zSFTPTestProcess.__init__cCsd|_g|_d|_dS)zR Clear any buffered data received from stdout. Should be private. rN)buffer_linesReceived _lineBufferr)r+r+r,r6s 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) rmsgr=rrr<rr7callbackr;_checkForCommand)r*rlinesrr+r+r, outReceiveds     zSFTPTestProcess.outReceivedcCsfd}|jr/|j|kr1d|j}||r|t|d}||jd}|_||dSdSdS)Nscftp> r>)r9r=rr< startswithrr6r@)r*promptbufrr+r+r,rA.s  z SFTPTestProcess._checkForCommandcCstd|dS)zO Called by Twisted when the cftp client prints data to stderr. zerr: %sN)rr?)r*rr+r+r, errReceived8szSFTPTestProcess.errReceivedcCs|jS)zQ Return the contents of the buffer of data received from stdout. )r;r)r+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. r2r>) r Deferredr9r6 isinstancestrr4rrr*commandr+r+r, runCommandDs   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. rcsg|] }j|qSr+)runrN).0rMr*semr+r, bsz-SFTPTestProcess.runScript..)r DeferredSemaphore gatherResults)r*commandsdlr+rQr, runScriptVs  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)r:r rrIr8r signalProcessr)r+r+r, killProcesses    zSFTPTestProcess.killProcesscCs,d|_|jr|jd}|_|ddSdS)zF Called by Twisted when the cftp client process ends. TN)r:r8r@)r*reasonrr+r+r, processEndedts zSFTPTestProcess.processEndedN)r;r<r=r>rtr6rCrArGrHrNrXr[r]r+r+r+r,r5s   r5c@s4eZdZddZddZddZddZd d Zd S) CFTPClientTestBasecCstdd}|tjWdn1swYtdd}|tjWdn1s1wYtddtdd}|dtjWdn1sTwYt|S)N rsa_test.pubrrsa_testikh_tests 127.0.0.1 ) rrrpublicRSA_opensshprivateRSA_opensshrEchmodr r-r*fr+r+r,r-s     zCFTPClientTestBase.setUpcCsFt|j}t|}|tt}||_tj d|dd|_ dS)Nrz 127.0.0.1) interface) r.r/r PortalregisterCheckerrconchTestPublicKeyCheckerConchTestServerFactoryr listenTCPserver)r*realmpfacr+r+r, startServers  zCFTPClientTestBase.startServercCsHt|jjds |dSd|jjj_t|jjjjj }| |j|S)Nprotor) hasattrrmfactory _cbStopServerrrexpectedLoseConnectionr maybeDeferredrloseConnectionrrr+r+r, stopServers    zCFTPClientTestBase.stopServercCst|jjSr$)r rwrm stopListeningr*ignoredr+r+r,rusz CFTPClientTestBase._cbStopServerc Cs4dD]}zt|WqtyYqwt|S)N)r_r`ra)rEremove BaseExceptionr tearDownrer+r+r,rs  zCFTPClientTestBase.tearDownN)r;r<r=r-rqryrurr+r+r+r,r^~s   r^c@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 rsa_test -a -v 127.0.0.1rmodrunning rc jSr$)processProtocolr6_r)r+r,r0 z3OurServerCmdLineClientTests.setUp.. PYTHONPATHr2env)r^r-rqrmgetHostportr _makeArgsrrr?rrr rIr5rrrErFcopypathseprr}rJrKr4rr spawnProcess) r*cmdsrrr encodedCmds encodedEnvcmdvarvalr+r)r,r-s>               z!OurServerCmdLineClientTests.setUpcs}|fdd|S)Ncrr$)rr[rr)r+r,r0rz6OurServerCmdLineClientTests.tearDown..)ryrrr+r)r,rsz$OurServerCmdLineClientTests.tearDowncCs,z |jjdWdStjyYdSw)NrY)rrrZr ProcessExitedAlreadyr{r+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. )rrNrLr+r+r,rN 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. )rrX)r*rVr+r+r,rXrz%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. r2N)rJbytesrr)rrrr+r+r, cmdOutputs    z8OurServerCmdLineClientTests.testCdPwd..cmdOutputr)r/rXrr6r}rEgetcwd)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--rrrs ----------r)flushLoggedErrorsr/rDr6resultsr)r+r,_checks z7OurServerCmdLineClientTests.testChAttrs.._checkzls -l testfile1zchmod 0 testfile1zchmod 644 testfile1rXrr*rrr+r)r, 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 testfile1rrrrr)s.testHiddenFilerrrrr6rr)r+r,r*s z4OurServerCmdLineClientTests.testList.._checkrzls ../zls *Filez ls -a *Filezls -l testDirectorycSsdd|DS)NcSsg|]}|dqS)r>)r)rPxr+r+r,rS@zJOurServerCmdLineClientTests.testList....r+)xsr+r+r,r0@sz6OurServerCmdLineClientTests.testList..)rXr/rrrr+r)r,testList$s   z$OurServerCmdLineClientTests.testListcCsD|d}tdd}t|tr|d}||j ||S)zB Check that running the '?' command returns help. ?Nrr2) rNrrcmd_HELPrrJrKr4rr6)r*rhelpTextr+r+r,testHelpCs   z$OurServerCmdLineClientTests.testHelpNcCs||||dS)zg Assert that the files at C{name1} and C{name2} contain exactly the same data. N)r6 getContent)r*name1name2r?r+r+r,assertFilesEqualOsz,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 file2r2cs8|jdjdddS)N testfile1 test file2z get failedzrm "test file2")r/endswithrr/childrNrrr*r+r, _checkGetds   z6OurServerCmdLineClientTests.testGet.._checkGetzget testfile1 "z /test file2"cjdS)Nr assertFalser/rexistsrr)r+r,r0prz5OurServerCmdLineClientTests.testGet..)r,r/r}rJrKr4rNr)r*rrr+rr,testGetVs    z#OurServerCmdLineClientTests.testGetcs fdd}d}||S)zQ Test that 'get' works correctly when given wildcard parameters. cs8jdtddjdtdddS)NtestRemoveFiletestRemoveFile get failedtestRenameFiletestRenameFile get failed)rr/rr)r|r)r+r,rys  z;OurServerCmdLineClientTests.testWildcardGet.._checkz get testR*)rNrrr+r)r,testWildcardGetts  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/rrNrrr+r, _checkPuts  z6OurServerCmdLineClientTests.testPut.._checkPutput z/testfile1 "test\"file2"cr)Nrrrr)r+r,r0rz5OurServerCmdLineClientTests.testPut..)r/ asBytesModer}rNr)r*rrr+rr,testPuts"    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/rrr)r+r,rszEOurServerCmdLineClientTests.test_putOverLongerFile.._checkPutrz/shorterFile longerFile)r/rrrrNr}rr*rfrrr+r)r,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/rcreateDirectoryrrrNr}rrr+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)r6rr/rr%rr)r+r,checks  z:OurServerCmdLineClientTests.testWildcardPut..checkrrz/testR*zcd %s)rXr/r}rr)r*rrr+r)r,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)Nrrrlz link failedz rm testLink)rr6r/rDrNrr)r+r,rs z4OurServerCmdLineClientTests.testLink.._checkzln testLink testfile1zls -l testLinkrrXrr6rr+r)r,testLinks   z$OurServerCmdLineClientTests.testLinkcr)zV Test that we can create and remove directories with the cftp client. cs.|dd|dddS)Nrrrdzrmdir testMakeDirectory)r6r/rDrNrr)r+r,rs z?OurServerCmdLineClientTests.testRemoteDirectory.._checkmkdir testMakeDirectoryzls -l testMakeDirector?rrrr+r)r,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)Nrrrsremote error 11: mkdir failedrrr)r+r,rszHOurServerCmdLineClientTests.test_existingRemoteDirectory.._checkrrrr+r)r,test_existingRemoteDirectory s   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)rNrr)r+r,r0 rz@OurServerCmdLineClientTests.testLocalDirectory..)rNr/r}rr6rr+r)r,testLocalDirectorys z.OurServerCmdLineClientTests.testLocalDirectorycr)z1 Test that we can rename a file. cs*|dd|dddS)Nrrrs testfile2zrename testfile2 testfile1)r6rNrr)r+r,r)s z6OurServerCmdLineClientTests.testRename.._checkzrename testfile1 testfile2z ls testfile?rrrr+r)r, testRename$rz&OurServerCmdLineClientTests.testRenamer$)r;r<r=r>r-rrrNrXrrrrrrrrrrrrrrrrr+r+r+r,rs.*  rc@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||dSr$r^r-rqr)r+r+r,r-;s  zOurServerBatchFileTests.setUpcCt||Sr$)r^rryr)r+r+r,r? 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 rsa_test -a -v -b %s 127.0.0.1rrrrrrrcst|Sr$)rEr}resfnr+r,_cleanupZs z9OurServerBatchFileTests._getBatchOutput.._cleanupcSs|dS)Nrr+rr+r+r,r0^sz9OurServerBatchFileTests._getBatchOutput..)rrrrmrrrrrrr?rrrErFrrrr}rtrvrraddBoth)r*rffprrrrrr+rr,_getBatchOutputCs$        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)Nr>zRES %srrr) rrr?reprassertInr/rr}r6rr)r+r,_cbCheckResultms  z=OurServerBatchFileTests.testBatchFile.._cbCheckResultrrr*rrrr+r)r, testBatchFilecs   z%OurServerBatchFileTests.testBatchFilecr)zO Test that an error in the batch file stops running the batch. zchown 0 missingFile pwd exit cjj|dSr$) assertNotInr/rr}rr)r+r,rz9OurServerBatchFileTests.testError.._cbCheckResultrrr+r)r, testErrorzs    z!OurServerBatchFileTests.testErrorcr)z_ Test that a minus sign '-' at the front of a line ignores any errors. z-chown 0 missingFile pwd exit crr$)rr/rr}rr)r+r,rrz@OurServerBatchFileTests.testIgnoredError.._cbCheckResultrrr+r)r,testIgnoredErrors    z(OurServerBatchFileTests.testIgnoredErrorN) r;r<r=r>r-rrrrrr+r+r+r,r4s  rsshz$no ssh command-line client availablesftpz%no sftp command-line client availablec@rx) OurServerSftpClientTestsz@ Test the sftp server against sftp command line client. cCrr$rr)r+r+r,r-rzOurServerSftpClientTests.setUpcCs|Sr$)ryr)r+r+r,rsz!OurServerSftpClientTests.tearDownc sttj}}t|d }|dWdn1swYjj}t j fdd} t d|djj _ dd d d d d d d d dd d|fd|df}fdd}td||}||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.._getAttrsr Tz-Fz /dev/nullz-ozIdentitiesOnly=yeszIdentityFile=rsa_testzUserKnownHostsFile=kh_testzHostKeyAlgorithms=ssh-rsazPort=%iz-bztestuser@127.0.0.1cs:|dd|dddD] }||dqdS)Nrrrasciir)r6rr)rir)r+r,rsz?OurServerSftpClientTests.test_extendedAttributes..checkr)dictrErFrrrrmrrr!r rIrtrvrr) r*rrrfrr argsrrr+)r r*r,test_extendedAttributess:        z0OurServerSftpClientTests.test_extendedAttributesN)r;r<r=r>r-rrr+r+r+r,r s  r )Qr>rrerKrErrrCiorrunittestrzope.interfacer twisted.conchrtwisted.conch.interfacesr$twisted.conch.test.test_filetransferrr twisted.credr twisted.internetr r r rrtwisted.internet.taskrtwisted.internet.testingrtwisted.internet.utilsrtwisted.pythonrtwisted.python.fakepwdrtwisted.python.filepathrtwisted.python.procutilsrtwisted.python.reflectrtwisted.trial.unittestrrunixtwisted.conch.scriptsrtwisted.conch.scripts.cftprtwisted.conch.sshrtwisted.conch.ssh.connectionrtwisted.conch.testrrtwisted.conch.test.test_conchr r! ImportError skipTestsIReactorProcessr"r?rpryrrrr.ProcessProtocolr5r^rrr r+r+r+r,s                    '#&2u {'g