File: //lib/python3/dist-packages/twisted/python/test/__pycache__/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