CasperSecurity
o
�b// � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
G dd� d�Z G dd � d e e
j �Z
G d
d � d e e
j �ZG d d
� d
e
j �ZdS )z'
Tests for L{twisted.python.zipstream}
� N)�md5)�filepath� zipstream)�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 sZ | � � }t�|d| j��}|�d|� W d � n1 s w Y t�|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 �D/usr/lib/python3/dist-packages/twisted/python/test/test_zipstream.py� getFileEntry s �
zFileEntryMixin.getFileEntryc C s> | � d��}| �|�� � W d � dS 1 sw Y dS )zK
zip files should not be ttys, so isatty() should be false
� N)r � assertFalse�isatty�r � fileEntryr r r � test_isatty s "�zFileEntryMixin.test_isattyc C sF | � d��}| �|j� W d � n1 sw Y | �|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 s` | � d��!}| �|�� d� | �|�� d� | �|�� d� W d � dS 1 s)w Y 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_readline0 s
"�z FileEntryMixin.test_readlinec C s^ | � d�� }| �|�� d� | �|�� d� | �t|j� W d � dS 1 s(w Y 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 sD | � d��}| �|�� g d�� W d � dS 1 sw Y dS )zG
C{readlines()} should return a list of all the lines.
s ho
ho
ho)r& r& r! N)r r# � readlinesr r r r �test_readlinesC s "�z FileEntryMixin.test_readlinesc C sP | � d��}| �t|�|� | �|�� |� W d � dS 1 s!w Y dS )zJ
C{__iter__()} and C{xreadlines()} should return C{self}.
r N)r �assertIs�iter�
xreadlinesr r r r �test_iterationJ s "�z FileEntryMixin.test_iterationc C sD d}| � |��}| �|�� |� W d � dS 1 sw Y dS )z9
C{.read()} should read the entire file.
s
Hello, world!N)r r# �read�r r �entryr r r �test_readWholeR s "�z FileEntryMixin.test_readWholec C s^ d}| � |��}|�d�}|�d�}W d � n1 s w Y | �|d� | �|d� dS )zD
C{.read(num)} should read num bytes from the file.
�
0123456789� �� Ns 0123s 456789)r r1 r# )r r r3 �one�twor r r �test_readPartialZ s
� z FileEntryMixin.test_readPartialc C sh d}| � |��#}|�d� | �|�� d� |�d� | �|�� d� W d � dS 1 s-w Y dS )zb
C{.tell()} should return the number of bytes that have been read so
far.
�dxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx� r6 � N)r r1 r# �tellr2 r r r � test_telle s
"�zFileEntryMixin.test_tellN)�__name__�
__module__� __qualname__�__doc__r r r r% r* r, r0 r4 r: r? r r r r r s
r c @ � e Zd ZdZejZdS )�DeflatedZipFileEntryTestsz2
DeflatedZipFileEntry should be file-like
N)r@ rA rB rC r � ZIP_DEFLATEDr
r r r r rE r �
rE c @ rD )�ZipFileEntryTestsz*
ZipFileEntry should be file-like
N)r@ rA rB rC r �
ZIP_STOREDr
r r r r rH z rG rH 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
�unzippedN)r �FilePathr
�testdir�makedirs�child�unzipdir�r r r r �setUp� s
zZipstreamTests.setUpr c C sv | j �d�j}t�|d��$}t|�D ]\}}t|�}|r"|d | }|�||� qW d � |S 1 s4w Y |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)rM rO �pathr r � enumerate�strr )r r � directory�
zpfilename�zpfile�ir r r r r � makeZipFile� s �
��zZipstreamTests.makeZipFilec C sF t �| �� d��}| �t|jd� W d � dS 1 s w Y 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.
r\ r N)r r r[ �closer( r] r r^ r r r �test_closedArchive� s z!ZipstreamTests.test_closedArchivec C s | � ddg�}t�|d��}|�d�j}W d � n1 s w Y t|d��}|�|d� |�d� W d � n1 s>w Y t� |��7}| �
tj |j d� |� d ��}| �
|�� d
� W d � n1 sjw Y W d � dS W d � dS 1 s�w Y dS ) 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+br s 0000�1�
more contents)r[ r r �getinfo�
header_offset�open�seek�writer r r( �
BadZipfiler r# r1 )r �fn�zf�
zeroOffset�scribbler_ �zfer r r �test_invalidHeader� s � � ��"�z!ZipstreamTests.test_invalidHeaderc C s | � ddg�}t�|d��}|�d�}d|_W d � n1 s w Y t|d��}|�|jd� |�|� � � W d � n1 sCw Y t
� |��7}| � tj
|jd� |�d ��}| �|�� d� W d � n1 sow Y W d � dS W d � dS 1 s�w Y dS )
z�
A zipfile entry with a different filename than is found in the central
directory should raise BadZipfile.
s
test contentsrf r rc znot zeroNrd r re )r[ r r rg r ri rj rh rk �
FileHeaderr r r( rl r r# r1 )r rm rn �inforp r_ rq r r r �test_filenameMismatch� s"
� � ��"�z$ZipstreamTests.test_filenameMismatchc C s� | � � }t�|d��}t�d�}|�|d� d|_W d � n1 s#w Y t�|��}| �tj |j
d� W d � dS 1 sBw Y dS )zq
A zipfile which describes an unsupported compression mechanism should
raise BadZipfile.
r rc z some datai� N) r
r r �ZipInfor �
compress_typer r r( rl r )r rm rn �zir_ r r r �test_unsupportedCompression� s
� "�z*ZipstreamTests.test_unsupportedCompressionc C s� | � � }t�|d��!}t�d�}d}t�ddt|��| |_|�|d� W d � n1 s.w Y t �
|��.}|� d��}| � |�
� d� W d � n1 sQw Y W d � dS W d � dS 1 siw Y dS )zW
readfile() should skip over 'extra' data present in the zip metadata.
r rc s hello, extraz<hh�* s
the real dataN)r
r r rv �struct�pack�len�extrar r r r r# r1 )r rm rn rx �
extra_datar_ rq r r r �test_extraData� s
�P�z ZipstreamTests.test_extraDatac C s� d}dd� t |�D �}dd� |D �}| �|�}tt�|| jj�� | �t| j� � �tt
t t |���� | j� � D ]}t
|�� �}| �|�� || � q8dS )z�
L{twisted.python.zipstream.unzipIterChunky} returns an iterator which
must be exhausted to completely unzip the input archive.
�
c S � g | ]}d | �qS �zThis is test file %d!r ��.0rZ r r r �
<listcomp>� � z7ZipstreamTests.test_unzipIterChunky.<locals>.<listcomp>c S � g | ]}|� d ��qS ��ascii��encoder� r r r r� � � N)�ranger[ �listr �unzipIterChunkyrP rT r# �set�listdir�maprV �children�int�basename�
getContent)r �numfilesr rX rO �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 |�
� �}| �|�� || � 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.
r� c S r� r� r r� r r r r� r� z@ZipstreamTests.test_unzipIterChunkyDirectory.<locals>.<listcomp>c S r� r� r� r� r r r r� r� �fooc S s h | ] }t |��d ��qS r� )rV r� )r� r� r r r � <setcomp> s z?ZipstreamTests.test_unzipIterChunkyDirectory.<locals>.<setcomp>s fooN)r� r[ r� r r� rP rT r# r� rO r� r� r� r� r� )r r� r rX � fileContentsrO r� r r r � test_unzipIterChunkyDirectory s �z,ZipstreamTests.test_unzipIterChunkyDirectoryc C s d}t d�D ]}tt�� d�}t|��d�}|d| 7 }qt|��� } t�| � � �}
|
�
� |
� d�j } | �
| ||� tj| |
j |d�} t| �}
||
k oQ|k n }| �|� | D ]}
q[| �|
d� |
� d ��� �}t|�� ��� }| �|| � W d
� d
S 1 s�w Y d
S ) zW
unzipIterChunky should unzip the given number of bytes per iteration.
r" i� � r� � z bigfile.zip)� chunksizer �
zipstreamjunkN)r� �round�randomrV r� r � hexdigestr rL r
rN rO rT � _makebigfiler r� r' r r# ri r1 )r r
r� �lower�upper�junk�nr� �
numEncoded�junkmd5�tempdir�zfpath�uziterr �approx�f�newmd5r r r �_unzipIterChunkyTest s*
"�z#ZipstreamTests._unzipIterChunkyTestc C � | � tjddd� dS )zs
unzipIterChunky should unzip the given number of bytes per iteration on
a stored archive.
i� �# �- N)r� r rI rQ r r r �test_unzipIterChunkyStored5 � z)ZipstreamTests.test_unzipIterChunkyStoredc C r� )zu
unzipIterChunky should unzip the given number of bytes per iteration on
a deflated archive.
i� � � N)r� r rF rQ r r r �test_chunkyDeflated< r� z"ZipstreamTests.test_chunkyDeflatedc C sb t �|d|�� }td�D ] }d| }|�|d� q |�d|� W d � dS 1 s*w Y dS )zT
Create a zip file with the given file name and compression scheme.
r r� z zipstream%dr r� N)r r r� r )r r r
r� rn rZ rm r r r r� C s "�zZipstreamTests._makebigfileN)r )r@ rA rB rC rR r[ r` rb rr ru ry r� r� r� r� r� r� r� r r r r rJ � s
rJ )rC r� r{ r �hashlibr �twisted.pythonr r �
twisted.trialr r �TestCaserE rH rJ r r r r �<module> s a