U «`”[0ã@s’dZddlZddlZddlZddlmZddlmZmZddl m Z Gdd„de ƒZ Gdd „d e e j ƒZGd d „d e e j ƒZGd d „d e j ƒZdS)z' Tests for L{twisted.python.zipstream} éN)Úmd5)Ú zipstreamÚfilepath)Ú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 c CsF| ¡}t |d|j¡}| d|¡W5QRXt |d¡}| d¡S)z6 Return an appropriate zip file entry ÚwÚcontentÚ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.getFileEntryc Cs(| d¡}| | ¡¡W5QRXdS)zK zip files should not be ttys, so isatty() should be false ÚN)rÚ assertFalseÚisatty©rZ fileEntryrrrÚ test_isatty s zFileEntryMixin.test_isattyc Cs2| d¡}| |j¡W5QRX| |j¡dS)zd The C{closed} attribute should reflect whether C{close()} has been called. rN)rrÚclosedÚ assertTruerrrrÚ test_closed(s zFileEntryMixin.test_closedc CsJ| d¡6}| | ¡d¡| | ¡d¡| | ¡d¡W5QRXdS)zk C{readline()} should mirror L{file.readline} and return up to a single delimiter. shoho hoshoho óhoóN)rÚ assertEqualÚreadlinerrrrÚ test_readline2s zFileEntryMixin.test_readlinec CsH| d¡4}| | ¡d¡| | ¡d¡| t|j¡W5QRXdS)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_nextc Cs0| d¡}| | ¡dddg¡W5QRXdS)zG C{readlines()} should return a list of all the lines. sho ho hor%r N)rr"Ú readlinesrrrrÚtest_readlinesGs zFileEntryMixin.test_readlinesc Cs:| d¡&}| t|ƒ|¡| | ¡|¡W5QRXdS)zJ C{__iter__()} and C{xreadlines()} should return C{self}. rN)rZassertIsÚiterZ xreadlinesrrrrÚtest_iterationOs zFileEntryMixin.test_iterationc Cs.d}| |¡}| | ¡|¡W5QRXdS)z9 C{.read()} should read the entire file. s Hello, world!N)rr"Úread©rrÚentryrrrÚtest_readWholeXs zFileEntryMixin.test_readWholec CsJd}| |¡}| d¡}| d¡}W5QRX| |d¡| |d¡dS)zD C{.read(num)} should read num bytes from the file. Ú 0123456789ééÈs0123s456789N)rr.r")rrr0ZoneZtworrrÚtest_readPartialas    zFileEntryMixin.test_readPartialc CsRd}| |¡:}| d¡| | ¡d¡| d¡| | ¡d¡W5QRXdS)zb C{.tell()} should return the number of bytes that have been read so far. Zdxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxér3éN)rr.r"Útellr/rrrÚ test_tellms    zFileEntryMixin.test_tellN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr$r)r+r-r1r5r9rrrrrs       rc@seZdZdZejZdS)ÚDeflatedZipFileEntryTestsz2 DeflatedZipFileEntry should be file-like N)r:r;r<r=r Ú ZIP_DEFLATEDr rrrrr>{sr>c@seZdZdZejZdS)ÚZipFileEntryTestsz( ZipFileEntry should be file-like N)r:r;r<r=r Ú ZIP_STOREDr rrrrr@ƒsr@c@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 ZunzippedN)rÚFilePathr ÚtestdirÚmakedirsÚchildÚunzipdir©rrrrÚsetUps zZipstreamTests.setUprc Cs`|j d¡j}t |d¡<}t|ƒD],\}}t|ƒ}|rD|d|}| ||¡q$W5QRX|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ú/)rDrFÚpathr r Ú enumerateÚstrr)rrZ directoryÚ zpfilenameZzpfileÚirrrrrÚ makeZipFilešs zZipstreamTests.makeZipFilec Cs0t | ¡d¡}| t|jd¡W5QRXdS)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©szZipstreamTests.test_invalidModecCs0t | dg¡d¡}| ¡| t|jd¡dS)zn A closed ChunkingZipFile should raise a L{RuntimeError} when .readfile() is invoked. rQr N)rrrPÚcloser'rRrrSrrrÚtest_closedArchive²sz!ZipstreamTests.test_closedArchivec Cs°| ddg¡}t |d¡}| d¡j}W5QRXt|dƒ}| |d¡| d¡W5QRXt  |¡>}|  tj |j d¡|  d¡}|  | ¡d ¡W5QRXW5QRXd S) 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 Ú0úr+brs0000Ú1ó more contentsN)rPr r ÚgetinfoÚ header_offsetÚopenÚseekÚwriterrr'Ú BadZipfilerr"r.)rÚfnÚzfZ zeroOffsetÚscribblerTÚzferrrÚtest_invalidHeader¼sÿ    z!ZipstreamTests.test_invalidHeaderc Csº| ddg¡}t |d¡}| d¡}d|_W5QRXt|dƒ"}| |jd¡| |  ¡¡W5QRXt   |¡>}|  tj |jd¡| d¡}| | ¡d¡W5QRXW5QRXd S) zƒ A zipfile entry with a different filename than is found in the central directory should raise BadZipfile. s test contentsr[r rXznot zerorYrrZN)rPr r r\rr^r_r]r`Z FileHeaderrrr'rarr"r.)rrbrcÚinfordrTrerrrÚtest_filenameMismatchÏsÿ    z$ZipstreamTests.test_filenameMismatchc Csh| ¡}t |d¡"}t d¡}| |d¡d|_W5QRXt |¡}| tj |j d¡W5QRXdS)zq A zipfile which describes an unsupported compression mechanism should raise BadZipfile. rrXz some dataiÒN) r r r ÚZipInforZ compress_typerrr'rar)rrbrcÚzirTrrrÚtest_unsupportedCompressionãs   z*ZipstreamTests.test_unsupportedCompressionc Cs’| ¡}t |d¡8}t d¡}d}t ddt|ƒ¡||_| |d¡W5QRXt   |¡,}|  d¡}|  |  ¡d¡W5QRXW5QRXdS)zW readfile() should skip over 'extra' data present in the zip metadata. rrXs hello, extraz sz7ZipstreamTests.test_unzipIterChunky..cSsg|]}| d¡‘qS©Úascii©Úencoderrrrrrt sN)ÚrangerPÚlistrÚunzipIterChunkyrGrKr"ÚsetÚlistdirÚmaprMÚchildrenÚintÚbasenameÚ getContent)rÚnumfilesrrNrFÚnumrrrÚtest_unzipIterChunkys  þ z#ZipstreamTests.test_unzipIterChunkycCs¬d}dd„t|ƒDƒ}dd„|Dƒ}| |d¡}tt ||jj¡ƒdd„t|ƒDƒ}| t|j  d¡  ¡ƒ|¡|j  d¡  ¡D]$}t |  ¡ƒ}| | ¡||¡q‚d 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. rpcSsg|] }d|‘qSrqrrrrrrrt sz@ZipstreamTests.test_unzipIterChunkyDirectory..cSsg|]}| d¡‘qSrurwrrrrrrt!sZfoocSsh|]}t|ƒ d¡’qSru)rMrx)rsr„rrrÚ $sz?ZipstreamTests.test_unzipIterChunkyDirectory..sfooN)ryrPrzrr{rGrKr"r|rFr}rr€rr‚)rrƒrrNZ fileContentsrFr„rrrÚtest_unzipIterChunkyDirectorys þ z,ZipstreamTests.test_unzipIterChunkyDirectoryc Csd}tdƒD],}tt ¡dƒ}t|ƒ d¡}|d|7}q t|ƒ ¡} t |  ¡¡} |   ¡|   d¡j } |  | ||¡tj| | j |d} t| ƒ} || ko¢|kn}| |¡| D]} q¶| | d¡|   d ¡ ¡"}t| ¡ƒ ¡}| || ¡W5QRXd S) zW unzipIterChunky should unzip the given number of bytes per iteration. r!ièé rvó z bigfile.zip)Ú chunksizerÚ zipstreamjunkN)ryÚroundÚrandomrMrxrZ hexdigestrrCr rErFrKÚ _makebigfilerr{r&rr"r^r.)rr rŠÚlowerÚupperÚjunkÚnr„Z numEncodedZjunkmd5ZtempdirZzfpathZuziterr ZapproxÚfZnewmd5rrrÚ_unzipIterChunkyTest2s,    ÿ  z#ZipstreamTests._unzipIterChunkyTestcCs| tjddd¡dS)zs unzipIterChunky should unzip the given number of bytes per iteration on a stored archive. iôé#é-N)r”r rArHrrrÚtest_unzipIterChunkyStoredQsz)ZipstreamTests.test_unzipIterChunkyStoredcCs| tjddd¡dS)zu unzipIterChunky should unzip the given number of bytes per iteration on a deflated archive. iÌééN)r”r r?rHrrrÚtest_chunkyDeflatedYsz"ZipstreamTests.test_chunkyDeflatedc CsLt |d|¡4}tdƒD]}d|}| |d¡q| d|¡W5QRXdS)zT Create a zip file with the given file name and compression scheme. rrpz zipstream%drr‹N)r r ryr)rrr r‘rcrOrbrrrrŽas  zZipstreamTests._makebigfileN)r)r:r;r<r=rIrPrUrWrfrhrkror…r‡r”r—ršrŽrrrrrB‹s   rB)r=rrmr ZhashlibrZtwisted.pythonrrZ twisted.trialrÚobjectrZTestCaser>r@rBrrrrÚs  j