Your IP : 216.73.216.213
�
f f�B � @ 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 Gd d � d e j � Z Gd d � d e j � Z Gd d � d e j � Z e d k r
e j � n d S)
� N)�support�
script_helperc @ s� 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 d S)�CompileallTestsc C s t j � | _ t j j | j d � | _ t j j | j � | _
t | j d � � } | j d � Wd QXt 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/python34/lib64/python3.4/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 sj t | j d � � } | j d � } Wd QXt 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 QXt | j d � �! } | j | � | j | � Wd QX| 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_check* s "
z CompileallTests.recreation_checkc C s& | j t j d t j j d � � d S)Nz<4sl� )r5 r% r& r
r r' )r r r r �
test_mtime9 s zCompileallTests.test_mtimec C s | j d � d S)Ns )r5 )r r r r �test_magic_number> 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 �unlinkr2 � compile_filer r4 r
�isfiler3 r )r �fnr r r �test_compile_filesB s z"CompileallTests.test_compile_filesc C s� t j j | j d � } t j j | d � } t j | � t | d � � Wd QXt j | � | j t j j t j j | d � � � d S)Nr( r r � __pycache__)
r r
r r r r r2 r: � assertFalse�exists)r Zdata_dirZ data_filer r r � test_no_pycache_in_non_packageS 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)Nr6 r r, T�optimize�debug_override)� __debug__r2 r3 r r
r r r r4 r r
r; r r )r rB �cachedZcached2Zcached3r r r �
test_optimize_ s
z CompileallTests.test_optimizeN)�__name__�
__module__� __qualname__r r r( �unittestZ
skipUnless�hasattrr r5 r7 r8 r= rA rF 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 si t j � | _ t j j | j d � | _ t | j d d d ��! } | j d � | j d � Wd QXd 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 s
zEncodingTest.setUpc C s t j | j � d S)N)r r r )r r r r r z s zEncodingTest.tearDownc C sL z; t j } t j t j � d d �t _ t j | j � Wd | t _ Xd S)NrM �ascii)�sys�stdout�io�
TextIOWrapper�BytesIOr2 r3 r )r Z orig_stdoutr r r �
test_error} s
zEncodingTest.test_errorN)rG rH rI �__doc__r r rT r r r r rL o s rL 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. 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@ S)A�CommandLineTestszTest compileall's CLI.c C s� x� t t j j t j � D]� } d } t j | � d } | d } zh yH | j � sg | j � d } n | j d � � } | j
d � Wd QXWn t k
r� d } PYn XWd t j
t | � � | r� | j � n Xq Wd } | | _ d S)NFr> ztest.tryTr z# for test_compileall)�filterr r
�isdirrO �pathlibZPathZis_dirr r r �OSErrorr r9 �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 � � n d S)Nz(not all entries on sys.path are writable)r] rJ ZSkipTest)r r r r � _skip_if_sys_path_not_writable� s z/CommandLineTests._skip_if_sys_path_not_writablec C sU d g } t j j r= | j i d d 6d d 6t j j � n | d d g t | � S)Nz-Sz-Or6 z-OO� z-mr2 )rO �flagsrB �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_okrf r/ )r re �env_vars�rc�out�errr r r � assertRunOK� s z CommandLineTests.assertRunOKc O s1 t j | j | � | � \ } } } | | | f S)N)r �assert_python_failurerf )r re ri rj rk rl r r r �assertRunNotOK� s z CommandLineTests.assertRunNotOKc C s/ t j j | � } | j t j j | � � d S)N)r
r r r4 r r
r@ )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 r? r r
r@ )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)NZfoor> �__init__� 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�bazrs �
PYTHONPATH) r` r rv r rm rp rq rw rx )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)Nry rs rz �< ra z-f)r` r rv r rm r
r r r �utime�timer# r$ r/ r1 )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)Nry rs rz s Listing z-q)r` r rv r rm ZassertInZ assertNotIn)r �noisyr, r r r � test_no_args_respects_quiet_flag� s
z1CommandLineTests.test_no_args_respects_quiet_flagZnormal�pycrB Zpyoz-OZdoubleoptimizez-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-mr2 z-qc 3 s' | ] } | j t j j � � Vq d S)N)�formatrO �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 rh rt r4 r r
r@ ru �sortedr/ �listdirr? )r r� �switch�expectedr )r� r �f� s
"zCommandLineTests.fZtest_pep3147_paths_c C s� | j d d | j � | j t j j | j � � t r> d n d } t d d | d d | g � } | j t t j
| j � � | � d S)Nz-bz-q�c�oz __init__.pyzbar.py) rm rt r? r r
r@ ru rD r� r/ r� )r Zoptr� 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-qr> ) rm rt r4 r r
r@ ru r r? )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} ra z-f)
rm rt r
r r rx r r~ r r# r$ r/ r1 )r r� r) r� r r r �
test_force 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)NZspamrr rs Zhamz-qz-lr> ) r r
r rt r r rv rm rq r? r@ rp )r Z
subpackageZ subinitfnZhamfnr r r �test_recursion_control$ s
(
z'CommandLineTests.test_recursion_controlc C sK | j | j � } | j d | j � } | j d | � | j d | � d S)Nz-qrg )rm rt r1 r/ )r r� r, r r r �
test_quiet0 s zCommandLineTests.test_quietc C s= | j d d d | j � | j | j � | j | j � d S)Nz-qz-xz
ba[^\\/]*$)rm rt rq rx rp rw )r r r r � test_regexp6 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)NZfoo2rr rs Zbar2z-q) r r
r r r r rv rm rt rp rw rx )r Zpkgdir2Zinit2fnZbar2fnr 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)NZ crunchyfrogz
bad(syntaxz-qz-d�dinsdales File "dinsdale)r rv rt ro � assertRegex)r rj rk rl r r r �test_d_compile_errorF 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) Nry zraise Exceptionz-qz-dr� �bingz
import bazzbaz.pycZ
__isolatedFs File "dinsdale)r rv rt rm r
r r r �renamer
r �removern r� )r r{ r< r� rj rk rl r r r �test_d_runtime_errorK 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)ro r r
r r rt r� ZassertNotRegexr? r@ r
r r ru )r rj rk rl r r r �test_include_bad_fileU s -z&CommandLineTests.test_include_bad_filec C s3 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 QX| j
d
t j j | j d � | � | j | � | j | � | j | � | j | � d S) N�f1rs �f2�f3�f4�l1r zf1.pyzf2.pyz-i)
r rv rt r r r
r r r �lineseprm rp rq )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 QX| j
d t j j | j d � � | j | � | j | � | j | � | j | � d S)
Nr� rs r� r� r� r� r zf2.pyz-i)
r rv rt r r r
r r r r� rm rq rp )r r� r� r� r� r� r r r �test_include_file_no_argk 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� rs r� r� r� z-i�-rN )
r rv rt Z spawn_pythonrf �stdinr r r� �encodeZ kill_pythonrq rp )r r� r� r� r� �pr r r �test_include_on_stdinx 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 rv rt ro rw rx r� rq rp )r Zbingfnrj rk rl 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')rm r� )r rk r r r �!test_invalid_arg_produces_message� s z2CommandLineTests.test_invalid_arg_produces_messageN)$rG rH rI rU � classmethodr_ r` rf rm ro rp rq 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 r rV � sB
"
rV �__main__)rO r2 �importlib.utilr
r rY r- r r% r r rJ rQ Ztestr r ZTestCaser rL rV rG �mainr r r r �<module> s$ _�