o ¯b//ã@sdZddlZddlZddlZddlmZddlmZmZddl m Z Gdd„dƒZ Gdd „d e e j ƒZ Gd d „d e e j ƒZGd d „d e j ƒZdS)z' Tests for L{twisted.python.zipstream} éN)Úmd5)ÚfilepathÚ zipstream)Úunittestc@s`eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dS)ÚFileEntryMixinz? File entry classes should behave as file-like objects cCsZ| ¡}t |d|j¡}| d|¡Wdƒn1swYt |d¡}| d¡S)z6 Return an appropriate zip file entry ÚwÚcontentNÚr)ÚmktempÚzipfileÚZipFileÚ compressionÚwritestrrÚChunkingZipFileÚreadfile)ÚselfÚcontentsÚfilenameÚz©rúD/usr/lib/python3/dist-packages/twisted/python/test/test_zipstream.pyÚ getFileEntrys ÿ  zFileEntryMixin.getFileEntrycCs>| d¡}| | ¡¡WdƒdS1swYdS)zK zip files should not be ttys, so isatty() should be false ÚN)rÚ assertFalseÚisatty©rÚ fileEntryrrrÚ test_isatty s "ÿzFileEntryMixin.test_isattycCsF| d¡}| |j¡Wdƒn1swY| |j¡dS)zd The C{closed} attribute should reflect whether C{close()} has been called. rN)rrÚclosedÚ assertTruerrrrÚ test_closed's ÿzFileEntryMixin.test_closedcCs`| d¡!}| | ¡d¡| | ¡d¡| | ¡d¡WdƒdS1s)wYdS)zk C{readline()} should mirror L{file.readline} and return up to a single delimiter. shoho hoshoho óhoóN)rÚ assertEqualÚreadlinerrrrÚ test_readline0s "ýzFileEntryMixin.test_readlinecCs^| d¡ }| | ¡d¡| | ¡d¡| t|j¡WdƒdS1s(wYdS)zV Zip file entries should implement the iterator protocol as files do. sho hohoóho shohoN)rr#ÚnextÚ assertRaisesÚ StopIterationrrrrÚ test_next:s "ýzFileEntryMixin.test_nextcCsD| d¡}| | ¡gd¢¡WdƒdS1swYdS)zG C{readlines()} should return a list of all the lines. sho ho ho)r&r&r!N)rr#Ú readlinesrrrrÚtest_readlinesCs "ÿzFileEntryMixin.test_readlinescCsP| d¡}| t|ƒ|¡| | ¡|¡WdƒdS1s!wYdS)zJ C{__iter__()} and C{xreadlines()} should return C{self}. rN)rÚassertIsÚiterÚ xreadlinesrrrrÚtest_iterationJs "þzFileEntryMixin.test_iterationcCsDd}| |¡}| | ¡|¡WdƒdS1swYdS)z9 C{.read()} should read the entire file. s Hello, world!N)rr#Úread©rrÚentryrrrÚtest_readWholeRs "ÿzFileEntryMixin.test_readWholecCs^d}| |¡}| d¡}| d¡}Wdƒn1swY| |d¡| |d¡dS)zD C{.read(num)} should read num bytes from the file. Ú 0123456789ééÈNs0123s456789)rr1r#)rrr3ÚoneÚtworrrÚtest_readPartialZs   þ zFileEntryMixin.test_readPartialcCshd}| |¡#}| d¡| | ¡d¡| d¡| | ¡d¡WdƒdS1s-wYdS)zb C{.tell()} should return the number of bytes that have been read so far. Údxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxér6éN)rr1r#Útellr2rrrÚ test_telles   "üzFileEntryMixin.test_tellN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr r%r*r,r0r4r:r?rrrrrs     rc@óeZdZdZejZdS)ÚDeflatedZipFileEntryTestsz2 DeflatedZipFileEntry should be file-like N)r@rArBrCr Ú ZIP_DEFLATEDr rrrrrEró rEc@rD)ÚZipFileEntryTestsz* ZipFileEntry should be file-like N)r@rArBrCr Ú ZIP_STOREDr rrrrrHzrGrHc@s‚eZdZdZdd„Zd dd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„ZdS)!ÚZipstreamTestsz, Tests for twisted.python.zipstream cCs6t | ¡¡|_|j ¡|j d¡|_|j ¡dS)zz Creates junk data that can be compressed and a test directory for any files that will be created ÚunzippedN)rÚFilePathr ÚtestdirÚmakedirsÚchildÚunzipdir©rrrrÚsetUp‡s zZipstreamTests.setUprcCsv|j d¡j}t |d¡$}t|ƒD]\}}t|ƒ}|r"|d|}| ||¡qWdƒ|S1s4wY|S)z¤ Makes a zip file archive containing len(contents) files. Contents should be a list of strings, each string being the content of one file. z zipfile.ziprú/N)rMrOÚpathr r Ú enumerateÚstrr)rrÚ directoryÚ zpfilenameÚzpfileÚirrrrrÚ makeZipFile‘s ü ÿúzZipstreamTests.makeZipFilecCsFt | ¡d¡}| t|jd¡WdƒdS1swYdS)z€ A ChunkingZipFile opened in write-mode should not allow .readfile(), and raise a RuntimeError instead. rÚ somethingN)rrr r(Ú RuntimeErrorr©rÚczfrrrÚtest_invalidModeŸs"ÿzZipstreamTests.test_invalidModecCs0t | dg¡d¡}| ¡| t|jd¡dS)zn A closed ChunkingZipFile should raise a L{RuntimeError} when .readfile() is invoked. r\r N)rrr[Úcloser(r]rr^rrrÚtest_closedArchive§sz!ZipstreamTests.test_closedArchivec Cs| ddg¡}t |d¡}| d¡j}Wdƒn1swYt|dƒ}| |d¡| d¡Wdƒn1s>wYt  |¡7}|  tj |j d¡|  d ¡}|  | ¡d ¡Wdƒn1sjwYWdƒdSWdƒdS1s‚wYdS) z¤ A zipfile entry with the wrong magic number should raise BadZipfile for readfile(), but that should not affect other files in the archive. z test contentsz more contentsr Ú0Núr+brs0000Ú1ó more contents)r[r r ÚgetinfoÚ header_offsetÚopenÚseekÚwriterrr(Ú BadZipfilerr#r1)rÚfnÚzfÚ zeroOffsetÚscribbler_ÚzferrrÚtest_invalidHeader°s ÿ   þ  ÿþ"þz!ZipstreamTests.test_invalidHeaderc Cs| ddg¡}t |d¡}| d¡}d|_Wdƒn1s wYt|dƒ}| |jd¡| |  ¡¡Wdƒn1sCwYt   |¡7}|  tj |jd¡| d ¡}| | ¡d¡Wdƒn1sowYWdƒdSWdƒdS1s‡wYdS) zƒ A zipfile entry with a different filename than is found in the central directory should raise BadZipfile. s test contentsrfr rcznot zeroNrdrre)r[r r rgrrirjrhrkÚ FileHeaderrrr(rlrr#r1)rrmrnÚinforpr_rqrrrÚtest_filenameMismatchÁs" þ þ  ÿþ"þz$ZipstreamTests.test_filenameMismatchcCs’| ¡}t |d¡}t d¡}| |d¡d|_Wdƒn1s#wYt |¡}| tj |j d¡WdƒdS1sBwYdS)zq A zipfile which describes an unsupported compression mechanism should raise BadZipfile. rrcz some dataiÒN) r r r ÚZipInforÚ compress_typerrr(rlr)rrmrnÚzir_rrrÚtest_unsupportedCompressionÓs  ú "ÿz*ZipstreamTests.test_unsupportedCompressionc Csà| ¡}t |d¡!}t d¡}d}t ddt|ƒ¡||_| |d¡Wdƒn1s.wYt   |¡.}|  d¡}|  |  ¡d¡Wdƒn1sQwYWdƒdSWdƒdS1siwYdS)zW readfile() should skip over 'extra' data present in the zip metadata. rrcs hello, extraz÷óz7ZipstreamTests.test_unzipIterChunky..cSóg|]}| d¡‘qS©Úascii©Úencoder„rrrr†øóN)Úranger[ÚlistrÚunzipIterChunkyrPrTr#ÚsetÚlistdirÚmaprVÚchildrenÚintÚbasenameÚ getContent)rÚnumfilesrrXrOÚnumrrrÚtest_unzipIterChunkyñs $ þz#ZipstreamTests.test_unzipIterChunkycCs¬d}dd„t|ƒDƒ}dd„|Dƒ}| |d¡}tt ||jj¡ƒdd„t|ƒDƒ}| t|j  d¡  ¡ƒ|¡|j  d¡  ¡D]}t |  ¡ƒ}| | ¡||¡qAd S) zä 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. rcSr‚rƒrr„rrrr†r‡z@ZipstreamTests.test_unzipIterChunkyDirectory..cSrˆr‰r‹r„rrrr† rÚfoocSsh|] }t|ƒ d¡’qSr‰)rVrŒ)r…r™rrrÚ sz?ZipstreamTests.test_unzipIterChunkyDirectory..sfooN)rŽr[rrrrPrTr#r‘rOr’r”r•r–r—)rr˜rrXÚ fileContentsrOr™rrrÚtest_unzipIterChunkyDirectorys  þz,ZipstreamTests.test_unzipIterChunkyDirectorycCsd}tdƒD]}tt ¡dƒ}t|ƒ d¡}|d|7}qt|ƒ ¡} t |  ¡¡} |   ¡|   d¡j } |  | ||¡tj| | j |d} t| ƒ} || koQ|kn}| |¡| D]} q[| | d¡|   d ¡ ¡}t| ¡ƒ ¡}| || ¡Wd ƒd S1s…wYd S) zW unzipIterChunky should unzip the given number of bytes per iteration. r"ièé rŠó z bigfile.zip)Ú chunksizerÚ zipstreamjunkN)rŽÚroundÚrandomrVrŒrÚ hexdigestrrLr rNrOrTÚ _makebigfilerrr'rr#rir1)rr r¡ÚlowerÚupperÚjunkÚnr™Ú numEncodedÚjunkmd5ÚtempdirÚzfpathÚuziterr ÚapproxÚfÚnewmd5rrrÚ_unzipIterChunkyTests*     "þz#ZipstreamTests._unzipIterChunkyTestcCó| tjddd¡dS)zs unzipIterChunky should unzip the given number of bytes per iteration on a stored archive. iôé#é-N)r³r rIrQrrrÚtest_unzipIterChunkyStored5óz)ZipstreamTests.test_unzipIterChunkyStoredcCr´)zu unzipIterChunky should unzip the given number of bytes per iteration on a deflated archive. iÌééN)r³r rFrQrrrÚtest_chunkyDeflated<r¸z"ZipstreamTests.test_chunkyDeflatedcCsbt |d|¡ }tdƒD] }d|}| |d¡q | d|¡WdƒdS1s*wYdS)zT Create a zip file with the given file name and compression scheme. rrz zipstream%drr¢N)r r rŽr)rrr r©rnrZrmrrrr¦Cs  "üzZipstreamTests._makebigfileN)r)r@rArBrCrRr[r`rbrrruryr€ršržr³r·r»r¦rrrrrJ‚s     rJ)rCr¤r{r ÚhashlibrÚtwisted.pythonrrÚ twisted.trialrrÚTestCaserErHrJrrrrÚs  a