Sindbad~EG File Manager
a
��g�? � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z G dd� dej �Z
G dd� dej �Zedkr�e�� dS ) � N)�
requires_zlib)�patchc @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
� 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� Ze � e j d1kd2�d3d4� �Z e � e j d1kd2�d5d6� �Z!d7S )8�
ZipAppTestc C s& t �� }| �|j� t�|j�| _d S �N��tempfile�TemporaryDirectoryZ
addCleanup�cleanup�pathlib�Path�name�tmpdir��selfr
� r �,/usr/local/lib/python3.9/test/test_zipapp.py�setUp s 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��0}| � d|� � � | � d |� � � W d � n1 s�0 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 � n1 s�0 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 r2 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
��D}| �t|� � �d � | �
d|� � � | �
d|� � � W d � n1 s�0 Y d S ) Nc S s | j d dkS )Nr �dummy)�partsr0 r r r �skip_dummy_dirO s zIZipAppTest.test_create_archive_filter_exclude_dir.<locals>.skip_dummy_dirr r r3 r7 ztest2.pyr r4 r$ � ) r
r r r r r( r) � assertEqual�lenr+ r* )r r9 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��0}dD ]}| �|�|�j tj
� qVW d � n1 s�0 Y d S ) Nr r r3 r T)�
compressedr$ )r r3 ) 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 sl | j d }|�� |d �� | j d }| �tj��$ t�t|�t|�� W d � n1 s^0 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 sp | j d }|�� |d �� | j d }| �tj��( tjt|�t|�dd� W d � n1 sb0 Y d S )Nr r r �
pkg.mod:fn��mainrC 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��2}| �d|� � � | �d|�
d�� W d � n1 s�0 Y d S ) Nr rB r rG rH 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 � n1 s�0 Y d S )
Nr rB zbar.pyr rG rH 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 ]t}| j|d��T | �tj��( tjt|�t|�|d� W d � n1 st0 Y W d � q(1 s�0 Y q(d S )Nr r )
� r# zfoo:z:barz12:barza.b.c.:dz.a:bza:b.za:.bz a:silly namerH )r
r ZsubTestrD r rE r r )r r r ZproblemsrI 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 � n1 sn0 Y d S )Nr r r �rbr: � #!) r
r r r r r �openZassertNotEqualrK �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��2}| �|�d�d� | �d |� � � W d � n1 s�0 Y d S )
Nr r r �python�Z interpreterrR r: rS s python
)
r
r r r r r rT r; rK �readlinerU 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 rX rY 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 rX rY �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 r` 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 rX rY � changed.pyz� python2.7r` �r r r �
new_targetr r r �test_modify_shebang� s
z ZipAppTest.test_modify_shebangc C st | j d }|�� |d �� | j d }tjt|�t|�dd� t�� }tjt|�|dd� | �|� � �
d�� d S )Nr r r rX rY re � #!python2.7
) r
r r r r r r% r& r r\ r] rf 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.pyzrX rY re )r
r r r r r; ra )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 rX rY r re ri ) r
r r r% r& r r r r' r r\ r] )r r r Z temp_archiverg 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 rX rY rd r` rf r r r �test_remove_shebang s
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 � n1 s�0 Y d S )Nr r rX rY r r$ )r
r r r% r& r r r r' r( r) r; �setr+ )r r r rg 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 r_ )
r
r r r r r r �stat�st_mode�S_IEXECr r r r �test_shebang_is_executable. s
z%ZipAppTest.test_shebang_is_executablec C sZ | j d }|�� |d �� | j d }tjt|�t|�d d� | �|�� jtj @ � d S )Nr r r rY )
r
r r r r r Z assertFalserq rr rs r r r r �!test_no_shebang_is_not_executable9 s
z,ZipAppTest.test_no_shebang_is_not_executableN)"�__name__�
__module__� __qualname__r r! r" r. r6 r= r> r rA rF rJ rL rO rQ rW r[ r^ rb rc rh rj rk rl rm ro �unittestZskipIf�sys�platformrt ru r r r r r sB
�
�r c @ sZ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze d
e
j d�dd� �Z dd� Z
dS )�ZipAppCmdlineTestc C s& t �� }| �|j� t�|j�| _d S r r r r r r r I s 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 rI � 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 rI 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 sN0 Y | �|jj � d S r� �
r} r
r rD �
SystemExitr rI 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 sR0 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>0 Y | �|jjd� | �|� � d� d S )N�--infor zInterpreter: <none>
)
r} r rD r� r rI 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@0 Y | �|jj� d S )Nz dummy.pyzr� ) r
r rD r� r rI r r� r� )r r r r� r r r �test_info_error� s
(z!ZipAppCmdlineTest.test_info_errorN)rv rw rx r r} r� r� r� r� r r% �StringIOr� r� r r r r r| E s
r| �__main__)r% r
rq rz r ry r r( Z test.supportr Z
unittest.mockr ZTestCaser r| rv rI r r r r �<module> s 8M
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists