Your IP : 216.73.216.213
��YfuR � @ 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 Z d d l
Z
d d l Z d d l Z d d l m
Z
m Z y d d l m Z d Z Wn e k
r� d Z Yn Xd d l m Z d d l m Z Gd d � d e j � Z Gd
d � d e j � Z Gd d
� d
e j � Z e d k r\e j � d S)� N)�mock�
skipUnless)�ProcessPoolExecutorTF)�support)�
script_helperc @ s3 e Z d Z d d � Z d d � Z d d � Z e j e e d � d � d d
� � Z
d d � Z d
d � Z d d � Z
d d � Z d d � Z e j d � d d � � Z d d � Z e j d � d d � � Z e j d � e j d � d d � � � Z e j d d d �e j d � d! d" � � � Z d S)#�CompileallTestsc
C s t j � | _ t j j | j d � | _ t j j | j � | _
t | j d � � } | j d � Wd QRXt j j | j d � | _
t j j | j
� | _ t j | j | j
� t j j | j d � | _ t j | j � t j j | j d � | _ t j | j | j � d S)Nz_test.py�wzx = 123
z _test2.pyZ_subdirz _test3.py)�tempfile�mkdtemp� directory�os�path�join� source_path� importlib�util�cache_from_source�bc_path�open�write� source_path2�bc_path2�shutilZcopyfileZ subdirectory�mkdir� source_path3)�self�file� r �9/opt/alt/python35/lib64/python3.5/test/test_compileall.py�setUp s zCompileallTests.setUpc C s t j | j � d S)N)r �rmtreer )r r r r �tearDown( s zCompileallTests.tearDownc C sk t | j d � � } | j d � } Wd QRXt t j | j � j � } t j d t
j j | � } | | f S)N�rb� z<4sl)
r r �read�intr �statr �st_mtime�struct�packr r � MAGIC_NUMBER)r r �data�mtimeZcomparer r r r+ + s
zCompileallTests.datar& ztest needs os.stat()c C s� t j | j � | j | j � � t | j d � �# } | j � t | � d � } Wd QRXt | j d � �! } | j | � | j | � Wd QRX| j
| j � � t j | j
d d d d �| j | j � � d S)zOCheck that compileall recreates bytecode when the new metadata is
used.r" N�wb�forceF�quietT)�
py_compile�compiler � assertEqualr+ r r r$ �lenr �assertNotEqual�
compileall� compile_dirr �
assertTrue)r Zmetadatar Zbcr r r �recreation_check2 s #
z CompileallTests.recreation_checkc C s& | j t j d t j j d � � d S)Nz<4sl� )r8 r( r) r r r* )r r r r �
test_mtimeA s zCompileallTests.test_mtimec C s | j d � d S)Ns )r8 )r r r r �test_magic_numberF s z!CompileallTests.test_magic_numberc
C s x6 | j | j f D]" } y t j | � Wq Yq Xq Wt j | j d d d d �| j t j j | j � o� t j j | j � � t j | j � t j
| j d d d d �| j t j j | j � o� t j j | j � � t j | j � t j | j � d S)Nr. Fr/ T) r r r �unlinkr5 � compile_filer r7 r
�isfiler6 r )r �fnr r r �test_compile_filesJ s z"CompileallTests.test_compile_filesc C s� t j j | j d � } t j j | d � } t j | � t | d � � Wd QRXt j | � | j t j j t j j | d � � � d S)Nr+ r r � __pycache__)
r r
r r r r r5 r= � assertFalse�exists)r Zdata_dirZ data_filer r r � test_no_pycache_in_non_package[ s
z.CompileallTests.test_no_pycache_in_non_packagec C s� t r d n d \ } } t j | j d d d | �t j j | j d | �} | j t j
j | � � t j j | j d | �} | j t j
j | � � t j j | j
d | �} | j t j
j | � � d S)
Nr9 r � r/ T�optimize� optimization)r9 r9 )r rE )� __debug__r5 r6 r r r r r r7 r r
r> r r )r rF Zopt�cachedZcached2Zcached3r r r �
test_optimizeg s z CompileallTests.test_optimizez compileall.ProcessPoolExecutorc C s0 t j | j d d d d �| j | j � d S)Nr/ T�workers� )r5 r6 r r7 �called)r � pool_mockr r r �test_compile_pool_calledv s z(CompileallTests.test_compile_pool_calledc C s4 | j t d � � t j | j d d �Wd QRXd S)Nz%workers must be greater or equal to 0rK r9 ���)ZassertRaisesRegex�
ValueErrorr5 r6 r )r r r r �!test_compile_workers_non_positive{ s
z1CompileallTests.test_compile_workers_non_positivec C s; t j | j d d d d �| j | j d d d � d S)Nr/ TrK r r9 Z max_workers)r5 r6 r r2 � call_args)r rN r r r � test_compile_workers_cpu_count� s z.CompileallTests.test_compile_workers_cpu_countzcompileall.compile_filec C s: t j | j d d �| j | j � | j | j � d S)Nr/ T)r5 r6 r rB rM r7 )r �compile_file_mockrN r r r �test_compile_one_worker� s z'CompileallTests.test_compile_one_worker�newNc C s0 t j | j d d d d �| j | j � d S)Nr/ TrK rL )r5 r6 r r7 rM )r rU r r r �$test_compile_missing_multiprocessing� s z4CompileallTests.test_compile_missing_multiprocessing)�__name__�
__module__� __qualname__r r! r+ �unittestr �hasattrr r8 r: r; r@ rD rJ r �patchrO rR rT rV rX r r r r r s ' r c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) � EncodingTestzTIssue 6716: compileall should escape source code when printing errors
to stdout.c C sj t j � | _ t j j | j d � | _ t | j d d d ��! } | j d � | j d � Wd QRXd S)Nz_test.pyr �encodingzutf-8z# -*- coding: utf-8 -*-
u
print u"€"
) r r
r r r
r r r r )r r r r r r � s
zEncodingTest.setUpc C s t j | j � d S)N)r r r )r r r r r! � s zEncodingTest.tearDownc C sL z; t j } t j t j � d d �t _ t j | j � Wd | t _ Xd S)Nr` �ascii)�sys�stdout�io�
TextIOWrapper�BytesIOr5 r6 r )r Z orig_stdoutr r r �
test_error� s
zEncodingTest.test_errorN)rY rZ r[ �__doc__r r! rg r r r r r_ � s r_ c @ s e Z d Z d Z e 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 xd d d g f d d d g f d d d g f g D]2 \ Z Z Z e e d d! � Z e e � d" e <q� Wd# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d; d< � Z! d= d> � Z" d? d@ � Z# dA dB � Z$ dC dD � Z% e& e' dE � dF dG � � Z( e) j* dH � dI dJ � � Z+ dK S)L�CommandLineTestszTest compileall's CLI.c C s� x� t t j j t j � D]� } d } t j | � d } | d } zf yF | j � sd | j � d } | j d � � } | j
d � Wd QRXWn t k
r� d } PYn XWd t j
t | � � | r� | j � Xq Wd } | | _ d S)NFrA ztest.tryTr z# for test_compileall)�filterr r
�isdirrb �pathlibZPath�is_dirr r r �OSErrorr r< �str�rmdir�_sys_path_writable)�clsr
Zdirectory_createdr r Zsys_path_writabler r r �
setUpClass� s$
zCommandLineTests.setUpClassc C s | j s t j d � � d S)Nz(not all entries on sys.path are writable)rq r\ ZSkipTest)r r r r � _skip_if_sys_path_not_writable� s z/CommandLineTests._skip_if_sys_path_not_writablec C sP d g } t j j r8 | j d d d d i t j j � | d d g t | � S)Nz-Sr9 z-O� z-OOz-mr5 )rb �flagsrF �append�list)r �argsZ interp_argsr r r �
_get_run_args� s #z CommandLineTests._get_run_argsc O s8 t j | j | � | � \ } } } | j d | � | S)N� )r �assert_python_okrz r2 )r ry �env_vars�rc�out�errr r r � assertRunOK� s z CommandLineTests.assertRunOKc O s1 t j | j | � | � \ } } } | | | f S)N)r �assert_python_failurerz )r ry r} r~ r r� r r r �assertRunNotOK� s z CommandLineTests.assertRunNotOKc C s/ t j j | � } | j t j j | � � d S)N)r r r r7 r r
rC )r r? r
r r r �assertCompiled� s z CommandLineTests.assertCompiledc C s/ t j j | � } | j t j j | � � d S)N)r r r rB r r
rC )r r? r
r r r �assertNotCompiled� s z"CommandLineTests.assertNotCompiledc C s� t j � | _ | j t j | j � t j j | j d � | _ t j
| j � t j j | j d � | _ t j
| j d d � | _ t j
| j d d � | _ d S)NZfoorA �__init__rE Zbar)r r
r Z
addCleanupr r r r
r �pkgdirr �pkgdir_cachedirr � make_script�initfn�barfn)r r r r r � s zCommandLineTests.setUpc C sf | j � t j | j d d � } | j d | j � | j | � | j | j � | j | j � d S)N�bazrE �
PYTHONPATH) rt r r� r r� r� r� r� r� )r �bazfnr r r �test_no_args_compiles_path� s
z+CommandLineTests.test_no_args_compiles_pathc C s� | j � t j | j d d � } | j d | j � t j j | � } t j | t
j
� d f d � t j | � j } | j d | j � t j | � j } | j
| | � | j d d | j �t j | � j } | j | | � d S)Nr� rE r� �< ru z-f)rt r r� r r� r r r r �utime�timer& r' r2 r4 )r r� �pycpathr, �mtime2r r r � test_no_args_respects_force_flag� s
!z1CommandLineTests.test_no_args_respects_force_flagc C sq | j � t j | j d d � | j d | j � } | j d | � | j d d | j �} | j d | � d S)Nr� rE r� s Listing z-q)rt r r� r r� ZassertInZ assertNotIn)r �noisyr/ r r r � test_no_args_respects_quiet_flag s
z1CommandLineTests.test_no_args_respects_quiet_flagZnormal�pycrF z opt-1.pycz-OZdoubleoptimizez opt-2.pycz-OOc s� t j | d d d | j g � | j t j j | j � � t � f d d � d
D� � } | j t t j
| j � � | � | j � f d d � t j
| j � D� � d S) Nz-mr5 z-qc 3 s' | ] } | j t j j � � Vq d S)N)�formatrb �implementation� cache_tag)�.0�base)�extr r � <genexpr> s z%CommandLineTests.f.<locals>.<genexpr>�__init__.{}.{}� bar.{}.{}c s% g | ] } | j � � r | � q Sr )�endswith)r� r? )r� r r �
<listcomp> s z&CommandLineTests.f.<locals>.<listcomp>)r� r� ) r r| r� r7 r r
rC r� �sortedr2 �listdirrB )r r� �switch�expectedr )r� r �f s
"zCommandLineTests.fZtest_pep3147_paths_c C sp | j d d | j � | j t j j | j � � t d d d d g � } | j t t j | j � � | � d S)Nz-bz-qz __init__.pyz __init__.pyczbar.pyzbar.pyc)
r� r� rB r r
rC r� r� r2 r� )r r� r r r �test_legacy_paths s
z"CommandLineTests.test_legacy_pathsc C s� | j d | j � | j t j j | j � � t j j | j d � } | j t j j | � � | j d | j � | j t j j | j � � | j t j j | � � d S)Nz-qrA ) r� r� r7 r r
rC r� r rB )r Z
cachecachedirr r r �test_multiple_runs* s z#CommandLineTests.test_multiple_runsc C s� | j d | j � t j j | j � } t j | t j � d f d � t j | � j
} | j d | j � t j | � j
} | j | | � | j d d | j � t j | � j
} | j | | � d S)Nz-qr� ru z-f)
r� r� r r r r� r r� r� r& r' r2 r4 )r r� r, r� r r r �
test_force7 s !zCommandLineTests.test_forcec C s� t j j | j d � } t j | � t j | d d � } t j | d d � } | j d d | j � | j | � | j t j j
t j j | d � � � | j d | j � | j | � | j | � d S)N�spamr� rE �hamz-qz-lrA ) r r
r r� r r r� r� r� rB rC r� )r �
subpackage� subinitfn�hamfnr r r �test_recursion_controlF s
(
z'CommandLineTests.test_recursion_controlc C s� t j j | j d � } t j j | d � } t j j | d � } x$ | | | f D] } t j | � qR Wt j j | d � } t j | d d � } t j | d d � } t j | d d � } | j d d | j � | j | � | j t j j
t j j | d � � � | j d d
| j � | j | � | j | � | j | � | j d d | j � | j | � | j | � | j | � | j | � | j d d | j � | j | � | j | � | j | � | j | � d S)
Nr� r� Zeggsz __init__.pyrE Zeggz-qz-r 0rA z-r 1z-r 2z-r 5) r r
r r� r Zmake_pkgr� r� r� rB rC r� ) r r� Z subpackage2Z subpackage3Zpkgr� r� ZspamfnZeggfnr r r �test_recursion_limitR s6
"
z%CommandLineTests.test_recursion_limitc C sK | j | j � } | j d | j � } | j d | � | j d | � d S)Nz-qr{ )r� r� r4 r2 )r r� r/ r r r �
test_quiett s zCommandLineTests.test_quietc C sv t j | j d d � | j d | j � \ } } } | j d | j � \ } } } | j d | � | j d | � d S)N� crunchyfrogz
bad(syntaxz-qz-qqr{ )r r� r� r� r4 r2 )r �_r/ Zsilentr r r � test_silentz s
z CommandLineTests.test_silentc C s= | j d d d | j � | j | j � | j | j � d S)Nz-qz-xz
ba[^\\/]*$)r� r� r� r� r� r� )r r r r � test_regexp� s z CommandLineTests.test_regexpc C s� t j j | j d � } t j | � t j | d d � } t j | d d � } | j d | j | � | j | j
� | j | j � | j | � | j | � d S)NZfoo2r� rE �bar2z-q) r r
r r r r r� r� r� r� r� r� )r Zpkgdir2Zinit2fn�bar2fnr r r �test_multiple_dirs� s
z#CommandLineTests.test_multiple_dirsc C sN t j | j d d � | j d d d | j � \ } } } | j | d � d S)Nr� z
bad(syntaxz-qz-d�dinsdales File "dinsdale)r r� r� r� � assertRegex)r r~ r r� r r r �test_d_compile_error� s $z%CommandLineTests.test_d_compile_errorc C s� t j | j d d � } | j d d d | j � t j | j d d � } t j j | � } t j | t j j
| j d � � t j | � t j | d d
�\ } } } | j
| d � d S) Nr� zraise Exceptionz-qz-dr� �bingz
import bazzbaz.pycZ
__isolatedFs File "dinsdale)r r� r� r� r r r r �renamer
r �remover� r� )r r� r? r� r~ r r� r r r �test_d_runtime_error� s "
z%CommandLineTests.test_d_runtime_errorc C s | j d t j j | j d � | j � \ } } } | j | d � | j | d � | j t j j t
j j | j
� � � d S)Nz-i�
nosuchfiles rror.*nosuchfiles Traceback)r� r r
r r r� r� ZassertNotRegexrB rC r r r r� )r r~ r r� r r r �test_include_bad_file� s -z&CommandLineTests.test_include_bad_filec C s4 t j | j d d � } t j | j d d � } t j | j d d � } t j | j d d � } t t j j | j d � d � �S } | j t j j | j d � t j � | j t j j | j d � t j � Wd QRX| j
d
t j j | j d � | � | j | � | j | � | j | � | j | � d S) N�f1rE �f2�f3�f4�l1r zf1.pyzf2.pyz-i)
r r� r� r r r
r r r �linesepr� r� r� )r r� r� r� r� r� r r r �test_include_file_with_arg� s $&-%
z+CommandLineTests.test_include_file_with_argc C s t j | j d d � } t j | j d d � } t j | j d d � } t j | j d d � } t t j j | j d � d � �- } | j t j j | j d � t j � Wd QRX| j
d t j j | j d � � | j | � | j | � | j | � | j | � d S)
Nr� rE r� r� r� r� r zf2.pyz-i)
r r� r� r r r
r r r r� r� r� r� )r r� r� r� r� r� r r r �test_include_file_no_arg� s $-"
z)CommandLineTests.test_include_file_no_argc C s� t j | j d d � } t j | j d d � } t j | j d d � } t j | j d d � } t j | j f � d d g � } | j j | t j j d � � t j
| � | j | � | j | � | j | � | j | � d S) Nr� rE r� r� r� z-i�-ra )
r r� r� Z spawn_pythonrz �stdinr r r� �encodeZ kill_pythonr� r� )r r� r� r� r� �pr r r �test_include_on_stdin� s "
z&CommandLineTests.test_include_on_stdinc C s� t j | j d d � } | j d | j | | j � \ } } } | j | d � | j | � | j | j � | j | j � d S)Nr� z syntax(errorr� s rror) r r� r� r� r� r� r� r� r� )r Zbingfnr~ r r� r r r �!test_compiles_as_much_as_possible� s
z2CommandLineTests.test_compiles_as_much_as_possiblec C s# | j d � } | j | d � d S)NZ badfilenames Can't list 'badfilename')r� r� )r r r r r �!test_invalid_arg_produces_message� s z2CommandLineTests.test_invalid_arg_produces_messagezrequires multiprocessingc C s� t j | j d d � } g } xs t d � D]e } t j j | j d j | � � } t j | � t j | d d � } | j t j | d d � � q+ W| j
| j d d � | j | � x | D] } | j | � q� Wd S)Nr� rE rL zfoo{}r� z-j�0) r r� r �ranger r
r r� r rw r� r� )r r� �files�suffixr� r? r r r r � test_workers� s !
z CommandLineTests.test_workerszcompileall.compile_dirc
C se t j d d t j | j d g ��: t j � | j | j � | j | j
d d d � Wd QRXd S)Nzsys.argvrW z-j0r9 rK rP ) r r^ rb �
executabler r5 �mainr7 rM r2 rS )r r6 r r r � test_workers_available_cores� s
z-CommandLineTests.test_workers_available_coresN),rY rZ r[ rh � classmethodrs rt rz r� r� r� r� r r� r� r� �namer� r� r� �localsr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r �_have_multiprocessingr� r r^ r� r r r r ri � sJ
"
"
ri �__main__) rb r5 �importlib.utilr r rl r0 r r( r r� r\ rd r r Zconcurrent.futuresr r� � ImportErrorZtestr Z test.supportr ZTestCaser r_ ri rY r� r r r r �<module> s2
z� M