Your IP : 216.73.217.129
���]�6 � @ s� 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)� N)�patchc @ sZ e 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, � d/ d0 � � Z d1 S)2�
ZipAppTestc C s5 t j � } | j | j � t j | j � | _ d S)N)�tempfile�TemporaryDirectory�
addCleanup�cleanup�pathlib�Path�name�tmpdir)�selfr � r
�0/opt/alt/python35/lib64/python3.5/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__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 s2
r c @ s� e 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)�ZipAppCmdlineTestc 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)rX r r r r* r r )r �originalr rY 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)
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 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)
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.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>
)
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 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 rY r` r
r
r �test_info_errorR s
z!ZipAppCmdlineTest.test_info_errorN)rQ rR rS r rX rZ r\ ra rb r r �StringIOrc rd r
r
r
r rW s
! rW �__main__)r r rL rU r rT r r Z
unittest.mockr ZTestCaser rW rQ r* r
r
r
r �<module> s � M