shell bypass 403
o
a b�? � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z G dd� dej
�ZG dd� dej
�Zed krJe�� dS dS )
z#Test harness for the zipapp module.� N)�
requires_zlib)�patchc @ s
e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
e � 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.d/� Zd0d1� Z e � e j d2kd3�d4d5� �Z!e � e j d2kd3�d6d7� �Z"d8S )9�
ZipAppTestz!Test zipapp module functionality.c C �& t �� }| �|j� t�|j�| _d S �N��tempfile�TemporaryDirectoryZ
addCleanup�cleanup�pathlib�Path�name�tmpdir��selfr � r �9/usr/local/python-3.10/lib/python3.10/test/test_zipapp.py�setUp � zZipAppTest.setUpc C sN | j d }|�� |d �� | j d }t�t|�t|�� | �|�� � d S �N�source� __main__.py�
source.pyz�r �mkdir�touch�zipapp�create_archive�str�
assertTrue�is_file�r r �targetr r r �test_create_archive s
z ZipAppTest.test_create_archivec C sF | j d }|�� |d �� | j d }t�||� | �|�� � d S r )r r r r r r r r! r r r � test_create_archive_with_pathlib! s
z+ZipAppTest.test_create_archive_with_pathlibc C s� | j d }|�� |d �� |d �� |d �� |d d �� t�� }t�t|�|� |�d� t �
|d��}| � d|� � � | � d |� � � W d � d S 1 sVw Y d S )
Nr r �fooZbarz __init__.pyr �rzfoo/zbar/)
r r r �io�BytesIOr r r �seek�zipfile�ZipFile�assertIn�namelist�r r r" �zr r r � test_create_archive_with_subdirs* s
"�z+ZipAppTest.test_create_archive_with_subdirsc C s� dd� }| j d }|�� |d �� |d �� |d �� | j d }tj|||d� t�|d ��!}| �d|�� � | �d|�� � | � d|�� � W d � d S 1 sVw Y d S )
Nc S s
| j dkS )Nz.pyc)�suffix��pathr r r �skip_pyc_files< s
zBZipAppTest.test_create_archive_with_filter.<locals>.skip_pyc_filesr r �test.pyztest.pycr ��filterr& )
r r r r r r* r+ r, r- Z assertNotIn)r r4 r r" r/ r r r � test_create_archive_with_filter9 s
"�z*ZipAppTest.test_create_archive_with_filterc C s� dd� }| j d }|�� |d �� |d �� |d �� |d d �� | j d }tj|||d � t�|d
��#}| �t|� � �d � | �
d|� � � | �
d|� � � W d � d S 1 s`w Y d S ) Nc S s | j d dkS )Nr �dummy)�partsr2 r r r �skip_dummy_dirO s zIZipAppTest.test_create_archive_filter_exclude_dir.<locals>.skip_dummy_dirr r r5 r9 ztest2.pyr r6 r& � ) r r r r r r* r+ � assertEqual�lenr- r, )r r; r r" r/ r r r �&test_create_archive_filter_exclude_dirK s
"�z1ZipAppTest.test_create_archive_filter_exclude_dirc C sH | j d }|�� |d �� t�t|�� | j d }| �|�� � d S r r )r r Zexpected_targetr r r �"test_create_archive_default_target_ s
z-ZipAppTest.test_create_archive_default_targetc C s� | j d }|�� |d �� |d �� | j d }tj||dd� t�|d��}dD ]
}| �|�|�j tj
� q+W d � d S 1 sDw Y d S ) Nr r r5 r T)�
compressedr& )r r5 ) r r r r r r* r+ r= ZgetinfoZ
compress_type� ZIP_DEFLATED)r r r" r/ r
r r r �$test_create_archive_with_compressionh s
��"�z/ZipAppTest.test_create_archive_with_compressionc C sn | j d }|�� |d �� | j d }| �tj�� t�t|�t|�� W d � d S 1 s0w Y d S )Nr �foo.pyr �r r r � assertRaisesr � ZipAppErrorr r r! r r r � test_no_mainw s
"�zZipAppTest.test_no_mainc C sr | j d }|�� |d �� | j d }| �tj�� tjt|�t|�dd� W d � d S 1 s2w Y d S )Nr r r �
pkg.mod:fn��mainrE r! r r r �test_main_and_main_py� s
"�z ZipAppTest.test_main_and_main_pyc C s� | j d }|�� |d �� | j d }tjt|�t|�dd� t�t|�d��}| �d|� � � | �d|�
d�� W d � d S 1 sEw Y d S ) Nr rD r rI rJ r& r s pkg.mod.fn()) r r r r r r r* r+ r, r- �readr. r r r �test_main_written� s
"�z ZipAppTest.test_main_writtenc C s� | j d }|�� |d �� |d �� | j d }tjt|�t|�dd� t�t|�d��}| �d|� � �
d �� W d � d S 1 sEw Y d S )
Nr rD zbar.pyr rI rJ r& � r ) r r r r r r r* r+ r= r- �countr. r r r �test_main_only_written_once� s
"�z&ZipAppTest.test_main_only_written_oncec
C s� | j d }|�� | j d }g d�}|D ]:}| j|d��* | �tj�� tjt|�t|�|d� W d � n1 s:w Y W d � n1 sIw Y qd S )Nr r )
� r% zfoo:z:barz12:barza.b.c.:dz.a:bza:b.za:.bz a:silly namerJ )r r ZsubTestrF r rG r r )r r r" ZproblemsrK r r r �test_main_validation� s
�� ���z ZipAppTest.test_main_validationc C s~ | j d }|�� |d �� | j d }t�t|�t|�� |�d��}| �|�d�d� W d � d S 1 s8w Y d S )Nr r r �rbr< � #!) r r r r r r �openZassertNotEqualrM �r r r" �fr r r �test_default_no_shebang� s
"�z"ZipAppTest.test_default_no_shebangc C s� | j d }|�� |d �� | j d }tjt|�t|�dd� |�d��}| �|�d�d� | �d |� � � W d � d S 1 sBw Y d S )
Nr r r �python�Z interpreterrT r< rU s python
)
r r r r r r rV r= rM �readlinerW r r r �test_custom_interpreter� s
"�z"ZipAppTest.test_custom_interpreterc C sR | j d }|�� |d �� t�� }tjt|�|dd� | �|� � �
d�� d S )Nr r rZ r[ s #!python
) r r r r' r( r r r r �getvalue�
startswithr! r r r �test_pack_to_fileobj� s
z ZipAppTest.test_pack_to_fileobjc C sZ | j d }|�� |d �� | j d }tjt|�t|�dd� | �t�t|��d� d S �Nr r r rZ r[ �r r r r r r r= �get_interpreterr! r r r �test_read_shebang� s
z ZipAppTest.test_read_shebangc C sV | j d }|�� |d �� | j d }t�t|�t|�� | �t�t|��d � d S r rb r! r r r �test_read_missing_shebang� s
z$ZipAppTest.test_read_missing_shebangc C s| | j d }|�� |d �� | j d }tjt|�t|�dd� | j d }tjt|�t|�dd� | �t�t|��d� d S )Nr r r rZ r[ � changed.pyz� python2.7rb �r r r" �
new_targetr r r �test_modify_shebang� �
z ZipAppTest.test_modify_shebangc C st | j d }|�� |d �� | j d }tjt|�t|�dd� t�� }tjt|�|dd� | �|� � �
d�� d S )Nr r r rZ r[ rg � #!python2.7
) r r r r r r r' r( r r^ r_ rh r r r � test_write_shebang_to_fileobj� s
z(ZipAppTest.test_write_shebang_to_fileobjc C sh | j d }|�� |d �� | j d }| j d }tj||dd� tj||dd� | �t�|�d� d S )Nr r z target1.pyzz target2.pyzrZ r[ rg )r r r r r r= rc )r r Ztarget1Ztarget2r r r �test_read_from_pathobj� s
z!ZipAppTest.test_read_from_pathobjc C s~ | j d }|�� |d �� | j d }t�� }tjt|�|dd� t�� }|�d� tj||dd� | � |�
� � d�� d S ) Nr r r rZ r[ r rg rl ) r r r r' r( r r r r) r r^ r_ )r r r" Z temp_archiveri r r r �test_read_from_fileobj s
z!ZipAppTest.test_read_from_fileobjc C s| | j d }|�� |d �� | j d }tjt|�t|�dd� | j d }tjt|�t|�d d� | �t�t|��d � d S )Nr r r rZ r[ rf rb rh r r r �test_remove_shebang rk z ZipAppTest.test_remove_shebangc C s� | j d }|�� |d �� t�� }tjt|�|dd� t�� }|�d� tj||d d� |�d� t �
|d��}| � t |�
� �dh� W d � d S 1 sPw Y d S )Nr r rZ r[ r r& )r r r r' r( r r r r) r* r+ r= �setr- )r r r" ri r/ r r r � test_content_of_copied_archive s
"�z)ZipAppTest.test_content_of_copied_archive�win32z*Windows does not support an executable bitc C sZ | j d }|�� |d �� | j d }tjt|�t|�dd� | �|�� jtj @ � d S ra )
r r r r r r r �stat�st_mode�S_IEXECr! r r r �test_shebang_is_executable. �
z%ZipAppTest.test_shebang_is_executablec C sZ | j d }|�� |d �� | j d }tjt|�t|�d d� | �|�� jtj @ � d S )Nr r r r[ )
r r r r r r Z assertFalsert ru rv r! r r r �!test_no_shebang_is_not_executable9 rx z,ZipAppTest.test_no_shebang_is_not_executableN)#�__name__�
__module__� __qualname__�__doc__r r# r$ r0 r8 r? r@ r rC rH rL rN rQ rS rY r] r` rd re rj rm rn ro rp rr �unittestZskipIf�sys�platformrw ry r r r r r sF
�
�r c @ s^ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
de j d�dd� �Z
dd� ZdS )�ZipAppCmdlineTestz$Test zipapp module command line API.c C r r r r r r r r I r zZipAppCmdlineTest.setUpc C s8 | j d }|�� |d �� | j d }t�||� |S r )r r r r r r! r r r � make_archiveN s
z ZipAppCmdlineTest.make_archivec C sN | j d }|�� |d �� t|�g}t�|� |�d�}| �|�� � d S )Nr r z.pyz) r r r r r rK � with_suffixr r )r r �argsr" r r r �test_cmdline_createW s
z%ZipAppCmdlineTest.test_cmdline_createc C s@ | � � }| jd }t|�dt|�g}t�|� | �|�� � d S �N�
target.pyz�-o)r� r r r rK r r )r �originalr" r� r r r �test_cmdline_copya s
z#ZipAppCmdlineTest.test_cmdline_copyc C sj | � � }| jd }t|�dt|�g}| �t��
}t�|� W d � n1 s'w Y | �|jj � d S r� �
r� r r rF �
SystemExitr rK r � exception�code�r r� r" r� �cmr r r �test_cmdline_copy_inplacei s
�z+ZipAppCmdlineTest.test_cmdline_copy_inplacec C sn | � � }| jd }t|�dt|�ddg}| �t��
}t�|� W d � n1 s)w Y | �|jj � d S )Nr� r� z-mzfoo:barr� r� r r r � test_cmdline_copy_change_mains s
�z/ZipAppCmdlineTest.test_cmdline_copy_change_mainz
sys.stdout)Z new_callablec C sl | � � }t|�dg}| �t��
}t�|� W d � n1 s w Y | �|jjd� | �|� � d� d S )N�--infor zInterpreter: <none>
)
r� r rF r� r rK r= r� r� r^ )r Z mock_stdoutr" r� r� r r r �test_info_command} s �z#ZipAppCmdlineTest.test_info_commandc C s\ | j d }t|�dg}| �t��
}t�|� W d � n1 s w Y | �|jj� d S )Nz dummy.pyzr� ) r r rF r� r rK r r� r� )r r" r� r� r r r �test_info_error� s
�z!ZipAppCmdlineTest.test_info_errorN)rz r{ r| r} r r� r� r� r� r� r r' �StringIOr� r� r r r r r� E s
r� �__main__)r} r' r rt r r r~ r r* Z test.supportr Z
unittest.mockr ZTestCaser r� rz rK r r r r �<module> s$ 8M �