File: //lib64/python3.6/test/__pycache__/test_compileall.cpython-36.pyc
3
\Z[ � @ s
d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dl
mZmZ yd dlmZ dZW n ek
r� dZY nX d dlmZ d dlmZ G dd � d e
j�ZG d
d � d e
j�ZG d d
� d
e
j�Ze dk�re
j � dS )� N)�mock�
skipUnless)�ProcessPoolExecutorTF)�support)�
script_helperc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zeje e
d �d
�d d � �Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zejd �d d!� �Zd"d#� Zejd �d$d%� �Zejd �ejd&�d'd(� ��Zejd d)d*�ejd&�d+d,� ��Zd)S )-�CompileallTestsc
C s� t j� | _tjj| jd�| _tjj | j�| _
t | jd��}|j d� W d Q R X tjj| jd�| _
tjj | j
�| _tj| j| j
� tjj| jd�| _tj| j� tjj| jd�| _tj| 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 �'/usr/lib64/python3.6/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 s8 t jj| jd�| _t| jd��}|jd� W d Q R X d S )Nz _test_bad.pyr zx (
)r r
r r �bad_source_pathr r )r r r r r �add_bad_source_file, s z#CompileallTests.add_bad_source_filec C sN t | jd��}|jd�}W d Q R X ttj| j�j�}tj dt
j j |�}||fS )N�rb� z<4sl)
r r �read�intr �statr �st_mtime�struct�packr r � MAGIC_NUMBER)r r �data�mtimeZcomparer r r r- 1 s
zCompileallTests.datar( ztest needs os.stat()c C s� t j| j� | j| j� � t| jd��}|j� t|�d� }W dQ R X t| jd��}|j |� |j |� W dQ R X | j
| j� � t j | j
ddd� | j| j� � dS )zOCheck that compileall recreates bytecode when the new metadata is
used.r$ N�wbFT)�force�quiet)�
py_compile�compiler � assertEqualr- r r r& �lenr �assertNotEqual�
compileall� compile_dirr �
assertTrue)r Zmetadatar Zbcr r r �recreation_check8 s
z CompileallTests.recreation_checkc C s | j tjdtjjd�� d S )Nz<4sl� )r: r* r+ r r r, )r r r r �
test_mtimeG s zCompileallTests.test_mtimec C s | j d� d S )Ns )r: )r r r r �test_magic_numberL s z!CompileallTests.test_magic_numberc
C s x0| j | jfD ] }ytj|� W q Y qX qW | jtj| jddd�� | jtjj | j �ojtjj | j� � tj| j � | jtj
| j ddd�� | jtjj | j �o�tjj | j�� tj| j � tj| j� | j � | j
tj| jddd�� | j
tj
| j ddd�� d S )NFT)r0 r1 � )r r r �unlinkr9 r7 � compile_filer r
�isfiler8 r r# � assertFalser" )r �fnr r r �test_compile_filesP s*
z"CompileallTests.test_compile_filesc C sh | j tjj| j�� tj� � }| jtj t
j | j ��� W d Q R X | j
|j� d� | jtjj| j�� d S )Nz&Compiling ([^WindowsPath|PosixPath].*))rB r r
rA r r �captured_stdoutr9 r7 r@ �pathlib�Pathr � assertRegex�getvalue)r �stdoutr r r �test_compile_file_pathlikei s
"z*CompileallTests.test_compile_file_pathlikec C sP | j tjj| j�� | jtjtj | j
�tj d�dd�� | jtjj| j�� d S )NZ ddir_pathr> )�ddirr1 ) rB r r
rA r r9 r7 r@ rF rG r )r r r r � test_compile_file_pathlike_ddirq s
z/CompileallTests.test_compile_file_pathlike_ddirc C st t jjj| jgd�� | jtjdd�� W d Q R X t jjj| jgd��$ | j� | j tjdddd�� W d Q R X d S )N)r
r> )r1 FT)� skip_curdirr0 r1 )
�testZtest_importlibr Z import_stater r9 r7 � compile_pathr# rB )r r r r �test_compile_pathx s
z!CompileallTests.test_compile_pathc C sh t jj| jd�}t jj|d�}t j|� t|d�� W d Q R X tj|� | jt jj t jj|d��� d S )Nr- r r � __pycache__)
r r
r r r r r7 r@ rB �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� dnd\}}t j| jd|d� tjj| j|d�}| jtj j
|�� tjj| j |d�}| jtj j
|�� tjj| j |d�}| jtj j
|�� d S ) NTr; r � )r1 �optimize)� optimization)r; r; )r rU )
r7 r8 r r r r r r9 r r
rA r r )r rV �opt�cachedZcached2Zcached3r r r �
test_optimize� s
z CompileallTests.test_optimizec
C sn | j tjj| j�� tj� �}tjt j
| j �� W d Q R X |j � j
� d }| j|d� | jtjj| j�� d S )Nr z$Listing ([^WindowsPath|PosixPath].*))rB r r
rA r r rE r7 r8 rF rG r rI �
splitlinesrH r9 )r rJ �liner r r �test_compile_dir_pathlike� s
z)CompileallTests.test_compile_dir_pathlikez&concurrent.futures.ProcessPoolExecutorc C s" t j| jddd� | j|j� d S )NT� )r1 �workers)r7 r8 r r9 �called)r � pool_mockr r r �test_compile_pool_called� s z(CompileallTests.test_compile_pool_calledc C s, | j td�� tj| jdd� W d Q R X d S )Nz%workers must be greater or equal to 0r; )r_ ���)ZassertRaisesRegex�
ValueErrorr7 r8 r )r r r r �!test_compile_workers_non_positive� s z1CompileallTests.test_compile_workers_non_positivec C s, t j| jddd� | j|jd d d � d S )NTr )r1 r_ r; � max_workers)r7 r8 r r4 � call_args)r ra r r r � test_compile_workers_cpu_count� s z.CompileallTests.test_compile_workers_cpu_countzcompileall.compile_filec C s, t j| jdd� | j|j� | j|j� d S )NT)r1 )r7 r8 r rB r` r9 )r �compile_file_mockra r r r �test_compile_one_worker� s z'CompileallTests.test_compile_one_workerN)�newc C s" t j| jddd� | j|j� d S )NTr^ )r1 r_ )r7 r8 r r9 r` )r ri r r r �$test_compile_missing_multiprocessing� s z4CompileallTests.test_compile_missing_multiprocessing) �__name__�
__module__� __qualname__r r! r# r- �unittestr �hasattrr r: r< r= rD rK rM rQ rT rZ r] r �patchrb re rh rj rl r r r r r s* r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � EncodingTestzTIssue 6716: compileall should escape source code when printing errors
to stdout.c C sP t j� | _tjj| jd�| _t| jddd��}|jd� |jd� W d Q R X d S )Nz_test.pyr zutf-8)�encodingz# -*- 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 s8 z*t j}tjtj� dd�t _tj| j� W d |t _X d S )N�ascii)rt )�sysrJ �io�
TextIOWrapper�BytesIOr7 r8 r )r Z orig_stdoutr r r �
test_error� s
zEncodingTest.test_errorN)rm rn ro �__doc__r r! rz r r r r rs � s rs c @ sd e Zd ZdZedd� �Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZxJddg fddd gfd d d gfgD ]&\ZZZeefd d!�Zee� d"e < q�W d#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z!d=d>� Z"d?d@� Z#dAdB� Z$dCdD� Z%e&e'dE�dFdG� �Z(e)j*dH�dIdJ� �Z+dKS )L�CommandLineTestszTest compileall's CLI.c C s� x�t tjjtj�D ]�}d}tj|�d }|d }zXy8|j� sH|j� d}|j d��}|j
d� W d Q R X W n t k
r� d}P Y nX W d t j
t|�� |r�|j� X qW d}|| _d S )NFrR ztest.tryTr z# for test_compileall)�filterr r
�isdirrv rF rG �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 stjd��d S )Nz(not all entries on sys.path are writable)r� rp ZSkipTest)r r r r � _skip_if_sys_path_not_writable� s z/CommandLineTests._skip_if_sys_path_not_writablec C s t j� d|�S )N�-S�-mr7 )r� r� r7 )r Z!optim_args_from_interpreter_flags)r �argsr r r �
_get_run_args� s z CommandLineTests._get_run_argsc O s( t j| j|�|�\}}}| jd|� |S )N� )r �assert_python_okr� r4 )r r� �env_vars�rc�out�errr r r � assertRunOK� s z CommandLineTests.assertRunOKc O s" t j| j|�|�\}}}|||fS )N)r �assert_python_failurer� )r r� r� r� r� r� r r r �assertRunNotOK s z CommandLineTests.assertRunNotOKc C s" t jj|�}| jtjj|�� d S )N)r r r r9 r r
rS )r rC r
r r r �assertCompiled s z CommandLineTests.assertCompiledc C s" t jj|�}| jtjj|�� d S )N)r r r rB r r
rS )r rC r
r r r �assertNotCompiled s z"CommandLineTests.assertNotCompiledc C sr t j� | _| jtj| j� tjj| jd�| _ tj
| j � tjj| j d�| _ t j
| j dd�| _t j
| j dd�| _d S )NZfoorR �__init__rU 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 sL | j � tj| jdd�}| j| jd� | j|� | j| j� | j| j� d S )N�bazrU )�
PYTHONPATH) r� 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 � tj| jdd�}| j| jd� tjj|�}tj |t
j
� d fd � tj |�j }| j| jd� tj |�j }| j
||� | jd| jd� tj |�j }| j||� d S )Nr� rU )r� �< r> z-f)r� r r� r r� r r r r �utime�timer( r) r4 r6 )r r� �pycpathr. �mtime2r r r � test_no_args_respects_force_flag s z1CommandLineTests.test_no_args_respects_force_flagc C sR | j � tj| jdd� | j| jd�}| jd|� | jd| jd�}| jd|� d S )Nr� rU )r� s Listing z-q)r� r r� r r� ZassertInZ assertNotIn)r �noisyr1 r r r � test_no_args_respects_quiet_flag1 s z1CommandLineTests.test_no_args_respects_quiet_flagZnormal�pycrV z opt-1.pycz-OZdoubleoptimizez opt-2.pycz-OOc s~ t j|ddd| jg � | jtjj| j�� t� fdd�d
D ��}| j ttj
| j��|� | j � fdd �tj
| j�D �� d S ) Nz-mr7 z-qc 3 s | ]}|j tjj� �V qd S )N)�formatrv �implementation� cache_tag)�.0�base)�extr r � <genexpr>E s z%CommandLineTests.f.<locals>.<genexpr>�__init__.{}.{}� bar.{}.{}c s g | ]}|j � �r|�qS r )�endswith)r� rC )r� r r �
<listcomp>I s z&CommandLineTests.f.<locals>.<listcomp>)r� r� ) r r� r� r9 r r
rS r� �sortedr4 �listdirrB )r r� �switch�expectedr )r� r �f@ s
zCommandLineTests.fZtest_pep3147_paths_c C sP | j dd| j� | jtjj| j�� tddddg�}| jttj | j��|� d S )Nz-bz-qz __init__.pyz __init__.pyczbar.pyzbar.pyc)
r� r� rB r r
rS r� r� r4 r� )r r� r r r �test_legacy_pathsM s
z"CommandLineTests.test_legacy_pathsc C s| | j d| j� | jtjj| j�� tjj| jd�}| jtjj|�� | j d| j� | jtjj| j�� | jtjj|�� d S )Nz-qrR ) r� r� r9 r r
rS r� r rB )r Z
cachecachedirr r r �test_multiple_runsW s z#CommandLineTests.test_multiple_runsc C s� | j d| j� tjj| j�}tj|tj� d fd � tj |�j
}| j d| j� tj |�j
}| j ||� | j dd| j� tj |�j
}| j ||� d S )Nz-qr� r> z-f)
r� r� r r r r� r r� r� r( r) r4 r6 )r r� r. r� r r r �
test_forced s zCommandLineTests.test_forcec C s� t jj| jd�}t j|� tj|dd�}tj|dd�}| jdd| j� | j|� | j t jj
t jj|d��� | jd| j� | j |� | j |� d S )N�spamr� rU �hamz-qz-lrR ) r r
r r� r r r� r� r� rB rS r� )r �
subpackage� subinitfn�hamfnr r r �test_recursion_controls s
z'CommandLineTests.test_recursion_controlc C sZ t jj| jd�}t jj|d�}t jj|d�}x |||fD ]}tj|� q8W t jj|d�}tj|dd�}tj|dd�}tj|dd�}| jdd| j� | j|� | j t jj
t jj|d ��� | jdd
| j� | j |� | j |� | j|� | jdd | j� | j |� | j |� | j |� | j|� | jdd | j� | j |� | j |� | j |� | j |� d S )
Nr� r� Zeggsz __init__.pyrU Zeggz-qz-r 0rR z-r 1z-r 2z-r 5) r r
r r� r Zmake_pkgr� r� r� rB rS r� ) r r� Z subpackage2Z subpackage3Zpkgr� r� ZspamfnZeggfnr r r �test_recursion_limit s6
z%CommandLineTests.test_recursion_limitc C s6 | j | j�}| j d| j�}| jd|� | jd|� d S )Nz-qr� )r� r� r6 r4 )r r� r1 r r r �
test_quiet� s zCommandLineTests.test_quietc C sT t j| jdd� | jd| j�\}}}| jd| j�\}}}| jd|� | jd|� d S )N� crunchyfrogz
bad(syntaxz-qz-qqr� )r r� r� r� r6 r4 )r �_r1 Zsilentr r r � test_silent� s
z CommandLineTests.test_silentc C s. | j ddd| 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 sv t jj| jd�}t j|� tj|dd�}tj|dd�}| jd| j|� | j | j
� | j | j � | j |� | j |� d S )NZfoo2r� rU �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 s8 t j| jdd� | jddd| j�\}}}| j|d� d S )Nr� z
bad(syntaxz-qz-d�dinsdales File "dinsdale)r r� r� r� rH )r r� r� r� r r r �test_d_compile_error� s z%CommandLineTests.test_d_compile_errorc C s� t j| jdd�}| jddd| j� t j| jdd�}tjj|�}tj|tj j
| jd�� tj |� t j |d d
�\}}}| j
|d � d S ) Nr� zraise Exceptionz-qz-dr� �bingz
import bazzbaz.pycF)Z
__isolateds File "dinsdale)r r� r� r� r r r r �renamer
r �remover� rH )r r� rC r� r� r� r� r r r �test_d_runtime_error� s
z%CommandLineTests.test_d_runtime_errorc C sZ | j dtjj| jd�| j�\}}}| j|d� | j|d� | jtjj t
j j | j
��� d S )Nz-i�
nosuchfiles rror.*nosuchfiles Traceback)r� r r
r r r� rH ZassertNotRegexrB rS r r r r� )r r� r� r� r r r �test_include_bad_file� s z&CommandLineTests.test_include_bad_filec C s� t j| jdd�}t j| jdd�}t j| jdd�}t j| jdd�}ttjj| jd�d��>}|jtjj| jd�tj � |jtjj| jd �tj � W d Q R X | j
d
tjj| jd�|� | j |� | j |� | j |� | j |� d S ) N�f1rU �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| jdd�}t j| jdd�}t j| jdd�}t j| jdd�}ttjj| jd�d��"}|jtjj| jd�tj � W d Q R X | j
d tjj| jd�� | j |� | j |� | j |� | j |� d S )
Nr� rU 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| jdd�}t j| jdd�}t j| jdd�}t j| jdd�}t j| jf �ddg � }|jj|tj j d�� t j
|� | j |� | j |� | j |� | j |� d S ) Nr� rU r� r� r� z-i�-ru )
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| jdd�}| jd| j|| j�\}}}| j|d� | j|� | j| j� | j| j� d S )Nr� z syntax(errorr� s rror) r r� r� r� r� r� rH 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� rH )r r� r r r �!test_invalid_arg_produces_message s
z2CommandLineTests.test_invalid_arg_produces_messagezrequires multiprocessingc C s� t j| jdd�}g }xRtd�D ]F}tjj| jdj|��}tj|� t j|dd�}|j t j|dd�� q W | j
| jdd� | j |� x|D ]}| j |� q�W d S )Nr� rU r^ zfoo{}r� z-j�0) r r� r �ranger r
r r� r �appendr� r� )r r� �files�suffixr� rC r r r r � test_workers
s
z CommandLineTests.test_workerszcompileall.compile_dirc
C sR t jdtj| jdgd��0 tj� | j|j� | j |j
d d d � W d Q R X d S )Nzsys.argvz-j0)rk r; r_ rc ) r rr rv �
executabler r7 �mainr9 r` r4 rg )r r8 r r r � test_workers_available_cores s
z-CommandLineTests.test_workers_available_coresN),rm rn ro r{ � classmethodr� r� 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 rr r� r r r r r| � sJ
"
r| �__main__) rv r7 �importlib.utilr Ztest.test_importlib.utilrO r rF r2 r r* r r� rp rw r r �concurrent.futuresr r� � ImportErrorr Z test.supportr ZTestCaser rs r| rm r� r r r r �<module> s8
( M