Your IP : 216.73.216.213
3
� f�6 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
G dd� dej �Z G dd� dej �Z
edkr�ej� dS ) z#Test harness for the zipapp module.� N)�patchc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � 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+� Zejej d,kd-�d.d/� �Z ejej d,kd-�d0d1� �Z d2S )3�
ZipAppTestz!Test zipapp module functionality.c C s& t j� }| j|j� tj|j�| _d S )N)�tempfile�TemporaryDirectory�
addCleanup�cleanup�pathlib�Path�name�tmpdir)�selfr � r
�5/opt/alt/python36/lib64/python3.6/test/test_zipapp.py�setUp s zZipAppTest.setUpc C sN | j d }|j� |d j� | j d }tjt|�t|�� | j|j� � 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 }|j� |d j� | j d }tj||� | j|j� � 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 }|j� |d j� |d j� |d j� |d d j� tj� }tjt|�|� |jd� t j
|d��&}| j d|j � � | j d |j � � 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 sH | j d }|j� |d j� tjt|�� | j d }| j|j� � 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_target8 s
z-ZipAppTest.test_create_archive_default_targetc C sX | j d }|j� |d j� | j d }| jtj�� tjt|�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_mainA s
zZipAppTest.test_no_mainc C s\ | j d }|j� |d j� | j d }| jtj�� tjt|�t|�dd� W d Q R X d S )Nr z __main__.pyz
source.pyzz
pkg.mod:fn)�main)r r r r' r r( r r )r r r r
r
r �test_main_and_main_pyJ s
z ZipAppTest.test_main_and_main_pyc C s� | j d }|j� |d j� | j d }tjt|�t|�dd� tjt|�d��(}| jd|j � � | jd|j
d�� W d Q R X d S ) Nr zfoo.pyz
source.pyzz
pkg.mod:fn)r* 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_writtenS s
z ZipAppTest.test_main_writtenc C s� | j d }|j� |d j� |d j� | j d }tjt|�t|�dd� tjt|�d�� }| jd|j � j
d �� W d Q R X d S )
Nr zfoo.pyzbar.pyz
source.pyzz
pkg.mod:fn)r* r � z __main__.py) r r r r r r r r! � assertEqualr# �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� | j d }ddddddd d
d d g
}xT|D ]L}| j|d
��6 | jtj�� 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)r* )r r ZsubTestr' r r( r r )r r r Zproblemsr* r
r
r �test_main_validationo s
z ZipAppTest.test_main_validationc
C sh | j d }|j� |d j� | j d }tjt|�t|�� |jd��}| j|jd�d� W d Q R X d S )Nr z __main__.pyz
source.pyz�rb� 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 }|j� |d j� | j d }tjt|�t|�dd� |jd��(}| j|jd�d� | jd |j � � W d Q R X d S )
Nr z __main__.pyz
source.pyz�python)� interpreterr4 r5 s #!s python
)
r r r r r r r6 r/ r, �readline)r r r r7 r
r
r �test_custom_interpreter� s
z"ZipAppTest.test_custom_interpreterc C sR | j d }|j� |d j� tj� }tjt|�|dd� | j|j � j
d�� d S )Nr z __main__.pyr9 )r: 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 }|j� |d j� | j d }tjt|�t|�dd� | jtjt|��d� d S )Nr z __main__.pyz
source.pyzr9 )r: )r r r r r r r/ �get_interpreter)r r r r
r
r �test_read_shebang� s
z ZipAppTest.test_read_shebangc C sV | j d }|j� |d j� | j d }tjt|�t|�� | jtjt|��d � d S )Nr z __main__.pyz
source.pyz)r r r r r r r/ r@ )r r r r
r
r �test_read_missing_shebang� s
z$ZipAppTest.test_read_missing_shebangc C s| | j d }|j� |d j� | j d }tjt|�t|�dd� | j d }tjt|�t|�dd� | jtjt|��d� d S )Nr z __main__.pyz
source.pyzr9 )r: z changed.pyzz python2.7)r r r r r r r/ r@ )r r r �
new_targetr
r
r �test_modify_shebang� s
z ZipAppTest.test_modify_shebangc C st | j d }|j� |d j� | j d }tjt|�t|�dd� tj� }tjt|�|dd� | j|j � j
d�� d S )Nr z __main__.pyz
source.pyzr9 )r: z python2.7s #!python2.7
) r r r r r r r r r r= r> )r r r rC r
r
r � test_write_shebang_to_fileobj� s
z(ZipAppTest.test_write_shebang_to_fileobjc C sh | j d }|j� |d j� | j d }| j d }tj||dd� tj||dd� | jtj|�d� d S )Nr z __main__.pyz target1.pyzz target2.pyzr9 )r: z python2.7)r r r r r r/ r@ )r r Ztarget1Ztarget2r
r
r �test_read_from_pathobj� s
z!ZipAppTest.test_read_from_pathobjc C s~ | j d }|j� |d j� | j d }tj� }tjt|�|dd� tj� }|jd� tj||dd� | j |j
� j d�� d S ) Nr z __main__.pyz
source.pyzr9 )r: r z python2.7s #!python2.7
) r r r r r r r r r r r= r> )r r r Z temp_archiverC r
r
r �test_read_from_fileobj� s
z!ZipAppTest.test_read_from_fileobjc C s| | j d }|j� |d j� | j d }tjt|�t|�dd� | j d }tjt|�t|�d d� | jtjt|��d � d S )Nr z __main__.pyz
source.pyzr9 )r: z changed.pyz)r r r r r r r/ r@ )r r r rC r
r
r �test_remove_shebang� s
z ZipAppTest.test_remove_shebangc
C s� | j d }|j� |d j� tj� }tjt|�|dd� tj� }|jd� tj||d d� |jd� t j
|d�� }| j t |j
� �dh� W d Q R X d S )Nr z __main__.pyr9 )r: r r )r r r r r r r r r r r! r/ �setr# )r r r rC r$ r
r
r � test_content_of_copied_archive� s
z)ZipAppTest.test_content_of_copied_archiveZwin32z*Windows does not support an executable bitc C sZ | j d }|j� |d j� | j d }tjt|�t|�dd� | j|j� jtj @ � d S )Nr z __main__.pyz
source.pyzr9 )r: )
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 }|j� |d j� | j d }tjt|�t|�d d� | j|j� jtj @ � d S )Nr z __main__.pyz
source.pyz)r: )
r r r r r r Z assertFalserK rL rM )r r r r
r
r �!test_no_shebang_is_not_executable s
z,ZipAppTest.test_no_shebang_is_not_executableN) �__name__�
__module__� __qualname__�__doc__r r r r% r& r) r+ r- r1 r3 r8 r< r? rA rB rD rE rF rG rH rJ �unittestZskipIf�sys�platformrN rO r
r
r
r r s4
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� }| j|j� tj|j�| _d S )N)r r r r r r r
r )r r r
r
r r s zZipAppCmdlineTest.setUpc C s8 | j d }|j� |d j� | j d }tj||� |S )Nr z __main__.pyz
source.pyz)r r r r r )r r r r
r
r � make_archive s
z ZipAppCmdlineTest.make_archivec C sN | j d }|j� |d j� t|�g}tj|� |jd�}| j|j� � d S )Nr z __main__.pyz.pyz) r r r r r r* Z with_suffixr r )r r �argsr r
r
r �test_cmdline_create! s
z%ZipAppCmdlineTest.test_cmdline_createc C s@ | j � }| jd }t|�dt|�g}tj|� | j|j� � d S )Nz
target.pyzz-o)rX r r r r* r r )r �originalr rY r
r
r �test_cmdline_copy+ s
z#ZipAppCmdlineTest.test_cmdline_copyc C sV | j � }| jd }t|�dt|�g}| jt��}tj|� W d Q R X | j|jj � d S )Nz
target.pyzz-o)
rX r r r' �
SystemExitr r* r � exception�code)r r[ r rY �cmr
r
r �test_cmdline_copy_inplace3 s
z+ZipAppCmdlineTest.test_cmdline_copy_inplacec C sZ | j � }| jd }t|�dt|�ddg}| jt��}tj|� W d Q R X | j|jj � d S )Nz
target.pyzz-oz-mzfoo:bar)
rX r r r' r] r r* r r^ r_ )r r[ r rY r` r
r
r � test_cmdline_copy_change_main= s
z/ZipAppCmdlineTest.test_cmdline_copy_change_mainz
sys.stdout)Z new_callablec
C sX | j � }t|�dg}| jt��}tj|� W d Q R X | j|jjd� | j|j � d� d S )Nz--infor zInterpreter: <none>
)
rX r r' r] r r* r/ r^ r_ r= )r Z mock_stdoutr rY r` r
r
r �test_info_commandG s z#ZipAppCmdlineTest.test_info_commandc C sH | j d }t|�dg}| jt��}tj|� W d Q R X | j|jj� d S )Nz dummy.pyzz--info) r r r' r] r r* r r^ r_ )r r rY r` r
r
r �test_info_errorR s
z!ZipAppCmdlineTest.test_info_errorN)rP rQ rR rS r rX rZ r\ ra rb r r �StringIOrc rd r
r
r
r rW s
rW �__main__)rS r r rK rU r rT r r Z
unittest.mockr ZTestCaser rW rP r* r
r
r
r �<module> s M