ó âT€Mc@sØdZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddd„ƒYZ d e e jfd „ƒYZd e e jfd „ƒYZd e jfd„ƒYZdS(s' Tests for L{twisted.python.zipstream} iÿÿÿÿN(tset(t zipstreamtfilepath(tmd5(tunittesttFileEntryMixincBsheZdZd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z RS( s? File entry classes should behave as file-like objects cCs]|jƒ}tj|d|jƒ}|jd|ƒ|jƒtj|dƒ}|jdƒS(s6 Return an appropriate zip file entry twtcontenttr( tmktemptzipfiletZipFilet compressiontwritestrtcloseRtChunkingZipFiletreadfile(tselftcontentstfilenametz((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt getFileEntrys   cCs#|j|jdƒjƒtƒdS(sK zip files should not be ttys, so isatty() should be false tN(t assertEqualsRtisattytFalse(R((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt test_isatty scCsC|jdƒ}|j|jtƒ|jƒ|j|jtƒdS(sd The C{closed} attribute should reflect whether C{close()} has been called. RN(RRtclosedRRtTrue(Rt fileEntry((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt test_closed's cCsU|jdƒ}|j|jƒdƒ|j|jƒdƒ|j|jƒdƒdS(sl C{readline()} should mirror L{file.readline} and return up to a single deliminter. shoho hoshoho thoRN(RRtreadline(RR((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt test_readline2scCsR|jdƒ}|j|jƒdƒ|j|jƒdƒ|jt|jƒdS(sV Zip file entries should implement the iterator protocol as files do. sho hohosho thohoN(RRtnextt assertRaisest StopIteration(RR((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt test_next=scCs2|jdƒ}|j|jƒdddgƒdS(sG C{readlines()} should return a list of all the lines. sho ho hosho RN(RRt readlines(RR((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_readlinesGscCs?|jdƒ}|jt|ƒ|ƒ|j|jƒ|ƒdS(sJ C{__iter__()} and C{xreadlines()} should return C{self}. RN(RtassertIdenticaltitert xreadlines(RR((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_iterationOscCs/d}|j|ƒ}|j|jƒ|ƒdS(s9 C{.read()} should read the entire file. s Hello, world!N(RRtread(RRtentry((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_readWholeXscCsWd}|j|ƒ}|jdƒ}|jdƒ}|j|dƒ|j|dƒdS(sD C{.read(num)} should read num bytes from the file. t 0123456789iiÈt0123t456789N(RR-R(RRR.tonettwo((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_readPartialas cCscdd}|j|ƒ}|jdƒ|j|jƒdƒ|jdƒ|j|jƒdƒdS(sb C{.tell()} should return the number of bytes that have been read so far. txidiiiN(RR-Rttell(RRR.((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt test_tellms    ( t__name__t __module__t__doc__RRRR!R&R(R,R/R5R8(((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyRs   tDeflatedZipFileEntryTestcBseZdZejZRS(s2 DeflatedZipFileEntry should be file-like (R9R:R;R t ZIP_DEFLATEDR (((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyR<{stZipFileEntryTestcBseZdZejZRS(s( ZipFileEntry should be file-like (R9R:R;R t ZIP_STOREDR (((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyR>ƒst ZipstreamTestcBsæeZdZd„Zdd„Zd„Zd„Zd„Zd„Zd„Z e j dkrxd Z e e_ e e _ nd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„ZRS(s, Tests for twisted.python.zipstream cCsKtj|jƒƒ|_|jjƒ|jjdƒ|_|jjƒdS(sz Creates junk data that can be compressed and a test directory for any files that will be created tunzippedN(RtFilePathR ttestdirtmakedirstchildtunzipdir(R((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pytsetUps RcCs…|jjdƒj}tj|dƒ}xMt|ƒD]?\}}t|ƒ}|rc|d|}n|j||ƒq4W|jƒ|S(s¤ Makes a zip file archive containing len(contents) files. Contents should be a list of strings, each string being the content of one file. s zipfile.zipRt/( RCREtpathR R t enumeratetstrR R(RRt directoryt zpfilenametzpfiletiRR((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt makeZipFilešs  csS|jdddddgƒ‰|jtdt‡fd†ƒ}|j|dƒd S( s Make sure the deprecated L{countZipFileEntries} returns the correct number of entries for a zip file. R3R4tthreetfourtfives"countZipFileEntries is deprecated.cs tjˆƒS(N(RtcountZipFileEntries((tname(sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt²siN(RPt assertWarnstDeprecationWarningt__file__R(Rtresult((RUsF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_countEntriesªs  cCs2tj|jƒdƒ}|jt|jdƒdS(s€ A ChunkingZipFile opened in write-mode should not allow .readfile(), and raise a RuntimeError instead. Rt somethingN(RRR R$t RuntimeErrorR(Rtczf((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_invalidMode·scCsBtj|jdgƒdƒ}|jƒ|jt|jdƒdS(sn A closed ChunkingZipFile should raise a L{RuntimeError} when .readfile() is invoked. R\RN(RRRPRR$R]R(RR^((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_closedArchiveÀs cCsÎ|jddgƒ}tj|dƒ}|jdƒj}|jƒt|dƒ}|j|dƒ|jt dƒdƒ|jƒt j |ƒ}|j tj |jdƒ|j|jdƒjƒdƒd S( s¤ A zipfile entry with the wrong magic number should raise BadZipfile for readfile(), but that should not affect other files in the archive. s test contentss more contentsRt0sr+biit1N(RPR R tgetinfot header_offsetRtfiletseektwritetchrRRR$t BadZipfileRRR-(Rtfntzft zeroOffsettscribbleR^((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_invalidHeaderÊs    cCsÓ|jddgƒ}tj|dƒ}|jdƒ}d|_|jƒt|dƒ}|j|jdƒ|j |j ƒƒ|jƒt j |ƒ}|j tj|jdƒ|j|jdƒjƒdƒd S( sƒ A zipfile entry with a different filename than is found in the central directory should raise BadZipfile. s test contentss more contentsRRasnot zerosr+biRbN(RPR R RcRRReRfRdRgt FileHeaderRRR$RiRRR-(RRjRktinfoRmR^((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_filenameMismatchÞs     iis+Consistency-checking only necessary in 2.5.cCs||jƒ}tj|dƒ}tjdƒ}|j|dƒd|_|jƒtj|ƒ}|j tj |j dƒdS(sq A zipfile which describes an unsupported compression mechanism should raise BadZipfile. RRas some dataiÒN( R R R tZipInfoR t compress_typeRRRR$RiR(RRjRktziR^((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_unsupportedCompressionýs   cCs‚|jƒ}tj|dƒ}tjdƒ}d|_|j|dƒ|jƒtj|ƒ}|j |j dƒj ƒdƒdS(sW readfile() should skip over 'extra' data present in the zip metadata. RRas hello, extras the real dataN( R R R RrtextraR RRRRRR-(RRjRkRtR^((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_extraDatas   cCsd}gt|ƒD]}d|^q}|j|ƒ}tj||jjƒ}xWt|ƒD]I}|jtt|jj ƒƒƒ|ƒ|j|j ƒ||dƒq]W|jtt|jj ƒƒƒ|ƒxI|jj ƒD]8}t |j ƒƒ}|j|j ƒjƒ||ƒqßWdS(s¯ L{twisted.python.zipstream.unzipIter} should unzip a file for each iteration and yield the number of files left to unzip after that iteration i sThis is test file %d!iN(trangeRPRt unzipIterRFRIRtlentlisttchildrenR#tinttbasenametopenR-(RtnumfilesRORRMtuziterREtnum((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_unzipIters#%"%cCsÒd}gt|ƒD]}d|^q}|j|ƒ}ttj||jjƒƒ|jt|jj ƒƒtt t t|ƒƒƒƒxC|jj ƒD]2}t |jƒƒ}|j|jƒ||ƒq˜WdS(s˜ L{twisted.python.zipstream.unzipIterChunky} returns an iterator which must be exhausted to completely unzip the input archive. i sThis is test file %d!N(RxRPR{RtunzipIterChunkyRFRIRRtlistdirtmapRKR|R}R~t getContent(RR€RORRMRER‚((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_unzipIterChunky2s#cCsçd}gt|ƒD]}d|^q}|j|dƒ}ttj||jjƒƒ|jt|jj dƒj ƒƒtt t t|ƒƒƒƒxL|jj dƒj ƒD]2}t|jƒƒ}|j|jƒ||ƒq­WdS(sä The path to which a file is extracted by L{zipstream.unzipIterChunky} is determined by joining the C{directory} argument to C{unzip} with the path within the archive of the file being extracted. i sThis is test file %d!tfooN(RxRPR{RR„RFRIRRRER…R†RKR|R}R~R‡(RR€RORRMRER‚((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_unzipIterChunkyDirectoryDs#cCsÇd}|jgt|ƒD]}t|ƒ^qƒ}tj||jjƒ|jt|jj ƒƒtt tt|ƒƒƒƒxBt|ƒD]4}|j|jj t|ƒƒj ƒt|ƒƒq‹WdS(sg L{twisted.python.zipstream.unzip} should extract all files from a zip archive iN( RPRxRKRtunzipRFRIt assertEqualRR…R†RER‡(RR€RORM((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt test_unzipWs.cCsÜd}|jgt|ƒD]}t|ƒ^qdƒ}tj||jjƒ|jt|jj dƒj ƒƒtt tt|ƒƒƒƒxKt|ƒD]=}|j|jj dƒj t|ƒƒj ƒt|ƒƒq—WdS(sÚ The path to which a file is extracted by L{zipstream.unzip} is determined by joining the C{directory} argument to C{unzip} with the path within the archive of the file being extracted. iR‰N( RPRxRKRR‹RFRIRŒRRER…R†R‡(RR€RORM((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_unzipDirectoryfs1cCs.|jjdƒ}|jdgƒ}|jdƒtj||jjƒ|j|jƒj ƒdƒtj||jjdt ƒ|j|jƒj ƒdƒ|jdƒtj ||jjƒ}|j ƒ|j|jƒj ƒdƒtj ||jjdt ƒ}|j ƒ|j|jƒj ƒdƒdS(s­ L{twisted.python.zipstream.unzip} and L{twisted.python.zipstream.unzipIter} shouldn't overwrite files unless the 'overwrite' flag is passed Rat OVERWRITTENsNOT OVERWRITTENt overwriteN( RFRERPt setContentRR‹RIRRR-RRyR#(RttestfileRMR((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_overwritews     cCs6djgtdƒD]}ttjƒƒ^qƒ}t|ƒjƒ}tj|jƒƒ}|j ƒ|j dƒj } |j | ||ƒt j| |j d|ƒ} | jƒ} || koÈ|kn} |j| ƒx| D]} qáW|j| dƒt|j dƒjƒjƒƒjƒ} |j| |ƒdS(sW unzipIterChunky should unzip the given number of bytes per iteration. t iès bigfile.zipt chunksizeit zipstreamjunkN(tjointxrangeRKtrandomRt hexdigestRRBR RDRERIt _makebigfileRR„R#t failUnlessRŒRR-(RR R•tlowertuppertntjunktjunkmd5ttempdirtzfpathRRtapproxtnewmd5((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyt_unzipIterChunkyTest”s"4     $cCs|jtjdddƒdS(ss unzipIterChunky should unzip the given number of bytes per iteration on a stored archive. iôi#i-N(R¦R R?(R((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_unzipIterChunkyStored®scCs|jtjdddƒdS(su unzipIterChunky should unzip the given number of bytes per iteration on a deflated archive. iÌiiN(R¦R R=(R((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyttest_chunkyDeflated¶scCsdtj|d|ƒ}x.tdƒD] }d|}|j|dƒq"W|jd|ƒ|jƒdS(sT Create a zip file with the given file name and compression scheme. Ri s zipstream%dRR–N(R R RxR R(RRR R RkRORj((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyR›¾s  (ii(R9R:R;RGRPR[R_R`RnRqtsyst version_infotmessagetskipRuRwRƒRˆRŠRRŽR“R¦R§R¨R›(((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyR@‹s0                ((R;R©R™R ttwisted.python.compatRttwisted.pythonRRttwisted.python.hashlibRt twisted.trialRRtTestCaseR<R>R@(((sF/usr/lib/python2.7/dist-packages/twisted/python/test/test_zipstream.pyts   k