Your IP : 216.73.216.213
��YfY � @ s
d Z d d l Z d d l m Z d d l m Z m Z d d l Z d d l Z d d l Z d d l
Z
e j d � Z d Z
d Z Gd d � d e j � Z Gd
d � d e j � Z Gd d
� d
e � Z Gd d � d e � Z d d d � Z e d k r e d d � d S)z!Test script for the gzip module.
� N)�support)�
bigmemtest�_4G�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 */
c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� UnseekableIOc C s d S)NF� )�selfr r �3/opt/alt/python35/lib64/python3.5/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 @ s1 e Z d Z e j Z d d � Z d d � Z d S)�BaseTestc C s t j | j � d S)N)r �unlink�filename)r r r r �setUp( s zBaseTest.setUpc C s t j | j � d S)N)r r r )r r r r �tearDown+ s zBaseTest.tearDownN)r r r r ZTESTFNr r r r r r r r % s r c @ s� e Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
e d e d d � d d � � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d; d< � Z d= d> � Z! d? d@ � Z" dA dB � Z# dC dD � Z$ dE dF � Z% dG dH � Z& dI dJ � Z' dK dL � Z( dM dN � Z) dO S)P�TestGzip�bc C s� t | � } t j | j d | � � } | j | � } Wd QRX| j | t | � � t j | j d | � � } | j | j � | � Wd QRXd S)N�w�r)�bytesr �GzipFiler �write� assertEqual�len�read)r �data�modeZb_data�f�lr r r �write_and_read_back0 s z TestGzip.write_and_read_backc
C s~ t j | j d � �X } | j t d � | j � | j � t t d � r_ t j | j � � | j
� Wd QRX| j
� d S)N�wb�2 �fsync) r r r r �data1�flush�fileno�hasattr�osr) �close)r r$ r r r �
test_write8 s
zTestGzip.test_writec C s^ | j t t d � � t t t d � � � } | j d d d d d g �} | j | � d S)Nr( � �B�shape� � )r& �
memoryviewr* r �range�cast)r �mr" r r r �test_write_memoryviewI s z TestGzip.test_write_memoryviewc C s | j t t d � � d S)Nr( )r& � bytearrayr* )r r r r �test_write_bytearrayO s z TestGzip.test_write_bytearrayc C s! | j t j d t d � � d S)N�I�( )r& �arrayr* )r r r r �test_write_arrayR s zTestGzip.test_write_arrayc C s� t j | j d � �\ } | j t � � | j d � Wd QRX| j t � � | j g � Wd QRX| j t � Wd QRXt j | j d � � } | j | j � t � Wd QRXd S)Nr' � �rb) r r r � assertRaises� TypeErrorr r* r r! )r r$ r r r � test_write_incompatible_typeU s z%TestGzip.test_write_incompatible_typec C sM | j � t j | j d � � } | j � } Wd QRX| j | t d � d S)Nr r( )r0 r r r r! r r* )r r$ �dr r r � test_readb s
zTestGzip.test_readc C s� | j � g } d } t j | j d � �T } xJ | j � } | sD P| j | � | t | � 7} | j | j � | � q1 WWd QRX| j d j | � t
d � d S)Nr r � r( ) r0 r r r �read1�appendr r r
�joinr* )r Zblocks�nreadr$ rF r r r �
test_read1i s
!zTestGzip.test_read1�sizeZmemuse� c C sS t j t d d �} t j d t j | � d d � } | j | j | � t � d S)N�
compresslevelrO �fileobjr# rB )r �compressr* r r �BytesIOr r! )r rN �
compressedr$ r r r �test_read_largex s !zTestGzip.test_read_largec % C sL | j � t j | j d � } | j } | j | j � | j � | j | j � | j t
� � | j d � Wd QRX| j t
� � | j d � Wd QRX| j t
� � | j
� Wd QRXt j | j d � } | j } | j | j � | j � | j | j � | j t
� � | j d � Wd QRX| j t
� � | j � Wd QRXd S)Nr rO r r rH )r0 r r r rQ Z assertFalse�closedr/ �
assertTruerC �
ValueErrorr! r r
r r+ )r r$ rQ r r r �test_io_on_closed_object� s*
z!TestGzip.test_io_on_closed_objectc C s� | j � t j | j d � � } | j t d � Wd QRXt j | j d � � } | j � } Wd QRX| j | t d t d � d S)N�ab� rB r( ) r0 r r r r �data2r! r r* )r r$ rF r r r � test_append� s
zTestGzip.test_appendc C s� t j | j d d � � } | j d � Wd QRXxF t d d � D]5 } t j | j d d � � } | j d � Wd QRXq? Wt j | j d � �4 } d } x$ | j d � } | | 7} | s� Pq� WWd QRX| j | d d
� d S) Nr' � � ar �� rZ rB rH i �� )r r r r r7 r! r )r r$ �iZzgfile�contentsZztxtr r r �test_many_append� s
zTestGzip.test_many_appendc C s� t j | j d � � } | j t d � Wd QRXt j | j d � �! } | j | j � t d � Wd QRX| j t � � t j | j d � Wd QRXd S)N�xbr( rB ) r r r r r* r r! rC �FileExistsError)r r$ r r r �test_exclusive_write� s !z TestGzip.test_exclusive_writec C sy | j � t j | j d � �3 } t j | � � } d d � | D� } Wd QRXWd QRX| j | d t j d d � � d S)NrB c S s g | ] } | � q Sr r )�.0�liner r r �
<listcomp>� s z1TestGzip.test_buffered_reader.<locals>.<listcomp>r( �keependsT) r0 r r r r �BufferedReaderr r* �
splitlines)r r$ r �linesr r r �test_buffered_reader� s
!z TestGzip.test_buffered_readerc
C s� | j � t j | j d � �^ } d } xN | j | � } | rN | d k rN P| j t | � | k � | d d } q+ WWd QRXd S)NrB r rO r( )r0 r r r �readlinerW r )r r$ Z line_length�Lr r r �
test_readline� s
zTestGzip.test_readlinec C s{ | j � t j | j d � � } | j � } Wd QRXt j | j d � �* } x | j d � } | g k rP PqP WWd QRXd S)NrB � )r0 r r r � readlines)r r$ rq r r r �test_readlines� s
zTestGzip.test_readlinesc C s� | j � t j | j � �� } x� | j � } | j � } | sA P| j � } | j | � t | � d k ru d } n t | � } | j | � } | j | d | � | � | j | � q" WWd QRXd S)N�
)
r0 r r r r
rp r r r! r )r r$ ZoldposZline1ZnewposZamountZline2r r r �test_seek_read� s
zTestGzip.test_seek_readc C ss | j � t j | j � �6 } | j d � | j d d d �| j d � } Wd QRX| j | t d d � � d S)Nrv �whencerO � � )r0 r r r r! r r r* )r r$ �yr r r �test_seek_whence� s
zTestGzip.test_seek_whencec C sZ t j | j d � �> } x4 t d d d � D] } | j | � | j d � q+ WWd QRXd S)Nr r r1 � s GZ
)r r r r7 r r )r r$ �posr r r �test_seek_write s
zTestGzip.test_seek_writec C s� | j � t j | j d � � } | j | j j d � Wd QRXt j | j � t j | j d � � } | j | j j d � Wd QRXd S)Nr rB �xre ) r0 r r r r Z myfileobjr# r r )r r$ r r r � test_mode s
zTestGzip.test_modec C s` xY d D]Q } t j | j | � �3 } | j t | d � � | j | j | j � Wd QRXq Wd S)Nr' rB �name)r' rB )r r r rW r- r r� )r r# r$ r r r � test_1647484 s
zTestGzip.test_1647484c C s_ | j � t j | j d � �9 } | j t | j d � � | j | j j | j � Wd QRXd S)NrB r� ) r0 r r r rW r- rQ r r� )r r$ r r r �test_paddedfile_getattr s
z TestGzip.test_paddedfile_getattrc C s� d } t j | j d d | �� } | j t � Wd QRXt j | j � �\ } | j t | d � � | j | j � | j � } | j
| t � | j
| j | � Wd QRXd S)Ni�[r �mtime) r r r r r* rW r- Z assertIsNoner� r! r )r r� �fWrite�fReadZdataReadr r r �
test_mtime s zTestGzip.test_mtimec C s� d } t j | j d d | �� } | j t � Wd QRXt | j d � �x} | j d � } | j | d � | j d � } | j | d � | j d � } | j | d � | j d � } | j | t j d
| � � | j d � } | j | d � | j d � } | j | d � | j j
d
� d }
| j t |
� � } | j | |
� | j t
j | j � j d � | j d � } | j | d � | j d � }
| j |
t j d
t t � � � Wd QRXd S)Ni�[r r� rB � s �rO s r5 z<is s �zLatin-1s r4 s ��d�)r r r r r* �openr! r �structZpack�encoder r r. �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 | j d � � } | j d � Wd QRXt j | j d � } | j � y | � Wd QRXWn t k
rq Yn X| j d � y+ t j | j d � � } d d Wd QRXWn t k
r� Yn X| j d � d S)Nr' s xxxrB z4__enter__ on a closed file didn't raise an exceptionrO r z 1/0 didn't raise an exception)r r r r r/ rX Zfail�ZeroDivisionError)r r$ r r r �test_with_openT s
zTestGzip.test_with_openc C s� t j | j d � � } | j t d � Wd QRXt | j d � � } | j d d � Wd QRXt j | j d � �* } | j � } | j | t d d � Wd QRXd S)Nr' r( rZ s rB zIncorrect data in file)r r r r r* r� r! r )r r$ rF r r r �test_zero_padded_filei s z TestGzip.test_zero_padded_filec C s� t d } t � } t j d | d d � � } | j | � Wd QRX| j � } t | � } t j d | d d � � } | j | j � | � Wd QRXd S)Nr( rQ r# r' rB )r* r r r r �getvaluer r! )r � uncompressed�bufr$ rT r r r �test_non_seekable_fileu s
z TestGzip.test_non_seekable_filec C s t d } t j | j d � � } | j | � Wd QRXd d � } t j | j d � �� } d | _ d } x\ | � D]Q } | j | � } | d k r� P| j | j t | � � | � | t | � 7} qs W| j | j d � d � | j | t | � � Wd QRXd S)
Nr` r' c s s- x& x t d d d � D] } | Vq Wq Wd S)N� r( rv )r7 )�nr r r �sizes� s z!TestGzip.test_peek.<locals>.sizesrB �! r rH �d )
r* r r r r Zmax_read_chunk�peekr r! r )r r� r$ r� rL r� �sr r r � test_peek s
zTestGzip.test_peekc C s� t d j d � j d d � } | j � t j | j d � �< } t j | d d �� } | j | j
� | � Wd QRXWd QRXd S)Nr( �asciirk Tr �encoding) r* �decoderm r0 r r r r �
TextIOWrapperr rt )r rn r$ �tr r r �test_textio_readlines� s
z TestGzip.test_textio_readlinesc C sa t j | j t j t j B� } t j | d � �) } t j d | d d � � } Wd QRXWd QRXd S)Nr' rQ r# r )r. 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� | j } y | j d � } Wn t k
r= | j d � Yn Xt j | d � � } | j t d � Wd QRXt j | d � �! } | j | j � t d � Wd QRXt j | d � �! } | j | j � t d � Wd QRXd S)Nr� z%Temporary file name needs to be ASCIIr' r( rB )
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 s� t j t t d � � d d �} | j t | � t j � t j | � } t j d | � } | j
t d � | j d � � d t j } | j | j
j j � | d � d S)z-Decompressed data buffering should be limitedg ��>ArP r^ rQ rO z)Excessive amount of data was decompressedN)r rR r �intZ
assertLessr r �DEFAULT_BUFFER_SIZErS r r r! ZassertLessEqual�_buffer�rawr
)r ZbombZdecompZ
max_decompr r r �test_decompress_limited� s !
z TestGzip.test_decompress_limitedc C s� x� t t g D]� } x� f d d d g D]o } t j | | � } | j t | � t � t j d t j | � d d � � } | j | j
� | � Wd QRXq& Wq
Wd S)
NrO � r^ rQ r# rB )rO )r� )r^ ) r* r\ r rR r �typer r r rS r! )r r" r �datacr$ r r r �
test_compress� s $zTestGzip.test_compressc C s� x� t t f D]� } t j � } t j d | d d � � } | j | � Wd QRX| j t j | j � � | � t j
| � } | j t j | � | � q
Wd S)NrQ r# r' ) r* r\ r rS r r r r �
decompressr� rR )r r" r� r$ r� r r r �test_decompress� s zTestGzip.test_decompressc C s t d } t j | � d d � } t j d t j | � � � } | j t | j � Wd QRXt j d t j | � � �< } | j | j t
| � � | � | j t | j d � Wd QRXx\ t d d � D]K } t j d t j | d | � � � � } | j t | j d � Wd QRXq� Wd S)Nr( r4 rQ rO r� rv i����) r* r rR r r rS rC �EOFErrorr! r r r7 )r r" Z truncatedr$ rb r r r �test_read_truncated� s
(z TestGzip.test_read_truncatedc
C sE d } t j d t j | � � � } | j | j � d � Wd QRXd S)Ns ��cQ� Extra I-. 2�Mx rQ s Test)r r r rS r r! )r Zgzdatar$ r r r �test_read_with_extra� s z TestGzip.test_read_with_extrac C sb t j | j d � � } | j t � Wd QRXt j | j d � � } | j j j j � Wd QRXd S)Nr' rB ) 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& r0 r: r< r@ rE rG rM r r rU rY r] rd rg ro rr ru rw 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 / sJ
!
-
r c @ sp e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d S)�TestOpenc ; C s� t d } t j | j d � � } | j | � Wd QRXt | j d � �, } t j | j � � } | j | | � Wd QRXt j | j d � � } | j | j � | � Wd QRXt j | j d � � } | j | � Wd QRXt | j d � �0 } t j | j � � } | j | | d � Wd QRX| j t � � t j | j d � Wd QRXt
j | j � t j | j d � � } | j | � Wd QRXt | j d � �, } t j | j � � } | j | | � Wd QRXd S)Nr( r' rB rZ r� re ) r* r r� r r r� r! r rC rf r r )r r� r$ � file_datar r r �test_binary_modes� s*
zTestOpen.test_binary_modesc ; C s� t d } t j | j d � � } | j | � Wd QRXt | j d � �, } t j | j � � } | j | | � Wd QRXt j | j d � � } | j | j � | � Wd QRXt j | j d � � } | j | � Wd QRXt | j d � �0 } t j | j � � } | j | | d � Wd QRX| j t � � t j | j d � Wd QRXt
j | j � t j | j d � � } | j | � Wd QRXt | j d � �, } t j | j � � } | j | | � Wd QRXd S)Nr( r rB r �ar� r� ) r* r r� r r r� r! r rC rf r r )r r� r$ r� r r r �test_implicit_binary_modes s*
z#TestOpen.test_implicit_binary_modesc ' C sQ t j d � d } | j d t j � } t j | j d � � } | j | � Wd QRXt | j d � �5 } t j | j
� � j d � } | j | | � Wd QRXt j | j d � � } | j | j
� | � Wd QRXt j | j d � � } | j | � Wd QRXt | j d � �9 } t j | j
� � j d � } | j | | d � Wd QRXd S) Nr� r( �
�wtrB �rtZatr� ) r* r� �replacer. �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 } | j d � } t j | � } t j t j | � d � � } | j | j � | � Wd QRXt j t j | � d � � } | j | j � | � Wd QRXt j t j | � d � � } | j | j � | � Wd QRXd S)Nr( r� r rB r� ) r* r� r rR r� r rS r r! )r Zuncompressed_bytesZuncompressed_strrT r$ r r r � test_fileobj8 s
zTestOpen.test_fileobjc / C s | j t � � t j d � Wd QRX| j t � � t j | j d � Wd QRX| j t � � t j | j d � Wd QRX| j t � � t j | j d d d �Wd QRX| j t � � t j | j d d d �Wd QRX| j t � � t j | j d d d
�Wd QRXd S) Ngw��/�^@ZwbtZxbtrB r� zutf-8�errors�ignore�newliner� )rC rD r r� rX r )r r r r �test_bad_paramsC s zTestOpen.test_bad_paramsc C s� t j d � d } | j d t j � } t j | j d d d �� } | j | � Wd QRXt | j d � �5 } t j | j
� � j d � } | j | | � Wd QRXt j | j d d d �� } | j | j
� | � Wd QRXd S) Nr� r( r� r� r� zutf-16rB r� ) r* r� r� r. r� r r� r r r� r! r )r r� r� r$ r� r r r �
test_encodingR s zTestOpen.test_encodingc C sq t j | j d � � } | j d � Wd QRXt j | j d d d d d �� } | j | j � d � Wd QRXd 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 s� t j d � d } t j | j d d d �� } | j | � Wd QRXt j | j d d d �� } | j | j � | g � Wd QRXd S)Nr� r( r� r� r� r� �
)r* r� r r� r r r rt )r r� r$ r r r � test_newlinef s
zTestOpen.test_newlineN) r r r r� r� r� r� r� r� r� r� r r r r r� � s r� c C s t j t t � d S)N)r Z run_unittestr r� )�verboser r r � test_mainn s r� �__main__r� T)�__doc__ZunittestZtestr Z test.supportr r r. r r� r? �
import_moduler r* r\ rS r ZTestCaser r r� r� r r r r r �<module> s$
� �~