Your IP : 216.73.216.213
���]uR � @ 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�shutil�copyfileZ subdirectory�mkdir� source_path3)�self�file� r �4/opt/alt/python35/lib64/python3.5/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)Nr# �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� )r9 r) r* r r r+ )r r r r �
test_mtimeA s zCompileallTests.test_mtimec C s | j d � d S)Ns )r9 )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/ Fr0 T) r r r �unlinkr6 � compile_filer r8 r
�isfiler7 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 r6 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)
Nr: r � r0 T�optimize� optimization)r: r: )r rF )� __debug__r6 r7 r r r r r r8 r r
r? r r )r rG �opt�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)Nr0 T�workers� )r6 r7 r r8 �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 0rM r: ���)ZassertRaisesRegex�
ValueErrorr6 r7 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)Nr0 TrM r r: � max_workers)r6 r7 r r3 � call_args)r rP 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)Nr0 T)r6 r7 r rC rO r8 )r �compile_file_mockrP 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)Nr0 TrM rN )r6 r7 r r8 rO )r rX r r r �$test_compile_missing_multiprocessing� s z4CompileallTests.test_compile_missing_multiprocessing)�__name__�
__module__� __qualname__r r" r, �unittestr �hasattrr r9 r; r< rA rE rL r �patchrQ rT rW rY r[ r r r r r s ' r c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� EncodingTestc 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)Nrc �ascii)�sys�stdout�io�
TextIOWrapper�BytesIOr6 r7 r )r Z orig_stdoutr r r �
test_error� s
zEncodingTest.test_errorN)r\ r] r^ r r" rj r r r r rb � s rb c @ s e 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. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z d8 d9 � Z d: d; � Z d< d= � Z! d> d? � Z" d@ dA � Z# dB dC � Z$ e% e& dD � dE dF � � Z' e( j) dG � dH dI � � Z* dJ S)K�CommandLineTestsc 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)NFrB ztest.tryTr z# for test_compileall)�filterr r
�isdirre �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)rs 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-Sr: z-O� z-OOz-mr6 )re �flagsrG �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_okr| r3 )r r{ �env_vars�rc�out�errr r r � assertRunOK� s z CommandLineTests.assertRunOKc O s1 t j | j | � | � \ } } } | | | f S)N)r �assert_python_failurer| )r r{ 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 r8 r r
rD )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 rC r r
rD )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)NZfoorB �__init__rF 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�bazrF �
PYTHONPATH) rv 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� rF r� �< rw z-f)rv r r� r r� r r r r �utime�timer' r( r3 r5 )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� rF r� s Listing z-q)rv r r� r r� ZassertInZ assertNotIn)r �noisyr0 r r r � test_no_args_respects_quiet_flag s
z1CommandLineTests.test_no_args_respects_quiet_flagZnormal�pycrG 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-mr6 z-qc 3 s' | ] } | j t j j � � Vq d S)N)�formatre �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� r8 r r
rD r� �sortedr3 �listdirrC )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� rC r r
rD r� r� r3 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-qrB ) r� r� r8 r r
rD r� r rC )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� rw z-f)
r� r� r r r r� r r� r� r' r( r3 r5 )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� rF �hamz-qz-lrB ) r r
r r� r r r� r� r� rC rD 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__.pyrF Zeggz-qz-r 0rB z-r 1z-r 2z-r 5) r r
r r� r Zmake_pkgr� r� r� rC rD 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� r5 r3 )r r� r0 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� r5 r3 )r �_r0 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� rF �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� ZassertNotRegexrC rD 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�f1rF �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� rF 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� rF r� r� r� z-i�-rd )
r r� r� Z spawn_pythonr| �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� rF rN zfoo{}r� z-j�0) r r� r �ranger r
r r� r ry 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.argvrZ z-j0r: rM rR ) r ra re �
executabler r6 �mainr8 rO r3 rV )r r7 r r r � test_workers_available_cores� s
z-CommandLineTests.test_workers_available_coresN)+r\ r] r^ � classmethodru rv r| 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 ra r� r r r r rk � sH
"
"
rk �__main__) re r6 �importlib.utilr r rn r1 r r) r r� r_ rg r r �concurrent.futuresr r� � ImportErrorZtestr Z test.supportr ZTestCaser rb rk r\ r� r r r r �<module> s2
z� M