Your IP : 216.73.216.213
��YfbT � @ sS 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 d d l m Z 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/ Gd0 d1 � d1 e j$ � Z0 Gd2 d3 � d3 � Z1 Gd4 d5 � d5 e1 e j$ � Z2 e Gd6 d7 � d7 e1 e j$ � � Z3 e Gd8 d9 � d9 e1 e j$ � � Z4 e Gd: d; � d; e1 e j$ � � Z5 Gd< d= � d= e j$ � Z6 Gd> d? � d? � Z7 Gd@ dA � dA e7 e j$ � Z8 e GdB dC � dC e7 e j$ � � Z9 e GdD dE � dE e7 e j$ � � Z: e GdF dG � dG e7 e j$ � � Z; GdH dI � dI � Z< GdJ dK � dK � Z= GdL dM � dM e j$ � Z> e GdN dO � dO e j$ � � Z? GdP dQ � dQ e j$ � Z@ GdR dS � dS � ZA GdT dU � dU eA e j$ � ZB e GdV dW � dW eA e j$ � � ZC e GdX dY � dY eA e j$ � � ZD e GdZ d[ � d[ eA e j$ � � ZE Gd\ d] � d] e j$ � ZF eG d^ k rOe jH � d S)_� N)�
TemporaryFile)�randint�random� getrandbits)�
script_helper)
�TESTFN�findfile�unlink�rmtree�temp_dir�
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/python35/lib64/python3.5/test/test_zipfile.py� getrandbytes s r c c s\ t Vt � � } | V| j | j � Wd QRXt j � � } | V| j | j � Wd QRXd 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 QRXd S)N� �rU)r �DeprecationWarning�open)�zipfp�fnr r r �openU* s r( c @ sZ 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 d6 � Z d7 S)8�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>1 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 QRXd S)N�wb)r% r �writer4 )�self�fpr r r �setUp6 s z!AbstractTestsWithSourceFile.setUpc
C sV t j | d | � �: } | j t d � | j t t � | j d | j � Wd QRXd S)N�wz another.name�strfile)�zipfile�ZipFiler8 r �writestrr4 )r9 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 QRXd 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. e s z8AbstractTestsWithSourceFile.zip_test.<locals>.<listcomp>)rB r> r? � assertEqual�readr r4 r �StringIO�printdir�getvalue�
splitlines�len�assertIn�split�
assertTrue�time�strptime�str�namelist�infolist� file_size�getinforJ �testzip)r9 r rA r& r: � directory�linesr' �date�time_r �names�infosr- �nm�infor r r �zip_testB sH
z$AbstractTestsWithSourceFile.zip_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! re rA )r9 r r r r �
test_basicv s z&AbstractTestsWithSourceFile.test_basicc C s� | j | | � t j | d | � �� } g } | j t � �1 } x' | j d � } | sY P| j | � qC WWd QRXg } | j d � �1 } x' | j d � } | s� P| j | � q� WWd QRX| j d j | � | j � | j d j | � | j � Wd QRXd S)NrC � z another.namer/ )
rB r> r? r% r rL �appendrK r3 r4 ) r9 r rA r& �zipdata1�zipopen1� read_data�zipdata2�zipopen2r r r �
zip_open_testz s$ z)AbstractTestsWithSourceFile.zip_open_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! rn rA )r9 r r r r � test_open� s z%AbstractTestsWithSourceFile.test_openc C s� | j | | � t j | d | � �u } g } | j t � �: } x0 | j t d d � � } | sb P| j | � qC WWd QRX| j d j | � | j
� Wd QRXd S)NrC rH i r/ ) rB r> r? r% r rL r rh rK r3 r4 )r9 r rA r& ri rj rk 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! rp rA )r9 r r r r �test_random_open� s z,AbstractTestsWithSourceFile.test_random_openc C s� | j | | � t j | d � �P } | j t � �7 } g } x' | j d � } | sV P| j | � q@ WWd QRXWd QRX| j d j | � | j � d S)NrC rH r/ ���)
rB r> r? r% r �read1rh rK r3 r4 )r9 r rA r& �zipopen�zipdatark r r r �zip_read1_test� s z*AbstractTestsWithSourceFile.zip_read1_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! rv rA )r9 r r r r �
test_read1� s z&AbstractTestsWithSourceFile.test_read1c C s� | j | | � t j | d � �f } | j t � �M } g } x= | j d � } | j t | � d � | sl P| j | � q@ WWd QRXWd QRX| j d j
| � | j � d S)NrC �
r/ ) rB r> r? r% r rs ZassertLessEqualrQ rh rK r3 r4 )r9 r rA r& rt ru rk 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! ry rA )r9 r r r r �
test_read1_10� s z)AbstractTestsWithSourceFile.test_read1_10c C s� | j | | � t j | d � �j } | j t � �Q } d } xA | j � } | sS P| | 7} | j d � } | ss P| | 7} q@ WWd QRXWd QRX| j | | j � d S)NrC r/ �d ) rB r> r? r% r �readlinerL rK r4 )r9 r rA r& rt r4 rL 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} rA )r9 r r r r �test_readline_read� s z.AbstractTestsWithSourceFile.test_readline_readc C sy | j | | � t j | d � �P } | j t � �7 } x- | j D]" } | j � } | j | | � qA WWd QRXWd QRXd S)NrC )rB r> r? r% r r2 r| rK )r9 r rA r& rt �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� rA )r9 r r r r �
test_readline� s z)AbstractTestsWithSourceFile.test_readlinec C s� | j | | � t j | d � �_ } | j t � � } | j � } Wd QRXx0 t | j | � D] \ } } | j | | � q] WWd QRXd S)NrC ) rB r> r? r% r � readlines�zipr2 rK )r9 r rA r& rt �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� rA )r9 r r r r �test_readlines s z*AbstractTestsWithSourceFile.test_readlinesc C s| | j | | � t j | d � �S } | j t � �: } x0 t | j | � D] \ } } | j | | � qJ WWd QRXWd QRXd S)NrC )rB r> r? r% r r� r2 rK )r9 r rA r& rt 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� rA )r9 r r r r �test_iterlines s z*AbstractTestsWithSourceFile.test_iterlinesc C s� t j t d | j � � } | j d d � Wd QRXt j t d | j � �R } | j d � �9 } | j | j d � d � | j | j d � d � Wd QRXWd QRXd S) z>Check for cases where compressed data is larger than original.r< r= Z12NrC rH � 1� 2)r> r? r rA r@ r% rK rL )r9 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@ rA r[ rK r� )r9 r& rd r r r �test_writestr_compression s z5AbstractTestsWithSourceFile.test_writestr_compressionc C s� x� d D]� } | d } t | � } t j t j � d | j � �X } | j d | � | j d d � �, } | j | � } | j t
| � | � Wd QRXWd QRXq Wd S)
NrH � � � � @ r< �foorC )rH r� r� r� r� ) r r> r? r r rA r@ r% rL rK rQ )r9 Z test_sizerZ Zjunk�zipfr: Zbufr r r �test_read_return_size$ s
!z1AbstractTestsWithSourceFile.test_read_return_sizec H C s� t j � } t j | d d ��/ } | j d | j d | j �| j � } Wd QRX| j � } t j | � } t j | � �R } | j d � �9 } | j
| d � | j t � � | j
� Wd QRXWd QRXWd QRXt j | � } t j | � �^ } | j d � �E } | j
| d � | j t � � x | j
d � r;q)WWd QRXWd QRXWd QRXt j | � } t j | � �^ } | j d � �E } | j
| d � | j t � � x | j d � r�q�WWd QRXWd QRXWd QRXd S)N�moder< r= r� � r{ )r r r> r? r@ r4 rA �tellrO r% �truncate� assertRaises�EOFErrorrL rs )r9 r: r� Z
end_offset� zipfiledatart r r r �test_truncated_zipfile0 s2 z2AbstractTestsWithSourceFile.test_truncated_zipfilec C s� d } x�t | � D]�} t j | d | j � �3 } | j t | � t | � } | j d | � Wd QRXt j | d � �Y} t | � } t | t � r� | j d | | � n | j d | | � | j d | � t | j
| � � } | j d | | � | j d | � | j d | � | j t j k rM| j d
| � | j d | � | j | � �Y } t | � } | j d | | � | j d | � | j t j k r�| j d
| � Wd QRX| j d
t | � � Wd QRX| j d
t | � � q Wd S)Nz file.namer< zmode='w'rC z filename=%rzfile=%rzmode='r'z filemode=z
file_size=zcompress_type=zcompress_size=zname=%rz[closed])
r! r> r? rA r8 r �reprrR �
isinstancerW r[ �
ZIP_STOREDr% )r9 �fnamer r& rC rt r r r � test_reprN s6 z%AbstractTestsWithSourceFile.test_reprc C s t t � t t � d S)N)r r r )r9 r r r �tearDownm s
z$AbstractTestsWithSourceFile.tearDownN) �__name__�
__module__� __qualname__� classmethodr6 r; rB re rf rn ro rp rq rv rw ry rz r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r r r r r) . s6 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 d � Z d d � Z d S)�StoredTestsWithSourceFileNc C sR | j | | � t j | d � �) } | j d � } | j | j d � Wd QRXd S)NrC r= i� � i �)rB r> r? r[ rK �
external_attr)r9 r rA 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> r� )r9 r r r r �test_writestr_permissions� s z3StoredTestsWithSourceFile.test_writestr_permissionsc C sq t j t d t j � � } | j t d � Wd QRXt j t d t j � � } | j | j � d g � Wd QRXd S)Nr< z /absoluterC �absolute)r> r? r r� r8 r rK rX )r9 r& r r r �test_absolute_arcnames� s z0StoredTestsWithSourceFile.test_absolute_arcnamesc C s� t j t d t j � � } | j t t � Wd QRXt j t d t j � �6 } | j d | j � | j | j � t d g � Wd QRXd S)z&Test appending to an existing zipfile.r< N�ar= )
r> r? r r� r8 r r@ r4 rK rX )r9 r& r r r �test_append_to_zip_file� s
z1StoredTestsWithSourceFile.test_append_to_zip_filec 4 C s� d d } t t d � � } | j | � Wd QRXt j t d t j � � } | j t t � Wd QRXt t d � �k } | j t | � � t j | d � �< } | j | j
� t g � | j | j t � | j � Wd QRXWd QRXt t d � �2 } | j | j t | � � | � | j � } Wd QRXt
j | � �U } t j | � �< } | j | j
� t g � | j | j t � | j � Wd QRXWd QRXd S)z9Test appending to an existing file that is not a zipfile.s I am not a ZipFile!rx r7 Nr� �rbrC )r% r r8 r> r? r� r �seekrQ rK rX rL r4 r r )r9 r4 r r& r� �bior r r �test_append_to_non_zip_file� s
* $z5StoredTestsWithSourceFile.test_append_to_non_zip_filec C s� t j � �E } t j | d t j � � } | j t t � Wd QRX| j � } Wd QRXd d } t t d � �! } | j | � | j | � Wd QRXt j t � �< } | j
| j � t g � | j
| j t � | j
� Wd QRXd S)Nr< s I am not a ZipFile!rx r7 )r r r> r? r� r8 r rO r% r rK rX rL r4 )r9 r� r& r� r4 r r r r � test_read_concatenated_zip_file� s
z9StoredTestsWithSourceFile.test_read_concatenated_zip_filec 5 C s� t j � �E } t j | d t j � � } | j t t � Wd QRX| j � } Wd QRXd d } t t d � �! } | j | � | j | � Wd QRXt j t d � �3 } | j
| j � t g � | j d | j
� Wd QRXt t d � �2 } | j
| j t | � � | � | j � } Wd QRXt j | � �t } t j | � �[ } | j
| j � t d g � | j
| j t � | j
� | j
| j d � | j
� Wd QRXWd QRXd S)Nr< s I am not a ZipFile!i@B r7 r� r= r� )r r r> r? r� r8 r rO r% r rK rX r@ r4 rL rQ )r9 r� r& r� r4 r r r r �$test_append_to_concatenated_zip_file� s$
$ z>StoredTestsWithSourceFile.test_append_to_concatenated_zip_filec C s� t j t d t j � � } | j t t � Wd QRXt t d � � } | j d � Wd QRXt j t d � � } | j | t j � Wd QRXd S)Nr< r� z
rC )r> r? r r� r8 r r% �assertIsInstance)r9 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 QRXt t d � � } | j d � Wd QRXt j t d � �- } | j | t j � | j | j d � Wd QRXd S)Nr< s this is a commentr� zabcdef
rC )
r> r? r r� �commentr8 r r% r� rK )r9 r& r r r r �)test_ignores_stuff_appended_past_comments� s zCStoredTestsWithSourceFile.test_ignores_stuff_appended_past_commentsc C se t j t d � �L } | j t � t t d � �& } | j | j t � | j � � Wd QRXWd QRXd S)z`Check that calling ZipFile.write without arcname specified
produces the expected result.r< r� N)r> r? r r8 r r% rK rL )r9 r& r r r r �test_write_default_name� s
z1StoredTestsWithSourceFile.test_write_default_namec C sh t j t d d �� } | j d d � Wd QRXt j t d d �� } | j t | j t � Wd QRXd S)z]Check that trying to call write() on a readonly ZipFile object
raises a RuntimeError.r� r< z somefile.txt�bogusNrC )r> r? r r@ r� � RuntimeErrorr8 r )r9 r& r r r �test_write_to_readonly� s z0StoredTestsWithSourceFile.test_write_to_readonlyc C sF t j t d � t j t d � � } | j t | j t � Wd QRXd S)Nr r< )r r ) �os�utimer r> r? r r� �
ValueErrorr8 )r9 r& r r r �test_add_file_before_1980� s z3StoredTestsWithSourceFile.test_add_file_before_1980)r� r� r� r> r� rA 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 QRXd S)zUCheck that files within a Zip archive can have different
compression options.r< ZstoremeZ deflatemeN)
r> r? r r8 r r� � ZIP_DEFLATEDr[ rK r� )r9 r& ZsinfoZdinfor r r �test_per_file_compression� s z4DeflateTestsWithSourceFile.test_per_file_compressionN)r� r� r� r> r� rA 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_BZIP2rA r r r r r� s r� c @ s e Z d Z e j Z d S)�LzmaTestsWithSourceFileN)r� r� r� r> �ZIP_LZMArA 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
)r0 r1 r3 r4 )r5 r2 r r r r6 s z(AbstractTestZip64InSmallFiles.setUpClassc C sW t j | _ t j | _ d t _ d t _ t t d � � } | j | j � Wd QRXd S)Ni� � r7 ) r> � ZIP64_LIMIT�_limit�ZIP_FILECOUNT_LIMIT�_filecount_limitr% r r8 r4 )r9 r: r r r r; s z#AbstractTestZip64InSmallFiles.setUpc C s t j | d | d d ��: } | j t d � | j t t � | j d | j � Wd QRXt 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 QRXd S)Nr< �
allowZip64Tz another.namer= rC rE z File Namer rF rG rH z%Y-%m-%dz%H:%M:%SrI c S s g | ] } | j � q Sr )rJ )r, r- r r r r. P s z:AbstractTestZip64InSmallFiles.zip_test.<locals>.<listcomp>)r> r? r8 r r@ r4 rK rL r rM rN rO rP rQ rR rS rT rU rV rW rX rY rZ r[ rJ r\ )r9 r rA r& r: r] r^ r' r_ r` r ra rb r- rc rd r r r re ( sN
z&AbstractTestZip64InSmallFiles.zip_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! re rA )r9 r r r r rf a 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� Tr{ � zfoo%08dz%drI �9 rC r* )
r> r? r rA �debugr0 r@ rK rQ rX �closerL �decode)r9 r� �numfilesr- �zipf2�contentr r r �test_too_many_filese 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 QRX| j t | j � � | � | j � t j t d | j d d �} d | _ | j t | j � � | � | j
t j � � | j d | d
� Wd QRX| 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 � qUW| j � d S)Nr< r� Fr{ r� zfoo%08dz%drI r� r/ r� Tr� rC r* )r> r? r rA r� r0 r@ rK rQ rX r� � LargeZipFiler� rL r� )r9 r� r� r- Z numfiles2r� r� r r r �test_too_many_files_appendx 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 )r9 r r r r� � s
z&AbstractTestZip64InSmallFiles.tearDownN) r� r� r� r� r6 r; re rf 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 sE t j | d | d d ��# } | j t j | j t d � Wd QRXd S)Nr< r� Fz another.name)r> r? r� r� r8 r )r9 r rA r& r r r �large_file_exception_test� s z5StoredTestZip64InSmallFiles.large_file_exception_testc C sH t j | d | d d ��& } | j t j | j d | j � Wd QRXd S)Nr< r� Fz another.name)r> r? r� r� r@ r4 )r9 r rA r& r r r �large_file_exception_test2� 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� )r9 r r r r �test_large_file_exception� s z5StoredTestZip64InSmallFiles.test_large_file_exceptionc C sw t j t d t j d d �� } | j t d � Wd QRXt j t d t j � � } | j | j � d g � Wd QRXd S)Nr< r� Tz /absoluterC r� )r> r? r r� r8 r rK rX )r9 r& r r r r� � s
z2StoredTestZip64InSmallFiles.test_absolute_arcnamesN)
r� r� r� r> r� rA 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� rA r r r r r� � s r� c @ s e Z d Z e j Z d S)�Bzip2TestZip64InSmallFilesN)r� r� r� r> r� rA r r r r r� � s r� c @ s e Z d Z e j Z d S)�LzmaTestZip64InSmallFilesN)r� r� r� r> r� rA 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 | � d S)N�o�c)rR )r9 �namerX r r r �assertCompiledIn� s z PyZipFileTests.assertCompiledInc C s� t j | t j d t j t j k �s4 | j d � 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�pathr3 r% �O_WRONLY�O_CREATr� � Exceptionr )r9 r� rJ �fdr r r �requiresWriteAccess� s
z"PyZipFileTests.requiresWriteAccessc C s� | j t j j t � � t � �� } t j | d � �� } t } | j d � r� | j t j
� } t j d k r� | j | j t j � � d | k r� t
j j | � } n | d d � } | j | � t j j | � } | j | | j � � | j | | j � � Wd QRXWd QRXt � �� } t j | d � �� } t } | j d � rY| d d � } | j | d � d d t j j | � f } | j | | j � � | j | | j � � Wd QRXWd QRXd S) Nr< z.pyc� __pycache__rH Z testpackagez%s/%srr rr )r� r� r� �dirname�__file__r r> � PyZipFile�endswithrS �sep�altsep�extend� importlib�util�source_from_cache�writepy�basename� assertNotInrX r� )r9 �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 QRXWd QRXd S)Nr r< zemail/__init__.pyzemail/mime/text.py) �emailr� r� r� r� r� r r> r� r� rX r� )r9 r �
packagedirr r& ra r r r �test_write_python_package s
!
z(PyZipFileTests.test_write_python_packagec . C sc d d l } t j j | j � } | j | � t � �%} t j | d � � } t � � } | j
| � Wd QRX| j � } | j d | k � t � � } | j
| d d d � �Wd QRX| j � } | j d | k � d d � } t � �1 } | j
t � � | j
| d | �Wd QRXWd QRX| j � } | r>t | � | j d | k � Wd QRXWd QRXd 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� rO rT � assertWarns� UserWarning�print)r9 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 } t � �j } t j | d d | ��H } | j
| � | j � } | j d | | � | j d | | � Wd QRXWd QRXd S)Nr rH z.pycr< �optimizezemail/__init__zemail/mime/text)
r r� r� r� r� r� � __debug__r r> r� r� rX rR )r9 r r ZoptlevelZextr r& ra r r r � test_write_with_optimization- s
z+PyZipFileTests.test_write_with_optimizationc , C s: t j t � zt t j j t d � d � � } | j d � Wd QRXt t j j t d � d � � } | j d � Wd QRXt t j j t d � d � � } | j d � Wd QRXt � �l } t j | d � �P } | j
t � | j � } | j d | � | j d | � | j
d | � Wd QRXWd QRXWd t t � Xd S)Nzmod1.pyr< z
print(42)
zmod2.pyzprint(42 * 42)
zmod2.txtz bla bla bla
)r� �mkdirr r% r� r3 r8 r r> r� r� rX r� r r
)r9 r: r r& ra 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 QRXt t j j t d � d � � } | j d � Wd QRXt � �h } t j | d � �L } | j
t d d d � �| j � } | j d | � | j
d | � Wd QRXWd QRXWd 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� r3 r8 r r> r� r� rX r� r r
)r9 r: r r& ra r r r �$test_write_python_directory_filteredS s
!!! "z3PyZipFileTests.test_write_python_directory_filteredc C sy t � �i } t j | d � �M } t t d � � } | j d � Wd QRX| j t | j t � t t � Wd QRXWd QRXd S)Nr< z!most definitely not a python file)
r r> r� r% r r8 r� r� r� r )r9 r r& r r r r �test_write_non_pyfileg s
!z$PyZipFileTests.test_write_non_pyfilec & C s� t j t � z� t t j j t d � d � � } | j d � Wd QRXt � �� } t j | d � �x } t
� �# } | j t j j t d � � Wd QRX| j d | j
� � | j � } | j d | � | j d | � Wd QRXWd QRXWd t t � Xd S)Nzmod1.pyr< zBad syntax in python file
r zmod1.pyc)r� r r r% r� r3 r8 r r> r� r r� rR rO rX r r
)r9 r: r r& �sra r r r � test_write_pyfile_bad_syntaxn 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 QRXt j t d � �� } x� t D]� \ } } | j | � } t j j t j
� | � } t j j | � } | j | | � t
| d � �# } | j | j � | j � � Wd QRXt | � qe WWd QRXt t j j t j
� d � � d S)Nr< rC r� � ziptest2dir)r> r? r r� �SMALL_TEST_DATAr@ �extractr� r� r3 �getcwd�normpathrK r% �encoderL r r
)r9 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 QRXt j t d � �� } | j � xn t D]f \ } } t j j t j
� | � } t | d � �# } | j | j
� | j � � Wd QRXt | � qo WWd QRXt t j j t j
� d � � d S)Nr< rC r� r )r> r? r r� r r@ �
extractallr� r� r3 r r% rK r rL r r
)r9 r& r r! Zoutfiler r r r �test_extract_all� s
#z ExtractTests.test_extract_allc
C sL | j t j j | � � t | d � � } | j | j � | � Wd QRXd S)Nr� )rT r� r� �isfiler% rK rL )r9 rJ r� r r r r �
check_file� 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_namerK )r9 Zsanr r r �test_sanitize_windows_name� 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- )r0 r- )r1 r- )r2 r- )r3 r- )r4 r- )� _test_extract_hackers_arcnames)r9 Zcommon_hacknamesr r r �*test_extract_hackers_arcnames_common_cases� 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)r8 r9 )r: r9 )r; r9 )r< r9 )r= r9 )r> r9 )r? r9 )r@ r9 )rA r9 )rB r9 )rC rD )rE rD )rF rD )rG rD )rH r9 )rI r9 )rJ rK )rL rM )rN rO )r5 )r9 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)rQ rR )rS rT )rU rU )r5 )r9 Zposix_hacknamesr r r �(test_extract_hackers_arcnames_posix_only� s z5ExtractTests.test_extract_hackers_arcnames_posix_onlyc
' C s[ xT| D]L\ } } d | j � } t j t d t j � �5 } t j � } | | _ d | _ | j | | � Wd QRX| 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 QRX| j | | � t d � t j t d � � } | j | � Wd QRX| j | | � t d � t
j j
t
j � | j d � � } t j t d � �0 } | j | � } | j | | d
d | �Wd QRX| j | | � t | j d � d
� t j t d � � } | j � Wd QRX| j | | � t | j d � d
� t t � q Wd S)Ns foobarr< i� r� r* �targetZsubdirZsubsubrC �msgzextract %r: %r != %rz
extract %rr i �)r r> r? r r� �ZipInforJ r� r@ �replacer� r� r� r3 rS r rK r( r
r% r r )
r9 Z hacknamesZarcnameZ fixednamer� r& r� Z
targetpathr# r" r r r r5 � sB
$z+ExtractTests._test_extract_hackers_arcnamesN)r� r� r� r$ r& r( r+ r6 �unittestZskipIfr� r� r� rP rV r5 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 dF � Z& dG S)H�
OtherTestsc C s8 t j t d t j � �[ } | j d d � | j t � � | j d d � Wd QRX| j | j � d g d � Wd QRXt j t d � �� } | j � } d } x7 | D]/ } | j
| � � } | | j � 7} Wd QRXq� W| j | d d h � d } x! | D] } | | j | � 7} q� W| j | d d h � Wd QRXd S)
Nr< r� r� Zbar� rC r/ s foobars barfoo)
r> r? r r� r@ r r
rK rX rY r% rL rR )r9 r& rb r4 rd rt 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 QRXt j | d � �K } xA d D]9 } | j t � � | j d | � } Wd QRX| j � qT WWd QRXd S)Nr< zspam.txts
ababagalamagarC �Ur# )r_ r# ) r r r> r? r@ r r$ r% r� )r9 r r& r� rt 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 QRXd } t j | d � �; } t | t � �" } x | D] } | | 7} q� WWd QRXWd QRX| j | | j d d � � d S) Ns a
r� i r< r/ rC s
s
)
r r r> r? r� r@ r r( rK rZ )r9 r r4 r& �data2rt r r r r �test_universal_readaheads8 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 QRXd S)Nr< Zabcdefghijklmnopr )r> r? r rY Z flag_bitsr@ )r9 Zorig_zipr4 r� r r r �-test_writestr_extended_local_header_issue1202G s
z8OtherTests.test_writestr_extended_local_header_issue1202c C s� t j t d � �A } x7 t D]/ \ } } | j | | � | j | j d � q WWd QRX| j | j d � t j t d � � } | j | j d � Wd QRX| j | j d � d S)z8Check that the zipfile is closed after the 'with' block.r< zzipfp is not openNzzipfp is not closedrC )r> r? r r r@ ZassertIsNotNoner: � assertIsNone)r9 r& r r! r r r �
test_closeN s zOtherTests.test_closec C s� t j t d � �. } x$ t D] \ } } | j | | � q WWd QRXy, t j t d � � } t j � � Wd QRXWn( 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< NrC zzipfp is not closed)r> r? r r r@ �
BadZipFilerd r: )r9 r& r r! Zzipfp2r r r �test_close_on_exceptionZ 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 / rC )r� �NotImplementedErrorr> r? r r )r9 r4 r r r �test_unsupported_versiong s z#OtherTests.test_unsupported_versionc
C sS t d � } t j | � �1 } x' | j � D] } | j | � j � q+ WWd QRXd S)Nzzip_cp437_header.zip)r r> r? rX r% r� )r9 r� r& r� r r r �test_read_unicode_filenamesr 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 QRXt j t d � �; } | j | j d j d � | j | j d j d � Wd QRXd S)Nr< zfoo.txtzTest for unicode filenameu ö.txtr rC rH )
r> r? r r@ r� rY rJ rW rK �filelist)r9 �zfr r r � test_write_unicode_filenamesz s $z'OtherTests.test_write_unicode_filenamesc C s� t t � d } d } t j t d t j � � } | j | | � Wd QRX| j t � � t j t d t j � Wd QRXt j t d � �9 } | j | j � | g � | j | j
| � | � Wd QRXd S)z&Test exclusive creating a new zipfile.z testfile.txts# hello, world. this is some content.�xNrC ) r r r> r? r� r@ r� �FileExistsErrorrK rX rL )r9 rJ r� r& r r r � test_exclusive_create_zip_file� s
z)OtherTests.test_exclusive_create_zip_filec C s� t j j t � r t j t � d } d } y0 t j t d � � } | j | | � Wd QRXWn t k
r| | j d � Yn X| j
t j j t � � t j t d � � } | j | j | � | � Wd QRXd S)Nz testfile.txts# hello, world. this is some content.r� z1Could not append data to a non-existent zip file.rC )
r� r� �existsr r r> r? r@ �OSError�failrT rK rL )r9 rJ r� rl r r r �(test_create_non_existent_file_for_append� s
z3OtherTests.test_create_non_existent_file_for_appendc C sU t t d � � } | j d � Wd QRXy t j t � } Wn t j k
rP Yn Xd S)Nr< z this is not a legal zip file
)r% r r8 r> r? rf )r9 r: rl r r r �test_close_erroneous_file� s z$OtherTests.test_close_erroneous_filec C s� t t d � � } | j d � Wd QRX| j t j t � � t t d � � } | j t j | � � Wd QRXt 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 r8 r r> �
is_zipfiler r r� )r9 r: 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 QRX| j � } xL t t | � � D]8 } t j | d | � � } | j t j t j | � qZ 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@ rO r0 rQ r� rf )r9 r: r� r� �Nr r r �test_damaged_zipfile� s z OtherTests.test_damaged_zipfilec C s� t j t d d �� } | j d d � Wd QRX| j t j t � � t t d � �9 } | j t j | � � | j d d � | j � } Wd QRXt 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@ rT rv r% r� rL r r r8 )r9 r� r: 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� rr r> r? r )r9 r r r �%test_non_existent_file_raises_OSError� s z0OtherTests.test_non_existent_file_raises_OSErrorc C su t t d � } | j � | j t j t j t � t t d � � } | j d � Wd QRX| j t j t j t � d S)Nr< z
short file)r% r r� r� r> rf r? r8 )r9 r r: 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 QRX| j t | j d � | j t | j d � | j t | j � | j t | j d d � t t
d � � } | j d � Wd QRX| 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� rL r% r\ r r8 )r9 r4 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 )r9 r r r �test_bad_constructor_mode s z$OtherTests.test_bad_constructor_modec C sx t j t d d �� } | j d d � Wd QRXt j t d d ��- } | j d � | j t | j d d � Wd QRXd S)z7Check that bad modes passed to ZipFile.open are caught.r� r< zfoo.txtzO, for a Muse of Fire!NrC r~ )r> r? r r@ rL r� r� r% )r9 r� r r r �test_bad_open_mode s
z OtherTests.test_bad_open_modec C s� t j t d d ��v } | j d d � | j d � �M } x- t t � D] } | j | j d � d � qG W| j | j � d � Wd QRXWd QRXd 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% r0 r1 rK rL )r9 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 QRXd S)zqCheck that attempting to call open() for an item that doesn't
exist in the archive raises a RuntimeError.r� r< zfoo.txtrC N)r> r? r r� �KeyErrorr% )r9 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< rH Nrr )r� r� r> r? r )r9 r r r �test_bad_compression_mode, s z$OtherTests.test_bad_compression_modec C sE d } t j t j | � d � � } | j t | j d � Wd QRXd S)Nsf PK. �C�@ x PK.. �C�@ � xPK / ! rC rn )r> r? r r r� rh r% )r9 r4 r� r r r � test_unsupported_compression1 s z'OtherTests.test_unsupported_compressionc
C sL t j t d d ��0 } | j d d � | j | j � d g � Wd QRXd 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@ rK rX )r9 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)rK r> ZsizeEndCentDirZsizeCentralDirZsizeEndCentDir64ZsizeEndCentDir64Locator)r9 r r r �test_struct_sizesC s z OtherTests.test_struct_sizesc l C s, t j t d d ��* } | j | j d � | j d d � Wd QRXt j t d d �� } | j | j d � Wd QRXd } t j t d d �� } | | _ | j d d � Wd QRXt j t d d �� } | j | j | � Wd QRXd j d
d � t d � D� � } | j d � } t j t d d �� } | | _ | j d d � Wd QRXt j t d d �� } | j | j | � Wd QRXt j t d d ��; } | j t
� � | d | _ Wd QRX| j d d � Wd QRXt j t d d �� } | j | j | � Wd QRXt j t d d �� } d | _ | j d d � Wd QRXt j t d d �� } d | _ Wd QRXt j t d d �� } | j | j d � Wd QRXt j t d d �� } d | _ | j d d � Wd QRXt j t d d �� } d | _ Wd QRXt j t d d �� } | j | j d � Wd QRXd S)z8Check that comments on the archive are handled properly.r� r< r/ zfoo.txtzO, for a Muse of Fire!NrC s9 Bravely taking to his feet, he beat a very brave retreat.r" c S s" g | ] } d | d d � q S)z%drI rx r )r, r- r r r r. ^ s z,OtherTests.test_comments.<locals>.<listcomp>rH r� r* s oopss original commentr� s an updated comments original comment that's longers shorter commenti i�� ) r> r? r rK r� r@ r3 r0 r r r
)r9 r� Zzipfrr� Zcomment2r r r �
test_commentsJ sL " zOtherTests.test_commentsc C sV t j t d t j � �7 } | j d d � | j t � � d | _ Wd QRXWd QRXd S)Nr< zfoo.txtzO, for a Muse of Fire!zthis is an error)r> r? r r� r@ r� � TypeErrorr� )r9 r� r r r �test_unicode_comment� s z OtherTests.test_unicode_commentc C sn t j t d t j � � } | j | j � d | _ Wd QRXt j t d � � } | j | j d � Wd QRXd S)Nr� s this is a commentrC )r> r? r r� r rk r� rK )r9 r� r r r �$test_change_comment_in_empty_archive� s
z/OtherTests.test_change_comment_in_empty_archivec C s� t j t d t j � � } | j d d � Wd QRXt j t d t j � � } | j | j � d | _ Wd QRXt j t d � � } | j | j d � Wd QRXd S)Nr< zfoo.txtzO, for a Muse of Fire!r� s this is a commentrC ) r> r? r r� r@ rT rk r� rK )r9 r� r r r �'test_change_comment_in_nonempty_archive� 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< rC z+Unable to create empty ZIP file in 'w' moder� z+Unable to create empty ZIP file in 'a' mode)r> r? r r� rf rs )r9 r� r r r �test_empty_zipfile� s
z OtherTests.test_empty_zipfilec C s9 t t d � } | j � | j t j t j t d � d S)Nr< rC )r% r r� r� r> rf r? )r9 r r r r �test_open_empty_file� s
z OtherTests.test_open_empty_filec
C s | j t t j d d � d S)NZ seventies� rH r )r� rH rH r r r )r� r� r> rY )r9 r r r � test_create_zipinfo_before_1980� s z*OtherTests.test_create_zipinfo_before_1980c C sB d } t j t j | � d � � } | j | j � � Wd QRXd S)z>If an extra field in the header is less than 4 bytes, skip it.sn PK ���@���� abc APK ���@���� �� abc PK 3 % rC N)r> r? r r rd r\ )r9 ru 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 )r9 r r r r� � s
zOtherTests.tearDownN)'r� r� r� r^ r` rb rc re rg ri r rj rm rp rt ru rw ry rz 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\ sF
7 r\ c @ s( e Z d Z d d � Z d d � Z d S)�AbstractBadCrcTestsc
C sK | j } t j t j | � d d �� } | j d | j � � Wd QRXd S)z>Tests that files with bad CRCs return their name from testzip.r� rC �afileN)�zip_with_bad_crcr> r? r r rK r\ )r9 ru r� r r r �test_testzip_with_bad_crc� s !z-AbstractBadCrcTests.test_testzip_with_bad_crcc , C s% | j } t j t j | � d d �� } | j t j | j d � Wd QRXt j t j | � d d ��9 } | j d d � � } | j t j | j � Wd QRXWd QRXt j t j | � d d ��\ } | j d d � �@ } d | _ | j t j � � x | j d � r q� WWd QRXWd QRXWd QRXd S)zFTests that files with bad CRCs raise a BadZipFile exception when read.r� rC r� Nr] )
r� r> r? r r r� rf rL r% Z
MIN_READ_SIZE)r9 ru 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 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� rA 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� rA 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� rA 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� rA 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 QRXt j t d � | _ t t d � � } | j | j � Wd QRXt j t d � | _ d S)Nr7 rC )
r% r r8 r4 r> r? r� r ra �zip2)r9 r: r r r r; 9 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 )r9 r r r r� A s
zDecryptionTests.tearDownc C s6 | j t | j j d � | j t | j j d � d S)Nztest.txt�zero)r� r� r� rL r� )r9 r r r �test_no_passwordG 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� rL r� )r9 r r r �test_bad_passwordM 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� rK rL �plainr� �plain2)r9 r r r �test_good_passwordS 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� rL r% r )r9 r r r �test_unicode_passwordZ s z%DecryptionTests.test_unicode_passwordN)r� r� r� �__doc__r4 ra r� r� r; 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 rH 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� d s z@AbstractTestsWithRandomBinaryFiles.setUpClass.<locals>.<genexpr>)r r3 r0 r4 )r5 Z datacountr r r r6 a s z-AbstractTestsWithRandomBinaryFiles.setUpClassc C s- t t d � � } | j | j � Wd QRXd S)Nr7 )r% r r8 r4 )r9 r: r r r r; g s z(AbstractTestsWithRandomBinaryFiles.setUpc C s t t � t t � d S)N)r r r )r9 r r r r� l s
z+AbstractTestsWithRandomBinaryFiles.tearDownc
C sC t j | d | � �' } | j t d � | j t t � Wd QRXd S)Nr< z another.name)r> r? r8 r )r9 r rA r& r r r rB p 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 QRXd S)NrC z another.name)rB r> r? rL r rK rQ r4 )r9 r rA r& �testdatar r r re v s z+AbstractTestsWithRandomBinaryFiles.zip_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! re rA )r9 r r r r � test_read� s z,AbstractTestsWithRandomBinaryFiles.test_readc C sG | j | | � t j | d | � �} g } | j t � �1 } x' | j d � } | sY P| j | � qC WWd QRXg } | j d � �1 } x' | j d � } | s� P| j | � q� WWd QRXd j | � } | j t | � t | j
� � | j | | j
� d j | � }
| j t |
� t | j
� � | j |
| j
� Wd QRXd S)NrC rg z another.namer/ ) rB r> r? r% r rL rh r3 rK rQ r4 ) r9 r rA r& ri rj rk rl rm Z testdata1Z testdata2r r r rn � s, z0AbstractTestsWithRandomBinaryFiles.zip_open_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! rn rA )r9 r r r r ro � s z,AbstractTestsWithRandomBinaryFiles.test_openc C s� | j | | � t j | d | � �� } g } | j t � �: } x0 | j t d d � � } | sb P| j | � qC WWd QRXd j | � } | j t
| � t
| j � � | j | | j � Wd QRXd S)NrC rH i r/ ) rB r> r? r% r rL r rh r3 rK rQ r4 )r9 r rA r& ri rj rk r� r r r rp � s z7AbstractTestsWithRandomBinaryFiles.zip_random_open_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! rp rA )r9 r r r r rq � s z3AbstractTestsWithRandomBinaryFiles.test_random_openN)r� r� r� r� r6 r; r� rB re r� rn ro rp rq r r r r r� ` s
r� c @ s e Z d Z e j Z d S)� StoredTestsWithRandomBinaryFilesN)r� r� r� r> r� rA r r r r r� � s r� c @ s e Z d Z e j Z d S)�!DeflateTestsWithRandomBinaryFilesN)r� r� r� r> r� rA r r r r r� � s r� c @ s e Z d Z e j Z d S)� Bzip2TestsWithRandomBinaryFilesN)r� r� r� r> r� rA r r r r r� � s r� c @ s e Z d Z e j Z d S)� LzmaTestsWithRandomBinaryFilesN)r� r� r� r> r� rA 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 S)
�Tellablec C s | | _ d | _ d S)Nr )r: �offset)r9 r: r r r �__init__� s zTellable.__init__c C s% | j j | � } | j | 7_ | S)N)r: r8 r� )r9 r4 �nr r r r8 � s zTellable.writec C s | j S)N)r� )r9 r r r r� � s z
Tellable.tellc C s | j j � d S)N)r: �flush)r9 r r r r� � s zTellable.flushN)r� r� r� r� r8 r� r� r r r r r� � s r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�
Unseekablec C s
| | _ d S)N)r: )r9 r: r r r r� � s zUnseekable.__init__c C s | j j | � S)N)r: r8 )r9 r4 r r r r8 � s zUnseekable.writec C s | j j � d S)N)r: r� )r9 r r r r� � s zUnseekable.flushN)r� r� r� r� r8 r� r r r r r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)�UnseekableTestsc ' C sK xDd d � t t f D]-} | j d | � �t j � } | j d � t j | � } t j | | � d t j � �' } | j
d d � | j
d d � Wd QRX| j | j � d d
� d � t j | d d
��e } | j
d � � } | j | j � d � Wd QRX| j
d � � } | j | j � d � Wd QRXWd QRXWd QRXq Wd S)Nc S s | S)Nr )r r r r r � s z/UnseekableTests.test_writestr.<locals>.<lambda>�wrappers abcr< �oness 111�twoss 222� s abcPKr� rC )r� r� �subTestr r r8 �BufferedWriterr> r? r� r@ rK rO r% rL )r9 r� r �bfr& r� �zopenr r r �
test_writestr� s
! z UnseekableTests.test_writestrc 5 C s� x�d d � t t f D]�} | j d | � �nt j � } | j d � t j | � } t j | | � d t j � �� } | j
t t � t
t d � � } | j d � Wd QRX| j t d � t
t d � � } | j d � Wd QRX| j t d
� Wd QRX| j | j � d d � d � t j | d
d ��e } | j
d � � } | j | j � d � Wd QRX| j
d
� � } | j | j � d � Wd QRXWd QRXWd QRXq Wd S)Nc S s | S)Nr )r r r r r � s z,UnseekableTests.test_write.<locals>.<lambda>r� s abcr< r7 s 111r� s 222r� r� s abcPKr� rC )r� r� r� r r r8 r� r> r? r� �
addCleanupr r r% rK rO rL )r9 r� r r� r& �f2r� r� r r r �
test_write� s&
! zUnseekableTests.test_writeN)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)�TestsWithMultipleOpensc C s* d t d � | _ d t d � | _ d S)Ns 111i' s 222)r �data1ra )r5 r r r r6 s z!TestsWithMultipleOpens.setUpClassc
C sL t j | d t j � �- } | j d | j � | j d | j � Wd QRXd S)Nr< r� r� )r> r? r� r@ r� ra )r9 r r& r r r rB s z(TestsWithMultipleOpens.make_test_archivec C s� x� t | � D]� } | j | � t j | d d ��� } | j d � �^ } | j d � �E } | j d � } | j d � } | | j � 7} | | j � 7} Wd QRXWd QRX| j | | � | j | | j � Wd QRXq
Wd S)Nr� rC r� i� )r! rB r> r? r% rL rK r� )r9 r r� �zopen1�zopen2r� ra r r r �test_same_file s
$ z%TestsWithMultipleOpens.test_same_filec C s� x� t | � D]� } | j | � t j | d d ��� } | j d � �^ } | j d � �E } | j d � } | j d � } | | j � 7} | | j � 7} Wd QRXWd QRX| j | | j � | j | | j � Wd QRXq
Wd S)Nr� rC r� r� i� ) r! rB r> r? r% rL rK r� ra )r9 r r� r� r� r� ra r r r �test_different_file, s
$ z*TestsWithMultipleOpens.test_different_filec C s� x� t | � D]� } | j | � t j | d d ��� } | j d � �^ } | j d � } | j d � �6 } | j d � } | | j � 7} | | j � 7} Wd QRXWd QRX| j | | j � | j | | j � Wd QRXq
Wd S)Nr� rC r� i� r� ) r! rB r> r? r% rL rK r� ra )r9 r r� r� r� r� ra r r r �test_interleaved: s
z'TestsWithMultipleOpens.test_interleavedc C s� x� t | � D]� } | j | � t j � �� } t j | d � �7 } | j | j d � � } | j | j d � � } Wd QRX| j d � } | j d � } | | j � 7} | | j � 7} Wd QRX| j | | j
� | j | | j � q
Wd S)NrC r� r� i� ) r! rB �
contextlib� ExitStackr> r? �
enter_contextr% rL rK r� ra )r9 r �stackr� r� r� r� ra r r r �test_read_after_closeI s
z,TestsWithMultipleOpens.test_read_after_closec C s xt | � D]� } t j | d t j � �U } | j d | j � | j d | j � | j d � � } | j d � } Wd QRXWd QRX| j | | j d d � � t j | d � �% } | j d � } | j d � } Wd QRX| j | | j � | j | | j � q
Wd S)Nr< r� r� i� rC )
r! r> r? r� r@ r� ra r% rL rK )r9 r r� r� r� ra r r r �test_read_after_writeW s z,TestsWithMultipleOpens.test_read_after_writec C s� x� t | � D]� } t j | d t j � �S } | j d | j � | j d � �' } | j d � | j d | j � Wd QRXWd QRXt j | d � �% } | j d � } | j d � } Wd QRX| j | | j � | j | | j � q
Wd S)Nr< r� i� r� rC )
r! r> r? r� r@ r� r% rL ra rK )r9 r r� r� r� ra r r r �test_write_after_reade s
!z,TestsWithMultipleOpens.test_write_after_readc C s� | j t � t j t d d ��D } x: t d � D], } | j d � | j d � � } Wd QRXq2 WWd QRXt t j � � } | j | j
� d � Wd QRXd S)Nr� rC r{ r� ) rB r r> r? r0 rL r% r� �devnullZ
assertLess�fileno)r9 r� rn r� r r r r �test_many_opensr s
z&TestsWithMultipleOpens.test_many_opensc C s t t � d S)N)r r )r9 r r r r� ~ s z TestsWithMultipleOpens.tearDownN)r� r� r� r� r6 rB r� r� r� r� r� r� r� r� r r r r r� 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 )r9 r r r r; � s zTestWithDirectory.setUpc C s� t j t d � � � } | j t � Wd QRX| 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 rT r� r� �isdirr3 rq )r9 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� r3 r r� )r9 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 QRXt 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 QRXd S)
Nrn i�� r< r z/x/r� �yrH zy/rC rW r] )r� r� r3 r r �stat�st_moder> r? r r8 rk rT rJ r� rK r� r% r� rQ �listdir)r9 �dirpathr� r� r� rW r r r �test_write_dir� s0
"
(z TestWithDirectory.test_write_dirc C sG t j t j j t d � � t j t d � �J } | j d d � | j d } | j
| j d � | j
| j d � Wd QRXt 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 QRXd S)Nrn r< zx/r/ r i�A r� rC rW i �Ai �Ai �Ai �A)r� r r� r3 r r> r? r r@ rk rK rJ r� rT r� r% r� r� )r9 r� r� rW r r r �test_writestr_dir� s
(z#TestWithDirectory.test_writestr_dirc C s* t t � t j j t � r& t t � d S)N)r
r r� r� rq r r )r9 r r r r� � s
zTestWithDirectory.tearDownN) r� r� r� 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 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� )r0 r1 r2 �seps�arcdata� enumerater3 )r5 r� r r r r r6 � s z(AbstractUniversalNewlineTests.setUpClassc C sx i | _ xh t | j � D]W \ } } d t | f | j | <t | j | d � � } | j | j | � Wd QRXq Wd S)Nz%s-%dr7 )�arcfilesr� r� r r% r8 r� )r9 r� r r r r r r; � s
z#AbstractUniversalNewlineTests.setUpc C sM t j | d | � �1 } x' | j j � D] } | j | | � q( WWd QRXd S)Nr< )r> r? r� �valuesr8 )r9 r rA r& r' r r r rB � s z/AbstractUniversalNewlineTests.make_test_archivec C s� | j | | � t j | d � �c } xY | j j � D]H \ } } t | | � � } | j � } Wd QRX| j | j | | � q5 WWd QRXd S)NrC ) rB r> r? r� �itemsr( rL rK r� )r9 r rA r& r� r' r: ru r r r � read_test� s z'AbstractUniversalNewlineTests.read_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! r� rA )r9 r r r r r� � s z'AbstractUniversalNewlineTests.test_readc C s� | j | | � t j | d � �� } x� | j j � D]o \ } } t | | � �Q } d } xA | j � } | so P| | 7} | j d � } | s� P| | 7} q\ WWd QRXq5 W| j | | j d � Wd QRXd S)NrC r/ r� s
)
rB r> r? r� r� r( r| rL rK r� ) r9 r rA r& r� r' rt r4 rL r r r �readline_read_test� s
z0AbstractUniversalNewlineTests.readline_read_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! r� rA )r9 r r r r r~ � s z0AbstractUniversalNewlineTests.test_readline_readc C s� | j | | � t j | d � �t } xj | j j � D]Y \ } } t | | � �; } x1 | j D]& } | j � } | j | | d � q] WWd QRXq5 WWd QRXd S)NrC s
) rB r> r? r� r� r( r2 r| rK ) r9 r rA r& r� r' rt r r� r r r �
readline_test s z+AbstractUniversalNewlineTests.readline_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! r� rA )r9 r r r r r� s z+AbstractUniversalNewlineTests.test_readlinec
C s� | j | | � t j | d � �� } xy | j j � D]h \ } } t | | � � } | j � } Wd QRXx4 t | j | � D] \ } } | j | | d � qy Wq5 WWd QRXd S)NrC s
)
rB r> r? r� r� r( r� r� r2 rK )
r9 r rA r& r� r' r: r� r r� r r r �readlines_test s z,AbstractUniversalNewlineTests.readlines_testc C s. x' t | � D] } | j | | j � q
Wd S)N)r! r� rA )r9 r r r r r� s z,AbstractUniversalNewlineTests.test_readlinesc C s� | j | | � t j | d � �w } xm | j j � D]\ \ } } t | | � �> } x4 t | j | � D] \ } } | j | | d � qf WWd QRXq5 WWd QRXd S)NrC s
) rB r> r? r� r� r( r� r2 rK ) r9 r rA r& r� r' r: 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� rA )r9 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 )r9 r� r' r r r r� - s
z&AbstractUniversalNewlineTests.tearDownN)r� r� r� r� r6 r; rB 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� rA r r r r r� 4 s r� c @ s e Z d Z e j Z d S)� DeflateUniversalNewlineTestsN)r� r� r� r> r� rA r r r r r� 8 s r� c @ s e Z d Z e j Z d S)�Bzip2UniversalNewlineTestsN)r� r� r� r> r� rA r r r r r� = s r� c @ s e Z d Z e j Z d S)�LzmaUniversalNewlineTestsN)r� r� r� r> r� rA r r r r r� B s r� c @ s^ e Z d Z d d � Z d d � Z d d � Z d d � Z e d d
� � Z d d � Z d
S)�CommandLineTestc O s: t j d d | | � \ } } } | j t j j � d � S)Nz-mr> s
)r Zassert_python_okrZ r� �linesepr )r9 �args�kwargs�rc�out�errr r r �
zipfilecmdJ s zCommandLineTest.zipfilecmdc G s t j d d | � S)Nz-mr> )r Zassert_python_failure)r9 r� r r r �zipfilecmd_failureO s z"CommandLineTest.zipfilecmd_failurec C so t d � } | j d | � } | j | j � d � t d � } | j d | � \ } } } | j | d � d S)Nz
zipdir.zipz-ts Done testingz testtar.tarr/ )r r� rK �rstripr� )r9 �zip_namer� r� r� r r r �test_test_commandR s z!CommandLineTest.test_test_commandc C s� t d � } t j � } t j | d � � } | j | � Wd QRX| j � j d d � } | j d | d d �} | j | | � d S)Nz
zipdir.ziprC r* �backslashreplacez-lZPYTHONIOENCODINGzascii:backslashreplace)
r r rM r> r? rN rO r r� rK )r9 r r ZtfZexpectedr� r r r �test_list_commandZ s z!CommandLineTest.test_list_commandc C sO | j t t � t t d � � } | j d � Wd QRXt j t � | j t t � t t j j
t d � d � � } | j d � Wd QRXt t g } t t d t d g } z� | j d t | � } | j
| d � t j t � �W } | j
| j � | � | j
| j | d � d
� | j
| j | d � d � Wd QRXWd t t � Xd S)
Nr< ztest 1zfile.txtztest 2r* z /file.txtz-cr/ r s test 1r] s test 2)r� r r r% r8 r� r � TESTFNDIRr
r� r3 r� r rK r> r? rX rL )r9 r �filesrX r� rl r r r �test_create_commandd s"
! (z#CommandLineTest.test_create_commandc C s t d � } t � �} | j d | | � } | j | d � t j | � �� } x� | j � D]� } t j j | | j
j d t j � � } | j
j
d � r� | j t j j | � � q\ | j t j j | � � t | d � �& } | j | j � | j | � � Wd QRXq\ WWd QRXWd QRXd S)Nz
zipdir.zipz-er/ r* r� )r r r� rK r> r? rY r� r� r3 rJ rZ r� r� rT r� r' r% rL )r9 r Zextdirr� rl Zzir� r r r r �test_extract_commandy s z$CommandLineTest.test_extract_commandN)
r� r� r� r� r� r r r r r r r r r r� H s
r� �__main__)Ir� r r� �sys�importlib.utilr� rU r� r> r[ Ztempfiler r r r Z test.supportr r r r r
r r r
r r r r r r1 Z
DATAFILES_DIRr r r! r( r) ZTestCaser� 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� r� r� r� r� �mainr r r r �<module> s� F
� E �� ��� � A[
'pAq A