o i@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+bC:\Users\psant\Downloads\PAF backend\school\venv\Lib\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&Z eofReceived assertTruer%Z 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&Z extReceivedr assertEqualr'getvalue)r*Z errorTextZ errorBytesr+r+r,test_extReceivedStderrNs z&SSHSessionTests.test_extReceivedStderr)r#N)__name__ __module__ __qualname____doc__r-r/r5r+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)Nr?)osenvironKeyErrorr>r;r+r+r+r,cleanup|s    z#ListingTests.setUp..cleanupN) r<patchrr@rA addCleanupoperatorsetitemr>r;)r*r=rCr+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. r?fooN)r@rAr>r;rZlsLine)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 fooNr<r@ stat_resultr3rKr*thenrJr+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. rMirrNz;!--------- 0 0 0 0 May 01 1973 foorOz;!--------- 0 0 0 0 May 02 1973 fooNrPrRr+r+r,test_oldSingleDigitDayOfMonth  z*ListingTests.test_oldSingleDigitDayOfMonthc CrL)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. 逛zrrN;!--------- 0 0 0 0 Aug 28 17:33 foorO;!--------- 0 0 0 0 Aug 29 09:33 fooNrPrRr+r+r, test_newFilerUzListingTests.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. rZrr^rNr[rOr\N) r<r@rQlocale getlocale setlocaleLC_ALLrEr3rK)r*rSrJ currentLocaler+r+r,test_localeIndependents   z#ListingTests.test_localeIndependentc CrV) rWrZiFrrNz;!--------- 0 0 0 0 Sep 01 17:33 foorOz;!--------- 0 0 0 0 Sep 02 09:33 fooNrPrRr+r+r,test_newSingleDigitDayOfMonthrYz*ListingTests.test_newSingleDigitDayOfMonth)r6r7r8r9getattrr>skipr-rKrTrXr]r_r`rcrarbErrorZ localeSkiprrdrer+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)connZ localClosedsuper__init__)r*rj __class__r+r,rlszInMemorySSHChannel.__init__)r6r7r8r9rl __classcell__r+r+rmr,risric@(eZdZdZddZddZddZdS) FilesystemAccessExpectationszC A test helper used to support expected filesystem access. cCs i|_dSr$_cacher)r+r+r,rl 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} Nrr)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} N)rspop)r*rurvr+r+r,ry(s z FilesystemAccessExpectations.popN)r6r7r8r9rlrxryr+r+r+r,rqs  rqc@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)ri transportoptions_availableFiles)r*ZavailableFilesr+r+r,rlBs  zInMemorySFTPClient.__init__cCs|j||S)z} @see: L{filetransfer.FileTransferClient.openFile}. Retrieve and remove cached file based on flags. N)rry)r*filenamervattrsr+r+r,openFileJszInMemorySFTPClient.openFileN)r6r7r8r9rlrr+r+r+r,rz8s rzc@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)namerrl)r*rr+r+r,rlYszInMemoryRemoteFile.__init__cCs||||t|S)z/ @see: L{ISFTPFile.writeChunk} N)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. N)rr4r)r+r+r,r4}rzInMemoryRemoteFile.getvalueN) r6r7r8r9rlrrrrrr4r+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) rqfakeFilesystemrzr StdioClientclientZcurrentDirectoryrdatabase_pwdsetKnownConsoleSizer)r*Z 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. secretrHbarzexec print(1 + 2)s3 N) raddUsergetpassgetuserr@getuidsys executabler_dispatchCommand addCallbackr3r*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}. rrrHrzexec echo hellohello N rrrrr@rrrrr3rr+r+r,test_execWithoutShells  z&StdioClientTests.test_execWithoutShellc Cr) zO The I{exec} command is run for lines which start with C{"!"}. rrrHrz/bin/shz !echo hellorNrrr+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.Z4Hr)Z TIOCGWINSZZfailstructpack)r*fdoptZmutateheightttywidthr+r,ioctls  z=StdioClientTests.setKnownConsoleSize..FakeFcntl.ioctlN)r6r7r8rr+rr+r, FakeFcntlsrfcntl)rrDr)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 FileWrappersizesecondsZadvancetotal_printProgressBarr3rvaluer*ZclockwrappedwrapperZ 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. r|rrrs! b'sample' 0% 0.0B 0.0Bps 00:00 N) rrrrrrrrrrr3rrrr+r+r,test_printProgressBarNoProgresss  z0StdioClientTests.test_printProgressBarNoProgresscCsL|ddt}d|_t|}|j|dd}|||jj dS)z5 Print the progress for empty files. r|rs empty-filers% b'empty-file'100% 0.0B 0.0Bps 00:00 N) rrrrrrrr3rr)r*rrrr+r+r,test_printProgressBarEmptyFiles  z/StdioClientTests.test_printProgressBarEmptyFilecC|jd}|d|dS)zK Returns empty value for both filename and remaining data. z )rrNrZ _getFilenamer3r*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*rucontentfiler+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 r0z  z Transferred z to r N r{z5There are still put responses which were not checked.) rrrdecodesplitappendrystripextendrsplitr3sortedlen)r*Z transfers randomOrderoutputexpectedOutputZ actualOutputrremoteexpectedZexpectedTransferlineZ progressPartsactuallastZactualTransferr+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)rrr|r~)z 76% 10.0B 100% 13.0BrN)rr FXF_WRITE FXF_CREAT FXF_TRUNCr@rujoinbasenamerrrxr rrrcmd_PUTsuccessResultOfr3r4r.rr)r*r localPathrv 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)rrrrrrrrxr rrrrrr.rr3r4)r*rrvrrrr+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-namerur*rrTrN)rr@rurdirnamerrrrrrrrxr rrrrr.rr3r4r) r*first firstName secondNameparentsecondrvfirstRemotePathsecondRemotePathfirstRemoteFilesecondRemoteFilerr+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)rr@rurr rrrrrrrrxr rrrformatrr.rr3r4r) r*r r r r rrvrrrrrr+r+r,"test_cmd_PUTMultipleWithRemotePaths4           z3StdioClientTests.test_cmd_PUTMultipleWithRemotePath)Nr)F)r6r7r8r9r-rrrrrrrrrrrrrrrrrr+r+r+r,rs(     = #rc@seZdZddZddZdS)FileTransferTestRealmcCs ||_dSr$)testDir)r*rr+r+r,rlrtzFileTransferTestRealm.__init__cGst|j}|d|ddfS)NrcSrr$r+r+r+r+r,sz5FileTransferTestRealm.requestAvatar..)rr)r*ZavatarIDZmindr ar+r+r, requestAvatars z#FileTransferTestRealm.requestAvatarN)r6r7r8rlrr+r+r+r,rs rc@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*rr+r+r,rl 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,rs 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#rryr"rrcallbackr!_checkForCommand)r*rlinesrr+r+r, outReceiveds     zSFTPTestProcess.outReceivedcCsfd}|jr/|j|kr1d|j}||r|t|d}||jd}|_||dSdSdS)Nscftp> r$)rr#rr" startswithrrr&)r*promptbufrr+r+r,r'.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. N)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. r0r$N) r Deferredrr isinstancestrr2rrr*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. r{csg|] }j|qSr+)runr4).0r3r*Zsemr+r, bsz-SFTPTestProcess.runScript..N)r ZDeferredSemaphoreZ gatherResults)r*commandsdlr+r7r, 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 rr/rr signalProcessr)r+r+r, killProcesses    zSFTPTestProcess.killProcesscCs,d|_|jr|jd}|_|ddSdS)zF Called by Twisted when the cftp client process ends. TN)r rr&)r*reasonrr+r+r, processEndedts zSFTPTestProcess.processEndedN)r6r7r8r9rlrr)r'r-r.r4r;r>r@r+r+r+r,rs   rc@s4eZdZddZddZddZddZd d Zd S) CFTPClientTestBasecCstdd}|tjWdn1swYtdd}|tjWdn1s1wYtddtdd}|dtjWdn1sTwYt|S)N rsa_test.pubrrsa_testikh_tests 127.0.0.1 ) rrrZpublicRSA_opensshZprivateRSA_opensshr@chmodr r-r*fr+r+r,r-s     zCFTPClientTestBase.setUpcCsFt|j}t|}|tt}||_tj d|dd|_ dS)Nrz 127.0.0.1)Z interface) rrr ZPortalZregisterCheckerrZconchTestPublicKeyCheckerZConchTestServerFactoryrZ listenTCPserver)r*realmpZfacr+r+r, startServers  zCFTPClientTestBase.startServercCsHt|jjds |dSd|jjj_t|jjjjj }| |j|S)Nprotor{) hasattrrHfactory _cbStopServerrLexpectedLoseConnectionr maybeDeferredrZloseConnectionrrr+r+r, stopServers    zCFTPClientTestBase.stopServercCst|jjSr$)r rQrH stopListeningr*ignoredr+r+r,rOsz CFTPClientTestBase._cbStopServerc Cs4dD]}zt|WqtyYqwt|S)N)rBrCrD)r@remove BaseExceptionr tearDownrFr+r+r,rXs  zCFTPClientTestBase.tearDownN)r6r7r8r-rKrRrOrXr+r+r+r,rA~s   rAc@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$)processProtocolr_r)r+r,r z3OurServerCmdLineClientTests.setUp.. PYTHONPATHr0env)rAr-rKrHgetHostportr _makeArgsrrr%rrr r/rr^rr@rAcopypathseprrur0r1r2rrZ spawnProcess) r*cmdsrfrrdZ encodedCmdsZ encodedEnvcmdvarvalr+r)r,r-s>               z!OurServerCmdLineClientTests.setUpcs}|fdd|S)Ncr]r$)r^r>r_r)r+r,rraz6OurServerCmdLineClientTests.tearDown..)rRrrr+r)r,rXsz$OurServerCmdLineClientTests.tearDowncCs,z |jjdWdStjyYdSw)Nr<)r^rr=r ZProcessExitedAlreadyrTr+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. N)r^r4r2r+r+r,r4 z&OurServerCmdLineClientTests.runCommandcGro)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. N)r^r;)r*r9r+r+r,r;rpz%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. pwdZlpwdzcd testDirectorycd ..cSsDg}|D]}t|tr|d}||q|dd|ddS)zH Callback function for handling command output. r0N)r0bytesrr)rrjrkr+r+r, cmdOutputs    z8OurServerCmdLineClientTests.testCdPwd..cmdOutputrN)rr;rr3rur@getcwd)r*ZhomeDirrrvr+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--r{rrs ----------rs)flushLoggedErrorsr.r*r3resultsr)r+r,_checks z7OurServerCmdLineClientTests.testChAttrs.._checkzls -l testfile1zchmod 0 testfile1zchmod 644 testfile1Nr;rr*r|rr+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 testfile1r{rrrrs)s.testHiddenFilerrrtrr3rzr)r+r,r|*s z4OurServerCmdLineClientTests.testList.._checkrzls ../zls *Filez ls -a *Filezls -l testDirectorycSsdd|DS)NcSsg|]}|dqS)r$)r)r6xr+r+r,r8@zJOurServerCmdLineClientTests.testList....r+)xsr+r+r,r@sz6OurServerCmdLineClientTests.testList..N)r;rrrr~r+r)r,testList$s   z$OurServerCmdLineClientTests.testListcCsD|d}tdd}t|tr|d}||j ||S)zB Check that running the '?' command returns help. ?Nrr0) r4rrZcmd_HELPrr0r1r2rr3)r*rZhelpTextr+r+r,testHelpCs   z$OurServerCmdLineClientTests.testHelpNcCs||||dS)zg Assert that the files at C{name1} and C{name2} contain exactly the same data. N)r3Z getContent)r*Zname1Zname2r%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 file2r0cs8|jdjdddS)N testfile1 test file2z get failedzrm "test file2")r.endswithrrchildr4rrr*r+r, _checkGetds   z6OurServerCmdLineClientTests.testGet.._checkGetzget testfile1 "z /test file2"cjdS)NrZ assertFalserrexistsr_r)r+r,rprz5OurServerCmdLineClientTests.testGet..N)rrrur0r1r2r4r)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)rrrr)rUr)r+r,r|ys  z;OurServerCmdLineClientTests.testWildcardGet.._checkz get testR*N)r4rr~r+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")rrrr.rr4rrr+r, _checkPuts  z6OurServerCmdLineClientTests.testPut.._checkPutput z/testfile1 "test\"file2"cr)Nrrr_r)r+r,rrz5OurServerCmdLineClientTests.testPut..N)r asBytesModerur4r)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)rrrrr)r+r,rszEOurServerCmdLineClientTests.test_putOverLongerFile.._checkPutrz/shorterFile longerFile)rrrrr4rurr*rGrrr+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)rrrrr*ZsomeDirr+r,rszMOurServerCmdLineClientTests.test_putMultipleOverLongerFile.._checkPutrz/dir/*)rrZcreateDirectoryrrr4rurrr+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)r3rrrr rzr)r+r,checks  z:OurServerCmdLineClientTests.testWildcardPut..checkrrrz/testR*zcd %sN)r;rrurr)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)Nrrr{lz link failedz rm testLink)ryr3r.r*r4rzr)r+r,r|s z4OurServerCmdLineClientTests.testLink.._checkzln testLink testfile1zls -l testLinkrNr;rr3r~r+r)r,testLinks   z$OurServerCmdLineClientTests.testLinkcr)zV Test that we can create and remove directories with the cftp client. cs.|dd|dddS)Nrrr{dzrmdir testMakeDirectory)r3r.r*r4rzr)r+r,r|s z?OurServerCmdLineClientTests.testRemoteDirectory.._checkmkdir testMakeDirectoryzls -l testMakeDirector?rNrr~r+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)Nrrr{sremote error 11: mkdir failedrrzr)r+r,r|szHOurServerCmdLineClientTests.test_existingRemoteDirectory.._checkrNr}r~r+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)r4r_r)r+r,r raz@OurServerCmdLineClientTests.testLocalDirectory..N)r4rrurr3rr+r)r,testLocalDirectorys z.OurServerCmdLineClientTests.testLocalDirectorycr)z1 Test that we can rename a file. cs*|dd|dddS)Nrrr{s testfile2zrename testfile2 testfile1)r3r4rzr)r+r,r|)s z6OurServerCmdLineClientTests.testRename.._checkzrename testfile1 testfile2z ls testfile?rNrr~r+r)r, testRename$rz&OurServerCmdLineClientTests.testRenamer$)r6r7r8r9r-rXrnr4r;rxrrrrrrrrrrrrrrrr+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||dSr$rAr-rKr)r+r+r,r-;s  zOurServerBatchFileTests.setUpcCt||Sr$)rArXrRr)r+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 rsa_test -a -v -b %s 127.0.0.1rrZr{r\rrbrccst|Sr$)r@rVresfnr+r,_cleanupZs z9OurServerBatchFileTests._getBatchOutput.._cleanupcSs|dS)Nrr+rr+r+r,r^sz9OurServerBatchFileTests._getBatchOutput..)rrrrHrerfrrgrrr%rrr@rArhrirrurNrPrrZaddBoth)r*rGfprfrjrdrrr+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 %sr{rsr) rrr%reprassertInrrrur3rr)r+r,_cbCheckResultms  z=OurServerBatchFileTests.testBatchFile.._cbCheckResultNrrr*rjrrr+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$)Z assertNotInrrrurr)r+r,rz9OurServerBatchFileTests.testError.._cbCheckResultNrrr+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$)rrrrurr)r+r,rrz@OurServerBatchFileTests.testIgnoredError.._cbCheckResultNrrr+r)r,testIgnoredErrors    z(OurServerBatchFileTests.testIgnoredErrorN) r6r7r8r9r-rXrrrrr+r+r+r,r4s  rsshz$no ssh command-line client availablesftpz%no sftp command-line client availablec@rp) OurServerSftpClientTestsz@ Test the sftp server against sftp command line client. cCrr$rr)r+r+r,r-rzOurServerSftpClientTests.setUpcCs|Sr$)rRr)r+r+r,rXsz!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)NrZext_foor+)r*sr) oldGetAttrr+r, _getAttrss zCOurServerSftpClientTests.test_extendedAttributes.._getAttrsrTz-Fz /dev/nullz-ozIdentitiesOnly=yeszIdentityFile=rsa_testzUserKnownHostsFile=kh_testzHostKeyAlgorithms=ssh-rsazPort=%iz-bztestuser@127.0.0.1cs:|dd|dddD] }||dqdS)Nrrr{asciir)r3rr)rir)r+r,rsz?OurServerSftpClientTests.test_extendedAttributes..checkr)dictr@rArrrrHrerfr!rrDrNrPrr) r*rdrrGrfrargsrrr+)rr*r,test_extendedAttributess:        z0OurServerSftpClientTests.test_extendedAttributesN)r6r7r8r9r-rXrr+r+r+r,rs  r)Qr9rr_rFr@rrr>iorrZunittestrZzope.interfacerZ twisted.conchrZtwisted.conch.interfacesrZ$twisted.conch.test.test_filetransferrr Z twisted.credr Ztwisted.internetr r r rrZtwisted.internet.taskrZtwisted.internet.testingrZtwisted.internet.utilsrZtwisted.pythonrZtwisted.python.fakepwdrZtwisted.python.filepathrZtwisted.python.procutilsrZtwisted.python.reflectrZtwisted.trial.unittestrrunixZtwisted.conch.scriptsrZtwisted.conch.scripts.cftprZtwisted.conch.sshrZtwisted.conch.ssh.connectionrZtwisted.conch.testrrZtwisted.conch.test.test_conchr r! ImportErrorZ skipTestsZIReactorProcessr"r:rirqrzrrrZProcessProtocolrrArYrrr+r+r+r,s                    '#&2u {'g