Your IP : 216.73.216.213
3
� f�k � @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z ddl m
Z
ddlmZmZ ddlmZ e
jd�ZdZd Zejje
j�d
ZG d d � d ej�Z G d
d� dej �Z G dd� de �Z G dd� de �Z dd� Z!G dd� dej �Z"d dd�Z#e$dk�r e#dd� dS ) z!Test script for the gzip module.
� N)�PIPE�Popen)�support)�_4G�
bigmemtest)�assert_python_ok�gzipsy int length=DEFAULTALLOC, err = Z_OK;
PyObject *RetVal;
int flushmode = Z_FINISH;
unsigned long start_total_out;
s� /* zlibmodule.c -- gzip-compatible data compression */
/* See http://www.gzip.org/zlib/
/* See http://www.winimage.com/zLibDll for Windows */
z-gzdirc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� UnseekableIOc C s dS )NF� )�selfr
r
�3/opt/alt/python36/lib64/python3.6/test/test_gzip.py�seekable$ s zUnseekableIO.seekablec C s
t j�d S )N)�io�UnsupportedOperation)r r
r
r �tell' s zUnseekableIO.tellc G s
t j�d S )N)r r )r �argsr
r
r �seek* s zUnseekableIO.seekN)�__name__�
__module__� __qualname__r
r r r
r
r
r r # s r c @ s" e Zd ZejZdd� Zdd� ZdS )�BaseTestc C s t j| j� d S )N)r �unlink�filename)r r
r
r �setUp1 s zBaseTest.setUpc C s t j| j� d S )N)r r r )r r
r
r �tearDown4 s zBaseTest.tearDownN)r r r r �TESTFNr r r r
r
r
r r . s r c @ sR e Zd ZdSdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z e e
dd�dd� �Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLdM� Z)dNdO� Z*dPdQ� Z+dRS )T�TestGzip�bc C sr t |�}tj| jd| ��}|j|�}W d Q R X | j|t|�� tj| jd| ��}| j|j� |� W d Q R X d S )N�w�r)�bytesr �GzipFiler �write� assertEqual�len�read)r �data�modeZb_data�f�lr
r
r �write_and_read_back9 s z TestGzip.write_and_read_backc
C sd t j| jd��D}|jtd � |j� |j� ttd�rFtj |j� � |j
� W d Q R X |j
� d S )N�wb�2 �fsync) r r! r r" �data1�flush�fileno�hasattr�osr- �close)r r( r
r
r �
test_writeA s
zTestGzip.test_writec C s� t j| j�}tj|d��}|jtd � W d Q R X | j|jt � tj|d��}|jt� W d Q R X tj|��}|j
� }W d Q R X | j |td � | j|jt � d S )Nr r, �a�3 ) �pathlib�Pathr r r! r" r. ZassertIsInstance�name�strr% r# )r r r( �dr
r
r �"test_write_read_with_pathlike_fileO s z+TestGzip.test_write_read_with_pathlike_filec C sD | j ttd �� tttd���}|jddddgd�}| j |� d S )Nr, � �B� � )�shape)r* �
memoryviewr. r �range�cast)r �mr& r
r
r �test_write_memoryview^ s z TestGzip.test_write_memoryviewc C s | j ttd �� d S )Nr, )r* � bytearrayr. )r r
r
r �test_write_bytearrayd s z TestGzip.test_write_bytearrayc C s | j tjdtd �� d S )N�I�( )r* �arrayr. )r r
r
r �test_write_arrayg s zTestGzip.test_write_arrayc C s� t j| jd��P}| jt�� |jd� W d Q R X | jt�� |jg � W d Q R X |jt� W d Q R X t j| jd��}| j|j� t� W d Q R X d S )Nr+ � �rb) r r! r � assertRaises� TypeErrorr" r. r# r% )r r( r
r
r � test_write_incompatible_typej s z%TestGzip.test_write_incompatible_typec C s> | j � tj| jd��}|j� }W d Q R X | j|td � d S )Nr r, )r4 r r! r r% r# r. )r r( r; r
r
r � test_readw s zTestGzip.test_readc C s~ | j � g }d}tj| jd��@}x8|j� }|s0P |j|� |t|�7 }| j|j� |� q"W W d Q R X | jdj |�t
d � d S )Nr r � r, ) r4 r r! r �read1�appendr$ r# r �joinr. )r Zblocks�nreadr( r; r
r
r �
test_read1~ s
zTestGzip.test_read1� )�sizeZmemusec C s8 t jtdd�}t jtj|�dd�}| j|j|�t� d S )NrY )�
compresslevelrN )�fileobjr' )r �compressr. r! r �BytesIOr# r% )r rZ �
compressedr( r
r
r �test_read_large� s zTestGzip.test_read_largec % C s | j � tj| jd�}|j}| j|j� |j� | j|j� | j t
�� |j d� W d Q R X | j t
�� |j d� W d Q R X | j t
�� |j
� W d Q R X tj| jd�}|j}| j|j� |j� | j|j� | j t
�� |jd� W d Q R X | j t
�� |j� W d Q R X d S )Nr rY r r rS )r4 r r! r r\ � assertFalse�closedr3 �
assertTruerO �
ValueErrorr% r r r" r/ )r r( r\ r
r
r �test_io_on_closed_object� s* z!TestGzip.test_io_on_closed_objectc C sn | j � tj| jd��}|jtd � W d Q R X tj| jd��}|j� }W d Q R X | j|td td � d S )N�ab� rN r, ) r4 r r! r r" �data2r% r# r. )r r( r; r
r
r � test_append� s zTestGzip.test_appendc C s� t j| jdd��}|jd� W d Q R X x8tdd�D ]*}t j| jdd��}|jd� W d Q R X q2W t j| jd��(}d}x |jd �}||7 }|svP qvW W d Q R X | j|dd
� d S ) Nr+ � � ar �� rf rN rS i �� )r r! r r" rC r% r# )r r( �iZzgfile�contentsZztxtr
r
r �test_many_append� s
zTestGzip.test_many_appendc C s~ t j| jd��}|jtd � W d Q R X t j| jd��}| j|j� td � W d Q R X | jt�� t j| jd� W d Q R X d S )N�xbr, rN ) r r! r r" r. r# r% rO �FileExistsError)r r( r
r
r �test_exclusive_write� s z TestGzip.test_exclusive_writec C sb | j � tj| jd��*}tj|��}dd� |D �}W d Q R X W d Q R X | j|dtjdd� � d S )NrN c S s g | ]}|�qS r
r
)�.0�liner
r
r �
<listcomp>� s z1TestGzip.test_buffered_reader.<locals>.<listcomp>r, T)�keepends) r4 r r! r r �BufferedReaderr# r. �
splitlines)r r( r �linesr
r
r �test_buffered_reader� s
"z TestGzip.test_buffered_readerc
C sh | j � tj| jd��H}d}x<|j|�}| r8|dkr8P | jt|�|k� |d d }q W W d Q R X d S )NrN r rY r, )r4 r r! r �readlinerc r$ )r r( Z line_length�Lr
r
r �
test_readline� s
zTestGzip.test_readlinec C sb | j � tj| jd��}|j� }W d Q R X tj| jd�� }x|jd�}|g kr<P q<W W d Q R X d S )NrN � )r4 r r! r � readlines)r r( r} r
r
r �test_readlines� s
zTestGzip.test_readlinesc C s� | j � tj| j��v}xn|j� }|j� }|s.P |j� }|j|� t|�dkrRd}nt|�}|j|�}| j |d |� |� |j|� qW W d Q R X d S )N�
)
r4 r r! r r r| r r$ r% r# )r r( ZoldposZline1ZnewposZamountZline2r
r
r �test_seek_read� s
zTestGzip.test_seek_readc C sZ | j � tj| j��(}|jd� |jddd� |jd�}W d Q R X | j|tdd� � d S )Nr� rY )�whence� � )r4 r r! r r% r r# r. )r r( �yr
r
r �test_seek_whence s
zTestGzip.test_seek_whencec C sH t j| jd��0}x(tddd�D ]}|j|� |jd� q W W d Q R X d S )Nr r r= � s GZ
)r r! r rC r r" )r r( �posr
r
r �test_seek_write s
zTestGzip.test_seek_writec C sl | j � tj| jd��}| j|jjd� W d Q R X tj| j� tj| jd��}| j|jjd� W d Q R X d S )Nr rN �xrq ) r4 r r! r r# Z myfileobjr' r r )r r( r
r
r � test_mode s zTestGzip.test_modec C sL xFdD ]>}t j| j|��&}| jt|d�� | j|j| j� W d Q R X qW d S )Nr+ rN r9 )r+ rN )r r! r rc r1 r# r9 )r r' r( r
r
r � test_1647484% s
zTestGzip.test_1647484c C sJ | j � tj| jd��*}| jt|jd�� | j|jj| j� W d Q R X d S )NrN r9 ) r4 r r! r rc r1 r\ r# r9 )r r( r
r
r �test_paddedfile_getattr+ s z TestGzip.test_paddedfile_getattrc C s� d}t j| jd|d��}|jt� W d Q R X t j| j��D}| jt|d�� | j|j� |j � }| j
|t� | j
|j|� W d Q R X d S )Ni�[r )�mtimer� ) r r! r r" r. rc r1 Z assertIsNoner� r% r# )r r� �fWrite�fReadZdataReadr
r
r �
test_mtime1 s zTestGzip.test_mtimec C sP d}t j| jd|d��}|jt� W d Q R X t| jd��� }|jd�}| j|d� |jd�}| j|d� |jd�}| j|d� |jd �}| j|tj d
|�� |jd�}| j|d � |jd�} | j| d � | jj
d
�d }
|jt |
��} | j| |
� |j t
j| j�jd � |jd �} | j| d� |jd �}
| j|
tj d
t t��� W d Q R X d S )Ni�[r )r� rN � s �rY � r@ z<i� � �zLatin-1� r? s ��d�)r r! r r" r. �openr% r# �structZpack�encoder$ r r2 �stat�st_size)r r� r� r� ZidBytesZcmByteZ flagsByteZ
mtimeBytesZxflByteZosByteZexpectedZ nameBytesZ
crc32BytesZ
isizeBytesr
r
r �
test_metadata< s0
zTestGzip.test_metadatac # C s� t j| jd��}|jd� W d Q R X t j| jd�}|j� y|� W d Q R X W n tk
rb Y n X | jd� y&t j| jd��}dd W d Q R X W n tk
r� Y n X | jd� d S )Nr+ s xxxrN z4__enter__ on a closed file didn't raise an exceptionrY r z 1/0 didn't raise an exception)r r! r r" r3 rd Zfail�ZeroDivisionError)r r( r
r
r �test_with_openi s
zTestGzip.test_with_openc C s� t j| jd��}|jtd � W d Q R X t| jd��}|jdd � W d Q R X t j| jd�� }|j� }| j|td d� W d Q R X d S )Nr+ r, rf r� rN zIncorrect data in file)r r! r r" r. r� r% r# )r r( r; r
r
r �test_zero_padded_file~ s z TestGzip.test_zero_padded_filec C sp t d }t� }tj|dd��}|j|� W d Q R X |j� }t|�}tj|dd��}| j|j� |� W d Q R X d S )Nr, r+ )r\ r' rN )r. r r r! r" �getvaluer# r% )r � uncompressed�bufr( r_ r
r
r �test_non_seekable_file� s z TestGzip.test_non_seekable_filec C s� t d }tj| jd��}|j|� W d Q R X dd� }tj| jd��x}d|_d}xD|� D ]:}|j|�}|dkrnP | j|jt |��|� |t |�7 }qVW | j|jd �d� | j|t |�� W d Q R X d S )
Nrl r+ c s s& x xt ddd�D ]
} | V qW qW d S )N� r, r� )rC )�nr
r
r �sizes� s z!TestGzip.test_peek.<locals>.sizesrN �! r rS �d )
r. r r! r r" Zmax_read_chunk�peekr# r% r$ )r r� r( r� rW r� �sr
r
r � test_peek� s
zTestGzip.test_peekc C sf t d jd�jdd�}| j� tj| jd��0}tj|dd��}| j |j
� |� W d Q R X W d Q R X d S )Nr, �asciiT)rw r )�encoding) r. �decodery r4 r r! r r �
TextIOWrapperr# r� )r rz r( �tr
r
r �test_textio_readlines� s
z TestGzip.test_textio_readlinesc C sL t j| jt jt jB �}t j|d�� }tj|dd��}W d Q R X W d Q R X d S )Nr+ r )r\ r' )r2 r� r �O_WRONLY�O_CREAT�fdopenr r! )r �fdr( �gr
r
r �test_fileobj_from_fdopen� s z!TestGzip.test_fileobj_from_fdopenc < C s� t j| jd�j� t| jd���}t j|dd��}| j|jt j� W d Q R X t j|dd��}| j|jt j� W d Q R X t j|dd��}| j|jt j� W d Q R X t j|dd��}| j|jt j� W d Q R X | j t
�� t j|dd� W d Q R X W d Q R X xNd
D ]F}t| j|��.}t j|d
��}| j|jt j� W d Q R X W d Q R X � q�W xddD ]\}d|k�rft j | j� t| j|��.}t j|d
��}| j|jt j� W d Q R X W d Q R X �qLW d S )Nr+ �r+br )r\ r' r r5 r� �zrN )r\ rf rq )rN r� )r+ rf rq )
r r! r r3 r� r# r' ZREADZWRITErO rd r r )r r( r� r' r
r
r �test_fileobj_mode� s, "
*
zTestGzip.test_fileobj_modec ! C s� | j }y|jd�}W n tk
r2 | jd� Y nX tj|d��}|jtd � W d Q R X tj|d��}| j|j � td � W d Q R X tj|d��}| j|j � td � W d Q R X d S )Nr� z%Temporary file name needs to be ASCIIr+ r, rN )
r r� �UnicodeEncodeErrorZskipTestr r! r" r. r# r% )r Z str_filenameZbytes_filenamer( r
r
r �test_bytes_filename� s z TestGzip.test_bytes_filenamec C st t jdtd� dd�}| jt|�tj� tj|�}t j|d�}| j |j
d�d� dtj }| j |j j
j� |d� dS ) z-Decompressed data buffering should be limitedr� g ��>Arj )r[ )r\ rY z)Excessive amount of data was decompressedN)r r] �intZ
assertLessr$ r �DEFAULT_BUFFER_SIZEr^ r! r# r% ZassertLessEqual�_buffer�rawr )r ZbombZdecompZ
max_decompr
r
r �test_decompress_limited� s
z TestGzip.test_decompress_limitedc C s| xvt tgD ]j}xdf dddgD ]T}tj|f|�� }| jt|�t� tjtj |�dd��}| j|j
� |� W d Q R X q W q
W d S ) NrY � rj rN )r\ r' )rY )r� )rj ) r. rh r r] r# �typer r! r r^ r% )r r& r �datacr( r
r
r �
test_compress� s zTestGzip.test_compressc C st xnt tfD ]b}tj� }tj|dd��}|j|� W d Q R X | jtj|j � �|� tj
|�}| jtj|�|� q
W d S )Nr+ )r\ r' ) r. rh r r^ r r! r" r# �
decompressr� r] )r r& r� r( r� r
r
r �test_decompress� s
zTestGzip.test_decompressc C s� t d }tj|�d d� }tjtj|�d��}| jt|j� W d Q R X tjtj|�d��,}| j |jt
|��|� | jt|jd� W d Q R X xHt dd�D ]:}tjtj|d |� �d��}| jt|jd� W d Q R X q�W d S )Nr, r? )r\ rY r� r� i����) r. r r] r! r r^ rO �EOFErrorr% r# r$ rC )r r& Z truncatedr( rn r
r
r �test_read_truncated s z TestGzip.test_read_truncatedc
C s6 d}t jtj|�d��}| j|j� d� W d Q R X d S )Ns ��cQ� Extra I-. 2�Mx )r\ s Test)r r! r r^ r# r% )r Zgzdatar( r
r
r �test_read_with_extra s z TestGzip.test_read_with_extrac C sP t j| jd��}|jt� W d Q R X t j| jd��}|jjjj� W d Q R X d S )Nr+ rN ) r r� r r" r. r� r� Z_fpZprepend)r r( r
r
r �test_prepend_error s zTestGzip.test_prepend_errorN)r ),r r r r* r4 r< rF rH rL rQ rR rX r r r` re ri rp rs r{ r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r
r
r
r r 8 sN
-
r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestOpenc ; C sl t d }tj| jd��}|j|� W d Q R X t| jd�� }tj|j� �}| j||� W d Q R X tj| jd��}| j|j� |� W d Q R X tj| jd��}|j|� W d Q R X t| jd��$}tj|j� �}| j||d � W d Q R X | jt �� tj| jd� W d Q R X t
j | j� tj| jd��}|j|� W d Q R X t| jd�� }tj|j� �}| j||� W d Q R X d S )Nr, r+ rN rf r� rq ) r. r r� r r" r� r% r# rO rr r r )r r� r( � file_datar
r
r �test_binary_modes s* zTestOpen.test_binary_modesc C s� t j| j�}tj|d��}|jtd � W d Q R X tj|d��}|jt� W d Q R X tj|��}| j|j� td � W d Q R X d S )Nr+ r, rf r6 ) r7 r8 r r r� r" r. r# r% )r r r( r
r
r �test_pathlike_file9 s zTestOpen.test_pathlike_filec ; C sl t d }tj| jd��}|j|� W d Q R X t| jd�� }tj|j� �}| j||� W d Q R X tj| jd��}| j|j� |� W d Q R X tj| jd��}|j|� W d Q R X t| jd��$}tj|j� �}| j||d � W d Q R X | jt �� tj| jd� W d Q R X t
j | j� tj| jd��}|j|� W d Q R X t| jd�� }tj|j� �}| j||� W d Q R X d S )Nr, r rN r r5 r� r� ) r. r r� r r" r� r% r# rO rr r r )r r� r( r� r
r
r �test_implicit_binary_modesB s* z#TestOpen.test_implicit_binary_modesc ' C s t jd�d }|jdtj�}tj| jd��}|j|� W d Q R X t| jd��&}tj |j
� �jd�}| j ||� W d Q R X tj| jd��}| j |j
� |� W d Q R X tj| jd��}|j|� W d Q R X t| jd��*}tj |j
� �jd�}| j ||d � W d Q R X d S ) Nr� r, �
�wtrN �rtZatr� ) r. r� �replacer2 �linesepr r� r r" r� r% r# )r r� �uncompressed_rawr( r� r
r
r �test_text_modes^ s zTestOpen.test_text_modesc C s� t d }|jd�}tj|�}tjtj|�d��}| j|j� |� W d Q R X tjtj|�d��}| j|j� |� W d Q R X tjtj|�d��}| j|j� |� W d Q R X d S )Nr, r� r rN r� ) r. r� r r] r� r r^ r# r% )r Zuncompressed_bytesZuncompressed_strr_ r( r
r
r � test_fileobjn s
zTestOpen.test_fileobjc / C s� | j t�� tjd� W d Q R X | j t�� tj| jd� W d Q R X | j t�� tj| jd� W d Q R X | j t�� tj| jddd� W d Q R X | j t�� tj| jddd� W d Q R X | j t�� tj| jdd d
� W d Q R X d S ) Ngw��/�^@ZwbtZxbtrN zutf-8)r� �ignore)�errorsr� )�newline)rO rP r r� rd r )r r
r
r �test_bad_paramsy s zTestOpen.test_bad_paramsc C s� t jd�d }|jdtj�}tj| jddd��}|j|� W d Q R X t| jd��&}tj |j
� �jd�}| j ||� W d Q R X tj| jddd��}| j |j
� |� W d Q R X d S ) Nr� r, r� r� zutf-16)r� rN r� ) r. r� r� r2 r� r r� r r" r� r% r# )r r� r� r( r� r
r
r �
test_encoding� s zTestOpen.test_encodingc C sX t j| jd��}|jd� W d Q R X t j| jdddd��}| j|j� d� W d Q R X d S )Nr+ s foo�barr� r� r� )r� r� Zfoobar)r r� r r" r# r% )r r( r
r
r �test_encoding_error_handler� s
z$TestOpen.test_encoding_error_handlerc C sj t jd�d }tj| jddd��}|j|� W d Q R X tj| jddd��}| j|j� |g� W d Q R X d S )Nr� r, r� r� )r� r� �
)r. r� r r� r r" r# r� )r r� r( r
r
r � test_newline� s
zTestOpen.test_newlineN) r r r r� r� r� r� r� r� r� r� r� r
r
r
r r� s r� c s � fdd�}|S )Nc s t j� ��� fdd��}|S )Nc s& t j� � z
�| |�S tj� � X d S )N)r2 �makedirsr Zrmtree)r �kwargs)� directory�functionr
r �wrapper� s
z?create_and_remove_directory.<locals>.decorator.<locals>.wrapper)� functools�wraps)r� r� )r� )r� r � decorator� s z.create_and_remove_directory.<locals>.decoratorr
)r� r� r
)r� r �create_and_remove_directory� s r� c @ sP e Zd ZdZdd� Zee�dd� �Zdd� Zee�dd � �Z ee�d
d � �Z
d S )
�TestCommandLines This is a simple test with gzipc C s� t j� �h}tj|dd��}|j| j� W d Q R X tjdddf}t|t t t d��}|j
|j � �\}}W d Q R X W d Q R X | j |d� | j || j� d S )Nr+ )r\ r' z-mr z-d)�stdin�stdout�stderrrS )
r r^ r r! r" r& �sys�
executabler r � communicater� r# )r Zbytes_ioZ gzip_filer �proc�out�errr
r
r � test_decompress_stdin_stdout� s
& z,TestCommandLine.test_decompress_stdin_stdoutc C s� t jjtd�}| jt jj|�� tj|dd��}|j| j � W d Q R X t
ddd|�\}}}tt jjtd�d��}| j |j � | j � W d Q R X | j
t jj|�� | j |d � | j |d
� | j |d
� d S ) Nz testgzip.gzr+ )r' z-mr z-d�testgziprN r rS )r2 �pathrV �TEMPDIRra �existsr r� r" r& r r# r% rc )r �gzipname�fp�rcr� r� Zgunzipedr
r
r � test_decompress_infile_outfile� s z.TestCommandLine.test_decompress_infile_outfilec C s< t dddd�\}}}| jd|� | j|d� | j|d� d S )Nz-mr z-dzthisisatest.outs filename doesn't end in .gz:r rS )r ZassertInr# )r r� r� r� r
r
r �$test_decompress_infile_outfile_error� s z4TestCommandLine.test_decompress_infile_outfile_errorc C s\ t jddf}t|tttd��}|j| j�\}}W d Q R X | j|d� | j|d d� d� d S )Nz-mr )r� r� r� rS r� s �)r� r� r r r� r& r# )r r r� r� r� r
r
r �test_compress_stdin_outfile� s
z+TestCommandLine.test_compress_stdin_outfilec C s� t jjtd�}|d }| jt jj|�� t|d��}|j| j� W d Q R X t dd|�\}}}| j
t jj|�� | j |d� | j |d� | j |d� d S )Nr� z.gzr+ z-mr r rS ) r2 r� rV r� ra r� r� r" r& r rc r# )r Zlocal_testgzipr� r� r� r� r� r
r
r � test_compress_infile_outfile� s z,TestCommandLine.test_compress_infile_outfileN) r r r r& r� r� r� r� r� r r r
r
r
r r� � s r� c C s t jttt� d S )N)r Z run_unittestr r� r� )�verboser
r
r � test_main� s r �__main__T)r )N)%�__doc__rK r� r r2 r7 r� r� Zunittest�
subprocessr r Ztestr Z test.supportr r Ztest.support.script_helperr �
import_moduler r. rh r� �abspathr r� r^ r ZTestCaser r r� r� r� r r r
r
r
r �<module> s:
h
@