Nexus File Manager
v2.0
🏠
Dashboard
⚡
System
🌐
›
opt
›
alt
›
python38
›
lib64
›
python3.8
›
test
›
__pycache__
Quick:
⬆️ Parent
🌐 Root
🏠 Home
🌍 WWW
📁 Temp
⚙️ Etc
📤 Upload
📁 Create
⚡ WordPress Admin
🔄 Refresh
✏️ test_zipfile.cpython-38.opt-1.pyc
← Back
U i�f1� � @ sN d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl Z d dl Z d dl Z d dl Z d dlmZ d dlZd dlmZ d dlmZmZmZ d dlmZ d dlmZmZmZmZm Z m Z m Z m Z m Z m!Z! ed Z"ed Z#dZ$d Z%d d d d gZ&dd� Z'dd� Z(G dd� d�Z)G dd� de)e j*�Z+e G dd� de)e j*��Z,e G dd� de)e j*��Z-e G dd� de)e j*��Z.G d d � d �Z/G d d � d e/e j*�Z0e G d d!� d!e/e j*��Z1e G d"d#� d#e/e j*��Z2e G d$d%� d%e/e j*��Z3G d&d'� d'�Z4G d(d)� d)e4e j*�Z5e G d*d+� d+e4e j*��Z6e G d,d-� d-e4e j*��Z7e G d.d/� d/e4e j*��Z8G d0d1� d1e j*�Z9G d2d3� d3e j*�Z:G d4d5� d5e j*�Z;G d6d7� d7�Z<G d8d9� d9e<e j*�Z=e G d:d;� d;e<e j*��Z>e G d<d=� d=e<e j*��Z?e G d>d?� d?e<e j*��Z@G d@dA� dAe j*�ZAG dBdC� dC�ZBG dDdE� dEeBe j*�ZCe G dFdG� dGeBe j*��ZDe G dHdI� dIeBe j*��ZEe G dJdK� dKeBe j*��ZFG dLdM� dM�ZGG dNdO� dO�ZHG dPdQ� dQe j*�ZIe G dRdS� dSe j*��ZJG dTdU� dUe j*�ZKG dVdW� dWe j*�ZLG dXdY� dYe j*�ZMG dZd[� d[e j*�ZNeOZPG d\d]� d]�ZQd^d_� ZRd`da� ZSG dbdc� dce j*�ZTG ddde� deejU�ZVeWdfk�rJe �X� dS )g� N)� TemporaryFile)�randint�random� getrandbits)� script_helper) �TESTFN�findfile�unlink�rmtree�temp_dir�temp_cwd� requires_zlib� requires_bz2� requires_lzma�captured_stdout�2�d�� Zzipfile_datafiles)Z _ziptest1Z 1q2w3e4r5t)zziptest2dir/_ziptest2Z qawsedrftg)z!ziptest2dir/ziptest3dir/_ziptest3Z azsxdcfvgb)z-ziptest2dir/ziptest3dir/ziptest4dir/_ziptest3Z 6y7u8i9o0pc C s t d| ��| d�S )N� �little)r �to_bytes)�size� r �6/opt/alt/python38/lib64/python3.8/test/test_zipfile.py� getrandbytes$ s r c c sT t V t� �}|V | �|j� W 5 Q R X t�� �}|V | �|j� W 5 Q R X d S �N)�TESTFN2r � assertFalse�closed�io�BytesIO)�test�fr r r � get_files' s r# c @ s e Zd Zedd� �Zdd� ZdBdd�ZdCdd �Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dS )D�AbstractTestsWithSourceFilec C s& dd� t t�D �| _d�| j�| _d S )Nc S s g | ]}t d |t� f d��qS )z'Zipfile test line %d. random float: %f �ascii)�bytesr ��.0�ir r r � <listcomp>3 s ���z:AbstractTestsWithSourceFile.setUpClass.<locals>.<listcomp>� )�range�FIXEDTEST_SIZE�line_gen�join�data��clsr r r � setUpClass1 s �z&AbstractTestsWithSourceFile.setUpClassc C s& t td��}|�| j� W 5 Q R X d S �N�wb��openr �writer0 ��self�fpr r r �setUp8 s z!AbstractTestsWithSourceFile.setUpNc C s� ||d�}t j|df|��\}|�td� |�tt� |�d| j� |jddd�� }| jD ]}|�|� qXW 5 Q R X W 5 Q R X d S )N)� compression� compresslevel�w� another.name�strfile�written-open-w��mode)�zipfile�ZipFiler8 r �writestrr0 r7 r. )r: r" r= r> �kwargs�zipfp�liner r r �make_test_archive= s z-AbstractTestsWithSourceFile.make_test_archivec C s. | � |||� t�|d|���}| �|�t�| j� | �|�d�| j� | �|�d�| j� t�� }|j |d� |� � }|� � }| �t |�d� | � d|d � | � d|d � | � d |d � |d �� \}} } } | �|d� | �t�| d �� | �t�| d �� | �| tt | j��� |�� } | �t | �d � | � t| � | � d| � | � d| � | � d| � |�� } dd� | D �} | �t | �d � | � t| � | � d| � | � d| � | � d| � | D ]}| �|jt | j�� �q�tdddfD ]2}|�|�}| �|j|� | �|jt | j�� �q�|�� W 5 Q R X d S )N�rr@ rA )�file� � File Namer �Modified�Size� �%Y-%m-%d�%H:%M:%S� rB c S s g | ] }|j �qS r ��filenamer' r r r r* l s z8AbstractTestsWithSourceFile.zip_test.<locals>.<listcomp>)rK rE rF � assertEqual�readr r0 r �StringIO�printdir�getvalue� splitlines�len�assertIn�split� assertTrue�time�strptime�str�namelist�infolist� file_size�getinforW �testzip)r: r" r= r> rI r; � directory�lines�fn�date�time_r �names�infosr) �nm�infor r r �zip_testH sL z$AbstractTestsWithSourceFile.zip_testc C s t | �D ]}| �|| j� qd S r �r# rs r= �r: r" r r r � test_basic~ s z&AbstractTestsWithSourceFile.test_basicc C s� | � ||� t�|d|���}g }|�t��"}|�d�}|s<qH|�|� q,W 5 Q R X g }|�d��"}|�d�}|srq~|�|� qbW 5 Q R X | �d�|�| j � | �d�|�| j � W 5 Q R X d S �NrL � r@ r+ ) rK rE rF r7 r rY �appendrX r/ r0 ) r: r" r= rI �zipdata1�zipopen1� read_data�zipdata2�zipopen2r r r � zip_open_test� s z)AbstractTestsWithSourceFile.zip_open_testc C s t | �D ]}| �|| j� qd S r �r# r r= ru r r r � test_open� s z%AbstractTestsWithSourceFile.test_openc C sF t �t�}| �|| j� t�|d| j��}| �|jt � W 5 Q R X d S �NrL ) �pathlib�Pathr r r= rE rF �assertIsInstancerW rd )r: �pathrI r r r �test_open_with_pathlike� s z3AbstractTestsWithSourceFile.test_open_with_pathlikec C sz | � ||� t�|d|��V}g }|�t��(}|�tdd��}|sBqN|�|� q,W 5 Q R X | �d� |�| j � W 5 Q R X d S �NrL rR � r+ ) rK rE rF r7 r rY r ry rX r/ r0 )r: r" r= rI rz r{ r| r r r �zip_random_open_test� s z0AbstractTestsWithSourceFile.zip_random_open_testc C s t | �D ]}| �|| j� qd S r �r# r� r= ru r r r �test_random_open� s z,AbstractTestsWithSourceFile.test_random_openc C sr | � ||� t�|d��<}|�t��&}g }|�d�}|s:qF|�|� q*W 5 Q R X W 5 Q R X | �d�|�| j � d S )NrL ���r+ ) rK rE rF r7 r �read1ry rX r/ r0 �r: r" r= rI �zipopen�zipdatar| r r r �zip_read1_test� s � z*AbstractTestsWithSourceFile.zip_read1_testc C s t | �D ]}| �|| j� qd S r )r# r� r= ru r r r � test_read1� s z&AbstractTestsWithSourceFile.test_read1c C s� | � ||� t�|d��L}|�t��6}g }|�d�}| �t|�d� |sJqV|�|� q*W 5 Q R X W 5 Q R X | � d� |�| j � d S )NrL � r+ ) rK rE rF r7 r r� ZassertLessEqualr^ ry rX r/ r0 r� r r r �zip_read1_10_test� s � z-AbstractTestsWithSourceFile.zip_read1_10_testc C s t | �D ]}| �|| j� qd S r )r# r� r= ru r r r � test_read1_10� s z)AbstractTestsWithSourceFile.test_read1_10c C s� | � ||� t�|d��P}|�t��:}d}|�� }|s8qZ||7 }|�d�}|sPqZ||7 }q*W 5 Q R X W 5 Q R X | �|| j� d S )NrL r+ �d ) rK rE rF r7 r �readlinerY rX r0 )r: r" r= rI r� r0 rY r r r �zip_readline_read_test� s � z2AbstractTestsWithSourceFile.zip_readline_read_testc C s t | �D ]}| �|| j� qd S r )r# r� r= ru r r r �test_readline_read� s z.AbstractTestsWithSourceFile.test_readline_readc C s^ | � ||� t�|d��<}|�t��&}| jD ]}|�� }| �||� q,W 5 Q R X W 5 Q R X d S r� )rK rE rF r7 r r. r� rX )r: r" r= rI r� rJ Zlinedatar r r �zip_readline_test� s z-AbstractTestsWithSourceFile.zip_readline_testc C s t | �D ]}| �|| j� qd S r )r# r� r= ru r r r � test_readline s z)AbstractTestsWithSourceFile.test_readlinec C sh | � ||� t�|d��F}|�t��}|�� }W 5 Q R X t| j|�D ]\}}| �||� qDW 5 Q R X d S r� ) rK rE rF r7 r � readlines�zipr. rX )r: r" r= rI r� ZziplinesrJ �zipliner r r �zip_readlines_test s z.AbstractTestsWithSourceFile.zip_readlines_testc C s t | �D ]}| �|| j� qd S r )r# r� r= ru r r r �test_readlines s z*AbstractTestsWithSourceFile.test_readlinesc C s` | � ||� t�|d��>}|�t��(}t| j|�D ]\}}| �||� q2W 5 Q R X W 5 Q R X d S r� )rK rE rF r7 r r� r. rX )r: r" r= rI r� rJ r� r r r �zip_iterlines_test s z.AbstractTestsWithSourceFile.zip_iterlines_testc C s t | �D ]}| �|| j� qd S r )r# r� r= ru r r r �test_iterlines s z*AbstractTestsWithSourceFile.test_iterlinesc C s� t �td| j��}|�dd� W 5 Q R X t �td| j��@}|�d��*}| �|�d�d� | �|�d�d� W 5 Q R X W 5 Q R X dS ) z>Check for cases where compressed data is larger than original.r? rA Z12rL rR � 1� 2N)rE rF r r= rG r7 rX rY )r: rI Zopenobjr r r �test_low_compression s z0AbstractTestsWithSourceFile.test_low_compressionc C s< t �td�}|jdd| jd� |�d�}| �|j| j� d S )Nr? �b.txt� hello world�� compress_type)rE rF r rG r= rh rX r� )r: rI rr r r r �test_writestr_compression, s z5AbstractTestsWithSourceFile.test_writestr_compressionc C s� t jtddd�}|jdd| jd� |jdd| jdd � |�d�}| �|j| j� | �|jd� |�d�}| �|j| j� | �|jd� d S ) Nr? rR �r> �a.txtr� r� r� � )r� r> ) rE rF r rG r= rh rX r� �_compresslevel)r: rI Za_infoZb_infor r r �test_writestr_compresslevel2 s � z7AbstractTestsWithSourceFile.test_writestr_compresslevelc C s| dD ]r}|d }t |�}t�t�� d| j��D}|�d|� |�dd�� }|�|�}| � t |�|� W 5 Q R X W 5 Q R X qd S )N)rR i� i i i @ rR r? �foorL ) r rE rF r r r= rG r7 rY rX r^ )r: Z test_sizerg Zjunk�zipfr; Zbufr r r �test_read_return_sizeB s z1AbstractTestsWithSourceFile.test_read_return_sizec C s| t �� }tj|dd��"}|jd| j| jd� |�� }W 5 Q R X |�� }t �|�}t�|��H}|� d��2}|� |d � | � t �� |� � W 5 Q R X W 5 Q R X W 5 Q R X t �|�}t�|��L}|� d��6}|� |d � | � t �� |� d�r�q�W 5 Q R X W 5 Q R X W 5 Q R X t �|�}t�|��P}|� d��:}|� |d � | � t �� |�d��rZ�qJW 5 Q R X W 5 Q R X W 5 Q R X d S )Nr? rC rA r� � r� )r r rE rF rG r0 r= �tellr\ r7 �truncate� assertRaises�EOFErrorrY r� )r: r; r� Z end_offset� zipfiledatar� r r r �test_truncated_zipfileN s2 & z2AbstractTestsWithSourceFile.test_truncated_zipfilec C s| d}t | �D �]h}t�|d| j��&}|�t|� t|�}| �d|� W 5 Q R X t�|d���}t|�}t|t �r�| �d| |� n| �d| |� | �d|� t|� |��}| �d| |� | �d|� | �d |� | jtj kr�| �d |� | �d |� |� |��D}t|�}| �d | |� | �d|� | jtj k�rB| �d |� W 5 Q R X | �d t|�� W 5 Q R X | �d t|�� q d S )Nz file.namer? zmode='w'rL z filename=%rzfile=%rzmode='r'z filemode=z file_size=zcompress_type=zcompress_size=zname=%rz[closed]) r# rE rF r= r8 r �reprr_ � isinstancerd rh � ZIP_STOREDr7 )r: �fnamer" rI rL r� r r r � test_reprl s6 z%AbstractTestsWithSourceFile.test_reprc C s$ t | �D ]}| j|| jdd� qd S )N� r� rt ru r r r �test_compresslevel_basic� s z4AbstractTestsWithSourceFile.test_compresslevel_basicc C sl t jtddd��R}|�td� |jtddd� |�d�}|�d�}| �|jd� | �|jd� W 5 Q R X dS )zTCheck that files within a Zip archive can have different compression levels.r? rR r� Z compress_1Z compress_9r� N)rE rF r r8 r rh rX r� )r: rI Zone_infoZ nine_infor r r �test_per_file_compresslevel� s z7AbstractTestsWithSourceFile.test_per_file_compresslevelc sl G � �fdd�dt j�}d�|� }d � t�|d| j���}|�dd��}|�d� W 5 Q R X d� zZz&|�dd��}|�d� W 5 Q R X W n tk r� �d 7 �Y nX W �W 5 Q R � �qW 5 d � X W 5 Q R X t�t �|�� ���*}| � |� � dg� | � |� d�d� W 5 Q R X q t�t �|�� ���>}| � |� � ddg� | � |� d�d� | � |� d�d� W 5 Q R X d S ) Nc s e Zd Z� ��fdd�Z� ZS )zCAbstractTestsWithSourceFile.test_writing_errors.<locals>.BrokenFilec s, �d k r ��krt ��d7 �t� �|� d S �NrR )�OSError�superr8 �r: r0 )� __class__�count�stopr r r8 � s zIAbstractTestsWithSourceFile.test_writing_errors.<locals>.BrokenFile.write)�__name__� __module__� __qualname__r8 � __classcell__r �r� r� �r� r � BrokenFile� s r� r r? Zfile1s data1Zfile2s data2rR ) r r rE rF r= r7 r8 r� r\ rX re rY )r: r� ZtestfilerI r" r r� r �test_writing_errors� s, z/AbstractTestsWithSourceFile.test_writing_errorsc C s t t� t t� d S r �r r r �r: r r r �tearDown� s z$AbstractTestsWithSourceFile.tearDown)N)N)$r� r� r� � classmethodr3 r< rK rs rv 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$ 0 sB 6 %r$ c @ s~ e Zd ZejZdZdd� Zdd� Zdd� Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS ) �StoredTestsWithSourceFileNc C sX | � ||� t�|d��6}|�d�}| �|jd� |�d�}| �|jd� W 5 Q R X d S )NrL rA � �rB )rK rE rF rh rX � external_attr)r: r" r= rI �zinfoZzinfo2r r r � zip_test_writestr_permissions� s z7StoredTestsWithSourceFile.zip_test_writestr_permissionsc C s t | �D ]}| �|tj� qd S r )r# r� rE r� ru r r r �test_writestr_permissions� s z3StoredTestsWithSourceFile.test_writestr_permissionsc C sZ t �tdt j��}|�td� W 5 Q R X t �tdt j��}| �|�� dg� W 5 Q R X d S )Nr? � /absoluterL �absolute�rE rF r r� r8 r rX re �r: rI r r r �test_absolute_arcnames� s z0StoredTestsWithSourceFile.test_absolute_arcnamesc C sj t �tdt j��}|�tt� W 5 Q R X t �tdt j��(}|�d| j� | �|� � tdg� W 5 Q R X dS )z&Test appending to an existing zipfile.r? �arA N) rE rF r r� r8 r rG r0 rX re r� r r r �test_append_to_zip_file� s z1StoredTestsWithSourceFile.test_append_to_zip_filec C s8 d}t td��}|�|� W 5 Q R X t�tdtj��}|�tt� W 5 Q R X t td��R}|�t|�� t�|d��,}| � |� � tg� | � |� t�| j � W 5 Q R X W 5 Q R X t td��$}| � |� t|��|� |� � }W 5 Q R X t �|��B}t�|��,}| � |� � tg� | � |� t�| j � W 5 Q R X W 5 Q R X dS )z9Test appending to an existing file that is not a zipfile.� I am not a ZipFile!I am not a ZipFile!I am not a ZipFile!I am not a ZipFile!I am not a ZipFile!I am not a ZipFile!I am not a ZipFile!I am not a ZipFile!I am not a ZipFile!I am not a ZipFile!r5 r� �rbrL N)r7 r r8 rE rF r� r �seekr^ rX re rY r0 r r )r: r0 r" rI r� �bior r r �test_append_to_non_zip_file� s ( z5StoredTestsWithSourceFile.test_append_to_non_zip_filec C s� t �� �6}t�|dtj��}|�tt� W 5 Q R X |�� }W 5 Q R X d}tt d��}|�|� |�|� W 5 Q R X t�t ��,}| � |� � tg� | � |� t�| j � W 5 Q R X d S )Nr? r� r5 )r r rE rF r� r8 r r\ r7 r rX re rY r0 �r: r� rI r� r0 r" r r r � test_read_concatenated_zip_file s z9StoredTestsWithSourceFile.test_read_concatenated_zip_filec C sN t �� �6}t�|dtj��}|�tt� W 5 Q R X |�� }W 5 Q R X dd }tt d��}|�|� |�|� W 5 Q R X t�t d��&}| � |� � tg� |� d| j � W 5 Q R X tt d��$}| � |�t|��|� |�� }W 5 Q R X t �|��X}t�|��B}| � |� � tdg� | � |�t�| j � | � |�d�| j � W 5 Q R X W 5 Q R X d S )Nr? s I am not a ZipFile!i@B r5 r� rA r� )r r rE rF r� r8 r r\ r7 r rX re rG r0 rY r^ r� r r r �$test_append_to_concatenated_zip_file s$ z>StoredTestsWithSourceFile.test_append_to_concatenated_zip_filec C sr t �tdt j��}|�tt� W 5 Q R X ttd��}|�d� W 5 Q R X t �td��}| �|t j� W 5 Q R X d S )Nr? r� z rL )rE rF r r� r8 r r7 r� �r: rI r" r r r �test_ignores_newline_at_end% s z5StoredTestsWithSourceFile.test_ignores_newline_at_endc C s� t �tdt j��}d|_|�tt� W 5 Q R X ttd��}|�d� W 5 Q R X t �td��"}| �|t j� | � |jd� W 5 Q R X d S )Nr? � this is a commentr� zabcdef rL ) rE rF r r� �commentr8 r r7 r� rX r� r r r �)test_ignores_stuff_appended_past_comments- s zCStoredTestsWithSourceFile.test_ignores_stuff_appended_past_commentsc C sR t �td��<}|�t� ttd�� }| �|�t�|�� � W 5 Q R X W 5 Q R X dS )z`Check that calling ZipFile.write without arcname specified produces the expected result.r? r� N)rE rF r r8 r r7 rX rY r� r r r �test_write_default_name7 s z1StoredTestsWithSourceFile.test_write_default_namec C s� t jtdd��}|�dd� W 5 Q R X t jtdd��}| �t|jt� W 5 Q R X t jtdd��*}| �t�� |jtdd� W 5 Q R X W 5 Q R X dS )z[Check that trying to call write() on a readonly ZipFile object raises a ValueError.r? rC z somefile.txt�bogusrL N) rE rF r rG r� � ValueErrorr8 r r7 r� r r r �test_write_to_readonly? s z0StoredTestsWithSourceFile.test_write_to_readonlyc C sv t �td� t�td��}| �t|jt� W 5 Q R X tjtddd��(}|�t� |� t�}| � |j d� W 5 Q R X d S )N)r r r? F�Zstrict_timestamps)i� rR rR r r r ) �os�utimer rE rF r r� r� r8 rh rX � date_time)r: rI r� r r r �test_add_file_before_1980L s z3StoredTestsWithSourceFile.test_add_file_before_1980c C s� d}zt �|� W n& tk r8 | �d|� d�� Y nX zt�t||f� W n tk rl | �d� Y nX t�t�j}|dkr�| �d|��� t � t d��}| � t j|jt� W 5 Q R X t j t ddd ��(}|�t� |�t�}| �|jd � W 5 Q R X d S ) Nl �&� ztime.localtime(z) raises OverflowErrorz/Host fs cannot set timestamp to required value.l �b� �f z,Linux VFS/XFS kernel bug detected: mtime_ns=r? Fr� )i; � � � �; r� )rb � localtime� OverflowError�skipTestr� r� r �stat� st_mtime_nsrE rF r r� �struct�errorr8 rh rX r� )r: ZtsZmtime_nsrI r� r r r �test_add_file_after_2107W s$ z2StoredTestsWithSourceFile.test_add_file_after_2107)r� r� r� rE 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 Zd ZejZdd� ZdS )�DeflateTestsWithSourceFilec C sp t �td��Z}|�tdt j� |�tdt j� |�d�}|�d�}| �|j t j� | �|j t j� W 5 Q R X dS )zUCheck that files within a Zip archive can have different compression options.r? ZstoremeZ deflatemeN) rE rF r r8 r r� � ZIP_DEFLATEDrh rX r� )r: rI ZsinfoZdinfor r r �test_per_file_compression} s z4DeflateTestsWithSourceFile.test_per_file_compressionN)r� r� r� rE r r= r r r r r r x s r c @ s e Zd ZejZdS )�Bzip2TestsWithSourceFileN�r� r� r� rE � ZIP_BZIP2r= r r r r r � s r c @ s e Zd ZejZdS )�LzmaTestsWithSourceFileN�r� r� r� rE �ZIP_LZMAr= r r r r r � s r c @ sH e Zd Zedd� �Zdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dS )� AbstractTestZip64InSmallFilesc C s$ dd� t dt�D �}d�|�| _d S )Nc s s | ]}t d | d�V qdS )zTest of zipfile line %d.r% N)r& r' r r r � <genexpr>� s �z;AbstractTestZip64InSmallFiles.setUpClass.<locals>.<genexpr>r � )r, r- r/ r0 )r2 r. r r r r3 � s �z(AbstractTestZip64InSmallFiles.setUpClassc C sB t j| _t j| _dt _dt _ttd��}|�| j� W 5 Q R X d S )Nr r� r5 ) rE � ZIP64_LIMIT�_limit�ZIP_FILECOUNT_LIMIT�_filecount_limitr7 r r8 r0 r9 r r r r< � s z#AbstractTestZip64InSmallFiles.setUpc C sH t j|d|dd��,}|�td� |�tt� |�d| j� W 5 Q R X t �|d|����}| �|�t�| j� | �|�d�| j� | �|�d�| j� t� � }|� |� |� � }|� � }| �t |�d� | �d|d � | �d |d � | �d |d � |d �� \}}} } | �|d� | �t�|d �� | �t�| d�� | �| tt | j��� |�� } | �t | �d� | �t| � | �d| � | �d| � |�� } dd� | D �} | �t | �d� | �t| � | �d| � | �d| � | D ]} | �| jt | j�� �q�tddfD ]2}|�|�}| �|j|� | �|jt | j�� �q�|�� W 5 Q R X d S )Nr? T�Z allowZip64r@ rA rL rU rO r rP rQ rR rS rT � c S s g | ] }|j �qS r rV r' r r r r* � s z:AbstractTestZip64InSmallFiles.zip_test.<locals>.<listcomp>)rE rF r8 r rG r0 rX rY r rZ r[ r\ r] r^ r_ r` ra rb rc rd re rf rg rh rW ri )r: r" r= rI r; rj rk rl rm rn r ro rp r) rq rr r r r rs � sN z&AbstractTestZip64InSmallFiles.zip_testc C s t | �D ]}| �|| j� qd S r rt ru r r r rv � s z(AbstractTestZip64InSmallFiles.test_basicc C s� t jtd| jdd�}d|_d}t|�D ] }|�d| d|d d � q&| �t|� � �|� |� � t �td | j�}| �t|� � �|� t|�D ]0}|� d| �� d �}| �|d|d d � q�|� � d S ) Nr? Tr r� � �foo%08d�%dr �9 rL r% ) rE rF r r= �debugr, rG rX r^ re �closerY �decode)r: r� �numfilesr) �zipf2�contentr r r �test_too_many_files� s � z1AbstractTestZip64InSmallFiles.test_too_many_filesc C s� t jtd| jdd�}d|_d}t|�D ] }|�d| d|d d � q&| �t|� � �|� | � t j �� |�d| d � W 5 Q R X | �t|� � �|� |� � t jtd | jdd�}d|_| �t|� � �|� | � t j �� |�d| d � W 5 Q R X | �t|� � �|� |� � t jtd | jd d�}d|_| �t|� � �|� d }t||�D ]"}|�d| d|d d � �qN| �t|� � �|� |� � t �td| j�}| �t|� � �|� t|�D ]2}|� d| ��d�}| �|d|d d � �q�|� � d S )Nr? Fr r� r� r r r r r+ r� Tr rL r% )rE rF r r= r r, rG rX r^ re r� � LargeZipFiler rY r )r: r� r r) Z numfiles2r r r r r �test_too_many_files_append� sJ � � � z8AbstractTestZip64InSmallFiles.test_too_many_files_appendc C s$ | j t_| jt_tt� tt� d S r )r rE r r r r r r r� r r r r� s z&AbstractTestZip64InSmallFiles.tearDownN) r� r� r� r� r3 r< rs rv r! r# r� r r r r r � s 9'r c @ sT e Zd ZejZdd� Zdd� Zdd� Zdd� Z d d � Z dd d �Z dd� Z dd� Z dS )�StoredTestZip64InSmallFilesc C s6 t j|d|dd��}| �t j|jtd� W 5 Q R X d S �Nr? Fr r@ )rE rF r� r" r8 r �r: r" r= rI r r r �large_file_exception_test) s �z5StoredTestZip64InSmallFiles.large_file_exception_testc C s8 t j|d|dd�� }| �t j|jd| j� W 5 Q R X d S r% )rE rF r� r" rG r0 r& r r r �large_file_exception_test2. s �z6StoredTestZip64InSmallFiles.large_file_exception_test2c C s. t | �D ] }| �|tj� | �|tj� qd S r )r# r' rE r� r( ru r r r �test_large_file_exception3 s z5StoredTestZip64InSmallFiles.test_large_file_exceptionc C s^ t jtdt jdd��}|�td� W 5 Q R X t �tdt j��}| �|�� dg� W 5 Q R X d S )Nr? Tr r� rL r� r� r� r r r r� 8 s �z2StoredTestZip64InSmallFiles.test_absolute_arcnamesc C s� t jtddd��}|�d| j� W 5 Q R X t jtddd��}|�d�}|j}W 5 Q R X t jtddd��}|�d| j� W 5 Q R X t jtddd�� }|�d�}| �|j|� W 5 Q R X d S )Nr? Tr rA rL r� Zstrfile2)rE rF r rG r0 rh �extrarX )r: rI r� r* r r r � test_append@ s z'StoredTestZip64InSmallFiles.test_appendFc C s� d}d}g } g } |} |r4d} |r4| � |� | � |� t�d| �} |} |rdd} |rd| � |� | � |� t�d| �} |} |r�d} |r�| � |� t�d| �} tjddt| � ddt| � f| �� }tjddt| � ddt| � f| �� }t�dd dt| � �}t�dd dt| � �}t�d d dt| � �}t�d d dt| � �}d }d}t�d t|��}d| | | | | | | d | | | | d | | | d d d | | d d d d }|S )z�Generate bytes sequence for a zip with (incomplete) zip64 data. The actual values (not the zip 64 0xffffffff values) stored in the file are: file_size: 8 compress_size: 8 header_offset: 0 r r l �� z<Lz<HH�QrR z<Q�: �2 z<HrU s test.txt� test1234s PK ! �%��s PK-- ! �%��s �s PK, - -s s s PK l s s PK : 2 s )ry r �packr^ )r: �file_size_64_set�file_size_extra�compress_size_64_set�compress_size_extra�header_offset_64_set�header_offset_extraZ actual_sizeZactual_header_offsetZlocal_zip64_fieldsZcentral_zip64_fieldsrg � compress_size� header_offsetZ local_extraZ central_extraZcentral_dir_sizeZoffset_to_central_dirZlocal_extra_lengthZcentral_extra_lengthrW r Zfilename_lengthZzip64_contentsr r r �make_zip64_fileN s� �� ��������� � � � � ������������ � ��!z+StoredTestZip64InSmallFiles.make_zip64_filec C s& | j dd�}| �tj��}t�t�|�� W 5 Q R X | �dt|j �� � � | j dddd�}| �tj��}t�t�|�� W 5 Q R X | �dt|j �� � � | j dd�}| �tj��}t�t�|�� W 5 Q R X | �dt|j �� � � | j dddddd�}| �tj��}t�t�|�� W 5 Q R X | �dt|j �� � � | j d dddd �}| �tj��}t�t�|�� W 5 Q R X | �dt|j �� � � | j ddd dd �}| �tj��}t�t�|�� W 5 Q R X | �dt|j �� � � | j d d dd �}| �tj��}t�t�|�� W 5 Q R X | �dt|j �� � � d S )a Missing zip64 extra records raises an exception. There are 4 fields that the zip64 format handles (the disk number is not used in this module and so is ignored here). According to the zip spec: The order of the fields in the zip64 extended information record is fixed, but the fields MUST only appear if the corresponding Local or Central directory record field is set to 0xFFFF or 0xFFFFFFFF. If the zip64 extra content doesn't contain enough entries for the number of fields marked with 0xFFFF or 0xFFFFFFFF, we raise an error. This test mismatches the length of the zip64 extra field and the number of fields set to indicate the presence of zip64 data. T)r1 z file size)r1 r2 r3 z compress size)r3 )r1 r2 r3 r4 r5 z header offsetF)r1 r3 r4 r5 )r1 r2 r3 r5 )r1 r3 r5 N) r9 r� rE � BadZipFilerF r r r_ rd Z exception�lower)r: Zmissing_file_size_extra�eZmissing_compress_size_extraZmissing_header_offset_extrar r r �test_bad_zip64_extra� sp �������z0StoredTestZip64InSmallFiles.test_bad_zip64_extrac C s� d}d}d}d}ddd�ddd�ddd�f}t dt|�d �D ]�}t�||�D ]�}i }|D ]} |�| � qZt�t�| j f |����N} | � � d } | � | j |� | � | j |� | � | j|� | � | �| �|� W 5 Q R X qNq>d S ) Nr r r/ T)r1 r2 )r3 r4 )r5 r6 rR )r, r^ � itertools� combinations�updaterE rF r r r9 rf rX rg r7 r8 rY ) r: Zexpected_file_sizeZexpected_compress_sizeZexpected_header_offsetZexpected_contentZparamsrL ZcomborH �c�zfr� r r r � test_generated_valid_zip64_extra s& � z<StoredTestZip64InSmallFiles.test_generated_valid_zip64_extraN)FFFFFF)r� r� r� rE r� r= r' r( r) r� r+ r9 r= rC r r r r r$ % s � c^r$ c @ s e Zd ZejZdS )� DeflateTestZip64InSmallFilesN�r� r� r� rE r r= r r r r rD + s rD c @ s e Zd ZejZdS )�Bzip2TestZip64InSmallFilesNr r r r r rF 0 s rF c @ s e Zd ZejZdS )�LzmaTestZip64InSmallFilesNr r r r r rG 5 s rG c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�AbstractWriterTestsc C s t t� d S r �r r r� r r r r� = s z AbstractWriterTests.tearDownc C st d}t �td| j��V}|�dd�}|�|� |�� | �|j� |�� | �|j� | � |� d�|� W 5 Q R X d S )N� contentr? r! ) rE rF r r= r7 r8 r ra r rX rY �r: r0 r� r? r r r �test_close_after_close@ s z*AbstractWriterTests.test_close_after_closec C sp d}t �td| j��R}|�dd�}|�|� |�� | �|j� | � t |jd� | � |� d�|� W 5 Q R X d S )NrJ r? r! r+ ) rE rF r r= r7 r8 r ra r r� r� rX rY rK r r r �test_write_after_closeK s z*AbstractWriterTests.test_write_after_closeN)r� r� r� r� rL rM r r r r rH ; s rH c @ s e Zd ZejZdS )�StoredWriterTestsN�r� r� r� rE r� r= r r r r rN U s rN c @ s e Zd ZejZdS )�DeflateWriterTestsNrE r r r r rP X s rP c @ s e Zd ZejZdS )�Bzip2WriterTestsNr r r r r rQ \ s rQ c @ s e Zd ZejZdS )�LzmaWriterTestsNr r r r r rR ` s rR c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dS )�PyZipFileTestsc C s |d |kr | � |d |� d S )N�orA )r_ )r: �namere r r r �assertCompiledInf s z PyZipFileTests.assertCompiledInc C s� t j|t jt jt jkd�s$| �d� t j�|d�}z"t �|t jt j B �}t � |� W n t k rr | �d� Y nX t |� d S )N)Z effective_idsz/requires write access to the installed locationztest_zipfile.try) r� �access�W_OK�supports_effective_idsr� r� r/ r7 �O_WRONLY�O_CREATr � Exceptionr )r: r� rW �fdr r r �requiresWriteAccessj s � z"PyZipFileTests.requiresWriteAccessc C sR | � tj�t�� t� ��}t�|d���}t}|�d�r�|� tj �}tj d k r^|� |� tj �� d|krtt j�|�}n |d d� }|�|� tj�|�}| �||�� � | �||�� � W 5 Q R X W 5 Q R X t� �x}t�|d��`}t}|�d�r�|d d� }|�|d� ddtj�|�f }| �||�� � | �||�� � W 5 Q R X W 5 Q R X d S )Nr? �.pyc� __pycache__r� Z testpackagez%s/%s)r^ r� r� �dirname�__file__r rE � PyZipFile�endswithr` �sep�altsep�extend� importlib�util�source_from_cache�writepy�basename� assertNotInre rV )r: �trI rl Z path_splitZbnr r r �test_write_pyfilew s, $ z PyZipFileTests.test_write_pyfilec C sx dd l }tj�|j�}| �|� t� �H}t�|d��0}|� |� |� � }| � d|� | � d|� W 5 Q R X W 5 Q R X d S )Nr r? zemail/__init__.pyzemail/mime/text.py) �emailr� r� ra rb r^ r rE rc rk re rV )r: rp � packagedirrn rI ro r r r �test_write_python_package� s z(PyZipFileTests.test_write_python_packagec C s dd l }tj�|j�}| �|� t� ��}t�|d���}t � �}|� |� W 5 Q R X |� � }| � d|k� t � �}|j |dd� d� W 5 Q R X |� � }| � d|k� dd� }t � �*}| � t�� |j ||d� W 5 Q R X W 5 Q R X |� � }|r�t|� | � d|k� W 5 Q R X W 5 Q R X d S ) Nr r? � SyntaxErrorc S s dS )NFr )Zwhateverr r r �<lambda>� r+ zCPyZipFileTests.test_write_filtered_python_package.<locals>.<lambda>�Z filterfuncc S s t j�| ��d� S )NZbad)r� r� rl � startswith)r� r r r �filter� s zAPyZipFileTests.test_write_filtered_python_package.<locals>.filter)r! r� r� ra rb r^ r rE rc r rk r\ ra � assertWarns� UserWarning�print)r: r! rq rn rI Z reportSIOZ reportStrrw r r r �"test_write_filtered_python_package� s& "z1PyZipFileTests.test_write_filtered_python_packagec C s� dd l }tj�|j�}| �|� dr(dnd}d}t� �T}tj|d|d��8}|� |� |� � }| � d| |� | � d| |� W 5 Q R X W 5 Q R X d S ) Nr FrR r_ r? )�optimizezemail/__init__zemail/mime/text) rp r� r� ra rb r^ r rE rc rk re r_ )r: rp rq ZoptlevelZextrn rI ro r r r � test_write_with_optimization� s � z+PyZipFileTests.test_write_with_optimizationc C s� t �t� z�tt j�td�d��}|�d� W 5 Q R X tt j�td�d��}|�d� W 5 Q R X tt j�td�d��}|�d� W 5 Q R X t� �T}t � |d��<}|� t� |� � }| � d|� | � d|� | �d|� W 5 Q R X W 5 Q R X W 5 tt� X d S )N�mod1.pyr? � print(42) �mod2.py�print(42 * 42) zmod2.txtz bla bla bla �r� �mkdirr r r7 r� r/ r8 r rE rc rk re rV rm �r: r; rn rI ro r r r �test_write_python_directory� s $z*PyZipFileTests.test_write_python_directoryc C s� t �t� z�tt j�td�d��}|�d� W 5 Q R X tt j�td�d��}|�d� W 5 Q R X t� �P}t � |d��8}|j tdd� d� |� � }| � d|� | �d|� W 5 Q R X W 5 Q R X W 5 tt� X d S ) Nr~ r? r r� r� c S s | � d� S )Nr� )rd )rl r r r rt � s zEPyZipFileTests.test_write_python_directory_filtered.<locals>.<lambda>ru r� r� r r r �$test_write_python_directory_filtered� s $z3PyZipFileTests.test_write_python_directory_filteredc C sf t � �V}t�|d��>}ttd��}|�d� W 5 Q R X | �t|jt� t t� W 5 Q R X W 5 Q R X d S )Nr? z!most definitely not a python file) r rE rc r7 r r8 r� � RuntimeErrorrk r )r: rn rI r" r r r �test_write_non_pyfile� s z$PyZipFileTests.test_write_non_pyfilec C s� t �t� z�tt j�td�d��}|�d� W 5 Q R X t� �t}t � |d��\}t � �}|� t j�td�� W 5 Q R X | � d|�� � |�� }| � d|� | �d|� W 5 Q R X W 5 Q R X W 5 tt� X d S )Nr~ r? zBad syntax in python file rs zmod1.pyc)r� r� r r r7 r� r/ r8 r rE rc r rk r_ r\ re rm )r: r; rn rI �sro r r r � test_write_pyfile_bad_syntax s $z+PyZipFileTests.test_write_pyfile_bad_syntaxc C s� t �t� z�tt j�td�d��}|�d� W 5 Q R X t� �F}t � |d��.}|� t � t�d � |�� }| �d|� W 5 Q R X W 5 Q R X W 5 tt� X d S )Nr~ r? r )r� r� r r r7 r� r/ r8 r rE rc rk r� r� re rV r� r r r �test_write_pathlike s $z"PyZipFileTests.test_write_pathlikeN)r� r� r� rV r^ ro rr r{ r} r� r� r� r� r� r r r r rS e s rS c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Ze�ejjdkd�dd � �Ze�ejjd kd �d d � �Zd!d"� Zd#S )$� ExtractTestsc C s: t �tdt j�� }tD ]\}}|�||� qW 5 Q R X d S )Nr? )rE rF r r� �SMALL_TEST_DATArG �r: rI �fpath�fdatar r r �make_test_file( s zExtractTests.make_test_filec C s� t � �� | �� t�td��z}tD ]n\}}|�|�}tj� t� � |�}tj� |�}| � ||� t |d��}| � |�� |�� � W 5 Q R X t|� q"W 5 Q R X W 5 Q R X d S �NrL r� )r r� rE rF r r� �extractr� r� r/ �getcwd�normpathrX r7 �encoderY r )r: rI r� r� � writtenfile� correctfiler" r r r � test_extract- s zExtractTests.test_extractc C s� | � � t�td���}tD ]z\}}|�||�}tj�||�}tj� |�}| � tj� ||�||f� t |d��}| � |�� |�� � W 5 Q R X t|� qW 5 Q R X tt� d S r� )r� rE rF r r� r� r� r� r/ r� ra �samefiler7 rX r� rY r )r: �targetrI r� r� r� r� r" r r r �_test_extract_with_target@ s z&ExtractTests._test_extract_with_targetc C s t � �}| �|� W 5 Q R X d S r )r r� �r: �extdirr r r �test_extract_with_targetS s z%ExtractTests.test_extract_with_targetc C s& t � �}| �t�|�� W 5 Q R X d S r )r r� r� r� r� r r r �!test_extract_with_target_pathlikeW s z.ExtractTests.test_extract_with_target_pathlikec C s� t � �� | �� t�td��`}|�� tD ]L\}}tj� t� � |�}t |d��}| � |� � |�� � W 5 Q R X t|� q*W 5 Q R X W 5 Q R X d S r� )r r� rE rF r � extractallr� r� r� r/ r� r7 rX r� rY r )r: rI r� r� �outfiler" r r r �test_extract_all[ s z ExtractTests.test_extract_allc C s� | � � t�td��^}|�|� tD ]H\}}tj�||�}t |d��}| � |� � |� � � W 5 Q R X t |� q$W 5 Q R X t t� d S r� )r� rE rF r r� r� r� r� r/ r7 rX r� rY r )r: r� rI r� r� r� r" r r r � _test_extract_all_with_targeth s z*ExtractTests._test_extract_all_with_targetc C s t � �}| �|� W 5 Q R X d S r )r r� r� r r r � test_extract_all_with_targetv s z)ExtractTests.test_extract_all_with_targetc C s& t � �}| �t�|�� W 5 Q R X d S r )r r� r� r� r� r r r �%test_extract_all_with_target_pathlikez s z2ExtractTests.test_extract_all_with_target_pathlikec C s< | � tj�|�� t|d��}| �|�� |� W 5 Q R X d S )Nr� )ra r� r� �isfiler7 rX rY )r: rW r r" r r r � check_file~ s zExtractTests.check_filec C sB t jj}| �|dd�d� | �|dd�d� | �|dd�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_i�../../foo../../ba..r�/� foo/ba..r)rE rF Z_sanitize_windows_namerX )r: Zsanr r r �test_sanitize_windows_name� s z'ExtractTests.test_sanitize_windows_namec C s" ddddddddg}| � |� d S ) N)z ../foo/bar�foo/bar)z foo/../barr� )z foo/../../barr� )z foo/bar/..r� )z ./../foo/barr� )z/foo/barr� )z /foo/../barr� )z/foo/../../barr� �� _test_extract_hackers_arcnames)r: 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 s8 ddddddddd d d d d ddddddg}| � |� dS )z>Test combination of path fixing and windows name sanitization.)z ..\foo\barr� )z ..\/foo\/barr� )z foo/\..\/barr� )z foo\/../\barr� )z C:foo/barr� )z C:/foo/barr� )z C://foo/barr� )z C:\foo\barr� )z//conky/mountpoint/foo/barr� )z\\conky\mountpoint\foo\barr� )z///conky/mountpoint/foo/bar�conky/mountpoint/foo/bar)z\\\conky\mountpoint\foo\barr� )z//conky//mountpoint/foo/barr� )z\\conky\\mountpoint\foo\barr� )z//?/C:/foo/barr� )z\\?\C:\foo\barr� )zC:/../C:/foo/barz C_/foo/bar)za:b\c<d>e|f"g?h*izb/c_d_e_f_g_h_i)r� r� Nr� )r: 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 dddg}| � |� d S )N)z //foo/barr� )r� z foo../ba..r)� foo/..\barr� r� )r: Zposix_hacknamesr r r �(test_extract_hackers_arcnames_posix_only� s �z5ExtractTests.test_extract_hackers_arcnames_posix_onlyc C s� |D �]�\}}d|� � }t�tdtj��&}t�� }||_d|_|�||� W 5 Q R X |� t j d�}t j � ddd�}t j j |f|�d��� }t�td��,}|�||�} | j| |d || |f d � W 5 Q R X | �||� td� t�td��}|�|� W 5 Q R X | �||� td� t j j t �� f|�d��� }t�td��$}|�|�} | j| |d | d � W 5 Q R X | �||� t|�d�d � t�td��}|�� W 5 Q R X | �||� t|�d�d � tt� qd S ) N� foobarr? r� r� r� �subdirZsubsubrL zextract %r: %r != %r)�msgz extract %rr )r� rE rF r r� �ZipInforW r� rG �replacer� re r� r/ r` r� rX r� r r� r� r ) r: Z hacknamesZarcnameZ fixednamer rI r� Z targetpathr� r� r r r r� � sH �� � z+ExtractTests._test_extract_hackers_arcnamesN)r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �unittestZskipIfr� r� re r� r� r� r r r r r� & s" r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zed d � �Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z!d=d>� Z"d?d@� Z#dAdB� Z$dCdD� Z%dEdF� Z&e'dGdH� �Z(edIdJ� �Z)edKdL� �Z*dMdN� Z+dOS )P� OtherTestsc C s� t �tdt j��J}|�dd� | �t�� |�dd� W 5 Q R X | �|�� dgd � W 5 Q R X t �td��z}|� � }d}|D ]&}|� |��}||� � 7 }W 5 Q R X q~| � |dd h� d}|D ]}||� |�7 }q�| � |dd h� W 5 Q R X d S ) Nr? rU r� �barr� rL r+ r� s barfoo) rE rF r r� rG rx ry rX re rf r7 rY r_ )r: rI rp r0 rr r� r r r �test_open_via_zip_info� s z!OtherTests.test_open_via_zip_infoc C sJ t �td��4}dD ](}t �|�}| jdO _|�||� qW 5 Q R X d S )Nr? Zabcdefghijklmnopr )rE rF r r� Z flag_bitsrG )r: Zorig_zipr0 r� r r r �-test_writestr_extended_local_header_issue1202� s z8OtherTests.test_writestr_extended_local_header_issue1202c C s� t �td��.}tD ]"\}}|�||� | �|jd� qW 5 Q R X | �|jd� t �td��}| �|jd� W 5 Q R X | �|jd� dS )z8Check that the zipfile is closed after the 'with' block.r? zzipfp is not open�zipfp is not closedrL N)rE rF r r� rG ZassertIsNotNoner; � assertIsNoner� r r r � test_close s zOtherTests.test_closec C s� t �td�� }tD ]\}}|�||� qW 5 Q R X z$t �td��}t �� �W 5 Q R X W n$ t jk rz | �|jd� Y nX dS )zWCheck that the zipfile is closed if an exception is raised in the 'with' block.r? rL r� N)rE rF r r� rG r: r� r; )r: rI r� r� Zzipfp2r r r �test_close_on_exception s z"OtherTests.test_close_on_exceptionc C s d}| � ttjt�|�d� d S )Nsd PKx !p�@ xPKxx !p�@ � xPK / rL )r� �NotImplementedErrorrE rF r r r� r r r �test_unsupported_version s �z#OtherTests.test_unsupported_versionc C s> t d�}t�|��"}|�� D ]}|�|��� q W 5 Q R X d S )Nzzip_cp437_header.zip)r rE rF re r7 r )r: r� rI rU r r r �test_read_unicode_filenames& s z&OtherTests.test_read_unicode_filenamesc C s� t �td��4}|�dd� |�dd� | �|�� d jt� W 5 Q R X t �td��.}| �|j d jd� | �|j d jd� W 5 Q R X d S )Nr? �foo.txtzTest for unicode filenameu ö.txtr rL rR ) rE rF r rG r� rf rW rd rX �filelist)r: rB r r r � test_write_unicode_filenames. s z'OtherTests.test_write_unicode_filenamesc C s: t �td��$}|�dd� | �|�d�d� W 5 Q R X d S )Nr? u прикладs sample)rE rF r rG rX rY r� r r r �'test_read_after_write_unicode_filenames8 s z2OtherTests.test_read_after_write_unicode_filenamesc C s� t t� d}d}t�tdtj��}|�||� W 5 Q R X | �t�� t�tdtj� W 5 Q R X t�td��*}| �|� � |g� | �|� |�|� W 5 Q R X dS )z&Test exclusive creating a new zipfile.� testfile.txt�# hello, world. this is some content.�xrL N) r r rE rF r� rG r� �FileExistsErrorrX re rY )r: rW r rI r r r � test_exclusive_create_zip_file= s z)OtherTests.test_exclusive_create_zip_filec C s� t j�t�rt �t� d}d}z(t�td��}|�||� W 5 Q R X W n tk rd | � d� Y nX | � t j�t�� t�td��}| � |� |�|� W 5 Q R X d S )Nr� r� r� z1Could not append data to a non-existent zip file.rL ) r� r� �existsr r rE rF rG r� �failra rX rY )r: rW r rB r r r �(test_create_non_existent_file_for_appendJ s z3OtherTests.test_create_non_existent_file_for_appendc C sJ t td��}|�d� W 5 Q R X zt�t�}W n tjk rD Y nX d S )Nr? � this is not a legal zip file )r7 r r8 rE rF r: )r: r; rB r r r �test_close_erroneous_file\ s z$OtherTests.test_close_erroneous_filec C s� t td��}|�d� W 5 Q R X | �t�t�� | �t�t�t��� t td��}| �t�|�� W 5 Q R X t� � }|�d� | �t�|�� |� dd� | �t�|�� dS )z;Check that is_zipfile() correctly identifies non-zip files.r? r� r� s this is not a legal zip file r N) r7 r r8 r rE � is_zipfiler� r� r r r� r9 r r r �test_is_zip_erroneous_filek s z%OtherTests.test_is_zip_erroneous_filec C sp t �� }tj|dd��}|�dd� W 5 Q R X |�� }tt|��D ](}t �|d|� �}| �tj tj|� qBdS )zCCheck that zipfiles with missing bytes at the end raise BadZipFile.r? rC r� � O, for a Muse of Fire!N) r r rE rF rG r\ r, r^ r� r: )r: r; r� r� �Nr r r �test_damaged_zipfile} s z OtherTests.test_damaged_zipfilec C s� t jtdd��}|�dd� W 5 Q R X | �t �t�� ttd��*}| �t �|�� |�dd� |�� }W 5 Q R X t � � }|� |� | �t �|�� |�dd� | �t �|�� dS )z7Check that is_zipfile() correctly identifies zip files.r? rC r� r� r� r N) rE rF r rG ra r� r7 r� rY r r r8 )r: r� r; Z zip_contentsr r r �test_is_zip_valid_file� s z!OtherTests.test_is_zip_valid_filec C s | � ttjt� d S r )r� r� rE rF r r� r r r �%test_non_existent_file_raises_OSError� s z0OtherTests.test_non_existent_file_raises_OSErrorc C sZ t td�}|�� | �tjtjt� t td��}|�d� W 5 Q R X | �tjtjt� d S )Nr? z short file)r7 r r r� rE r: rF r8 )r: r" r; r r r �!test_empty_file_raises_BadZipFile� s z,OtherTests.test_empty_file_raises_BadZipFilec C s� t �� }tj|dd��}|�dd� W 5 Q R X | �t|jd� | �t|jd� | �t|j � | �t|jdd� tt d��}|� d� W 5 Q R X | �t|j t � dS ) z?Verify that testzip() doesn't swallow inappropriate exceptions.r? rC r� �O, for a Muse of Fire!z bogus.txtr� zzipfile test dataN) r r rE rF rG r� r� rY r7 ri r r8 )r: r0 r� r" r r r �!test_closed_zip_raises_ValueError� s z,OtherTests.test_closed_zip_raises_ValueErrorc C s | � ttjtd� dS )z>Check that bad modes passed to ZipFile constructor are caught.�qN)r� r� rE rF r r� r r r �test_bad_constructor_mode� s z$OtherTests.test_bad_constructor_modec C s� t jtdd��}|�dd� W 5 Q R X t jtdd��F}|�d� | �t|jdd� | �t|jdd� | �t|jdd� W 5 Q R X d S ) z7Check that bad modes passed to ZipFile.open are caught.r? rC r� r� rL r� �UZrUN)rE rF r rG rY r� r� r7 �r: r� r r r �test_bad_open_mode� s z OtherTests.test_bad_open_modec C sp t jtdd��X}|�dd� |�d��6}tt�D ]}| �|�d�d� q0| �|�� d� W 5 Q R X W 5 Q R X dS ) zsCheck that calling read(0) on a ZipExtFile object returns an empty string and doesn't advance file pointer.r? rC r� r� r r+ r� N) rE rF r rG r7 r, r- rX rY )r: r� r" r) r r r � test_read0� s zOtherTests.test_read0c C s0 t jtdd��}| �t|jdd� W 5 Q R X dS )zqCheck that attempting to call open() for an item that doesn't exist in the archive raises a RuntimeError.r? rC r� rL N)rE rF r r� �KeyErrorr7 r� r r r �test_open_non_existent_item� s z&OtherTests.test_open_non_existent_itemc C s | � ttjtdd� dS )zMCheck that bad compression methods passed to ZipFile.open are caught.r? r� N)r� r� rE rF r r� r r r �test_bad_compression_mode� s z$OtherTests.test_bad_compression_modec C s6 d}t �t�|�d��}| �t|jd� W 5 Q R X d S )Nsf PK. �C�@ x PK.. �C�@ � xPK / ! rL r� )rE rF r r r� r� r7 )r: r0 r� r r r � test_unsupported_compression� s z'OtherTests.test_unsupported_compressionc C s< t jtdd��$}|�dd� | �|�� dg� W 5 Q R X dS )zLCheck that a filename containing a null byte is properly terminated.r? rC z foo.txt qqqr� r� N)rE rF r rG rX re r� r r r �test_null_byte_in_filename� s z%OtherTests.test_null_byte_in_filenamec C s< | � tjd� | � tjd� | � tjd� | � tjd� dS )zACheck that ZIP internal structure sizes are calculated correctly.� �. �8 r� N)rX rE ZsizeEndCentDirZsizeCentralDirZsizeEndCentDir64ZsizeEndCentDir64Locatorr� r r r �test_struct_sizes s z OtherTests.test_struct_sizesc C s� t jtdd�� }| �|jd� |�dd� W 5 Q R X t jtdd��}| �|jd� W 5 Q R X d}t jtdd��}||_|�dd� W 5 Q R X t jtdd��}| �|j|� W 5 Q R X d�d d � td �D ��}|�d �}t jtdd��}||_|�dd� W 5 Q R X t jtdd��}| �|j|� W 5 Q R X t jtdd��2}| � t �� |d |_W 5 Q R X |�dd� W 5 Q R X t jtdd��}| �|j|� W 5 Q R X t jtdd��}d|_|�dd� W 5 Q R X t jtdd�� }d|_W 5 Q R X t jtdd��}| �|jd� W 5 Q R X t jtdd��}d|_|�dd� W 5 Q R X t j � t�}t jtdd�� }d|_W 5 Q R X | �|t j � t�k� t jtdd��}| �|jd� W 5 Q R X dS )z8Check that comments on the archive are handled properly.r? rC r+ r� r� rL s9 Bravely taking to his feet, he beat a very brave retreat.� c S s g | ]}d |d d �qS )r r r� r r' r r r r* s z,OtherTests.test_comments.<locals>.<listcomp>� r% s oopss original commentr� s an updated comments original comment that's longers shorter commentN)rE rF r rX r� rG r/ r, r� rx ry r� r� �getsizera )r: r� Zzipfrr� Zcomment2Zoriginal_zip_sizer r r � test_comments sP zOtherTests.test_commentsc C sH t �tdt j��.}|�dd� | �t�� d|_W 5 Q R X W 5 Q R X d S )Nr? r� r� zthis is an error)rE rF r r� rG r� � TypeErrorr� r� r r r �test_unicode_commentB s z OtherTests.test_unicode_commentc C sX t �tdt j��}| �|j� d|_W 5 Q R X t �td��}| �|jd� W 5 Q R X d S )Nr� r� rL )rE rF r r� r r� r� rX r� r r r �$test_change_comment_in_empty_archiveH s z/OtherTests.test_change_comment_in_empty_archivec C s� t �tdt j��}|�dd� W 5 Q R X t �tdt j��}| �|j� d|_W 5 Q R X t �td��}| �|jd� W 5 Q R X d S )Nr? r� r� r� r� rL ) rE rF r r� rG ra r� r� rX r� r r r �'test_change_comment_in_nonempty_archiveO s z2OtherTests.test_change_comment_in_nonempty_archivec C s� t jtdd�}|�� zt jtdd�}W n t jk rH | �d� Y nX t jtdd�}|�� zt jtdd�}W n | �d� Y nX d S )Nr? rC rL z+Unable to create empty ZIP file in 'w' moder� z+Unable to create empty ZIP file in 'a' mode)rE rF r r r: r� r� r r r �test_empty_zipfileX s z OtherTests.test_empty_zipfilec C s* t td�}|�� | �tjtjtd� d S )Nr? rL )r7 r r r� rE r: rF ru r r r �test_open_empty_filei s z OtherTests.test_open_empty_filec C s | � ttjdd� d S )NZ seventies)i� rR rR r r r )r� r� rE r� r� r r r � test_create_zipinfo_before_1980q s �z*OtherTests.test_create_zipinfo_before_1980c C s4 d}t �t�|�d��}| �|�� � W 5 Q R X dS )z>If an extra field in the header is less than 4 bytes, skip it.sn PK ���@���� abc APK ���@���� �� abc PK 3 % rL N)rE rF r r r� ri �r: r� r� r r r �#test_zipfile_with_short_extra_fieldu s � z.OtherTests.test_zipfile_with_short_extra_fieldc C s� d}d}d}t �tdt j��� }|jddd��}|�|� W 5 Q R X |jddd���}| �t�� |jddd� W 5 Q R X | �t�� |jdd d� W 5 Q R X | �t�� |�d d � W 5 Q R X | �t�� |�t d � W 5 Q R X | �t�� |� � W 5 Q R X |�|� W 5 Q R X |jd dd��}|�|� W 5 Q R X W 5 Q R X t �td ��R}| � |� d�|� | � |� d�|� | � |� d �|� | � |� � ddd g� W 5 Q R X d S )Ns" It's fun to charter an accountant!s And sail the wide accountant seas# To find, explore the funds offshorer? r� rC r� ZhandlerL rd ZabcderM �baz)rE rF r r� r7 r8 r� r� rG rb r rX rY re )r: Zmsg1Zmsg2Zmsg3r� Zw2�w1r r r � test_open_conflicting_handles� s2 z(OtherTests.test_open_conflicting_handlesc C s: d}|� d�}t�td��}|�d|� W 5 Q R X t�td���}|�dd���}|�|tj� | � |� � |� |�| tj � | � |� � d� |�|tj � | � |� � |� | � |� d�|||d � � |�dtj � | � |� � t|�� |�dtj� | � |� � d� W 5 Q R X W 5 Q R X t�� }tj|dd��}|�d|� W 5 Q R X tj|dd���}|�dd���}|�|tj� | � |� � |� |�| tj � | � |� � d� |�|tj � | � |� � |� | � |� d�|||d � � |�dtj � | � |� � t|�� |�dtj� | � |� � d� W 5 Q R X W 5 Q R X d S ) Ns Where's Bruce?s Brucer? r� rL r rN rC )�findrE rF r rG r7 r� r� �SEEK_SETrX r� �SEEK_CURrY �SEEK_ENDr^ r r )r: �txt�blocr� r; r0 r r r �test_seek_tell� sB $ zOtherTests.test_seek_tellc C sz d}t �|�}tj|dtjd��}|�dd� W 5 Q R X t�dd ��, t�|��}| �t |j d� W 5 Q R X W 5 Q R X d S )Ns PK r? )r= r� � az zipfile.bz2) r r rE rF r rG �mockZpatchr� r� r� )r: r0 Zzip_filerB r r r �)test_decompress_without_3rd_party_library� s z4OtherTests.test_decompress_without_3rd_party_libraryc C s� d}t �t�|�d���}| �|�� ddg� |�d�}| �|jd� | �|jd� | �|j d� |�d�}| �|jd� | �|jd� | �|j d� | �t |� d��d� | � t j d�� |� d� W 5 Q R X W 5 Q R X d S ) Ns� PK �lH� 8� a� �0�K\d `PK �lH� 8� aPK �lH� 8� bPK ^ / rL r� �br � � zFile name.*differ)rE rF r r rX re rh r8 r7 rg r^ rY �assertRaisesRegexr: �r: r0 r� �zir r r �test_full_overlap� s � z OtherTests.test_full_overlapc C s� d}t �t�|�d���}| �|�� ddg� |�d�}| �|jd� | �|jd� | �|j d� |�d�}| �|jd� | �|jd � | �|j d � | � t j d �� |� d� W 5 Q R X | �t |� d��d � W 5 Q R X d S ) Ns� PK �lHY�84 ( a �PK �lH� 8� b� �0�K\d `PK �lHY�84 ( aPK �lH� 8� $ bPK ^ S rL r� r r �4 i( �$ r r zOverlapped entries)rE rF r r rX re rh r8 r7 rg r r: rY r^ r r r r �test_quoted_overlap� s � z OtherTests.test_quoted_overlapc C s t t� t t� d S r r� r� r r r r� s zOtherTests.tearDownN),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� r� r� r� r� r r r r r r� r r r r r� � sV : & r� c @ s e Zd Zdd� Zdd� ZdS )�AbstractBadCrcTestsc C s: | j }tjt�|�dd��}| �d|�� � W 5 Q R X dS )z>Tests that files with bad CRCs return their name from testzip.rL rC �afileN)�zip_with_bad_crcrE rF r r rX ri r� r r r �test_testzip_with_bad_crc s z-AbstractBadCrcTests.test_testzip_with_bad_crcc C s� | j }tjt�|�dd��}| �tj|jd� W 5 Q R X tjt�|�dd��.}|�dd��}| �tj|j� W 5 Q R X W 5 Q R X tjt�|�dd��H}|�dd��0}d|_ | �tj�� |�d�r�q�W 5 Q R X W 5 Q R X W 5 Q R X dS )zFTests that files with bad CRCs raise a BadZipFile exception when read.rL rC r r� N) r rE rF r r r� r: rY r7 � MIN_READ_SIZE)r: r� 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 Zd ZejZdZdS )�StoredBadCrcTestssx PK ��;:r�� afilehello,AworldPK ��;:r�� � afilePK 3 / N)r� r� r� rE r� r= r r r r r r 1 s �r c @ s e Zd ZejZdZdS )�DeflateBadCrcTestss| PK n} =FAKE n afile�H���W(�/�Iɠ= PK n} =FAKE n � afilePK 3 3 N)r� r� r� rE r r= r r r r r r = s �r c @ s e Zd ZejZdZdS )�Bzip2BadCrcTestss� PK nu =FAKE8 n afileBZh91AY&SYԨ� �@ D�� �P�!����u5:���"�(HjTe?�PK nu =FAKE8 n ��� afilePK 3 [ N)r� r� r� rE r r= r r r r r r J s �r c @ s e Zd ZejZdZdS )�LzmaBadCrcTestss� PK nu =FAKE n afile ] 4I���:3` q!.8 PK nu =FAKE n ��� afilePK 3 > N)r� r� r� rE r r= r r r r r r Z s �r c @ s\ e Zd ZdZdZdZdZdZdd� Zdd � Z d d � Z d d � Z e dd� �Z dd� Zdd� ZdS )�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 c C sd t td��}|�| j� W 5 Q R X t�td�| _t td��}|�| j� W 5 Q R X t�td�| _ d S )Nr5 rL ) r7 r r8 r0 rE rF r� r �data2�zip2r9 r r r r< � s zDecryptionTests.setUpc C s, | j �� t�t� | j�� t�t� d S r )r� r r� r r r r r� r r r r� � s zDecryptionTests.tearDownc C s( | � t| jjd� | � t| jjd� d S )N�test.txt�zero)r� r� r� rY r r� r r r �test_no_password� s z DecryptionTests.test_no_passwordc C s@ | j �d� | �t| j jd� | j�d� | �t| jjd� d S )Ns perlr r! )r� � setpasswordr� r� rY r r� r r r �test_bad_password� s z!DecryptionTests.test_bad_passwordc C sH | j �d� | �| j �d�| j� | j�d� | �| j�d�| j� d S )N� pythonr s 12345r! )r� r# rX rY �plainr �plain2r� r r r �test_good_password� s z"DecryptionTests.test_good_passwordc C sV | � t| jjd� | � t| jjdd� | j t| jjddd� | j t| jjddd� d S )NZunicoder �python)�pwd)r� r� r� r# rY r7 r� r� r r r �test_unicode_password� s z%DecryptionTests.test_unicode_passwordc C sr | j �d� | j}d}|�|�}t|�}| j �dd���0}|�|tj� | � |� � |� |�| tj � | � |� � d� |�|tj � | � |� � |� | � |� |�|||| � � |j }d|_ d|_d|_|�dtj� | � |� � d� |�|tj � | � |� |�|||| � � ||_ |�dtj� | � |� � t|�� |�dtj� | � |� � d� |� � W 5 Q R X d S )Nr% s encryptionr rL r rR r+ )r� r# r& r� r^ r7 r� r� r rX r� r rY r Z _readbufferZ_offsetr )r: r Z test_wordr Zbloc_lenr; Z old_read_sizer r r r � s6 z DecryptionTests.test_seek_tellN)r� r� r� �__doc__r0 r r& r' r< r� r" r$ r r( r+ r r r r r r i s � � r c @ s` e Zd Zedd� �Zdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dS )�"AbstractTestsWithRandomBinaryFilesc C s6 t dd�d t dd� }d�dd� t|�D ��| _d S )Nr �@ r� rR r+ c s s& | ] }t �d t� tdd� �V qdS )z<fi���r N)r r0 r r r' r r r r � s �z@AbstractTestsWithRandomBinaryFiles.setUpClass.<locals>.<genexpr>)r r/ r, r0 )r2 Z datacountr r r r3 � s �z-AbstractTestsWithRandomBinaryFiles.setUpClassc C s& t td��}|�| j� W 5 Q R X d S r4 r6 r9 r r r r< � s z(AbstractTestsWithRandomBinaryFiles.setUpc C s t t� t t� d S r r� r� r r r r� � s z+AbstractTestsWithRandomBinaryFiles.tearDownc C s6 t �|d|�� }|�td� |�tt� W 5 Q R X d S )Nr? r@ )rE rF r8 r r& r r r rK � s z4AbstractTestsWithRandomBinaryFiles.make_test_archivec C sl | � ||� t�|d|��H}|�t�}| �t|�t| j�� | �|| j� | �|�d�| j� W 5 Q R X d S )NrL r@ )rK rE rF rY r rX r^ r0 )r: r" r= rI �testdatar r r rs � s z+AbstractTestsWithRandomBinaryFiles.zip_testc C s t | �D ]}| �|| j� qd S r rt ru r r r � test_read� s z,AbstractTestsWithRandomBinaryFiles.test_readc C s� | � ||� t�|d|���}g }|�t��"}|�d�}|s<qH|�|� q,W 5 Q R X g }|�d��"}|�d�}|srq~|�|� qbW 5 Q R X d�|�} | �t | �t | j �� | �| | j � d�|�} | �t | �t | j �� | �| | j � W 5 Q R X d S rw ) rK rE rF r7 r rY ry r/ rX r^ r0 ) r: r" r= rI rz r{ r| r} r~ Z testdata1Z testdata2r r r r � s( z0AbstractTestsWithRandomBinaryFiles.zip_open_testc C s t | �D ]}| �|| j� qd S r r� ru r r r r� s z,AbstractTestsWithRandomBinaryFiles.test_openc C s� | � ||� t�|d|��p}g }|�t��(}|�tdd��}|sBqN|�|� q,W 5 Q R X d�|�}| � t |�t | j �� | � || j � W 5 Q R X d S r� ) rK rE rF r7 r rY r ry r/ rX r^ r0 )r: r" r= rI rz r{ r| r/ r r r r� s z7AbstractTestsWithRandomBinaryFiles.zip_random_open_testc C s t | �D ]}| �|| j� qd S r r� ru r r r r� & s z3AbstractTestsWithRandomBinaryFiles.test_random_openN)r� r� r� r� r3 r< r� rK rs r0 r r� r� r� r r r r r- � s r- c @ s e Zd ZejZdS )� StoredTestsWithRandomBinaryFilesNrO r r r r r1 + s r1 c @ s e Zd ZejZdS )�!DeflateTestsWithRandomBinaryFilesNrE r r r r r2 / s r2 c @ s e Zd ZejZdS )� Bzip2TestsWithRandomBinaryFilesNr r r r r r3 4 s r3 c @ s e Zd ZejZdS )� LzmaTestsWithRandomBinaryFilesNr r r r r r4 9 s r4 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S ) �Tellablec C s || _ d| _d S �Nr )r; �offsetr9 r r r �__init__A s zTellable.__init__c C s | j �|�}| j|7 _|S r )r; r8 r7 )r: r0 �nr r r r8 E s zTellable.writec C s | j S r )r7 r� r r r r� J s z Tellable.tellc C s | j �� d S r �r; �flushr� r r r r; M s zTellable.flushN)r� r� r� r8 r8 r� r; r r r r r5 @ s r5 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� Unseekablec C s || _ d S r )r; r9 r r r r8 Q s zUnseekable.__init__c C s | j �|�S r )r; r8 r� r r r r8 T s zUnseekable.writec C s | j �� d S r r: r� r r r r; W s zUnseekable.flushN)r� r� r� r8 r8 r; r r r r r< P s r< c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�UnseekableTestsc C s dd� t tfD ]�}| j|d��� t�� }|�d� t�|�}t�||�dtj �� }|� dd� |� dd � W 5 Q R X | � |� � d d � d � tj|d d ��R}|� d��}| � |�� d� W 5 Q R X |� d��}| � |�� d � W 5 Q R X W 5 Q R X W 5 Q R X qd S )Nc S s | S r r �r" r r r rt \ r+ z/UnseekableTests.test_writestr.<locals>.<lambda>��wrapper� abcr? �ones� 111�twos� 222rN � abcPKrL rC )r5 r< �subTestr r r8 �BufferedWriterrE rF r� rG rX r\ r7 rY )r: r@ r" �bfrI r� �zopenr r r � test_writestr[ s z UnseekableTests.test_writestrc C sR dd� t tfD �]<}| j|d���$ t�� }|�d� t�|�}t�||�dtj ��j}| � t t � t t d��}|�d� W 5 Q R X |�t d� t t d��}|�d � W 5 Q R X |�t d � W 5 Q R X | �|�� d d � d � tj|d d��R}|� d��}| �|�� d� W 5 Q R X |� d ��}| �|�� d � W 5 Q R X W 5 Q R X W 5 Q R X qd S )Nc S s | S r r r> r r r rt l r+ z,UnseekableTests.test_write.<locals>.<lambda>r? rA r? r5 rC rB rE rD rN rF rL rC )r5 r< rG r r r8 rH rE rF r� � addCleanupr r r7 rX r\ rY )r: r@ r" rI rI �f2r� rJ r r r � test_writek s& zUnseekableTests.test_writec C s dd� t tfD ]�}| j|d��� t�� }|�d� t�|�}t�||�dtj ��J}|� dd��}|�d� W 5 Q R X |� dd��}|�d � W 5 Q R X W 5 Q R X | � |� � d d � d � t�|��*}| � |� d�d� | � |� d�d � W 5 Q R X W 5 Q R X qd S ) Nc S s | S r r r> r r r rt � r+ z1UnseekableTests.test_open_write.<locals>.<lambda>r? rA r? rB rC rD rE rN rF )r5 r< rG r r r8 rH rE rF r� r7 rX r\ rY )r: r@ r" rI r� rJ r r r �test_open_write� s z UnseekableTests.test_open_writeN)r� r� r� rK rN rO r r r r r= Z s r= c @ sh e Zd Zedd� �Zdd� Zdd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� ZdS )�TestsWithMultipleOpensc C s dt d� | _dt d� | _d S )NrC �' rE )r �data1r r1 r r r r3 � s z!TestsWithMultipleOpens.setUpClassc C s< t �|dt j��"}|�d| j� |�d| j� W 5 Q R X d S )Nr? rB rD )rE rF r rG rR r )r: r" rI r r r rK � s z(TestsWithMultipleOpens.make_test_archivec C s� t | �D ]�}| �|� tj|dd��x}|�d��H}|�d��2}|�d�}|�d�}||�� 7 }||�� 7 }W 5 Q R X W 5 Q R X | �||� | �|| j� W 5 Q R X qd S )NrL rC rB �� )r# rK rE rF r7 rY rX rR �r: r" r� �zopen1�zopen2rR r r r r �test_same_file� s z%TestsWithMultipleOpens.test_same_filec C s� t | �D ]�}| �|� tj|dd��z}|�d��H}|�d��2}|�d�}|�d�}||�� 7 }||�� 7 }W 5 Q R X W 5 Q R X | �|| j� | �|| j� W 5 Q R X qd S )NrL rC rB rD rS � r# rK rE rF r7 rY rX rR r rT r r r �test_different_file� s z*TestsWithMultipleOpens.test_different_filec C s� t | �D ]�}| �|� tj|dd��z}|�d��H}|�d�}|�d��(}|�d�}||�� 7 }||�� 7 }W 5 Q R X W 5 Q R X | �|| j� | �|| j� W 5 Q R X qd S )NrL rC rB rS rD rX )r: r" r� rU rR rV r r r r �test_interleaved� s z'TestsWithMultipleOpens.test_interleavedc C s� t | �D ]�}| �|� t�� �j}t�|d��&}|�|�d��}|�|�d��}W 5 Q R X |�d�}|�d�}||�� 7 }||�� 7 }W 5 Q R X | � || j � | � || j � qd S )NrL rB rD rS ) r# rK � contextlib� ExitStackrE rF � enter_contextr7 rY rX rR r )r: r" �stackr� rU rV rR r r r r �test_read_after_close� s z,TestsWithMultipleOpens.test_read_after_closec C s� t | �D ]�}t�|dtj��B}|�d| j� |�d| j� |�d��}|�d�}W 5 Q R X W 5 Q R X | � || jd d� � t�|d��}|�d�}|�d�}W 5 Q R X | � || j� | � || j� qd S )Nr? rB rD rS rL ) r# rE rF r rG rR r r7 rY rX �r: r" r� rU rR r r r r �test_read_after_write� s z,TestsWithMultipleOpens.test_read_after_writec C s� t | �D ]�}t�|dtj��B}|�d| j� |�d�� }|�d� |�d| j� W 5 Q R X W 5 Q R X t�|d��}|�d�}|�d�}W 5 Q R X | � || j� | � || j� qd S )Nr? rB rS rD rL ) r# rE rF r rG rR r7 rY r rX r` r r r �test_write_after_read� s " z,TestsWithMultipleOpens.test_write_after_readc C s| | � t� tjtdd��4}td�D ]$}|�d� |�d��}W 5 Q R X q"W 5 Q R X ttj��}| � |� � d� W 5 Q R X d S )NrL rC r� rB ) rK r rE rF r, rY r7 r� �devnullZ assertLess�fileno)r: r� r� rU r" r r r �test_many_opens� s z&TestsWithMultipleOpens.test_many_opensc C s� t �tdt j��}|�d| j� W 5 Q R X t �tdt j��X}|�dd��@}|�d�}|�dd��}|�| j � W 5 Q R X ||�� 7 }W 5 Q R X W 5 Q R X | � || j� t �t��}| � |�d�| j � W 5 Q R X d S )Nr? rB r� rL rS rD ) rE rF r r rG rR r7 rY r8 r rX )r: r� Zr1rR r� r r r �test_write_while_reading� s z/TestsWithMultipleOpens.test_write_while_readingc C s t t� d S r rI r� r r r r� s z TestsWithMultipleOpens.tearDownN)r� r� r� r� r3 rK rW rY rZ r_ ra rb re rf r� r r r r rP � s rP c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�TestWithDirectoryc C s t �t� d S r )r� r� r r� r r r r< s zTestWithDirectory.setUpc C s� t �td���}|�t� W 5 Q R X | �tj�tj� td��� | �tj�tj� tdd��� | �tj� tj� tddd��� d S )N� zipdir.zipr� r rA ) rE rF r r� r ra r� r� �isdirr/ r� r� r r r �test_extract_dir s z"TestWithDirectory.test_extract_dirc C s t �t j�td�� | �� d S �Nr� )r� r� r� r/ r rj r� r r r � test_bug_6050 s z TestWithDirectory.test_bug_6050c C s� t j�td�}t �|� t �|�jd@ }t�t d��|}|� |� |j d }| � |j �d�� | �|j|d> dB � |� |d� |j d }| � |j d � | �|j|d> dB � W 5 Q R X t�t d ���}|j d }| � |j �d�� | �|j|d> dB � |j d }| � |j d � | �|j|d> dB � t j�td �}t �|� |�|� | � t j�t j�|d��� | �tt �|��d � W 5 Q R X d S ) Nr� r� r? r z/x/r �yrR zy/rL r� r� )r� r� r/ r r� r �st_moderE rF r r8 r� ra rW rd rX r� r� ri r^ �listdir)r: �dirpathrD r� r� r� r r r �test_write_dir s0 z TestWithDirectory.test_write_dirc C s� t �t j�td�� t�td��8}|�dd� |j d }| � |j d� | � |j d� W 5 Q R X t�td���}|j d }| � |j �d�� | � |j d� t j�td�}t �|� |�|� | � t j�t j�|d��� | � t �|�dg� W 5 Q R X d S ) Nr� r? zx/r+ r i �ArL r� )r� r� r� r/ r rE rF r rG r� rX rW r� ra rd r� ri ro )r: r� r� r� r r r �test_writestr_dir9 s z#TestWithDirectory.test_writestr_dirc C s t t� tj�t�r tt� d S r )r r r� r� r� r r r� r r r r� J s zTestWithDirectory.tearDownN) r� r� r� r< rj rl rq rr r� r r r r rg s rg c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd S ) � ZipInfoTestsc C sH t j�t�}| �t�|j�d� | �|� � � | �|j t j � t�� d S �Nztest_zipfile.py)rE r� � from_filerb rX � posixpathrl rW r �is_dirrg r� r� r� �r: r r r r �test_from_fileQ s zZipInfoTests.test_from_filec C sN t j�t�t��}| �t�|j �d� | � |� � � | �|j t j�t�� d S rt )rE r� ru r� r� rb rX rv rl rW r rw rg r� r� r� rx r r r �test_from_file_pathlikeW s z$ZipInfoTests.test_from_file_pathlikec C sP t j�t�t�d�}| �t�|j �d� | � |� � � | �|j tj �t�� d S )Nr! )rE r� ru r� �fsencoderb rX rv rl rW r rw rg r� r� rx r r r �test_from_file_bytes] s z!ZipInfoTests.test_from_file_bytesc C sd t td��P}tj�|�� d�}| �t�|j �d� | � |� � � | �|j t j�t�� W 5 Q R X d S )Nr� r! )r7 rb rE r� ru rd rX rv rl rW r rw rg r� r� r� )r: r" r r r r �test_from_file_filenoc s z"ZipInfoTests.test_from_file_filenoc C s` t j�t j�t��}tj�|d�}| �|j d� | � |� � � | �|j tj � | �|jd� d S )NZ stdlib_testsz stdlib_tests/r )r� r� ra �abspathrb rE r� ru rX rW ra rw r� r� rg )r: rp r r r r � test_from_dirj s zZipInfoTests.test_from_dirN)r� r� r� ry rz r| r} r r r r r rs P s rs c @ sH e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zed d � �Z d d� Z dS )�CommandLineTestc O s( t jd|�|�\}}}|�tj�� d�S )N�-mrE r )r� rE )r Zassert_python_okr� r� �linesepr� )r: �argsrH �rc�out�errr r r � zipfilecmdu s � zCommandLineTest.zipfilecmdc G s t jd|�� S )Nr� rE )r� rE )r Zassert_python_failure)r: r� r r r �zipfilecmd_failurez s z"CommandLineTest.zipfilecmd_failurec C s| | � � \}}}| �|d� | �d|�� � | �d|�� � | �d|�� � | � dd�\}}}| �|d� | �|�� d� d S )Nr+ s usages errors required�-lr� )r� rX r_ r; ZassertNotEqual�strip)r: r� r� r� r r r � test_bad_use} s z CommandLineTest.test_bad_usec C sX t d�}dD ] }| �||�}| �|�� d� q t d�}| �d|�\}}}| �|d� d S )Nrh )�-tz--tests Done testingz testtar.tarr� r+ )r r� rX �rstripr� )r: �zip_name�optr� r� r� r r r �test_test_command� s z!CommandLineTest.test_test_commandc C sl t d�}t�� }t�|d��}|�|� W 5 Q R X |�� �dd�}dD ] }| j||dd�}| � ||� qFd S )Nrh rL r% �backslashreplace)r� z--listzascii:backslashreplace)ZPYTHONIOENCODING) r r rZ rE rF r[ r\ r� r� rX )r: r� rn ZtfZexpectedr� r� r r r �test_list_command� s �z!CommandLineTest.test_list_commandc C s | � tt� ttd��}|�d� W 5 Q R X t�t� | � tt� ttj � td�d��}|�d� W 5 Q R X ttg}ttd td g}dD ]�}zt| j |t f|�� }| � |d� t�t ��B}| � |�� |� | � |�|d �d � | � |�|d �d � W 5 Q R X W 5 tt � X q�d S ) Nr? ztest 1zfile.txtztest 2r� z /file.txt)z-cz--creater+ r s test 1r� s test 2)rL r r r7 r8 r� r� � TESTFNDIRr r� r/ r r� rX rE rF re rY )r: r" �filesre r� r� rB r r r �test_create_command� s$ $z#CommandLineTest.test_create_commandc C s� t d�}dD ]�}t� ��}| �|||�}| �|d� t�|���}|�� D ]x}tj� ||j � dtj ��}|� � r�| �tj�|�� qF| �tj�|�� t|d�� }| �|�� |�|�� W 5 Q R X qFW 5 Q R X W 5 Q R X q d S )Nrh )z-ez --extractr+ r� r� )r r r� rX rE rF rf r� r� r/ rW r� re rw ra ri r� r7 rY ) r: r� r� r� r� rB r r� r" r r r �test_extract_command� s � z$CommandLineTest.test_extract_commandN) r� r� r� r� r� r� r� r� r r� r� r r r r r� s s r� c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Ze� e j d �e� e � d e j�d �d d� ��Ze� e j d �e� e � d e j�d �dd� ��ZdS )�TestExecutablePrependedZipz@Test our ability to open zip files with an executable prepended.c C s t ddd�| _t ddd�| _d S )NZ exe_with_zipZ ziptestdata)r� Z exe_with_z64)r �exe_zip� exe_zip64r� r r r r< � s z TestExecutablePrependedZip.setUpc C sV | � t�|�d|� �� t�|��*}|�� D ]}|�|�}| �d|� q,W 5 Q R X d S )Nzis_zipfile failed on s FAVORITE_NUMBER)ra rE r� rF re rY r_ )r: rU rI r9 r0 r r r �_test_zip_works� s � z*TestExecutablePrependedZip._test_zip_worksc C s | � | j� d S r )r� r� r� r r r � test_read_zip_with_exe_prepended� s z;TestExecutablePrependedZip.test_read_zip_with_exe_prependedc C s | � | j� d S r )r� r� r� r r r �"test_read_zip64_with_exe_prepended� s z=TestExecutablePrependedZip.test_read_zip64_with_exe_prependedzsys.executable required.z /bin/bashz#Test relies on #!/bin/bash working.c C s" t �| jtjg�}| �d|� d S �Ns number in executable: 5)� subprocess� check_outputr� �sys� executabler_ �r: �outputr r r �test_execute_zip2� s z,TestExecutablePrependedZip.test_execute_zip2c C s" t �| jtjg�}| �d|� d S r� )r� r� r� r� r� r_ r� r r r �test_execute_zip64� s z-TestExecutablePrependedZip.test_execute_zip64N)r� r� r� r, r< r� r� r� r� Z skipUnlessr� r� r� rW �X_OKr� r� r r r r r� � s � �r� c @ s e Zd ZG dd� d�ZdS )�jaracoc @ s e Zd ZG dd� d�ZdS )zjaraco.itertoolsc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )zjaraco.itertools.Counterc C s d| _ t|�| _d S r6 )r� �iter� _orig_iter)r: r) r r r r8 � s z!jaraco.itertools.Counter.__init__c C s | S r r r� r r r �__iter__� s z!jaraco.itertools.Counter.__iter__c C s t | j�}| jd7 _|S r� )�nextr� r� )r: �resultr r r �__next__� s z!jaraco.itertools.Counter.__next__N)r� r� r� r8 r� r� r r r r �Counter� s r� N)r� r� r� r� r r r r r> � s r> N)r� r� r� r> r r r r r� � s r� c C s& t j�| �� �D ]}| �|d� q| S )z} Given a writable zip file zf, inject directory entries for any directories implied by the presence of children. r+ )rE � CompleteDirs� _implied_dirsre rG )rB rU r r r �add_dirs� s r� c C sZ t �� } t�| d�}|�dd� |�dd� |�dd� |�dd � |�d d � d |_|S ) u: Create a zip file with this structure: . ├── a.txt ├── b │ ├── c.txt │ ├── d │ │ └── e.txt │ └── f.txt └── g └── h └── i.txt This fixture has the following key characteristics: - a file at the root (a) - a file two levels deep (b/d/e) - multiple files in a directory (b/c, b/f) - a directory containing only a directory (g/h) "alpha" because it uses alphabet "rep" because it's a representative example r? r� s content of azb/c.txts content of cz b/d/e.txts content of ezb/f.txts content of fz g/h/i.txts content of iz alpharep.zip)r r rE rF rG rW )r0 rB r r r �build_alpharep_fixture s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,S )-�TestPathc C s t �� | _| �| jj� d S r )r[ r\ �fixturesrL r r� r r r r< - s zTestPath.setUpc c s@ | � � � t� V W 5 Q R X | � � � tt� �V W 5 Q R X d S r )rG r� r� r� r r r �zipfile_alpharep1 s zTestPath.zipfile_alpharepc c sh t �| j�t� ��}| �� D ]F}|j}|�� ||j }|� d��}|� |� � � W 5 Q R X |V q d S r4 ) r� r� r� r] r r� r; r rW r7 r8 r\ )r: Ztmpdir�alpharep�bufferr� �strmr r r �zipfile_ondisk7 s zTestPath.zipfile_ondiskc C sV | � � D ]H}t�|�}|�� \}}}|�� \}}}|�� \} |�� \} | �� \} qd S r �r� rE r� �iterdir) r: r� �rootr� r �grA r" r r< �hr) r r r �test_iterdir_and_typesA s z TestPath.test_iterdir_and_typesc C s | � � D ]}t�|�}qd S r �r� rE r� �r: r� r� r r r �test_subdir_is_dirQ s zTestPath.test_subdir_is_dirc C sF | � � D ]8}t�|�}|�� \}}}|�� �}|�� }W 5 Q R X qd S r )r� rE r� r� r7 rY )r: r� r� r� r r� r� r0 r r r r� Y s zTestPath.test_openc C s* | � � D ] }t�|�}|�� \}}}qd S r r� )r: r� r� r� r r� r r r r0 a s zTestPath.test_readc C s< | � � D ].}t�|�}|�d�}|�d��d��d�}qd S �Nr� r r ze.txt)r� rE r� �joinpath�r: r� r� r� r< r r r � test_joinpathh s zTestPath.test_joinpathc C s4 | � � D ]&}t�|�}|d }|d d d }qd S r� r� r� r r r �test_traverse_truedivp s z TestPath.test_traverse_truedivc C s* | � � D ] }t�t|��}t�|� qdS )zN zipfile.Path should be constructable from a path-like object N)r� r� r� rd rE )r: r� Zpathliker r r �test_pathlike_constructionx s z#TestPath.test_pathlike_constructionc C s* | � � D ] }t�|�}|t�d� qd S rk )r� rE r� r� r� r r r �test_traverse_pathlike� s z TestPath.test_traverse_pathlikec C s | � � D ]}t�|�}qd S r r� r� r r r � test_parent� s zTestPath.test_parentc C s | � � D ]}t�|�}qd S r r� r� r r r �test_dir_parent� s zTestPath.test_dir_parentc C s | � � D ]}t�|�}qd S r r� r� r r r �test_missing_dir_parent� s z TestPath.test_missing_dir_parentc C sP | � � D ]B}t�|�}|�� \}}}|�dd� |�dd� |d �� \}qdS )zk If the underlying zipfile is changed, the Path object should reflect that change. r� r� z bar/baz.txtr� r� N)r� rE r� r� rG )r: r� r� r� r r� r� r r r �test_mutability� s zTestPath.test_mutabilityi c C s@ t �� }t�|d�}ttt| j��D ]}|�||� q$d|_ |S )zACreate a read-only zipfile with a huge number of entries entries.r? rL ) r r rE rF �maprd r, �HUGE_ZIPFILE_NUM_ENTRIESrG rD )r: r� rB �entryr r r � huge_zipfile� s zTestPath.huge_zipfilec C s6 t �| �� �}tj�|�� �}|D ]}|�d� q"dS )zE Ensure joinpath on items in zipfile is linear time. �suffixN)rE r� r� r� r> r� r� r� )r: r� �entriesr� r r r �test_joinpath_constant_time� s z$TestPath.test_joinpath_constant_timec C s" dd� t d�D �}tj�|� d S )Nc S s g | ]}d � tjt|� ��qS )r� )r/ �stringZascii_lowercaserd )r( r9 r r r r* � s z:TestPath.test_implied_dirs_performance.<locals>.<listcomp>rQ )r, rE r� r� r� r r r � test_implied_dirs_performance� s z&TestPath.test_implied_dirs_performancec C sL t �� }t�|d�}|�dd� |�dd� |�dd� d|_t�|�}dS )z� Path should handle malformed paths gracefully. Paths with leading slashes are not visible. Paths with dots are treated like regular files. r? z/one-slash.txtrJ z//two-slash.txtz ../parent.txtr� N)r r rE rF rG rW r� )r: r0 rB r� r r r �test_malformed_paths� s z TestPath.test_malformed_pathsc C sH t �� }t�|d�}|�dd� |�dd� d|_t�|�}|�� }dS )z� Path segments with special characters are readable. On some platforms or file systems, characters like ``:`` and ``?`` are not allowed, but they are valid in the zip file. r? zpath?rJ z V: NMS.flacs fLaC...r� N)r r rE rF rG rW r� r� )r: r0 rB r� �contentsr r r �test_unsupported_names� s z TestPath.test_unsupported_namesc C sF t �� }t�|d�}|�t�d|�d� d|_t�|�}|� � \}dS )z@ In a zip file, backslashes are not separators. r? zfoo\barrJ r� N) r r rE rF rG � DirtyZipInfo�for_namerW r� r� )r: r0 rB r� �firstr r r � test_backslash_not_separator� s z%TestPath.test_backslash_not_separatorN)r� r� r� r< r� r� r� r� r� r0 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 Zd ZdZ� fdd�Zedd� �Z� ZS )r� z# Bypass name sanitization. c s t � j|f|�|� || _d S r )r� r8 rW )r: rW r� rH r� r r r8 � s zDirtyZipInfo.__init__c C sZ | |t �t � � �dd� d�}|j|_|j|_|j�d�rPd|_| jdO _nd|_|S )zy Construct the same way that ZipFile.writestr does. TODO: extract this functionality and re-use N� )rW r� r� i �Ar r� ) rb r� r= r� r> Zcompress_levelrW rd r� )r2 rU �archiver: r r r r� � s zDirtyZipInfo.for_name)r� r� r� r, r8 r� r� r� r r r� r r� � s r� �__main__)Yr[ �importlib.utilrh r r> r� r� rv r� r r� r� rb r� Z unittest.mockr rE Ztempfiler r r r Z test.supportr r r r r r r r r r r r r� r- Z DATAFILES_DIRr� r r# r$ ZTestCaser� r r r r r$ rD rF rG rH rN rP rQ rR rS r� r� r r r r r r r- r1 r2 r3 r4 r5 r< r= rP rg rs r� r� �tupleZconsumer� r� r� r� r� r� r� �mainr r r r �<module> s 0� � 5 � � � � � � � B A 0 g[ � � � � 7}A#P' $ J
💻
System Info
Current path
/opt/alt/python38/lib64/python3.8/test/__pycache__
Contents
0 folders, 1461 files
Disk free
100.90 GB
PHP version
8.3.30
🚀
Quick Actions
📍 Script location
🌐 Web root
🗑️ Temp
📁 /opt/alt/python38/lib64/python3.8/test/__pycache__
⚡ Nexus File Manager • 1461 items