a )¶(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ƒn1s:0Yt |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úAlib/python3.9/site-packages/twisted/python/test/test_zipstream.pyÚ getFileEntrys * zFileEntryMixin.getFileEntrycCs<| d¡}| | ¡¡Wdƒn1s.0YdS)zK zip files should not be ttys, so isatty() should be false ÚN)rÚ assertFalseÚisatty©rZ fileEntryrrrÚ test_isatty s zFileEntryMixin.test_isattycCsF| d¡}| |j¡Wdƒn1s,0Y| |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ƒn1sP0YdS)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ƒn1sN0YdS)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_nextcCsB| d¡$}| | ¡gd¢¡Wdƒn1s40YdS)zG C{readlines()} should return a list of all the lines. sho ho ho)r%r%r N)rr"Ú readlinesrrrrÚtest_readlinesCs zFileEntryMixin.test_readlinescCsN| d¡0}| t|ƒ|¡| | ¡|¡Wdƒn1s@0YdS)zJ C{__iter__()} and C{xreadlines()} should return C{self}. rN)rZassertIsÚiterZ xreadlinesrrrrÚtest_iterationJs zFileEntryMixin.test_iterationcCsBd}| |¡ }| | ¡|¡Wdƒn1s40YdS)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ƒn1s80Y| |d¡| |d¡dS)zD C{.read(num)} should read num bytes from the file. Ú 0123456789ééÈNs0123s456789)rr.r")rrr0ZoneZtworrrÚtest_readPartialZs   ( zFileEntryMixin.test_readPartialcCsfd}| |¡D}| d¡| | ¡d¡| d¡| | ¡d¡Wdƒn1sX0YdS)zb C{.tell()} should return the number of bytes that have been read so far. Zdxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxér3éN)rr.r"Útellr/rrrÚ test_telles    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>rsr>c@seZdZdZejZdS)ÚZipFileEntryTestsz* ZipFileEntry should be file-like N)r:r;r<r=r Ú ZIP_STOREDr rrrrr@zsr@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ÚsetUp‡s zZipstreamTests.setUprcCst|j d¡j}t |d¡F}t|ƒD],\}}t|ƒ}|rD|d|}| ||¡q$Wdƒn1sf0Y|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)rDrFÚpathr r Ú enumerateÚstrr)rrZ directoryÚ zpfilenameZzpfileÚirrrrrÚ makeZipFile‘s ,zZipstreamTests.makeZipFilecCsDt | ¡d¡ }| t|jd¡Wdƒn1s60YdS)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}Wdƒn1s<0Yt|dƒ&}| |d¡| d¡Wdƒn1s|0Yt  |¡\}|  tj |j d¡|  d ¡ }|  | ¡d ¡Wdƒn1sÔ0YWdƒn1sò0YdS) 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)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|_Wdƒn1s@0Yt|dƒ,}| |jd¡| |  ¡¡Wdƒn1s†0Yt   |¡\}|  tj |jd¡| d ¡ }| | ¡d¡Wdƒn1sÞ0YWdƒn1sü0YdS) zƒ A zipfile entry with a different filename than is found in the central directory should raise BadZipfile. s test contentsr[r rXznot zeroNrYrrZ)rPr r r\rr^r_r]r`Z FileHeaderrrr'rarr"r.)rrbrcÚinfordrTrerrrÚtest_filenameMismatchÁs $ ,  z$ZipstreamTests.test_filenameMismatchcCs| ¡}t |d¡,}t d¡}| |d¡d|_Wdƒn1sF0Yt |¡"}| tj |j d¡Wdƒn1s‚0YdS)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¡B}t d¡}d}t ddt|ƒ¡||_| |d¡Wdƒn1s\0Yt   |¡J}|  d¡ }|  |  ¡d¡Wdƒn1s¢0YWdƒn1sÀ0YdS)zW readfile() should skip over 'extra' data present in the zip metadata. rrXs hello, extraz÷r!z7ZipstreamTests.test_unzipIterChunky..cSsg|]}| d¡‘qS©Úascii©Úencoderrrrrrtør!N)ÚrangerPÚlistrÚunzipIterChunkyrGrKr"ÚsetÚlistdirÚmaprMÚchildrenÚintÚbasenameÚ getContent)rÚnumfilesrrNrFÚ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 |  ¡ƒ}| | ¡||¡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|‘qSrqrrrrrrrtr!z@ZipstreamTests.test_unzipIterChunkyDirectory..cSsg|]}| d¡‘qSrurwrrrrrrt r!ZfoocSsh|]}t|ƒ d¡’qSru)rMrx)rsr„rrrÚ r!z?ZipstreamTests.test_unzipIterChunkyDirectory..sfooN)ryrPrzrr{rGrKr"r|rFr}rr€rr‚)rrƒrrNZ fileContentsrFr„rrrÚtest_unzipIterChunkyDirectorys  z,ZipstreamTests.test_unzipIterChunkyDirectorycCsd}tdƒD],}tt ¡dƒ}t|ƒ d¡}|d|7}q t|ƒ ¡} t |  ¡¡} |   ¡|   d¡j } |  | ||¡tj| | j |d} t| ƒ} || ko¢|kn}| |¡| D]} q¶| | d¡|   d ¡ ¡,}t| ¡ƒ ¡}| || ¡Wd ƒn1s 0Yd 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Ú_unzipIterChunkyTests(     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_unzipIterChunkyStored5sz)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_chunkyDeflated<sz"ZipstreamTests.test_chunkyDeflatedcCs`t |d|¡>}tdƒD]}d|}| |d¡q| d|¡Wdƒn1sR0YdS)zT Create a zip file with the given file name and compression scheme. rrpz zipstream%drr‹N)r r ryr)rrr r‘rcrOrbrrrrŽCs  zZipstreamTests._makebigfileN)r)r:r;r<r=rIrPrUrWrfrhrkror…r‡r”r—ršrŽrrrrrB‚s   rB)r=rrmr ZhashlibrZtwisted.pythonrrZ twisted.trialrrZTestCaser>r@rBrrrrÚs  a