o Óó·i//ã@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úhC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-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©rZ 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)rZassertIsÚiterZ 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)rr.r")rrr0ZoneZtworrrÚ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. Zdxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxér3éN)rr.r"Útellr/rrrÚ test_telles   "üzFileEntryMixin.test_tellN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr$r)r+r-r1r5r9rrrrrs     rc@óeZdZdZejZdS)ÚDeflatedZipFileEntryTestsz2 DeflatedZipFileEntry should be file-like N)r:r;r<r=r Ú ZIP_DEFLATEDr rrrrr?ró r?c@r>)ÚZipFileEntryTestsz* ZipFileEntry should be file-like N)r:r;r<r=r Ú ZIP_STOREDr rrrrrBzrArBc@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.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)rFrHÚpathr r Ú enumerateÚstrr)rrÚ directoryÚ zpfilenameZzpfileÚ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. rTr N)rrrSÚcloser'rUrrVrrrÚ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)rSr r ÚgetinfoÚ header_offsetÚopenÚseekÚwriterrr'Ú BadZipFilerr"r.)rÚfnÚzfZ zeroOffsetÚscribblerWÚ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 contentsr^r r[znot zeroNr\rr])rSr r r_rrarbr`rcÚ FileHeaderrrr'rdrr"r.)rrerfÚinforgrWrhrrrÚ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. rr[z some dataiÒN) r r r ÚZipInforÚ compress_typerrr'rdr)rrerfÚzirWrrrÚ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. rr[s hello, extraz÷óz7ZipstreamTests.test_unzipIterChunky..cSóg|]}| d¡‘qS©Úascii©Úencoder{rrrr}øóN)ÚrangerSÚlistrÚunzipIterChunkyrIrMr"ÚsetÚlistdirÚmaprOÚchildrenÚintÚbasenameÚ getContent)rÚnumfilesrrQrHÚ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. rxcSryrzrr{rrrr}r~z@ZipstreamTests.test_unzipIterChunkyDirectory..cSrr€r‚r{rrrr} r„ZfoocSsh|] }t|ƒ d¡’qSr€)rOrƒ)r|rrrrÚ sz?ZipstreamTests.test_unzipIterChunkyDirectory..sfooN)r…rSr†rr‡rIrMr"rˆrHr‰r‹rŒrrŽ)rrrrQZ fileContentsrHrrrrÚ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ÚrandomrOrƒrÚ hexdigestrrEr rGrHrMÚ _makebigfilerr‡r&rr"rar.)rr r–ÚlowerÚupperÚjunkÚnrZ numEncodedZjunkmd5ÚtempdirZzfpathZuziterr ZapproxÚfZnewmd5rrrÚ_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 rCrJrrrÚ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 r@rJrrrÚ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. rrxz zipstream%drr—N)r r r…r)rrr ržrfrRrerrrr›Cs  "üzZipstreamTests._makebigfileN)r)r:r;r<r=rKrSrXrZrirlrprwr‘r“r¢r¦rªr›rrrrrD‚s     rD)r=r™rrr ÚhashlibrZtwisted.pythonrrZ twisted.trialrrZTestCaser?rBrDrrrrÚs  a