Your IP : 216.73.216.213
�
f f�0 � @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
m Z d d l m
Z
m Z m Z d d l m Z m Z m Z m Z m Z m Z m Z m Z m Z e d Z e d Z d Z d Z d d
f d d f d
d f d d f g Z d d � Z d d � Z d d � Z Gd d � d � Z! Gd d � d e! e j" � Z# e Gd d � d e! e j" � � Z$ e Gd d � d e! e j" � � Z% e Gd d � d e! e j" � � Z& Gd! d" � d" � Z' Gd# d$ � d$ e' e j" � Z( e Gd% d& � d& e' e j" � � Z) e Gd' d( � d( e' e j" � � Z* e Gd) d* � d* e' e j" � � Z+ Gd+ d, � d, e j" � Z, Gd- d. � d. e j" � Z- Gd/ d0 � d0 e j" � Z. Gd1 d2 � d2 � Z/ Gd3 d4 � d4 e/ e j" � Z0 e Gd5 d6 � d6 e/ e j" � � Z1 e Gd7 d8 � d8 e/ e j" � � Z2 e Gd9 d: � d: e/ e j" � � Z3 Gd; d< � d< e j" � Z4 Gd= d> � d> � Z5 Gd? d@ � d@ e5 e j" � Z6 e GdA dB � dB e5 e j" � � Z7 e GdC dD � dD e5 e j" � � Z8 e GdE dF � dF e5 e j" � � Z9 e GdG dH � dH e j" � � Z: GdI dJ � dJ e j" � Z; GdK dL � dL � Z< GdM dN � dN e< e j" � Z= e GdO dP � dP e< e j" � � Z> e GdQ dR � dR e< e j" � � Z? e GdS dT � dT e< e j" � � Z@ eA dU k r�e jB � n d S)V� N)�
TemporaryFile)�randint�random� getrandbits) �TESTFN�findfile�unlink�rmtree�
requires_zlib� requires_bz2�
requires_lzma�captured_stdout�check_warnings�2�di� Zzipfile_datafilesZ _ziptest1Z
1q2w3e4r5tzziptest2dir/_ziptest2Z
qawsedrftgz!ziptest2dir/ziptest3dir/_ziptest3Z
azsxdcfvgbz-ziptest2dir/ziptest3dir/ziptest4dir/_ziptest3Z
6y7u8i9o0pc C s t d | � j | d � S)N� �little)r �to_bytes)�size� r �6/opt/alt/python34/lib64/python3.4/test/test_zipfile.py� getrandbytes s r c c sZ t Vt � � } | V| j | j � Wd QXt j � � } | V| j | j � Wd QXd S)N)�TESTFN2r � assertFalse�closed�io�BytesIO)�test�fr r r � get_files s r c
C s- t d t f � � | j | d � SWd QXd S)N� �rU)r �DeprecationWarning�open)�zipfp�fnr r r �openU) s r& c @ sN e Z d Z e 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 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 S)6�AbstractTestsWithSourceFilec C s5 d d � t t � D� | _ d j | j � | _ d S)Nc S s, g | ]" } t d | t � f d � � q S)z'Zipfile test line %d. random float: %f
�ascii)�bytesr )�.0�ir r r �
<listcomp>0 s z:AbstractTestsWithSourceFile.setUpClass.<locals>.<listcomp>� )�range�FIXEDTEST_SIZE�line_gen�join�data)�clsr r r �
setUpClass. s z&AbstractTestsWithSourceFile.setUpClassc C s, t t d � � } | j | j � Wd QXd S)N�wb)r# r �writer2 )�self�fpr r r �setUp5 s z!AbstractTestsWithSourceFile.setUpc
C sU t j | d | � �: } | j t d � | j t t � | j d | j � Wd QXd S)N�wz another.name�strfile)�zipfile�ZipFiler6 r �writestrr2 )r7 r � compressionr$ r r r �make_test_archive: s z-AbstractTestsWithSourceFile.make_test_archivec C s� | j | | � t j | d | � ��} | j | j t � | j � | j | j d � | j � | j | j d � | j � t j � } | j d | � | j
� } | j � } | j t | � d � | j
d | d � | j
d | d � | j
d | d � | d
j � \ } } } }
| j | d � | j t j | d � � | j t j | d � � | j |
t t | j � � � | j � } | j t | � d
� | j
t | � | j
d | � | j
d | � | j � } d d � | D� } | j t | � d
� | j
t | � | j
d | � | j
d | � x* | D]" }
| j |
j t | j � � q=WxU t d d f D]D } | j | � } | j | j | � | j | j t | j � � qsW| j � Wd QXd S)N�rz another.namer; �file� z File Namer �Modified�Size� z%Y-%m-%dz%H:%M:%S� c S s g | ] } | j � q Sr )�filename)r* r+ r r r r, d s z8AbstractTestsWithSourceFile.zip_test.<locals>.<listcomp>)r@ r< r= � assertEqual�readr r2 r �StringIO�printdir�getvalue�
splitlines�len�assertIn�split�
assertTrue�time�strptime�str�namelist�infolist� file_size�getinforH �testzip)r7 r r? r$ r8 � directory�linesr% �date�time_r �names�infosr+ �nm�infor r r �zip_testA sH
z$AbstractTestsWithSourceFile.zip_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r rc r? )r7 r r r r �
test_basicu s z&AbstractTestsWithSourceFile.test_basicc C s | j | | � t j | d | � �� } g } | j t � �4 } x* | j d � } | s\ Pn | j | � qC WWd QXg } | j d � �4 } x* | j d � } | s� Pn | j | � q� WWd QX| j d j | � | j � | j d j | � | j � Wd QXd S)NrA � z another.namer- )
r@ r< r= r# r rJ �appendrI r1 r2 ) r7 r r? r$ �zipdata1�zipopen1� read_data�zipdata2�zipopen2r r r �
zip_open_testy s$ z)AbstractTestsWithSourceFile.zip_open_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r rl r? )r7 r r r r � test_open� s z%AbstractTestsWithSourceFile.test_openc C s� | j | | � t j | d | � �w } g } | j t � �= } x3 | j t d d � � } | se Pn | j | � qC WWd QX| j d j | � | j
� Wd QXd S)NrA rF i r- ) r@ r< r= r# r rJ r rf rI r1 r2 )r7 r r? r$ rg rh ri r r r �zip_random_open_test� s z0AbstractTestsWithSourceFile.zip_random_open_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r rn r? )r7 r r r r �test_random_open� s z,AbstractTestsWithSourceFile.test_random_openc C s� | j | | � t j | d � �R } | j t � �: } g } x* | j d � } | sY Pn | j | � q@ WWd QXWd QX| j d j | � | j � d S)NrA rF r- ���)
r@ r< r= r# r �read1rf rI r1 r2 )r7 r r? r$ �zipopen�zipdatari r r r �zip_read1_test� s z*AbstractTestsWithSourceFile.zip_read1_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r rt r? )r7 r r r r �
test_read1� s z&AbstractTestsWithSourceFile.test_read1c C s� | j | | � t j | d � �h } | j t � �P } g } x@ | j d � } | j t | � d � | so Pn | j | � q@ WWd QXWd QX| j d j
| � | j � d S)NrA �
r- ) r@ r< r= r# r rq ZassertLessEqualrO rf rI r1 r2 )r7 r r? r$ rr rs ri r r r �zip_read1_10_test� s z-AbstractTestsWithSourceFile.zip_read1_10_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r rw r? )r7 r r r r �
test_read1_10� s z)AbstractTestsWithSourceFile.test_read1_10c C s� | j | | � t j | d � �o } | j t � �W } d } xG | j � } | sV Pn | | 7} | j d � } | sy Pn | | 7} q@ WWd QXWd QX| j | | j � d S)NrA r- �d ) r@ r< r= r# r �readlinerJ rI r2 )r7 r r? r$ rr r2 rJ r r r �zip_readline_read_test� s
z2AbstractTestsWithSourceFile.zip_readline_read_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r r{ r? )r7 r r r r �test_readline_read� s z.AbstractTestsWithSourceFile.test_readline_readc C sw | j | | � t j | d � �O } | j t � �7 } x- | j D]" } | j � } | j | | � qA WWd QXWd QXd S)NrA )r@ r< r= r# r r0 rz rI )r7 r r? r$ rr �line�linedatar r r �zip_readline_test� s z-AbstractTestsWithSourceFile.zip_readline_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r r r? )r7 r r r r �
test_readline� s z)AbstractTestsWithSourceFile.test_readlinec C s� | j | | � t j | d � �^ } | j t � � } | j � } Wd QXx0 t | j | � D] \ } } | j | | � q\ WWd QXd S)NrA ) r@ r< r= r# r � readlines�zipr0 rI )r7 r r? r$ rr �ziplinesr} �zipliner r r �zip_readlines_test� s z.AbstractTestsWithSourceFile.zip_readlines_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r r� r? )r7 r r r r �test_readlines s z*AbstractTestsWithSourceFile.test_readlinesc C sz | j | | � t j | d � �R } | j t � �: } x0 t | j | � D] \ } } | j | | � qJ WWd QXWd QXd S)NrA )r@ r< r= r# r r� r0 rI )r7 r r? r$ rr r} r� r r r �zip_iterlines_test s
z.AbstractTestsWithSourceFile.zip_iterlines_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r r� r? )r7 r r r r �test_iterlines
s z*AbstractTestsWithSourceFile.test_iterlinesc C s� t j t d | j � � } | j d d � Wd QXt j t d | j � �Q } | j d � �9 } | j | j d � d � | j | j d � d � Wd QXWd QXd S) z>Check for cases where compressed data is larger than original.r: r; Z12NrA rF � 1� 2)r< r= r r? r> r# rI rJ )r7 r$ Zopenobjr r r �test_low_compression s z0AbstractTestsWithSourceFile.test_low_compressionc C sT t j t d � } | j d d d | j �| j d � } | j | j | j � d S)Nr: zb.txtz hello world�
compress_type)r< r= r r> r? rY rI r� )r7 r$ rb r r r �test_writestr_compression s z5AbstractTestsWithSourceFile.test_writestr_compressionc C s� x� d D]� } | d } t | � } t j t j � d | j � �W } | j d | � | j d d � �, } | j | � } | j t
| � | � Wd QXWd QXq Wd S)
NrF � � � � @ r: �foorA )rF r� r� r� r� ) r r< r= r r r? r> r# rJ rI rO )r7 Z test_sizerX Zjunk�zipfr8 Zbufr r r �test_read_return_size# s
!z1AbstractTestsWithSourceFile.test_read_return_sizec A C s� t j � } t j | d d ��/ } | j d | j d | j �| j � } Wd QX| j � } t j | � } t j | � �P } | j d � �8 } | j
| d � | j t � � | j
� Wd QXWd QXWd QXt j | � } t j | � �\ } | j d � �D } | j
| d � | j t � � x | j
d � r7q%WWd QXWd QXWd QXt j | � } t j | � �\ } | j d � �D } | j
| d � | j t � � x | j d � r�q�WWd QXWd QXWd QXd S)N�moder: r; r� � ry )r r r< r= r> r2 r? �tellrM r# �truncate� assertRaises�EOFErrorrJ rq )r7 r8 r� Z
end_offset� zipfiledatarr r r r �test_truncated_zipfile/ s2 z2AbstractTestsWithSourceFile.test_truncated_zipfilec C s t t � t t � d S)N)r r r )r7 r r r �tearDownM s
z$AbstractTestsWithSourceFile.tearDownN) �__name__�
__module__� __qualname__� classmethodr4 r9 r@ rc rd rl rm rn ro rt ru rw rx r{ r| r r� r� r� r� r� r� r� r� r� r� r r r r r' - s4 4
r' c @ s� e Z d Z e j 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 d � Z d d � Z d S)�StoredTestsWithSourceFileNc C sQ | j | | � t j | d � �) } | j d � } | j | j d � Wd QXd S)NrA r; i� � i �)r@ r< r= rY rI �
external_attr)r7 r r? r$ �zinfor r r � zip_test_writestr_permissionsW s z7StoredTestsWithSourceFile.zip_test_writestr_permissionsc C s. x' t | � D] } | j | t j � q
Wd S)N)r r� r< �
ZIP_STORED)r7 r r r r �test_writestr_permissions` s z3StoredTestsWithSourceFile.test_writestr_permissionsc C so t j t d t j � � } | j t d � Wd QXt j t d t j � � } | j | j � d g � Wd QXd S)Nr: z /absoluterA �absolute)r< r= r r� r6 r rI rV )r7 r$ r r r �test_absolute_arcnamesd s z0StoredTestsWithSourceFile.test_absolute_arcnamesc C s� t j t d t j � � } | j t t � Wd QXt j t d t j � �6 } | j d | j � | j | j � t d g � Wd QXd S)z&Test appending to an existing zipfile.r: N�ar; )
r< r= r r� r6 r r> r2 rI rV )r7 r$ r r r �test_append_to_zip_filek s
z1StoredTestsWithSourceFile.test_append_to_zip_filec C s� d d } t t d � � } | j | � Wd QXt j t d t j � � } | j t t � Wd QXt t d � �N } | j t | � � t j | d � � } | j | j
� t g � Wd QXWd QXd S)z9Test appending to an existing file that is not a zipfile.s I am not a ZipFile!rv r5 Nr� �rbrA ) r# r r6 r< r= r� r �seekrO rI rV )r7 r2 r r$ r r r �test_append_to_non_zip_filet s
z5StoredTestsWithSourceFile.test_append_to_non_zip_filec C s� t j t d t j � � } | j t t � Wd QXt t d � � } | j d � Wd QXt j t d � � } | j | t j � Wd QXd S)Nr: r� z
rA )r< r= r r� r6 r r# �assertIsInstance)r7 r$ r r r r �test_ignores_newline_at_end� s z5StoredTestsWithSourceFile.test_ignores_newline_at_endc C s� t j t d t j � � } d | _ | j t t � Wd QXt t d � � } | j d � Wd QXt j t d � �- } | j | t j � | j | j d � Wd QXd S)Nr: s this is a commentr� zabcdef
rA )
r< r= r r� �commentr6 r r# r� rI )r7 r$ r r r r �)test_ignores_stuff_appended_past_comments� s zCStoredTestsWithSourceFile.test_ignores_stuff_appended_past_commentsc C sc t j t d � �K } | j t � t t d � �& } | j | j t � | j � � Wd QXWd QXd S)z`Check that calling ZipFile.write without arcname specified
produces the expected result.r: r� N)r< r= r r6 r r# rI rJ )r7 r$ r r r r �test_write_default_name� s
z1StoredTestsWithSourceFile.test_write_default_namec C sf t j t d d �� } | j d d � Wd QXt j t d d �� } | j t | j t � Wd QXd S)z]Check that trying to call write() on a readonly ZipFile object
raises a RuntimeError.r� r: z somefile.txt�bogusNrA )r< r= r r> r� � RuntimeErrorr6 r )r7 r$ r r r �test_write_to_readonly� s z0StoredTestsWithSourceFile.test_write_to_readonlyc C sE t j t d � t j t d � � } | j t | j t � Wd QXd S)Nr r: )r r ) �os�utimer r< r= r r� �
ValueErrorr6 )r7 r$ r r r �test_add_file_before_1980� s z3StoredTestsWithSourceFile.test_add_file_before_1980)r� r� r� r< r� r? r� r� r� r� r� r� r� r� r� r� r� r r r r r� R s
r� c @ s% e Z d Z e j Z d d � Z d S)�DeflateTestsWithSourceFilec C s� t j t d � �} } | j t d t j � | j t d t j � | j d � } | j d � } | j | j t j � | j | j t j � Wd QXd S)zUCheck that files within a Zip archive can have different
compression options.r: ZstoremeZ deflatemeN)
r< r= r r6 r r� � ZIP_DEFLATEDrY rI r� )r7 r$ ZsinfoZdinfor r r �test_per_file_compression� s z4DeflateTestsWithSourceFile.test_per_file_compressionN)r� r� r� r< r� r? r� r r r r r� � s r� c @ s e Z d Z e j Z d S)�Bzip2TestsWithSourceFileN)r� r� r� r< � ZIP_BZIP2r? r r r r r� � s r� c @ s e Z d Z e j Z d S)�LzmaTestsWithSourceFileN)r� r� r� r< �ZIP_LZMAr? r r r r r� � s r� c @ sj e Z d Z e d d � � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z
d S)� AbstractTestZip64InSmallFilesc C s2 d d � t d t � D� } d j | � | _ d S)Nc s s" | ] } t d | d � Vq d S)zTest of zipfile line %d.r( N)r) )r* r+ r r r � <genexpr>� s z;AbstractTestZip64InSmallFiles.setUpClass.<locals>.<genexpr>r s
)r. r/ r1 r2 )r3 r0 r r r r4 � s z(AbstractTestZip64InSmallFiles.setUpClassc C sV t j | _ t j | _ d t _ d t _ t t d � � } | j | j � Wd QXd S)Ni� � r5 ) r< � ZIP64_LIMIT�_limit�ZIP_FILECOUNT_LIMIT�_filecount_limitr# r r6 r2 )r7 r8 r r r r9 � s z#AbstractTestZip64InSmallFiles.setUpc C s t j | d | d d ��: } | j t d � | j t t � | j d | j � Wd QXt j | d | � ��} | j | j t � | j � | j | j d � | j � | j | j d � | j � t j � } | j
| � | j � } | j � } | j t
| � d � | j d | d � | j d
| d � | j d | d � | d j � \ } } } }
| j | d � | j t j | d
� � | j t j | d � � | j |
t t
| j � � � | j � } | j t
| � d � | j t | � | j d | � | j d | � | j � } d d � | D� } | j t
| � d � | j t | � | j d | � | j d | � x* | D]" }
| j |
j t
| j � � q�WxU t d d f D]D } | j | � } | j | j | � | j | j t
| j � � q�W| j � Wd QXd S)Nr: �
allowZip64Tz another.namer; rA rC z File Namer rD rE rF z%Y-%m-%dz%H:%M:%SrG c S s g | ] } | j � q Sr )rH )r* r+ r r r r, s z:AbstractTestZip64InSmallFiles.zip_test.<locals>.<listcomp>)r< r= r6 r r> r2 rI rJ r rK rL rM rN rO rP rQ rR rS rT rU rV rW rX rY rH rZ )r7 r r? r$ r8 r[ r\ r% r] r^ r r_ r` r+ ra rb r r r rc � sN
z&AbstractTestZip64InSmallFiles.zip_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r rc r? )r7 r r r r rd s z(AbstractTestZip64InSmallFiles.test_basicc C s t j t d | j d d �} d | _ d } x4 t | � D]& } | j d | d | d d � q: W| j t | j � � | � | j
� t j t d
| j � } | j t | j � � | � xL t | � D]> } | j d | � j d � } | j | d | d d � q� W| j
� d S) Nr: r� Try � zfoo%08dz%drG �9 rA r( )
r< r= r r? �debugr. r> rI rO rV �closerJ �decode)r7 r� �numfilesr+ �zipf2�contentr r r �test_too_many_files s $
z1AbstractTestZip64InSmallFiles.test_too_many_filesc C s� t j t d | j d d �} d | _ d } x4 t | � D]& } | j d | d | d d � q: W| j t | j � � | � | j
t j � � | j d | d
� Wd QX| j t | j � � | � | j � t j t d | j d d �} d | _ | j t | j � � | � | j
t j � � | j d | d
� Wd QX| j t | j � � | � | j � t j t d | j d d �} d | _ | j t | j � � | � d
} x7 t | | � D]& } | j d | d | d d � q�W| j t | j � � | � | j � t j t d | j � } | j t | j � � | � xL t | � D]> } | j
d | � j d � } | j | d | d d � qSW| j � d S)Nr: r� Fry r� zfoo%08dz%drG r� r- r� Tr� rA r( )r< r= r r? r� r. r> rI rO rV r� � LargeZipFiler� rJ r� )r7 r� r� r+ Z numfiles2r� r� r r r �test_too_many_files_append- sD $
$
z8AbstractTestZip64InSmallFiles.test_too_many_files_appendc C s0 | j t _ | j t _ t t � t t � d S)N)r� r< r� r� r� r r r )r7 r r r r� T s
z&AbstractTestZip64InSmallFiles.tearDownN) r� r� r� r� r4 r9 rc rd r� r� r� r r r r r� � s
9 'r� c @ sI e Z d Z e j Z d d � Z d d � Z d d � Z d d � Z d S)
�StoredTestZip64InSmallFilesc C sD t j | d | d d ��# } | j t j | j t d � Wd QXd S)Nr: r� Fz another.name)r< r= r� r� r6 r )r7 r r? r$ r r r �large_file_exception_test_ s z5StoredTestZip64InSmallFiles.large_file_exception_testc C sG t j | d | d d ��& } | j t j | j d | j � Wd QXd S)Nr: r� Fz another.name)r< r= r� r� r> r2 )r7 r r? r$ r r r �large_file_exception_test2d s z6StoredTestZip64InSmallFiles.large_file_exception_test2c C sA x: t | � D], } | j | t j � | j | t j � q
Wd S)N)r r� r< r� r� )r7 r r r r �test_large_file_exceptioni s z5StoredTestZip64InSmallFiles.test_large_file_exceptionc C su t j t d t j d d �� } | j t d � Wd QXt j t d t j � � } | j | j � d g � Wd QXd S)Nr: r� Tz /absoluterA r� )r< r= r r� r6 r rI rV )r7 r$ r r r r� n s
z2StoredTestZip64InSmallFiles.test_absolute_arcnamesN)
r� r� r� r< r� r? r� r� r� r� r r r r r� [ s
r� c @ s e Z d Z e j Z d S)� DeflateTestZip64InSmallFilesN)r� r� r� r< r� r? r r r r r� v s r� c @ s e Z d Z e j Z d S)�Bzip2TestZip64InSmallFilesN)r� r� r� r< r� r? r r r r r� { s r� c @ s e Z d Z e j Z d S)�LzmaTestZip64InSmallFilesN)r� r� r� r< r� r? r r r r r� � s r� c @ s� 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 d � Z d d � Z d S)�PyZipFileTestsc C s+ | d | k r' | j | d | � n d S)N�o�c)rP )r7 �namerV r r r �assertCompiledIn� s z PyZipFileTests.assertCompiledInc C s� t j | t j d t j t j k �s7 | j d � n t j j | d � } y- t j | t j t j B� } t j
| � Wn t k
r� | j d � Yn Xt | � d S)NZ
effective_idsz/requires write access to the installed locationztest_zipfile.try)
r� �access�W_OK�supports_effective_idsZskipTest�pathr1 r# �O_WRONLY�O_CREATr� � Exceptionr )r7 r� rH �fdr r r �requiresWriteAccess� s
z"PyZipFileTests.requiresWriteAccessc C s� | j t j j t � � t � �} t j | d � �� } t } | j d � s^ | j d � r� | j t j
� } t j d k r� | j | j t j � � n d | k r� t
j j | � } q� | d d � } n | j | � t j j | � } | j | | j � � | j | | j � � Wd QXWd QXt � �� } t j | d � �� } t } | j d � ro| d d
� } n | j | d � d d t j j | � f } | j | | j � � | j | | j � � Wd QXWd QXd S) Nr: �.pyc�.pyo� __pycache__rF Z testpackagez%s/%srp )r� r� rp )r� r� r� �dirname�__file__r r< � PyZipFile�endswithrQ �sep�altsep�extend� importlib�util�source_from_cache�writepy�basename� assertNotInrV r� )r7 �tr$ r% Z
path_splitZbnr r r �test_write_pyfile� s, !
"! z PyZipFileTests.test_write_pyfilec C s� d d l } t j j | j � } | j | � t � �[ } t j | d � �@ } | j | � | j
� } | j d | � | j d | � Wd QXWd QXd S)Nr r: zemail/__init__.pyzemail/mime/text.py) �emailr� r� r� r� r� r r< r� r� rV r� )r7 r� �
packagedirr� r$ r_ r r r �test_write_python_package� s
!
z(PyZipFileTests.test_write_python_packagec , C s` d d l } t j j | j � } | j | � t � �#} t j | d � �} t � � } | j
| � Wd QX| j � } | j d | k � t � � } | j
| d d d � �Wd QX| j � } | j d | k � d d � } t � �0 } | j
t � � | j
| d | �Wd QXWd QX| j � } | r=t | � n | j d | k � Wd QXWd QXd S) Nr r: � SyntaxError�
filterfuncc S s d S)NFr )Zwhateverr r r �<lambda>� s zCPyZipFileTests.test_write_filtered_python_package.<locals>.<lambda>c S s t j j | � j d � S)NZbad)r� r� r� �
startswith)r� r r r �filter� s zAPyZipFileTests.test_write_filtered_python_package.<locals>.filter)r r� r� r� r� r� r r< r� r
r� rM rR � assertWarns� UserWarning�print)r7 r r� r� r$ Z reportSIOZ reportStrr r r r �"test_write_filtered_python_package� s&
!
z1PyZipFileTests.test_write_filtered_python_packagec C s� d d l } t j j | j � } | j | � t r: d n d } | d k rR d n d } t � �i } t j | d d | ��H } | j
| � | j � } | j d | | � | j d | | � Wd QXWd QXd S) Nr rF z.pyoz.pycr: �optimizezemail/__init__zemail/mime/text)
r� r� r� r� r� r� � __debug__r r< r� r� rV rP )r7 r� r� ZoptlevelZextr� r$ r_ r r r � test_write_with_optimization� s
z+PyZipFileTests.test_write_with_optimizationc ) C s5 t j t � zt t j j t d � d � � } | j d � Wd QXt t j j t d � d � � } | j d � Wd QXt t j j t d � d � � } | j d � Wd QXt � �k } t j | d � �P } | j
t � | j � } | j d | � | j d | � | j
d | � Wd QXWd QXWd t t � Xd S)Nzmod1.pyr: z
print(42)
zmod2.pyzprint(42 * 42)
zmod2.txtz bla bla bla
)r� �mkdirr r# r� r1 r6 r r< r� r� rV r� r� r )r7 r8 r� r$ r_ r r r �test_write_python_directory� s
!!!!
z*PyZipFileTests.test_write_python_directoryc % C s� t j t � z� t t j j t d � d � � } | j d � Wd QXt t j j t d � d � � } | j d � Wd QXt � �g } t j | d � �L } | j
t d d d � �| j � } | j d | � | j
d | � Wd QXWd QXWd t t � Xd S) Nzmod1.pyr: z
print(42)
zmod2.pyzprint(42 * 42)
r c S s | j d � S)Nzmod2.py)r� )r% r r r r s zEPyZipFileTests.test_write_python_directory_filtered.<locals>.<lambda>)r� r r r# r� r1 r6 r r< r� r� rV r� r� r )r7 r8 r� r$ r_ r r r �$test_write_python_directory_filtered
s
!!! z3PyZipFileTests.test_write_python_directory_filteredc C sv t � �g } t j | d � �L } t t d � � } | j d � Wd QX| j t | j t � t t � Wd QXWd QXd S)Nr: z!most definitely not a python file)
r r< r� r# r r6 r� r� r� r )r7 r� r$ r r r r �test_write_non_pyfile s
!z$PyZipFileTests.test_write_non_pyfilec % C s t j t � z� t t j j t d � d � � } | j d � Wd QXt � �� } t j | d � �� } t
� �# } | j t j j t d � � Wd QX| j d | j
� � | j � } | j d | � | j d | � | j d | � Wd QXWd QXWd t t � Xd S)Nzmod1.pyr: zBad syntax in python file
r zmod1.pyczmod1.pyo)r� r r r# r� r1 r6 r r< r� r
r� rP rM rV r� r )r7 r8 r� r$ �sr_ r r r � test_write_pyfile_bad_syntax% s
!! " z+PyZipFileTests.test_write_pyfile_bad_syntaxN)
r� r� r� r� r� r� r� r r r
r r r r r r r r� � s
r� c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z e j e
j j d k d � d
d � � Z
e j e
j j d k d � d d � � Z d d � Z d S)� ExtractTestsc C s' t j t d t j � �. } x$ t D] \ } } | j | | � q" WWd QXt j t d � �� } x� t D]� \ } } | j | � } t j j t j
� | � } t j j | � } | j | | � t
| d � �# } | j | j � | j � � Wd QXt | � qd WWd QXt t j j t j
� d � � d S)Nr: rA r� � ziptest2dir)r< r= r r� �SMALL_TEST_DATAr> �extractr� r� r1 �getcwd�normpathrI r# �encoderJ r r )r7 r$ �fpath�fdata� writtenfile� correctfiler r r r � test_extract> s "zExtractTests.test_extractc C s t j t d t j � �. } x$ t D] \ } } | j | | � q" WWd QXt j t d � �� } | j � xm t D]e \ } } t j j t j
� | � } t | d � �# } | j | j
� | j � � Wd QXt | � qn WWd QXt t j j t j
� d � � d S)Nr: rA r� r )r< r= r r� r r> �
extractallr� r� r1 r r# rI r rJ r r )r7 r$ r r Zoutfiler r r r �test_extract_allV s
"z ExtractTests.test_extract_allc
C sK | j t j j | � � t | d � � } | j | j � | � Wd QXd S)Nr� )rR r� r� �isfiler# rI rJ )r7 rH r� r r r r �
check_fileh s zExtractTests.check_filec C s[ t j j } | j | d d � d � | j | d d � d � | j | d d � d � d S) Nz,,?,C:,foo,bar/z�,z_,C_,foo,bar/zza\b,c<d>e|f"g?h*iza\b,c_d_e_f_g_h_iz../../foo../../ba..r�/z foo/ba..r)r< r= Z_sanitize_windows_namerI )r7 Zsanr r r �test_sanitize_windows_namem s z'ExtractTests.test_sanitize_windows_namec C s/ d
d d d
d d d d g } | j | � d S)N�
../foo/bar�foo/bar�
foo/../bar�
foo/../../bar�
foo/bar/..� ./../foo/bar�/foo/bar� /foo/../bar�/foo/../../bar)r% r& )r' r& )r( r& )r) r& )r* r& )r+ r& )r, r& )r- r& )� _test_extract_hackers_arcnames)r7 Zcommon_hacknamesr r r �*test_extract_hackers_arcnames_common_casest s z7ExtractTests.test_extract_hackers_arcnames_common_cases�\z Requires \ as path separator.c C sP d d d d d d d d! d" d# d$ d% d& d' d( d) d* d+ d, g } | j | � d S)-z>Test combination of path fixing and windows name sanitization.�
..\foo\bar�foo/bar� ..\/foo\/bar� foo/\..\/bar� foo\/../\bar� C:foo/bar�
C:/foo/bar� C://foo/bar�
C:\foo\bar�//conky/mountpoint/foo/bar�\\conky\mountpoint\foo\bar�///conky/mountpoint/foo/bar�conky/mountpoint/foo/bar�\\\conky\mountpoint\foo\bar�//conky//mountpoint/foo/bar�\\conky\\mountpoint\foo\bar�//?/C:/foo/bar�\\?\C:\foo\bar�C:/../C:/foo/bar�
C_/foo/bar�a:b\c<d>e|f"g?h*i�b/c_d_e_f_g_h_i�../../foo../../ba..r� foo/ba..rN)r1 r2 )r3 r2 )r4 r2 )r5 r2 )r6 r2 )r7 r2 )r8 r2 )r9 r2 )r: r2 )r; r2 )r< r= )r> r= )r? r= )r@ r= )rA r2 )rB r2 )rC rD )rE rF )rG rH )r. )r7 Zwindows_hacknamesr r r �*test_extract_hackers_arcnames_windows_only� s( z7ExtractTests.test_extract_hackers_arcnames_windows_onlyr# z Requires / as path separator.c C s d d d g } | j | � d S) N� //foo/bar�foo/bar�../../foo../../ba..r� foo../ba..r�
foo/..\bar)rJ rK )rL rM )rN rN )r. )r7 Zposix_hacknamesr r r �(test_extract_hackers_arcnames_posix_only� s z5ExtractTests.test_extract_hackers_arcnames_posix_onlyc
" C sV xO| D]G\ } } d | j � } t j t d t j � �5 } t j � } | | _ d | _ | j | | � Wd QX| j t
j d � } t
j j
d d d � } t
j j
| | j d � � } t j t d � �< } | j | | � } | j | | d
d | | | f �Wd QX| j | | � t d � t j t d � � } | j | � Wd QX| j | | � t d � t
j j
t
j � | j d � � } t j t d � �0 } | j | � } | j | | d
d | �Wd QX| j | | � t | j d � d
� t j t d � � } | j � Wd QX| j | | � t | j d � d
� t t � q Wd S)Ns foobarr: i� r� r# �targetZsubdirZsubsubrA �msgzextract %r: %r != %rz
extract %rr i �)r r< r= r r� �ZipInforH r� r> �replacer� r� r� r1 rQ r rI r! r r r r )
r7 Z hacknamesZarcnameZ fixednamer� r$ r� Z
targetpathr r r r r r. � sB
$z+ExtractTests._test_extract_hackers_arcnamesN)r� r� r� r r r! r$ r/ �unittestZskipIfr� r� r� rI rO r. r r r r r = s
** r c @ s� 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 e
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) 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 S)F�
OtherTestsc C s4 t j t d t j � �Z } | j d d � | j t � � | j d d � Wd QX| j | j � d g d � Wd QXt j t d � �� } | j � } d } x6 | D]. } | j
| � � } | | j � 7} Wd QXq� W| j | d d h � d } x! | D] } | | j | � 7} q� W| j | d d h � Wd QXd S)
Nr: r� r� Zbar� rA r- s foobars barfoo)
r< r= r r� r> r r rI rV rW r# rJ rP )r7 r$ r` r2 rb rr r r r �test_open_via_zip_info� s #
z!OtherTests.test_open_via_zip_infoc C s� t j � } t j | d � � } | j d d � Wd QXt j | d � �J } x@ d D]8 } | j t � � | j d | � } Wd QX| j � qS WWd QXd S)Nr: zspam.txts
ababagalamagarA �Ur! )rX zrU) r r r< r= r> r r" r# r� )r7 r r$ r� rr r r r �test_universal_deprecation� s
z%OtherTests.test_universal_deprecationc C s� t j � } d d d } t j | d t j � � } | j t | � Wd QXd } t j | d � �: } t | t � �" } x | D] } | | 7} q WWd QXWd QX| j | | j d d � � d S) Ns a
r� i r: r- rA s
s
)
r r r< r= r� r> r r& rI rS )r7 r r2 r$ �data2rr r} r r r �test_universal_readaheads� s
z$OtherTests.test_universal_readaheadsc C s^ t j t d � �F } x< d D]4 } t j | � } | j d O_ | j | | � q WWd QXd S)Nr: Zabcdefghijklmnopr )r< r= r rR Z flag_bitsr> )r7 Zorig_zipr2 r� r r r �-test_writestr_extended_local_header_issue1202� s
z8OtherTests.test_writestr_extended_local_header_issue1202c C s� t j t d � �A } x7 t D]/ \ } } | j | | � | j | j d � q WWd QX| j | j d � t j t d � � } | j | j d � Wd QX| j | j d � d S)z8Check that the zipfile is closed after the 'with' block.r: zzipfp is not openNzzipfp is not closedrA )r< r= r r r> ZassertIsNotNoner8 � assertIsNone)r7 r$ r r r r r �
test_close s zOtherTests.test_closec C s� t j t d � �. } x$ t D] \ } } | j | | � q WWd QXy+ t j t d � � } t j � � Wd QXWn( t j k
r� | j | j d � Yn Xd S)zWCheck that the zipfile is closed if an exception is raised in the
'with' block.r: NrA zzipfp is not closed)r< r= r r r> �
BadZipFiler] r8 )r7 r$ r r Zzipfp2r r r �test_close_on_exception s z"OtherTests.test_close_on_exceptionc C s, d } | j t t j t j | � d � d S)Nsd PKx !p�@ xPKxx !p�@ � xPK / rA )r� �NotImplementedErrorr< r= r r )r7 r2 r r r �test_unsupported_version s z#OtherTests.test_unsupported_versionc
C sR t d � } t j | � �1 } x' | j � D] } | j | � j � q+ WWd QXd S)Nzzip_cp437_header.zip)r r< r= rV r# r� )r7 Zfnamer$ r� r r r �test_read_unicode_filenames* s z&OtherTests.test_read_unicode_filenamesc C s� t j t d � �D } | j d d � | j d d � | j | j � d j t � Wd QXt j t d � �; } | j | j d j d � | j | j d j d � Wd QXd S)Nr: zfoo.txtzTest for unicode filenameu ö.txtr rA rF )
r< r= r r> r� rW rH rU rI �filelist)r7 �zfr r r � test_write_unicode_filenames2 s #z'OtherTests.test_write_unicode_filenamesc C s� t j j t � r" t j t � n d } d } y/ t j t d � � } | j | | � Wd QXWn t k
r~ | j d � Yn X| j
t j j t � � t j t d � � } | j | j | � | � Wd QXd S)Nz testfile.txts# hello, world. this is some content.r� z1Could not append data to a non-existent zip file.rA )
r� r� �existsr r r< r= r> �OSError�failrR rI rJ )r7 rH r� re r r r �(test_create_non_existent_file_for_append< s
z3OtherTests.test_create_non_existent_file_for_appendc C sT t t d � � } | j d � Wd QXy t j t � } Wn t j k
rO Yn Xd S)Nr: z this is not a legal zip file
)r# r r6 r< r= r_ )r7 r8 re r r r �test_close_erroneous_fileN s z$OtherTests.test_close_erroneous_filec C s� t t d � � } | j d � Wd QX| j t j t � � t t d � � } | j t j | � � Wd QXt j � } | j d � | j t j | � � | j d d � | j t j | � � d S)z;Check that is_zipfile() correctly identifies non-zip files.r: z this is not a legal zip file
Nr� s this is not a legal zip file
r ) r# r r6 r r< �
is_zipfiler r r� )r7 r8 r r r �test_is_zip_erroneous_file] s
z%OtherTests.test_is_zip_erroneous_filec C s� t j � } t j | d d �� } | j d d � Wd QX| j � } xL t t | � � D]8 } t j | d | � � } | j t j t j | � qY Wd S)zCCheck that zipfiles with missing bytes at the end raise BadZipFile.r� r: zfoo.txts O, for a Muse of Fire!N)
r r r< r= r> rM r. rO r� r_ )r7 r8 r� r� �Nr r r �test_damaged_zipfilem s z OtherTests.test_damaged_zipfilec C s� t j t d d �� } | j d d � Wd QX| j t j t � � t t d � �9 } | j t j | � � | j d d � | j � } Wd QXt j
� } | j | � | j t j | � � | j d d � | j t j | � � d S)z7Check that is_zipfile() correctly identifies zip files.r� r: zfoo.txts O, for a Muse of Fire!Nr� r ) r< r= r r> rR rl r# r� rJ r r r6 )r7 r� r8 Z zip_contentsr r r �test_is_zip_valid_file{ s
z!OtherTests.test_is_zip_valid_filec C s | j t t j t � d S)N)r� rh r< r= r )r7 r r r �%test_non_existent_file_raises_OSError� s z0OtherTests.test_non_existent_file_raises_OSErrorc
C st t t d � } | j � | j t j t j t � t t d � � } | j d � Wd QX| j t j t j t � d S)Nr: z
short file)r# r r� r� r< r_ r= r6 )r7 r r8 r r r �!test_empty_file_raises_BadZipFile� s
z,OtherTests.test_empty_file_raises_BadZipFilec C s� t j � } t j | d d �� } | j d d � Wd QX| j t | j d � | j t | j d � | j t | j � | j t | j d d � t t
d � � } | j d � Wd QX| j t | j t
� d S) z?Verify that testzip() doesn't swallow inappropriate exceptions.r� r: zfoo.txtzO, for a Muse of Fire!Nz bogus.txtr� zzipfile test data) r r r< r= r> r� r� rJ r# rZ r r6 )r7 r2 r� r r r r �#test_closed_zip_raises_RuntimeError� s z.OtherTests.test_closed_zip_raises_RuntimeErrorc C s | j t t j t d � d S)z>Check that bad modes passed to ZipFile constructor are caught.�qN)r� r� r< r= r )r7 r r r �test_bad_constructor_mode� s z$OtherTests.test_bad_constructor_modec C sv t j t d d �� } | j d d � Wd QXt j t d d ��- } | j d � | j t | j d d � Wd QXd S)z7Check that bad modes passed to ZipFile.open are caught.r� r: zfoo.txtzO, for a Muse of Fire!NrA rt )r< r= r r> rJ r� r� r# )r7 r� r r r �test_bad_open_mode� s
z OtherTests.test_bad_open_modec C s� t j t d d ��u } | j d d � | j d � �M } x- t t � D] } | j | j d � d � qG W| j | j � d � Wd QXWd QXd S) zsCheck that calling read(0) on a ZipExtFile object returns an empty
string and doesn't advance file pointer.r� r: zfoo.txtzO, for a Muse of Fire!r r- s O, for a Muse of Fire!N) r< r= r r> r# r. r/ rI rJ )r7 r� r r+ r r r �
test_read0� s zOtherTests.test_read0c C s; t j t d d �� } | j t | j d d � Wd QXd S)zqCheck that attempting to call open() for an item that doesn't
exist in the archive raises a RuntimeError.r� r: zfoo.txtrA N)r< r= r r� �KeyErrorr# )r7 r� r r r �test_open_non_existent_item� s z&OtherTests.test_open_non_existent_itemc C s | j t t j t d d � d S)zMCheck that bad compression methods passed to ZipFile.open are
caught.r: rF Nrp )r� r� r< r= r )r7 r r r �test_bad_compression_mode� s z$OtherTests.test_bad_compression_modec C sD d } t j t j | � d � � } | j t | j d � Wd QXd S)Nsf PK. �C�@ x PK.. �C�@ � xPK / ! rA �x)r< r= r r r� ra r# )r7 r2 r� r r r � test_unsupported_compression� s z'OtherTests.test_unsupported_compressionc
C sK t j t d d ��0 } | j d d � | j | j � d g � Wd QXd S)zLCheck that a filename containing a null byte is properly
terminated.r� r: z foo.txt qqqs O, for a Muse of Fire!zfoo.txtN)r< r= r r> rI rV )r7 r� r r r �test_null_byte_in_filename� s z%OtherTests.test_null_byte_in_filenamec C sP | j t j d � | j t j d � | j t j d � | j t j d � d S)zACheck that ZIP internal structure sizes are calculated correctly.� �. �8 r� N)rI r< ZsizeEndCentDirZsizeCentralDirZsizeEndCentDir64ZsizeEndCentDir64Locator)r7 r r r �test_struct_sizes� s z OtherTests.test_struct_sizesc ^ C s t j t d d ��* } | j | j d � | j d d � Wd QXt j t d d �� } | j | j d � Wd QXd } t j t d d �� } | | _ | j d d � Wd QXt j t d d �� } | j | j | � Wd QXd j d
d � t d � D� � } | j d � } t j t d d �� } | | _ | j d d � Wd QXt j t d d �� } | j | j | � Wd QXt j t d d ��: } | j t
� � | d | _ Wd QX| j d d � Wd QXt j t d d �� } | j | j | � Wd QXt j t d d �� } d | _ | j d d � Wd QXt j t d d �� } d | _ Wd QXt j t d d �� } | j | j d � Wd QXt j t d d �� } d | _ | j d d � Wd QXt j t d d �� } d | _ Wd QXt j t d d �� } | j | j d � Wd QXd S)z8Check that comments on the archive are handled properly.r� r: r- zfoo.txtzO, for a Muse of Fire!NrA s9 Bravely taking to his feet, he beat a very brave retreat.r c S s" g | ] } d | d d � q S)z%drG rv r )r* r+ r r r r, s z,OtherTests.test_comments.<locals>.<listcomp>rF r� r( s oopss original commentr� s an updated comments original comment that's longers shorter commenti i�� ) r< r= r rI r� r> r1 r. r r r )r7 r� Zzipfrr� Zcomment2r r r �
test_comments� sL " zOtherTests.test_commentsc C sT t j t d t j � �6 } | j d d � | j t � � d | _ Wd QXWd QXd S)Nr: zfoo.txtzO, for a Muse of Fire!zthis is an error)r< r= r r� r> r� � TypeErrorr� )r7 r� r r r �test_unicode_comment, s z OtherTests.test_unicode_commentc C sl t j t d t j � � } | j | j � d | _ Wd QXt j t d � � } | j | j d � Wd QXd S)Nr� s this is a commentrA )r< r= r r� r rd r� rI )r7 r� r r r �$test_change_comment_in_empty_archive2 s
z/OtherTests.test_change_comment_in_empty_archivec C s� t j t d t j � � } | j d d � Wd QXt j t d t j � � } | j | j � d | _ Wd QXt j t d � � } | j | j d � Wd QXd S)Nr: zfoo.txtzO, for a Muse of Fire!r� s this is a commentrA ) r< r= r r� r> rR rd r� rI )r7 r� r r r �'test_change_comment_in_nonempty_archive9 s z2OtherTests.test_change_comment_in_nonempty_archivec C s� t j t d d �} | j � y t j t d d �} Wn" t j k
r\ | j d � Yn Xt j t d d �} | j � y t j t d d �} Wn | j d � Yn Xd S)Nr� r: rA z+Unable to create empty ZIP file in 'w' moder� z+Unable to create empty ZIP file in 'a' mode)r< r= r r� r_ ri )r7 r� r r r �test_empty_zipfileB s
z OtherTests.test_empty_zipfilec C s9 t t d � } | j � | j t j t j t d � d S)Nr: rA )r# r r� r� r< r_ r= )r7 r r r r �test_open_empty_fileS s
z OtherTests.test_open_empty_filec
C s | j t t j d d � d S)NZ seventies� rF r )r� rF rF r r r )r� r� r< rR )r7 r r r � test_create_zipinfo_before_1980[ s z*OtherTests.test_create_zipinfo_before_1980c C sA d } t j t j | � d � � } | j | j � � Wd QXd S)z>If an extra field in the header is less than 4 bytes, skip it.sn PK ���@���� abc APK ���@���� �� abc PK 3 % rA N)r< r= r r r] rZ )r7 rs r� r r r �#test_zipfile_with_short_extra_field_ s z.OtherTests.test_zipfile_with_short_extra_fieldc C s t t � t t � d S)N)r r r )r7 r r r r� n s
zOtherTests.tearDownN)&r� r� r� rW rY r[ r\ r^ r` rb r
rc rf rj rk rm ro rp rq rr rs ru rv rw ry rz r| r} r� r� r� r� r� r� r� r� r� r� r r r r rU � sD
7 rU c @ s( e Z d Z d d � Z d d � Z d S)�AbstractBadCrcTestsc
C sJ | j } t j t j | � d d �� } | j d | j � � Wd QXd S)z>Tests that files with bad CRCs return their name from testzip.r� rA �afileN)�zip_with_bad_crcr< r= r r rI rZ )r7 rs r� r r r �test_testzip_with_bad_crct s !z-AbstractBadCrcTests.test_testzip_with_bad_crcc ) C s | j } t j t j | � d d �� } | j t j | j d � Wd QXt j t j | � d d ��8 } | j d d � � } | j t j | j � Wd QXWd QXt j t j | � d d ��Z } | j d d � �? } d | _ | j t j � � x | j d � rq� WWd QXWd QXWd QXd S)zFTests that files with bad CRCs raise a BadZipFile exception when read.r� rA r� NrV )
r� r< r= r r r� r_ rJ r# Z
MIN_READ_SIZE)r7 rs r� Z corrupt_filer r r �test_read_with_bad_crc| s ! !"! z*AbstractBadCrcTests.test_read_with_bad_crcN)r� r� r� r� r� r r r r r� s s r� c @ s e Z d Z e j Z d Z d S)�StoredBadCrcTestssx PK ��;:r�� afilehello,AworldPK ��;:r�� � afilePK 3 / N)r� r� r� r< r� r? r� r r r r r� � s r� c @ s e Z d Z e j Z d Z d S)�DeflateBadCrcTestss| PK n} =FAKE n afile�H���W(�/�Iɠ= PK n} =FAKE n � afilePK 3 3 N)r� r� r� r< r� r? r� r r r r r� � s r� c @ s e Z d Z e j Z d Z d S)�Bzip2BadCrcTestss� PK nu =FAKE8 n afileBZh91AY&SYԨ� �@ D�� �P�!����u5:���"�(HjTe?�PK nu =FAKE8 n ��� afilePK 3 [ N)r� r� r� r< r� r? r� r r r r r� � s r� c @ s e Z d Z e j Z d Z d S)�LzmaBadCrcTestss� PK nu =FAKE n afile ] 4I���:3` q!.8 PK nu =FAKE n ��� afilePK 3 > N)r� r� r� r< r� r? r� r r r r r� � s r� c @ s� e Z d Z d Z d Z d Z d Z d d Z d d � Z d d
� Z d d � Z
d
d � Z e d d � � Z
d d � Z d S)�DecryptionTestsz�Check that ZIP decryption works. Since the library does not
support encryption at the moment, we use a pre-generated encrypted
ZIP file.s� PK n�i.#y�?& test.txt��gly|�-�=�y�r�Z}Lg���|�䋺�)���lPK n�i.#y�?& �� test.txtPK 6 L s� PK �}38xu�� zeroUT �Gڋ�GUx ���<M�a�X�Y&�{oEם���PKxu�� PK �}38xu��
�� zeroUT �GUx PK ? [ s zipfile.py encryption tests i c C s~ t t d � � } | j | j � Wd QXt j t d � | _ t t d � � } | j | j � Wd QXt j t d � | _ d S)Nr5 rA )
r# r r6 r2 r< r= r� r rZ �zip2)r7 r8 r r r r9 � s zDecryptionTests.setUpc C s8 | j j � t j t � | j j � t j t � d S)N)r� r� r� r r r� r )r7 r r r r� � s
zDecryptionTests.tearDownc C s6 | j t | j j d � | j t | j j d � d S)Nztest.txt�zero)r� r� r� rJ r� )r7 r r r �test_no_password� s z DecryptionTests.test_no_passwordc C sV | j j d � | j t | j j d � | j j d � | j t | j j d � d S)Ns perlztest.txtr� )r� � setpasswordr� r� rJ r� )r7 r r r �test_bad_password� s z!DecryptionTests.test_bad_passwordc C sb | j j d � | j | j j d � | j � | j j d � | j | j j d � | j � d S)Ns pythonztest.txts 12345r� )r� r� rI rJ �plainr� �plain2)r7 r r r �test_good_password� s z"DecryptionTests.test_good_passwordc C sw | j t | j j d � | j t | j j d d � | j t | j j d d d �| j t | j j d d d �d S)NZunicodeztest.txt�python�pwd)r� r� r� r� rJ r# r )r7 r r r �test_unicode_password s z%DecryptionTests.test_unicode_passwordN)r� r� r� �__doc__r2 rZ r� r� r9 r� r� r� r
r� r� r r r r r� � s
r� c @ s� e Z d Z e 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 S)�"AbstractTestsWithRandomBinaryFilesc C sI t d d � d t d d � } d j d d � t | � D� � | _ d S)Nr� �@ i rF r- c s s1 | ]' } t j d t � t d d � � Vq d S)z<fi� Ni���)�structZpackr r )r* r+ r r r r� s z@AbstractTestsWithRandomBinaryFiles.setUpClass.<locals>.<genexpr>)r r1 r. r2 )r3 Z datacountr r r r4 s z-AbstractTestsWithRandomBinaryFiles.setUpClassc C s, t t d � � } | j | j � Wd QXd S)Nr5 )r# r r6 r2 )r7 r8 r r r r9 s z(AbstractTestsWithRandomBinaryFiles.setUpc C s t t � t t � d S)N)r r r )r7 r r r r� s
z+AbstractTestsWithRandomBinaryFiles.tearDownc
C sB t j | d | � �' } | j t d � | j t t � Wd QXd S)Nr: z another.name)r< r= r6 r )r7 r r? r$ r r r r@ s z4AbstractTestsWithRandomBinaryFiles.make_test_archivec C s� | j | | � t j | d | � �d } | j t � } | j t | � t | j � � | j | | j � | j | j d � | j � Wd QXd S)NrA z another.name)r@ r< r= rJ r rI rO r2 )r7 r r? r$ �testdatar r r rc ! s z+AbstractTestsWithRandomBinaryFiles.zip_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r rc r? )r7 r r r r � test_read+ s z,AbstractTestsWithRandomBinaryFiles.test_readc C sJ | j | | � t j | d | � � } g } | j t � �4 } x* | j d � } | s\ Pn | j | � qC WWd QXg } | j d � �4 } x* | j d � } | s� Pn | j | � q� WWd QXd j | � } | j t | � t | j
� � | j | | j
� d j | � }
| j t |
� t | j
� � | j |
| j
� Wd QXd S)NrA re z another.namer- ) r@ r< r= r# r rJ rf r1 rI rO r2 ) r7 r r? r$ rg rh ri rj rk Z testdata1Z testdata2r r r rl / s, z0AbstractTestsWithRandomBinaryFiles.zip_open_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r rl r? )r7 r r r r rm L s z,AbstractTestsWithRandomBinaryFiles.test_openc C s� | j | | � t j | d | � �� } g } | j t � �= } x3 | j t d d � � } | se Pn | j | � qC WWd QXd j | � } | j t
| � t
| j � � | j | | j � Wd QXd S)NrA rF i r- ) r@ r< r= r# r rJ r rf r1 rI rO r2 )r7 r r? r$ rg rh ri r� r r r rn P s z7AbstractTestsWithRandomBinaryFiles.zip_random_open_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r rn r? )r7 r r r r ro a s z3AbstractTestsWithRandomBinaryFiles.test_random_openN)r� r� r� r� r4 r9 r� r@ rc r� rl rm rn ro r r r r r� s
r� c @ s e Z d Z e j Z d S)� StoredTestsWithRandomBinaryFilesN)r� r� r� r< r� r? r r r r r� f s r� c @ s e Z d Z e j Z d S)�!DeflateTestsWithRandomBinaryFilesN)r� r� r� r< r� r? r r r r r� j s r� c @ s e Z d Z e j Z d S)� Bzip2TestsWithRandomBinaryFilesN)r� r� r� r< r� r? r r r r r� o s r� c @ s e Z d Z e j Z d S)� LzmaTestsWithRandomBinaryFilesN)r� r� r� r< r� r? r r r r r� t s r� c @ s� e Z d Z e 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 S)�TestsWithMultipleOpensc C s* d t d � | _ d t d � | _ d S)Ns 111i' s 222)r �data1rZ )r3 r r r r4 | s z!TestsWithMultipleOpens.setUpClassc
C sK t j | d t j � �- } | j d | j � | j d | j � Wd QXd S)Nr: �ones�twos)r< r= r� r> r� rZ )r7 r r$ r r r r@ � s z(TestsWithMultipleOpens.make_test_archivec C s� | j t � t j t d d ��� } | j d � �] } | j d � �E } | j d � } | j d � } | | j � 7} | | j � 7} Wd QXWd QX| j | | � | j | | j � Wd QXd S)Nr� rA r� i� )r@ r r< r= r# rJ rI r� )r7 r� �zopen1�zopen2r� rZ r r r �test_same_file� s
$ z%TestsWithMultipleOpens.test_same_filec C s� | j t � t j t d d ��� } | j d � �] } | j d � �E } | j d � } | j d � } | | j � 7} | | j � 7} Wd QXWd QX| j | | j � | j | | j � Wd QXd S)Nr� rA r� r� i� ) r@ r r< r= r# rJ rI r� rZ )r7 r� r� r� r� rZ r r r �test_different_file� s
$ z*TestsWithMultipleOpens.test_different_filec C s� | j t � t j t d d ��� } | j d � �] } | j d � �E } | j d � } | j d � } | | j � 7} | | j � 7} Wd QXWd QX| j | | j � | j | | j � Wd QXd S)Nr� rA r� r� i� ) r@ r r< r= r# rJ rI r� rZ )r7 r� r� r� r� rZ r r r �test_interleaved� s
$ z'TestsWithMultipleOpens.test_interleavedc C s� | j t � t j � �� } t j t d � �7 } | j | j d � � } | j | j d � � } Wd QX| j d � } | j d � } | | j � 7} | | j � 7} Wd QX| j | | j
� | j | | j � d S)NrA r� r� i� ) r@ r �
contextlib� ExitStackr< r= �
enter_contextr# rJ rI r� rZ )r7 �stackr� r� r� r� rZ r r r �test_read_after_close� s
z,TestsWithMultipleOpens.test_read_after_closec C s� t j t d t j � �T } | j d | j � | j d | j � | j d � � } | j d � } Wd QXWd QX| j | | j d d � � t j t d � �% } | j d � } | j d � } Wd QX| j | | j � | j | | j � d S)Nr: r� r� i� rA )
r< r= r r� r> r� rZ r# rJ rI )r7 r� r� r� rZ r r r �test_read_after_write� s z,TestsWithMultipleOpens.test_read_after_writec C s� t j t d t j � �R } | j d | j � | j d � �' } | j d � | j d | j � Wd QXWd QXt j t d � �% } | j d � } | j d � } Wd QX| j | | j � | j | | j � d S)Nr: r� i� r� rA )
r< r= r r� r> r� r# rJ rZ rI )r7 r� r� r� rZ r r r �test_write_after_read� s
z,TestsWithMultipleOpens.test_write_after_readc C s� | j t � t j t d d ��C } x9 t d � D]+ } | j d � | j d � � } Wd QXq2 WWd QXt t j � � } | j | j
� d � Wd QXd S)Nr� rA ry r� ) r@ r r< r= r. rJ r# r� �devnullZ
assertLess�fileno)r7 r� r{ r� r r r r �test_many_opens� s
z&TestsWithMultipleOpens.test_many_opensc C s t t � d S)N)r r )r7 r r r r� � s z TestsWithMultipleOpens.tearDownN)r� r� r� r� r4 r@ r� r� r� r� r� r� r� r� r r r r r� z s
r� c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�TestWithDirectoryc C s t j t � d S)N)r� r r )r7 r r r r9 � s zTestWithDirectory.setUpc
C s� t j t d � � � } | j t � Wd QX| j t j j t j j t d � � � | j t j j t j j t d d � � � | j t j j
t j j t d d d � � � d S)Nz
zipdir.zipr� �br� ) r< r= r r r rR r� r� �isdirr1 rg )r7 r� r r r �test_extract_dir� s
(+z"TestWithDirectory.test_extract_dirc C s* t j t j j t d � � | j � d S)Nr� )r� r r� r1 r r� )r7 r r r �
test_bug_6050� s z TestWithDirectory.test_bug_6050c C s� t j j t d � } t j | � t j | � j d @} t j t d � �� } | j
| � | j d } | j | j
j d � � | j | j | d >d B� | j
| d � | j d } | j | j
d � | j | j | d >d B� Wd QXt j t d
� �� } | j d } | j | j
j d � � | j | j | d >d B� | j d } | j | j
d � | j | j | d >d B� t j j t d � } t j | � | j | � | j t j j t j j | d � � � | j t t j | � � d � Wd QXd S)
Nr{ i�� r: r z/x/r� �yrF zy/rA rP rV )r� r� r1 r r �stat�st_moder< r= r r6 rd rR rH r� rI r� r r� rO �listdir)r7 �dirpathr� r� r� rP r r r �test_write_dir� s0
!
(z TestWithDirectory.test_write_dirc C sE t j t j j t d � � t j t d � �J } | j d d � | j d } | j
| j d � | j
| j d � Wd QXt j t d � �� } | j d } | j
| j j d � � | j
| j d
� t j j t d � } t j | � | j | � | j
t j j t j j | d � � � | j
t j | � d g � Wd QXd S)Nr{ r: zx/r- r i�A r� rA rP i �Ai �Ai �Ai �A)r� r r� r1 r r< r= r r> rd rI rH r� rR r� r r� r� )r7 r� r� rP r r r �test_writestr_dir s
(z#TestWithDirectory.test_writestr_dirc C s- t t � t j j t � r) t t � n d S)N)r r r� r� rg r r )r7 r r r r� s
zTestWithDirectory.tearDownN) r� r� r� r9 r� r� r� r� r� r r r r r� � s r� c @ s� e Z d Z e 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 d � Z d S) � AbstractUniversalNewlineTestsc C so d d � t t � D� | _ d | _ i | _ x: t | j � D]) \ } } | j | j � | | j | <q> Wd S)Nc S s# g | ] } t d | d � � q S)zTest of zipfile line %d.r( )r) )r* r+ r r r r, ( s z<AbstractUniversalNewlineTests.setUpClass.<locals>.<listcomp>�
�
�
)r� r� r� )r. r/ r0 �seps�arcdata� enumerater1 )r3 �nr r r r r4 & s z(AbstractUniversalNewlineTests.setUpClassc C sw i | _ xg t | j � D]V \ } } d t | f | j | <t | j | d � � } | j | j | � Wd QXq Wd S)Nz%s-%dr5 )�arcfilesr� r� r r# r6 r� )r7 r� r r r r r r9 / s
z#AbstractUniversalNewlineTests.setUpc C sL t j | d | � �1 } x' | j j � D] } | j | | � q( WWd QXd S)Nr: )r< r= r� �valuesr6 )r7 r r? r$ r% r r r r@ 6 s z/AbstractUniversalNewlineTests.make_test_archivec C s� | j | | � t j | d � �b } xX | j j � D]G \ } } t | | � � } | j � } Wd QX| j | j | | � q5 WWd QXd S)NrA ) r@ r< r= r� �itemsr& rJ rI r� )r7 r r? r$ r� r% r8 rs r r r � read_test< s z'AbstractUniversalNewlineTests.read_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r r� r? )r7 r r r r r� F s z'AbstractUniversalNewlineTests.test_readc C s� | j | | � t j | d � �� } x� | j j � D]t \ } } t | | � �W } d } xG | j � } | sr Pn | | 7} | j d � } | s� Pn | | 7} q\ WWd QXq5 W| j | | j d � Wd QXd S)NrA r- � s
)
r@ r< r= r� r� r& rz rJ rI r� ) r7 r r? r$ r� r% rr r2 rJ r r r �readline_read_testJ s
z0AbstractUniversalNewlineTests.readline_read_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r r� r? )r7 r r r r r| _ s z0AbstractUniversalNewlineTests.test_readline_readc C s� | j | | � t j | d � �s } xi | j j � D]X \ } } t | | � �; } x1 | j D]& } | j � } | j | | d � q] WWd QXq5 WWd QXd S)NrA s
) r@ r< r= r� r� r& r0 rz rI ) r7 r r? r$ r� r% rr r} r~ r r r �
readline_testc s z+AbstractUniversalNewlineTests.readline_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r r� r? )r7 r r r r r� n s z+AbstractUniversalNewlineTests.test_readlinec
C s� | j | | � t j | d � �� } xx | j j � D]g \ } } t | | � � } | j � } Wd QXx4 t | j | � D] \ } } | j | | d � qx Wq5 WWd QXd S)NrA s
)
r@ r< r= r� r� r& r� r� r0 rI )
r7 r r? r$ r� r% r8 r� r} r� r r r �readlines_testr s z,AbstractUniversalNewlineTests.readlines_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r r� r? )r7 r r r r r� } s z,AbstractUniversalNewlineTests.test_readlinesc C s� | j | | � t j | d � �v } xl | j j � D][ \ } } t | | � �> } x4 t | j | � D] \ } } | j | | d � qf WWd QXq5 WWd QXd S)NrA s
) r@ r< r= r� r� r& r� r0 rI ) r7 r r? r$ r� r% r8 r} r� r r r �iterlines_test� s z,AbstractUniversalNewlineTests.iterlines_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r r� r? )r7 r r r r r� � s z,AbstractUniversalNewlineTests.test_iterlinesc C sB x' | j j � D] \ } } t | � q Wt t � t t � d S)N)r� r� r r r )r7 r� r% r r r r� � s
z&AbstractUniversalNewlineTests.tearDownN)r� r� r� r� r4 r9 r@ r� r� r� r| r� r� r� r� r� r� r� r r r r r� % s
r� c @ s e Z d Z e j Z d S)�StoredUniversalNewlineTestsN)r� r� r� r< r� r? r r r r r� � s r� c @ s e Z d Z e j Z d S)� DeflateUniversalNewlineTestsN)r� r� r� r< r� r? r r r r r� � s r� c @ s e Z d Z e j Z d S)�Bzip2UniversalNewlineTestsN)r� r� r� r< r� r? r r r r r� � s r� c @ s e Z d Z e j Z d S)�LzmaUniversalNewlineTestsN)r� r� r� r< r� r? r r r r r� � s r� �__main__)Cr� r r� �sys�importlib.utilr� rS r� r< rT Ztempfiler r r r Z test.supportr r r r r
r r r
r r Z TESTFNDIRr/ Z
DATAFILES_DIRr r r r& r' ZTestCaser� r� r� r� r� r� r� r� r� r� r rU r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �mainr r r r �<module> s� @
� & [� ��� � A[ iAq