File: //usr/local/lib/python3.7/test/__pycache__/test_zipapp.cpython-37.pyc
B
��g�? � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
ddl m Z G dd� dej
�ZG dd� dej
�Zed kr�e�� 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 s& t �� }| �|j� t�|j�| _d S )N)�tempfile�TemporaryDirectory�
addCleanup�cleanup�pathlib�Path�name�tmpdir)�selfr � r �,/usr/local/lib/python3.7/test/test_zipapp.py�setUp s zZipAppTest.setUpc C sN | j d }|�� |d �� | j d }t�t|�t|�� | �|�� � d S )N�sourcez __main__.pyz
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 )Nr z __main__.pyz
source.pyz)r 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 Q R X d S )
Nr z __main__.py�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 ��6}| �d|�� � | �d|�� � | � d|�� � W d Q R X 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 z __main__.pyztest.pyztest.pycz
source.pyz)�filterr )
r r r r r r! r" r# r$ Z assertNotIn)r
r) 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 Q R X d S ) Nc S s | j d dkS )Nr �dummy)�parts)r( r r r �skip_dummy_dirO s zIZipAppTest.test_create_archive_filter_exclude_dir.<locals>.skip_dummy_dirr z __main__.pyztest.pyr, ztest2.pyz
source.pyz)r* 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 )Nr z __main__.pyz
source.pyz)r r r r r r 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��*}x"dD ]}| �|�|�j tj
� qXW W d Q R X d S ) Nr z __main__.pyztest.pyz
source.pyzT)Z
compressedr )z __main__.pyztest.py) r r r r r r! r" r0 ZgetinfoZ
compress_typeZ ZIP_DEFLATED)r
r r r% r r r r �$test_create_archive_with_compressionh s
z/ZipAppTest.test_create_archive_with_compressionc C sX | j d }|�� |d �� | j d }| �tj�� t�t|�t|�� W d Q R X d S )Nr zfoo.pyz
source.pyz)r r r � assertRaisesr � ZipAppErrorr r )r
r r r r r � test_no_mainw s
zZipAppTest.test_no_mainc C s\ | j d }|�� |d �� | j d }| �tj�� tjt|�t|�dd� W d Q R X d S )Nr z __main__.pyz
source.pyzz
pkg.mod:fn)�main)r r r r5 r r6 r r )r
r 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 Q R X d S ) Nr zfoo.pyz
source.pyzz
pkg.mod:fn)r8 r z __main__.pys pkg.mod.fn()) r r r r r r r! r" r# r$ �read)r
r r r% 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 Q R X d S )
Nr zfoo.pyzbar.pyz
source.pyzz
pkg.mod:fn)r8 r � z __main__.py) r r r r r r r! r" r0 r$ �count)r
r r r% r r r �test_main_only_written_once� s
z&ZipAppTest.test_main_only_written_oncec C s� | j d }|�� | j d }ddddddd d
d d g
}xT|D ]L}| j|d
��6 | �tj�� tjt|�t|�|d
� W d Q R X W d Q R X q:W d S )Nr z
source.pyz� r zfoo:z:barz12:barza.b.c.:dz.a:bza:b.za:.bz a:silly name)r8 )r r ZsubTestr5 r r6 r r )r
r r Zproblemsr8 r r r �test_main_validation� s
z ZipAppTest.test_main_validationc C sh | j d }|�� |d �� | j d }t�t|�t|�� |�d��}| �|�d�d� W d Q R X d S )Nr z __main__.pyz
source.pyz�rbr/ s #!) r r r r r r �openZassertNotEqualr: )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 Q R X d S )
Nr z __main__.pyz
source.pyz�python)� interpreterrA r/ s #!s python
)
r r r r r r rB r0 r: �readline)r
r r rC r r r �test_custom_interpreter� s
z"ZipAppTest.test_custom_interpreterc C sR | j d }|�� |d �� t�� }tjt|�|dd� | �|� � �
d�� d S )Nr z __main__.pyrE )rF s #!python
) r r r r r r r r r �getvalue�
startswith)r
r r 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 z __main__.pyz
source.pyzrE )rF )r r r r r r r0 �get_interpreter)r
r r 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 )Nr z __main__.pyz
source.pyz)r r r r r r r0 rL )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 z __main__.pyz
source.pyzrE )rF z changed.pyzz python2.7)r r r r r r r0 rL )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 z __main__.pyz
source.pyzrE )rF z python2.7s #!python2.7
) r r r r r r r r r rI rJ )r
r r rO 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 z __main__.pyz target1.pyzz target2.pyzrE )rF z python2.7)r r r r r r0 rL )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 z __main__.pyz
source.pyzrE )rF r z python2.7s #!python2.7
) r r r r r r r r r r rI rJ )r
r r Z temp_archiverO 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 z __main__.pyz
source.pyzrE )rF z changed.pyz)r r r r r r r0 rL )r
r r rO 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 Q R X d S )Nr z __main__.pyrE )rF r r )r r r r r r r r r r! r" r0 �setr$ )r
r r rO 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 )Nr z __main__.pyz
source.pyzrE )rF )
r r r r r r r �stat�st_mode�S_IEXEC)r
r r 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 z __main__.pyz
source.pyz)rF )
r r r r r r Z assertFalserX rY rZ )r
r r r r r �!test_no_shebang_is_not_executable9 s
z,ZipAppTest.test_no_shebang_is_not_executableN)#�__name__�
__module__� __qualname__�__doc__r r r r& r+ r2 r3 r r4 r7 r9 r; r> r@ rD rH rK rM rN rP rQ rR rS rT rV �unittestZskipIf�sys�platformr[ r\ r r r r r s:
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 s& t �� }| �|j� t�|j�| _d S )N)r r r r r r
r r )r
r r r r r I s zZipAppCmdlineTest.setUpc C s8 | j d }|�� |d �� | j d }t�||� |S )Nr z __main__.pyz
source.pyz)r 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 z __main__.pyz.pyz) r r r r r r8 Z 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 )Nz
target.pyzz-o)re r r r r8 r r )r
�originalr rf r r r �test_cmdline_copya s
z#ZipAppCmdlineTest.test_cmdline_copyc C sV | � � }| jd }t|�dt|�g}| �t��}t�|� W d Q R X | �|jj � d S )Nz
target.pyzz-o)
re r r r5 �
SystemExitr r8 r � exception�code)r
rh r rf �cmr r r �test_cmdline_copy_inplacei s
z+ZipAppCmdlineTest.test_cmdline_copy_inplacec C sZ | � � }| jd }t|�dt|�ddg}| �t��}t�|� W d Q R X | �|jj � d S )Nz
target.pyzz-oz-mzfoo:bar)
re r r r5 rj r r8 r rk rl )r
rh r rf rm r r r � test_cmdline_copy_change_mains s
z/ZipAppCmdlineTest.test_cmdline_copy_change_mainz
sys.stdout)Z new_callablec C sX | � � }t|�dg}| �t��}t�|� W d Q R X | �|jjd� | �|� � d� d S )Nz--infor zInterpreter: <none>
)
re r r5 rj r r8 r0 rk rl rI )r
Z mock_stdoutr rf rm r r r �test_info_command} s z#ZipAppCmdlineTest.test_info_commandc C sH | j d }t|�dg}| �t��}t�|� W d Q R X | �|jj� d S )Nz dummy.pyzz--info) r r r5 rj r r8 r rk rl )r
r rf rm r r r �test_info_error� s
z!ZipAppCmdlineTest.test_info_errorN)r] r^ r_ r` r re rg ri rn ro r r �StringIOrp rq r r r r rd E s
rd �__main__)r` r r rX rb r ra r r! Z test.supportr Z
unittest.mockr ZTestCaser rd r] r8 r r r r �<module> s 8M