Your IP : 216.73.216.60
��Yf�6 � @ s� d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m
Z
Gd d � d e j � Z Gd d � d e j � Z
e d k r� e j � d S) z#Test harness for the zipapp module.� N)�patchc @ s` e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z d( d) � Z d* d+ � Z e j e j d, k d- � d. d/ � � Z e j e j d, k d- � d0 d1 � � Z d2 S)3�
ZipAppTestz!Test zipapp module functionality.c C s5 t j � } | j | j � t j | j � | _ d S)N)�tempfile�TemporaryDirectory�
addCleanup�cleanup�pathlib�Path�name�tmpdir)�selfr � r
�5/opt/alt/python35/lib64/python3.5/test/test_zipapp.py�setUp s zZipAppTest.setUpc C se | j d } | j � | d j � | j d } t j t | � 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 sY | j d } | j � | d j � | j d } t j | | � | 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 � t j � } t j t | � | � | j d � t j
| d � �3 } | j d | j � � | j d | j � � Wd QRXd 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\ | j d } | j � | d j � t j t | � � | 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 sl | j d } | j � | d j � | j d } | j t j � �! t j t | � t | � � Wd QRXd 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 sr | j d } | j � | d j � | j d } | j t j � �' t j t | � t | � d d �Wd QRXd S)Nr z __main__.pyz
source.pyz�mainz
pkg.mod:fn)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 } t j t | � t | � d d �t j t | � d � �6 } | j d | j � � | j d | j
d � � Wd QRXd S) Nr zfoo.pyz
source.pyzr* z
pkg.mod:fnr 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 } t j t | � t | � d d �t j t | � d � �& } | j d | j � j
d � � Wd QRXd S)
Nr zfoo.pyzbar.pyz
source.pyzr* z
pkg.mod:fnr � 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 } d d d d d d d d
d d g
} xd | D]\ } | j d
| � �A | j t j � �' t j t | � t | � d
| �Wd QRXWd QRXqO Wd S)Nr z
source.pyz� r zfoo:z:barz12:barza.b.c.:dz.a:bza:b.za:.bz a:silly namer* )r r ZsubTestr' r r( r r )r r r Zproblemsr* r
r
r �test_main_validationo s
z ZipAppTest.test_main_validationc
C s� | j d } | j � | d j � | j d } t j t | � t | � � | j d � � } | j | j d � d � Wd QRXd 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 } t j t | � t | � d d �| j d � �6 } | j | j d � d � | j d | j � � Wd QRXd S)
Nr z __main__.pyz
source.pyz� interpreter�pythonr4 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 sm | j d } | j � | d j � t j � } t j t | � | d d �| 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 sw | j d } | j � | d j � | j d } t j t | � t | � d d �| j t j t | � � 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 sq | j d } | j � | d j � | j d } t j t | � t | � � | j t j t | � � 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 } t j t | � t | � d d �| j d } t j t | � t | � d d �| j t j t | � � 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 s� | j d } | j � | d j � | j d } t j t | � t | � d d �t j � } t j t | � | d d �| 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 s� | j d } | j � | d j � | j d } | j d } t j | | d d �t j | | d d �| j t j | � 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 } t j � } t j t | � | d d �t j � } | j d � t j | | d d �| 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 } t j t | � t | � d d �| j d } t j t | � t | � d d �| j t j t | � � 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 � t j � } t j t | � | d d �t j � } | j d � t j | | d d �| j d � t j
| d � �& } | j t | j
� � d h � Wd QRXd 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_archive�win32z*Windows does not support an executable bitc C su | j d } | j � | d j � | j d } t j t | � t | � d d �| j | j � j t j @� 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 su | j d } | j � | d j � | j d } t j t | � t | � d d �| j | j � j t j @� d S)Nr z __main__.pyz
source.pyzr9 )
r r r r r r Z assertFalserL rM rN )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�platformrO rP r
r
r
r r s4
r c @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z e
d d e j �d d � � Z
d d � Z d S)�ZipAppCmdlineTestz$Test zipapp module command line API.c C s5 t j � } | j | j � t j | j � | _ d S)N)r r r r r r r
r )r r r
r
r r s zZipAppCmdlineTest.setUpc C sF | j d } | j � | d j � | j d } t j | | � | 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 sg | j d } | j � | d j � t | � g } t j | � | j d � } | 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 sX | j � } | j d } t | � d t | � g } t j | � | j | j � � d S)Nz
target.pyzz-o)rY r r r r* r r )r �originalr rZ r
r
r �test_cmdline_copy+ s
z#ZipAppCmdlineTest.test_cmdline_copyc C sq | j � } | j d } t | � d t | � g } | j t � � } t j | � Wd QRX| j | j j � d S)Nz
target.pyzz-o)
rY r r r' �
SystemExitr r* r � exception�code)r r\ r rZ �cmr
r
r �test_cmdline_copy_inplace3 s
z+ZipAppCmdlineTest.test_cmdline_copy_inplacec C sw | j � } | j d } t | � d t | � d d g } | j t � � } t j | � Wd QRX| j | j j � d S)Nz
target.pyzz-oz-mzfoo:bar)
rY r r r' r^ r r* r r_ r` )r r\ r rZ ra r
r
r � test_cmdline_copy_change_main= s
!z/ZipAppCmdlineTest.test_cmdline_copy_change_mainz
sys.stdoutZ new_callablec
C st | j � } t | � d g } | j t � � } t j | � Wd QRX| j | j j d � | j | j � d � d S)Nz--infor zInterpreter: <none>
)
rY r r' r^ r r* r/ r_ r` r= )r Z mock_stdoutr rZ ra r
r
r �test_info_commandG s z#ZipAppCmdlineTest.test_info_commandc C s\ | j d } t | � d g } | j t � � } t j | � Wd QRX| j | j j � d S)Nz dummy.pyzz--info) r r r' r^ r r* r r_ r` )r r rZ ra r
r
r �test_info_errorR s
z!ZipAppCmdlineTest.test_info_errorN)rQ rR rS rT r rY r[ r] rb rc r r �StringIOrd re r
r
r
r rX s
! rX �__main__)rT r r rL rV r rU r r Z
unittest.mockr ZTestCaser rX rQ r* r
r
r
r �<module> s � M