tf//dZddlZddlZddlZddlmZddlmZmZddl m Z GddZ Gdd e e j Z Gd d e e j ZGd d e j ZdS)z' Tests for L{twisted.python.zipstream} N)md5)filepath zipstream)unittestcNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) FileEntryMixinz? File entry classes should behave as file-like objects c|}tj|d|j5}|d|dddn #1swxYwYt j|d}|dS)z6 Return an appropriate zip file entry wcontentNr)mktempzipfileZipFile compressionwritestrrChunkingZipFilereadfile)selfcontentsfilenamezs a/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/python/test/test_zipstream.py getFileEntryzFileEntryMixin.getFileEntrys;;== _XsD,< = = , JJy( + + + , , , , , , , , , , , , , , ,  %h 4 4zz)$$$sAAAc|d5}||ddddS#1swxYwYdS)zK zip files should not be ttys, so isatty() should be false N)r assertFalseisattyr fileEntrys r test_isattyzFileEntryMixin.test_isatty s  r " " 1i   Y--// 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1s(A  AAc|d5}||jdddn #1swxYwY||jdS)zd The C{closed} attribute should reflect whether C{close()} has been called. rN)rrclosed assertTruers r test_closedzFileEntryMixin.test_closed's   r " " /i   Y- . . . / / / / / / / / / / / / / / /  ()))))s=AAcR|d5}||d||d||dddddS#1swxYwYdS)zk C{readline()} should mirror L{file.readline} and return up to a single delimiter. shoho hoshoho hoN)r assertEqualreadliners r test_readlinezFileEntryMixin.test_readline0s   { + + 8y   Y//119 = = =   Y//115 9 9 9   Y//113 7 7 7 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8 8sA9BB #B cB|d5}||d||d|t|jddddS#1swxYwYdS)zV Zip file entries should implement the iterator protocol as files do. sho hohoho shohoN)rr(next assertRaises StopIterationrs r test_nextzFileEntryMixin.test_next:s  { + + =y   Y^^--w 7 7 7   Y^^--w 7 7 7   mY^ < < < = = = = = = = = = = = = = = = = = =sA1BBBc|d5}||gdddddS#1swxYwYdS)zG C{readlines()} should return a list of all the lines. sho ho ho)r,r,r&N)rr( readlinesrs rtest_readlineszFileEntryMixin.test_readlinesCs  } - - O   Y00224M4M4M N N N O O O O O O O O O O O O O O O O O Os+AAAc|d5}|t|||||ddddS#1swxYwYdS)zJ C{__iter__()} and C{xreadlines()} should return C{self}. rN)rassertIsiter xreadlinesrs rtest_iterationzFileEntryMixin.test_iterationJs  r " " =i MM$y//9 5 5 5 MM)..00) < < < = = = = = = = = = = = = = = = = = =sA A//A36A3cd}||5}|||ddddS#1swxYwYdS)z9 C{.read()} should read the entire file. s Hello, world!N)rr(readrrentrys rtest_readWholezFileEntryMixin.test_readWholeRs$   x ( ( 5E   UZZ\\8 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5s)AAAcd}||5}|d}|d}dddn #1swxYwY||d||ddS)zD C{.read(num)} should read num bytes from the file. 0123456789Ns0123s456789)rr:r()rrr<onetwos rtest_readPartialzFileEntryMixin.test_readPartialZs    x ( ( "E**Q--C**S//C " " " " " " " " " " " " " " " g&&& i(((((s+AAAcZd}||5}|d||d|d||dddddS#1swxYwYdS)zb C{.tell()} should return the number of bytes that have been read so far. dxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxr@N)rr:r(tellr;s r test_tellzFileEntryMixin.test_telles    x ( ( .E JJqMMM   UZZ\\1 - - - JJqMMM   UZZ\\1 - - -  . . . . . . . . . . . . . . . . . .sA;B  B$'B$N)__name__ __module__ __qualname____doc__rr r$r*r0r3r8r=rDrJr'rrrs%%%111***888===OOO===555 ) ) ) . . . . .r'rc eZdZdZejZdS)DeflatedZipFileEntryTestsz2 DeflatedZipFileEntry should be file-like N)rKrLrMrNr ZIP_DEFLATEDrrOr'rrQrQrs"&KKKr'rQc eZdZdZejZdS)ZipFileEntryTestsz* ZipFileEntry should be file-like N)rKrLrMrNr ZIP_STOREDrrOr'rrTrTzs"$KKKr'rTcheZdZdZdZddZdZdZdZdZ d Z d Z d Z d Z d ZdZdZdZdS)ZipstreamTestsz, Tests for twisted.python.zipstream ctj||_|j|jd|_|jdS)zz Creates junk data that can be compressed and a test directory for any files that will be created unzippedN)rFilePathr testdirmakedirschildunzipdirrs rsetUpzZipstreamTests.setUpsc  (77   **:66       r'rc(|jdj}tj|d5}t |D]4\}}t |}|r|dz|z}|||5 dddn #1swxYwY|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)r[r]pathrr enumeratestrr)rr directory zpfilenamezpfileir rs r makeZipFilezZipstreamTests.makeZipFiles \'' 66; _Z - - 3'11 3 3 7q66:(39H'2222  3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 sABB B ctj|d5}|t|jdddddS#1swxYwYdS)z A ChunkingZipFile opened in write-mode should not allow .readfile(), and raise a RuntimeError instead. r somethingN)rrr r. RuntimeErrorrrczfs rtest_invalidModezZipstreamTests.test_invalidModes  &t{{}}c : : Gc   lCL+ F F F G G G G G G G G G G G G G G G G G Gs"AAActj|dgd}||t |jddS)zn A closed ChunkingZipFile should raise a L{RuntimeError} when .readfile() is invoked. rlr N)rrrjcloser.rmrrns rtest_closedArchivez!ZipstreamTests.test_closedArchivesS '(8(8+(G(GMM  , kBBBBBr'c|ddg}tj|d5}|dj}dddn #1swxYwYt |d5}||d|ddddn #1swxYwYtj |5}| tj |j d| d 5}| |d dddn #1swxYwYddddS#1swxYwYdS) 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 0Nr+brs00001 more contents)rjrrgetinfo header_offsetopenseekwriterrr. BadZipFilerr(r:)rfnzf zeroOffsetscribblerozfes rtest_invalidHeaderz!ZipstreamTests.test_invalidHeaders.   @ A A _R % % 7C6J 7 7 7 7 7 7 7 7 7 7 7 7 7 7 7"e__ % MM*a ( ( ( NN8 $ $ $ % % % % % % % % % % % % % % % &r * * ?c   g0#, D D Dc"" ?c  -=>>> ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?sYAAA/,B''B+.B+D; ?EEEc|ddg}tj|d5}|d}d|_dddn #1swxYwYt |d5}||jd|| dddn #1swxYwYtj |5}| tj |jd|d 5}||ddddn #1swxYwYddddS#1swxYwYdS) z A zipfile entry with a different filename than is found in the central directory should raise BadZipFile. s test contentsrxr ruznot zeroNrvrrw)rjrrryrr{r|rzr} FileHeaderrrr.r~rr(r:)rrrinforrors rtest_filenameMismatchz$ZipstreamTests.test_filenameMismatchsA   /1AB C C _R % % '::c??D&DM ' ' ' ' ' ' ' ' ' ' ' ' ' ' '"e__ . MM$,a 0 0 0 NN4??,, - - - . . . . . . . . . . . . . . . &r * * ?c   g0#, D D Dc"" ?c  -=>>> ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?sZAAA1ACCC>KBH KK, - - -  . . . . . . . . . . . . . . .  &r * * ;c3<<3D3D ;   SXXZZ)9 : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ;sHABBB-D)C8, D8C< <D?C< DDDc zd}dt|D}d|D}||}ttj||jj|t|j tttt||j D]Q}t|}||||RdS)z L{twisted.python.zipstream.unzipIterChunky} returns an iterator which must be exhausted to completely unzip the input archive. cg|]}d|zSzThis is test file %d!rO.0ris r z7ZipstreamTests.test_unzipIterChunky..IIIA+a/IIIr'c8g|]}|dSasciiencoders rrz7ZipstreamTests.test_unzipIterChunky..$888!AHHW%%888r'N)rangerjlistrunzipIterChunkyr^rcr(setlistdirmaprechildrenintbasename getContent)rnumfilesrrgr]nums rtest_unzipIterChunkyz#ZipstreamTests.test_unzipIterChunkys IIxIII88x888%%h//  Y &z4=3E F FGGG T]224455s3sE(OO;T;T7U7UVVV]++-- @ @Eenn&&''C   U--//# ? ? ? ? @ @r'cd}dt|D}d|D}||d}ttj||jjdt|D}|t|j d ||j d D]Q}t| }||||RdS)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. rcg|]}d|zSrrOrs rrz@ZipstreamTests.test_unzipIterChunkyDirectory..rr'c8g|]}|dSrrrs rrz@ZipstreamTests.test_unzipIterChunkyDirectory.. rr'foocRh|]$}t|d%Sr)rer)rrs r z?ZipstreamTests.test_unzipIterChunkyDirectory.. s*LLLSC00LLLr'sfooN)rrjrrrr^rcr(rr]rrrrr)rrrrg fileContentsr]rs rtest_unzipIterChunkyDirectoryz,ZipstreamTests.test_unzipIterChunkyDirectorys6 IIxIII88x888%%h66  Y &z4=3E F FGGGLLE(OOLLL  T]0088@@BBCC\RRR]((0099;; @ @Eenn&&''C   U--//# ? ? ? ? @ @r'cd}tdD]M}ttjd}t|d}|d|zz }Nt |} tj| } | | dj } | | ||tj| | j |} t!| } || cxko|knc}||| D]} || d| d 5}t |}||| d d d d S#1swxYwYd S) zW unzipIterChunky should unzip the given number of bytes per iteration. r'i r z bigfile.zip) chunksizer zipstreamjunkN)rroundrandomrerr hexdigestrrZr r\r]rc _makebigfilerrr-r#r(r{r:)rrrlowerupperjunknr numEncodedjunkmd5tempdirzfpathuziterr approxfnewmd5s r_unzipIterChunkyTestz#ZipstreamTests._unzipIterChunkyTestst & &A ,,CS11J D:% %DDd))%%''#DKKMM22}--2 &+t444*67<9UUU LL""""U""""   A  A ]]? + + 0 0 2 2 .a]],,..F   VW - - - . . . . . . . . . . . . . . . . . .s>A GGGcJ|tjddddS)zs unzipIterChunky should unzip the given number of bytes per iteration on a stored archive. i#-N)rrrUr_s rtest_unzipIterChunkyStoredz)ZipstreamTests.test_unzipIterChunkyStored5s' !!'"4c2rBBBBBr'cJ|tjddddS)zu unzipIterChunky should unzip the given number of bytes per iteration on a deflated archive. iN)rrrRr_s rtest_chunkyDeflatedz"ZipstreamTests.test_chunkyDeflated<s' !!'"6RDDDDDr'ctj|d|5}tdD]}d|z}||d|d|ddddS#1swxYwYdS)zT Create a zip file with the given file name and compression scheme. r rz zipstream%drrN)rrrr)rrrrrrirs rrzZipstreamTests._makebigfileCs_XsK 8 8 /B2YY $ $"Q& B#### KK . . .  / / / / / / / / / / / / / / / / / /sAA((A,/A,N)r)rKrLrMrNr`rjrprsrrrrrrrrrrrOr'rrWrWs !!!    GGGCCC???"???$EEE" ; ; ;@@@ @@@,...<CCCEEE/////r'rW)rNrrrhashlibrtwisted.pythonrr twisted.trialrrTestCaserQrTrWrOr'rrs8 ........""""""^.^.^.^.^.^.^.^.B'''''0A'''%%%%%(9%%%I/I/I/I/I/X&I/I/I/I/I/r'