← Back
Editing: test_zipstream.cpython-38.pyc
U �`�[0 � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m Z G dd� de �Z G dd � d e e j �ZG d d � d e e j �ZG d d � d e j �ZdS )z' Tests for L{twisted.python.zipstream} � N)�md5)� zipstream�filepath)�unittestc @ s` e Zd ZdZdd� 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 C sF | � � }t�|d| j��}|�d|� W 5 Q R X t�|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� getFileEntry s zFileEntryMixin.getFileEntryc C s( | � d��}| �|�� � W 5 Q R X dS )zK zip files should not be ttys, so isatty() should be false � N)r � assertFalse�isatty�r Z fileEntryr r r � test_isatty s zFileEntryMixin.test_isattyc C s2 | � d��}| �|j� W 5 Q R X | �|j� dS )zd The C{closed} attribute should reflect whether C{close()} has been called. r N)r r �closed� assertTruer r r r � test_closed( s zFileEntryMixin.test_closedc C sJ | � d��6}| �|�� d� | �|�� d� | �|�� d� W 5 Q R X dS )zk C{readline()} should mirror L{file.readline} and return up to a single delimiter. s hoho hos hoho � ho� N)r � assertEqual�readliner r r r � test_readline2 s z FileEntryMixin.test_readlinec C sH | � d��4}| �|�� d� | �|�� d� | �t|j� W 5 Q R X dS )zV Zip file entries should implement the iterator protocol as files do. s ho hoho� ho s hohoN)r r" �next� assertRaises� StopIterationr r r r � test_next= s zFileEntryMixin.test_nextc C s0 | � d�� }| �|�� dddg� W 5 Q R X dS )zG C{readlines()} should return a list of all the lines. s ho ho hor% r N)r r" � readlinesr r r r �test_readlinesG s z FileEntryMixin.test_readlinesc C s: | � d��&}| �t|�|� | �|�� |� W 5 Q R X dS )zJ C{__iter__()} and C{xreadlines()} should return C{self}. r N)r ZassertIs�iterZ xreadlinesr r r r �test_iterationO s z FileEntryMixin.test_iterationc C s. d}| � |��}| �|�� |� W 5 Q R X dS )z9 C{.read()} should read the entire file. s Hello, world!N)r r" �read�r r �entryr r r �test_readWholeX s z FileEntryMixin.test_readWholec C sJ d}| � |��}|�d�}|�d�}W 5 Q R X | �|d� | �|d� dS )zD C{.read(num)} should read num bytes from the file. � 0123456789� �� s 0123s 456789N)r r. r" )r r r0 ZoneZtwor r r �test_readPartiala s z FileEntryMixin.test_readPartialc C sR d}| � |��:}|�d� | �|�� d� |�d� | �|�� d� W 5 Q R X dS )zb C{.tell()} should return the number of bytes that have been read so far. Zdxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx� r3 � N)r r. r" �tellr/ r r r � test_tellm s zFileEntryMixin.test_tellN)�__name__� __module__� __qualname__�__doc__r r r r$ r) r+ r- r1 r5 r9 r r r r r s r c @ s e Zd ZdZejZdS )�DeflatedZipFileEntryTestsz2 DeflatedZipFileEntry should be file-like N)r: r; r< r= r � ZIP_DEFLATEDr r r r r r> { s r> c @ s e Zd ZdZejZdS )�ZipFileEntryTestsz( ZipFileEntry should be file-like N)r: r; r< r= r � ZIP_STOREDr r r r r r@ � s r@ c @ s� e Zd ZdZdd� Zd dd�Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd S )!�ZipstreamTestsz, Tests for twisted.python.zipstream c C s6 t �| �� �| _| 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�r r r r �setUp� s zZipstreamTests.setUpr c C s` | j �d�j}t�|d��<}t|�D ],\}}t|�}|rD|d | }|�||� q$W 5 Q R X |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 �/)rD rF �pathr r � enumerate�strr )r r Z directory� zpfilenameZzpfile�ir r r r r � makeZipFile� s zZipstreamTests.makeZipFilec C s0 t �| �� d��}| �t|jd� W 5 Q R X dS )z� A ChunkingZipFile opened in write-mode should not allow .readfile(), and raise a RuntimeError instead. r � somethingN)r r r r' � RuntimeErrorr �r �czfr r r �test_invalidMode� s z ZipstreamTests.test_invalidModec C s0 t �| �dg�d�}|�� | �t|jd� dS )zn A closed ChunkingZipFile should raise a L{RuntimeError} when .readfile() is invoked. rQ r N)r r rP �closer' rR r rS r r r �test_closedArchive� s z!ZipstreamTests.test_closedArchivec C s� | � ddg�}t�|d��}|�d�j}W 5 Q R X t|d�� }|�|d� |�d� W 5 Q R X t� |��>}| � tj |j d� |� d��}| � |�� d � W 5 Q R X W 5 Q R X d 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+br s 0000�1� more contentsN)rP r r �getinfo� header_offset�open�seek�writer r r' � BadZipfiler r" r. )r �fn�zfZ zeroOffset�scribblerT �zfer r r �test_invalidHeader� s � z!ZipstreamTests.test_invalidHeaderc C s� | � ddg�}t�|d��}|�d�}d|_W 5 Q R X t|d��"}|�|jd� |�|� � � W 5 Q R X t � |��>}| � tj |jd� |�d��}| �|�� d� W 5 Q R X W 5 Q R X d S ) z� A zipfile entry with a different filename than is found in the central directory should raise BadZipfile. s test contentsr[ r rX znot zerorY r rZ N)rP r r r\ r r^ r_ r] r` Z FileHeaderr r r' ra r r" r. )r rb rc �inford rT re r r r �test_filenameMismatch� s � z$ZipstreamTests.test_filenameMismatchc C sh | � � }t�|d��"}t�d�}|�|d� d|_W 5 Q R X t�|��}| �tj |j d� W 5 Q R X dS )zq A zipfile which describes an unsupported compression mechanism should raise BadZipfile. r rX z some datai� N) r r r �ZipInfor Z compress_typer r r' ra r )r rb rc �zirT r r r �test_unsupportedCompression� s z*ZipstreamTests.test_unsupportedCompressionc C s� | � � }t�|d��8}t�d�}d}t�ddt|��| |_|�|d� W 5 Q R X t � |��,}|� d��}| � |� � d� W 5 Q R X W 5 Q R X dS )zW readfile() should skip over 'extra' data present in the zip metadata. r rX s hello, extraz<hh�* s the real dataN)r r r ri �structZpack�lenZextrar r r r r" r. )r rb rc rj Z extra_datarT re r r r �test_extraData� s ��z ZipstreamTests.test_extraDatac C s� d}dd� t |�D �}dd� |D �}| �|�}tt�|| jj�� | �t| j� � �tt t t |���� | j� � D ]$}t |�� �}| �|�� || � qpdS )z� L{twisted.python.zipstream.unzipIterChunky} returns an iterator which must be exhausted to completely unzip the input archive. � c S s g | ] }d | �qS �zThis is test file %d!r ��.0rO r r r � <listcomp> s z7ZipstreamTests.test_unzipIterChunky.<locals>.<listcomp>c S s g | ]}|� d ��qS ��ascii��encoderr r r r rt s N)�rangerP �listr �unzipIterChunkyrG rK r" �set�listdir�maprM �children�int�basename� getContent)r �numfilesr rN rF �numr r r �test_unzipIterChunky s � z#ZipstreamTests.test_unzipIterChunkyc C s� 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. rp c S s g | ] }d | �qS rq r rr r r r rt s z@ZipstreamTests.test_unzipIterChunkyDirectory.<locals>.<listcomp>c S s g | ]}|� d ��qS ru rw rr r r r rt ! s Zfooc S s h | ]}t |��d ��qS ru )rM rx )rs r� r r r � <setcomp>$ s z?ZipstreamTests.test_unzipIterChunkyDirectory.<locals>.<setcomp>s fooN)ry rP rz r r{ rG rK r" r| rF r} r r� r� r� )r r� r rN Z fileContentsrF r� r r r � test_unzipIterChunkyDirectory s � z,ZipstreamTests.test_unzipIterChunkyDirectoryc C s d}t d�D ],}tt�� d�}t|��d�}|d| 7 }q t|��� } t�| � � �} | � � | � d�j } | � | ||� tj| | j |d�} t| �} || k o�|k n }| �|� | D ]} q�| �| d� | � d ��� �"}t|�� ��� }| �|| � W 5 Q R X d S ) zW unzipIterChunky should unzip the given number of bytes per iteration. r! i� � rv � z bigfile.zip)� chunksizer � zipstreamjunkN)ry �round�randomrM rx r Z hexdigestr rC r rE rF rK � _makebigfiler r{ r& r r" r^ r. )r r r� �lower�upper�junk�nr� Z numEncodedZjunkmd5ZtempdirZzfpathZuziterr Zapprox�fZnewmd5r r r �_unzipIterChunkyTest2 s, � z#ZipstreamTests._unzipIterChunkyTestc C s | � tjddd� dS )zs unzipIterChunky should unzip the given number of bytes per iteration on a stored archive. i� �# �- N)r� r rA rH r r r �test_unzipIterChunkyStoredQ s z)ZipstreamTests.test_unzipIterChunkyStoredc C s | � tjddd� dS )zu unzipIterChunky should unzip the given number of bytes per iteration on a deflated archive. i� � � N)r� r r? rH r r r �test_chunkyDeflatedY s z"ZipstreamTests.test_chunkyDeflatedc C sL t �|d|��4}td�D ]}d| }|�|d� q|�d|� W 5 Q R X dS )zT Create a zip file with the given file name and compression scheme. r rp z zipstream%dr r� N)r r ry r )r r r r� rc rO rb r r r r� a s zZipstreamTests._makebigfileN)r )r: r; r< r= rI rP rU rW rf rh rk ro r� r� r� r� r� r� r r r r rB � s rB )r= r� rm r Zhashlibr Ztwisted.pythonr r Z twisted.trialr �objectr ZTestCaser> r@ rB r r r r �<module> s j
Save File
Cancel