File: //usr/local/lib/python3.10/test/__pycache__/test_compileall.cpython-310.opt-1.pyc
o
�i1� � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Zd dlZd dlZd dlmZmZ d dlmZ z
d dlmZ e� dZW n eyb dZY nw d dlmZ d dlmZ d d lm Z d d
l m Z d d l m Z d d
� Z dd� Z!dd� Z"G dd� d�Z#G dd� de#ej$e dd�Z%G dd� de#ej$e dd�Z&G dd� dej$�Z'G dd � d �Z(G d d � d e(ej$e dd�Z)G d d � d e(ej$e dd�Z*G d!d"� d"�Z+G d#d$� d$e+ej$e dd�Z,G d%d&� d&e+ej$e dd�Z-e.d'k�re�/� dS dS )(� N)�mock�
skipUnless)�ProcessPoolExecutor)�_check_system_limitsTF)�support)� os_helper)�
script_helper)�without_source_date_epoch)�SourceDateEpochTestMetac C s |sd}t jj| |d�S )N� �� optimization)� importlib�util�cache_from_source)�script�opt� r �1/usr/local/lib/python3.10/test/test_compileall.py�get_pyc# s r c s � fdd�dD �S )Nc s g | ]}t � |��qS r )r )�.0r �r r r �
<listcomp>+ s z get_pycs.<locals>.<listcomp>�r � � r r r r r �get_pycs* s r c C s t �| �j}t �|�j}||kS )z8Returns True if two files have the same inode (hardlink))�os�stat�st_ino)Z filename1� filename2Zinode1Zinode2r r r � is_hardlink. s r! c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zeed+�e�d,�d-d.� ��Z d/d0� Z eed+�e�d,�d1d2� ��Z eed+�e�d,�e�d3�d4d5� ���Z ejd,d6d7�e�d3�d8d9� ��Z d:d;� Z!d<d=�d>d?�Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&dHdI� Z'dJdK� Z(dLdM� Z)dNdO� Z*dPdQ� Z+e,j-dRdS� �Z.d6S )T�CompileallTestsBasec C s� t �� | _tj�| jd�| _tj� | j�| _
t | jddd��
}|� d� W d � n1 s.w Y tj�| jd�| _
tj� | j
�| _t�| j| j
� tj�| jd�| _t�| j� tj�| jd�| _t�| j| j� d S ) N�_test.py�w�utf-8��encodingzx = 123
z _test2.pyZ_subdirz _test3.py)�tempfile�mkdtemp� directoryr �path�join� source_pathr r r �bc_path�open�write� source_path2�bc_path2�shutil�copyfileZ subdirectory�mkdir� source_path3��self�filer r r �setUp7 s
� zCompileallTestsBase.setUpc C � t �| j� d S �N�r3 �rmtreer* �r8 r r r �tearDownE � z CompileallTestsBase.tearDownc C sR t j�| jd�| _t| jddd��}|�d� W d � d S 1 s"w Y d S )Nz _test_bad.pyr$ r% r& zx (
)r r+ r, r* �bad_source_pathr/ r0 r7 r r r �add_bad_source_fileH s "�z'CompileallTestsBase.add_bad_source_filec C sh t | jd��
}|�d�}W d � n1 sw Y tt�| j�j�}t� dt
j j d|d@ �}||fS )N�rb� �<4sLLr � �� )
r/ r. �read�intr r r- �st_mtime�struct�packr r � MAGIC_NUMBER)r8 r9 �data�mtimeZcomparer r r �timestamp_metadataM s ��z&CompileallTestsBase.timestamp_metadatac C �z z t �| jd� W n ttfy | �d� Y nw t�t� � �� | �
t � | j�� W d � d S 1 s6w Y d S )N)rG rG z0filesystem doesn't support timestamps near 2**32�
r �utimer- �
OverflowError�OSErrorZskipTest�
contextlib�redirect_stdout�io�StringIO�
assertTrue�
compileall� compile_filer? r r r � test_year_2038_mtime_compilationU � �"�z4CompileallTestsBase.test_year_2038_mtime_compilationc C rQ )N)� r_ z+filesystem doesn't support large timestampsrR r? r r r � test_larger_than_32_bit_times_ r^ z1CompileallTestsBase.test_larger_than_32_bit_timesc C s� t j�d�r t�d��t�| j� | j| � � � t
| j d��}|� � t
|�d� }W d � n1 s3w Y t
| j d��}|�|� |�|� W d � n1 sSw Y | j| � � � tj| jddd� | j| � � � dS ) zOCheck that compileall recreates bytecode when the new metadata is
used.�SOURCE_DATE_EPOCHzSOURCE_DATE_EPOCH is setrD N�wbFT��force�quiet)r �environ�get�unittestZSkipTest�
py_compile�compiler- � assertEqualrP r/ r. rH �lenr0 �assertNotEqualr[ � compile_dirr* rZ )r8 Zmetadatar9 �bcr r r �recreation_checki s
�
�z$CompileallTestsBase.recreation_checkc C s | � t�dtjjdd�� d S )NrF r r )rp rK rL r r rM r? r r r �
test_mtimey s
�z CompileallTestsBase.test_mtimec C s | � d� d S )Ns )rp r? r r r �test_magic_number~ � z%CompileallTestsBase.test_magic_numberc C s | j | jfD ]}zt�|� W q Y q| �tj| jddd�� | �tj� | j �o1tj� | j� � t�| j � | �tj
| j ddd�� | �tj� | j �oTtj� | j�� t�| j � t�| j� | � � | �
tj| jddd�� | �
tj
| j ddd�� d S )NFTrc r )r. r2 r �unlinkrZ r[ r\ r- r+ �isfilern r* rC � assertFalserB )r8 �fnr r r �test_compile_files� s6 �� � � � �z&CompileallTestsBase.test_compile_filesc C s| | � tj�| j�� t�� �}| �t� t
� | j ��� W d � n1 s%w Y | �
|�� d� | �tj�| j�� d S )Nz&Compiling ([^WindowsPath|PosixPath].*))rv r r+ ru r. r �captured_stdoutrZ r[ r\ �pathlib�Pathr- � assertRegex�getvalue)r8 �stdoutr r r �test_compile_file_pathlike� s
�z.CompileallTestsBase.test_compile_file_pathlikec C �P | � tj�| j�� | �tjt� | j
�t� d�dd�� | �tj�| j�� d S )NZ ddir_pathr )�ddirre � rv r r+ ru r. rZ r[ r\ rz r{ r- r? r r r � test_compile_file_pathlike_ddir� � �z3CompileallTestsBase.test_compile_file_pathlike_ddirc C r� �NZ
stripdir_pathr )�stripdirre r� r? r r r �#test_compile_file_pathlike_stripdir� r� z7CompileallTestsBase.test_compile_file_pathlike_stripdirc C r� �NZprependdir_pathr )�
prependdirre r� r? r r r �%test_compile_file_pathlike_prependdir� r� z9CompileallTestsBase.test_compile_file_pathlike_prependdirc C s� t jjj| jgd�� | �tjdd�� W d � n1 s w Y t jjj| jgd�� | �� | � tjdddd�� W d � d S 1 sHw Y d S )N�r+ r �re FT)� skip_curdirrd re )
�test�test_importlibr Z import_stater* rZ r[ � compile_pathrC rv r? r r r �test_compile_path� s �
�"�z%CompileallTestsBase.test_compile_pathc C s| t j�| jd�}t j�|d�}t �|� t|d�� W d � n1 s$w Y t�|� | �t j� t j�|d��� d S )NrN r9 rb � __pycache__)
r r+ r, r* r5 r/ r[ r\ rv �exists)r8 Zdata_dirZ data_filer r r � test_no_pycache_in_non_package� s
�
z2CompileallTestsBase.test_no_pycache_in_non_packagec C sN | � � t�t�� �� | �t�| j�� W d � d S 1 s w Y d S r< ) rC rV rW rX rY rv r[ r\ rB r? r r r �#test_compile_file_encoding_fallback� s "�z7CompileallTestsBase.test_compile_file_encoding_fallbackc C s� d\}}t j| jd|d� tjj| j|d�}| �tj �
|�� tjj| j |d�}| �tj �
|�� tjj| j |d�}| �tj �
|�� d S )NF)r r )r r T�re �optimizer )
r[ rn r* r r r r- rZ r r+ ru r1 r6 )r8 r� r �cachedZcached2Zcached3r r r �
test_optimize� s
�
�
�z!CompileallTestsBase.test_optimizec C s� | � tj�| j�� t�� �}t�t �
| j �� W d � n1 s"w Y |� � �
� d }| �|d� | �tj�| j�� d S )Nr z$Listing ([^WindowsPath|PosixPath].*))rv r r+ ru r. r ry r[ rn rz r{ r* r} �
splitlinesr| rZ )r8 r~ �liner r r �test_compile_dir_pathlike� s
� z-CompileallTestsBase.test_compile_dir_pathlikec C r� r� � rv r r+ ru r. rZ r[ rn rz r{ r* r? r r r �"test_compile_dir_pathlike_stripdir� r� z6CompileallTestsBase.test_compile_dir_pathlike_stripdirc C r� r� r� r? r r r �$test_compile_dir_pathlike_prependdir� r� z8CompileallTestsBase.test_compile_dir_pathlike_prependdir�requires multiprocessingz&concurrent.futures.ProcessPoolExecutorc C �" t j| jddd� | �|j� d S �NT� �re �workers�r[ rn r* rZ �called�r8 � pool_mockr r r �test_compile_pool_called� � z,CompileallTestsBase.test_compile_pool_calledc C sB | � td�� tj| jdd� W d � d S 1 sw Y d S )Nz%workers must be greater or equal to 0���)r� )ZassertRaisesRegex�
ValueErrorr[ rn r* r? r r r �!test_compile_workers_non_positive s
�"�z5CompileallTestsBase.test_compile_workers_non_positivec C s, t j| jddd� | �|jd d d � d S )NTr r� r � max_workers)r[ rn r* rk � call_argsr� r r r � test_compile_workers_cpu_count s z2CompileallTestsBase.test_compile_workers_cpu_countzcompileall.compile_filec C s, t j| jdd� | �|j� | �|j� d S )NTr� )r[ rn r* rv r� rZ )r8 �compile_file_mockr� r r r �test_compile_one_worker s z+CompileallTestsBase.test_compile_one_workerN��newc C r� r� r� )r8 r� r r r �$test_compile_missing_multiprocessing r� z8CompileallTestsBase.test_compile_missing_multiprocessingc C s� d}| j }td|d �D ] }tj�|d|� ��}tj�|d�}t�|� t�| j|� q t j
� |�}t j
| j d|d d� | �tj�|�� t j
| j d|d� | �tj�|�� d S )N� r Zdir_z script.pyT)re � maxlevels)r* �ranger r+ r, r5 r3 r4 r- r r r r[ rn rv ru rZ )r8 �depthr+ �i�sourceZ pyc_filenamer r r �test_compile_dir_maxlevels s
z.CompileallTestsBase.test_compile_dir_maxlevelsT)�parallelc C s ddg}| j }g }|D ] }tj�||�}t�|� t�|dd� |�t�|dd�� q tj | j d||r5dnd d
� | �
|� |D ]H}| �
|� | j �|� t j
�|�}|t| j tj �d � } t�|�\}
}
} tj�|| �} tjj
�|�}
| �|
j| � | �d | � d �t�| �� q@d S )
z@Recursive compile_dir ddir must contain package paths; bpo39769.r� �foo�__init__r �modzdef fn(): 1/0
fn()
Tr r )re r� r� N�")r* r r+ r, r5 r � make_script�appendr[ rn rZ �
startswithr r r rl �sep�assert_python_failurer� r� Zget_code_from_pycrk � co_filename�assertIn�fsdecode)r8 r� r� �fullpathr+ ZmodsZsubdirr� Zmodcode�modpath�_�err� expected_inZ mod_code_objr r r �_test_ddir_only* s2
�
�
�z#CompileallTestsBase._test_ddir_onlyc C � | j ddd�S )�=Recursive compile_dir ddir= contains package paths; bpo39769.�
<a prefix>F�r� r� �r� r? r r r �test_ddir_only_one_workerC rs z-CompileallTestsBase.test_ddir_only_one_workerc C r� )r� r� Tr� r� r? r r r �test_ddir_multiple_workersG rs z.CompileallTestsBase.test_ddir_multiple_workersc C r� )�?Recursive compile_dir ddir='' contains package paths; bpo39769.r Fr� r� r? r r r � test_ddir_empty_only_one_workerK rs z3CompileallTestsBase.test_ddir_empty_only_one_workerc C r� )r� r Tr� r� r? r r r � test_ddir_empty_multiple_workersO rs z4CompileallTestsBase.test_ddir_empty_multiple_workersc
C s� g d�}t jj| jg|�R � }t �|� t�|dd�}tj� |�}t jj| jg|d d� �R � }t
j |d|d� t� |�\}}}t jj|dd � � } | �
| t|t�� d�� | �|t|t�� d�� d S )N�r� �build�realr+ r� �1 / 0r T)re r� r& �r r+ r, r* �makedirsr r� r r r r[ rn r� r� �str�sys�getdefaultencoding� assertNotIn)
r8 r� r+ r ro r� �rc�outr� r� r r r �test_strip_onlyS s"
��z#CompileallTestsBase.test_strip_onlyc
C s� g d�}t jj| jg|�R � }t �|� t�|dd�}tj� |�}d}t
j |d|d� t� |�\}}}t jj|| jg|�R � } | �
| t|t�� d�� d S )Nr� r� r� �/fooT)re r� r& )r r+ r, r* r� r r� r r r r[ rn r� r� r� r� r� )
r8 r� r+ r ro r� r� r� r� r� r r r �test_prepend_onlyf s
�z%CompileallTestsBase.test_prepend_onlyc C s� g d�}t jj| jg|�R � }t �|� t�|dd�}tj� |�}t jj| jg|d d� �R � }d}t
j |d||d� t� |�\}}} t jj|g|dd � �R � }
| �
|
t| t�� d�� | �|t| t�� d�� d S ) Nr� r� r� r r� T)re r� r� r& r� � r8 r� r+ r ro r� r� r� r� r� r� r r r �test_strip_and_prependu s(
� ��z*CompileallTestsBase.test_strip_and_prependc C sz g d�}t jj| jg|�R � }t �|� t�|dd� | �t�� t j
|ddddd� W d � d S 1 s6w Y d S )N)r� r� r� r+ r� r� r� Tz/barr� )re r� r� r� ) r r+ r, r* r� r r� � assertRaisesr� r[ rn )r8 r� r+ r r r �test_strip_prepend_and_ddir� s
�"�z/CompileallTestsBase.test_strip_prepend_and_ddirc
C s� t �| jdd�}g }dD ]
}|�tjj||d�� q ddgddgddgg d�g}|D ],}tj|d |d
� |D ] }| � t
j � || �� z t
�
|| � W q6 tyU Y q6w q*d S ) N�test_optimization�a = 0�r r r r� r r r r r Tr� )r r� r* r� r r r r[ r\ rZ r r+ ru rt � Exception)r8 r ro � opt_level�test_combinations�opt_combinationr r r �!test_multiple_optimization_levels� s. �
� � ���z5CompileallTestsBase.test_multiple_optimization_levelsc
C s� t j�| jddd�}t j�| jddd�}t j�| jddd�}t �|� t �|� t �|� t�|dd�}t�|dd�}t j�|d �}t j�|d
�}t �||� t �||� tj �
|�}tj �
|�} t j |d |d � | �
t j�|�� | �t j�| �� d S )
Nr� �dir�allowed�symlinks�
prohibited� test_allowedr� �test_prohibited�test_allowed.py�test_prohibited.pyT)re �
limit_sl_dest)r r+ r, r* r� r r� �symlinkr r r r[ rn rZ ru rv �
r8 Z allowed_pathZ
symlinks_pathZprohibited_pathZallowed_scriptZprohibited_scriptZallowed_symlinkZprohibited_symlinkZ
allowed_bcZ
prohibited_bcr r r � test_ignore_symlink_destination� s"
z3CompileallTestsBase.test_ignore_symlink_destination)/�__name__�
__module__� __qualname__r: r@ rC rP r] r` rp rq rr rx r r� r� r� r� r� r� r� r� r� r� r �_have_multiprocessingr �patchr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r �skip_unless_symlinkr r r r r r" 5 sb
r" c @ � e Zd ZdS )� CompileallTestsWithSourceEpochN�r r r r r r r r � � r )� metaclassZsource_date_epochc @ r )�!CompileallTestsWithoutSourceEpochNr
r r r r r
� r r
c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � EncodingTestzTIssue 6716: compileall should escape source code when printing errors
to stdout.c C sf t �� | _tj�| jd�| _t| jddd��}|�d� |�d� W d � d S 1 s,w Y d S )Nr# r$ r% r& z# -*- coding: utf-8 -*-
u
print u"€"
) r( r) r* r r+ r, r- r/ r0 r7 r r r r: � s
"�zEncodingTest.setUpc C r; r< r= r? r r r r@ � rA zEncodingTest.tearDownc C s< zt j}tjt�� dd�t _t�| j� W |t _d S |t _w )N�asciir& )r� r~ rX �
TextIOWrapper�BytesIOr[ rn r* )r8 Z orig_stdoutr r r �
test_error� s
zEncodingTest.test_errorN)r r r �__doc__r: r@ r r r r r r � s
r c @ s� e Zd ZdZdd� Zejdd� �Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
edd� �Zdd� Zddg fdddgfdd d gffD ]\ZZZeefd d �Zee� d e < qCd!d"� Zd#d$� Zed%d&� �Zd'd(� Zd)d*� Zej d+d,� �Z d-d.� Z d/d0� Z d1d2� Z d3d4� Z!d5d6� Z"d7d8� Z#d9d:� Z$d;d<� Z%d=d>� Z&d?d@� Z'dAdB� Z(dCdD� Z)dEdF� Z*e+e,dG�dHdI� �Z-e.�/dJ�dKdL� �Z0dMdN� Z1dOdP� Z2ej dQdR� �Z3dSdT� Z4dUdV� Z5dWS )X�CommandLineTestsBasezTest compileall's CLI.c C sr t �� | _| �tj| j� tj�| jd�| _ t�
| j � tj�| j d�| _ t �
| j dd�| _t �
| j dd�| _d S )Nr� r� r� r Zbar)r( r) r* Z
addCleanupr r> r r+ r, �pkgdirr5 �pkgdir_cachedirr r� �initfn�barfnr? r r r r: � s
zCommandLineTestsBase.setUpc c sB � t j}tj�| jd�}z|t _| j|d�V W |t _dS |t _w )z&Adjust and restore sys.pycache_prefix.Z
__testcache__)�
PYTHONPATHZPYTHONPYCACHEPREFIXN)r� �pycache_prefixr r+ r, r* )r8 Z
old_prefixZ
new_prefixr r r �temporary_pycache_prefix� s �
�z-CommandLineTestsBase.temporary_pycache_prefixc C s g t �� �d�d�d�|�S )Nz-S�-mr[ )r Z!optim_args_from_interpreter_flags)r8 �argsr r r �
_get_run_args s
����z"CommandLineTestsBase._get_run_argsc O s4 t j| �|�i |�ddi��\}}}| �d|� |S )N�PYTHONIOENCODINGr% � )r �assert_python_okr rk �r8 r Zenv_varsr� r� r� r r r � assertRunOK s ��� z CommandLineTestsBase.assertRunOKc O s. t j| �|�i |�ddi��\}}}|||fS )Nr r% )r r� r r" r r r �assertRunNotOK s ���
z#CommandLineTestsBase.assertRunNotOKc C �" t j�|�}| �tj�|�� d S r< )r r r rZ r r+ r� �r8 rw r+ r r r �assertCompiled � z#CommandLineTestsBase.assertCompiledc C r% r< )r r r rv r r+ r� r& r r r �assertNotCompiled r( z&CommandLineTestsBase.assertNotCompiledc C sp t �| jdd�}| �� �"}| jdi |�� | �|� | �| j� | �| j� W d � d S 1 s1w Y d S )N�bazr r ) r r� r* r r# r' r) r r )r8 �bazfn�envr r r �test_no_args_compiles_path s
"�z/CommandLineTestsBase.test_no_args_compiles_pathc C s� t �| jdd�}| �� �}| jdi |�� tj�|�}W d � n1 s%w Y t� |t
�
� d fd � t� |�j }| jdi |�� t� |�j }| �
||� | jdi |�� t� |�j }| �||� d S )Nr* r �< r �-fr )r/ )r r� r* r r# r r r r rS �timer rJ rk rm )r8 r+ r, �pycpathrO �mtime2r r r � test_no_args_respects_force_flag# s
� z5CommandLineTestsBase.test_no_args_respects_force_flagc C st t �| jdd� | �� �}| jdi |��}W d � n1 s w Y | �d|� | jdi |��}| �d|� d S )Nr* r s Listing �-qr )r4 )r r� r* r r# r� r� )r8 r, �noisyre r r r � test_no_args_respects_quiet_flag5 s
� z5CommandLineTestsBase.test_no_args_respects_quiet_flagZnormal�pycr� z opt-1.pyc�-OZdoubleoptimizez opt-2.pycz-OOc s~ t j|ddd| jg � | �tj�| j�� t� fdd�dD ��}| � tt�
| j��|� | � � fdd�t�
| j�D �� d S ) Nr r[ r4 c 3 s � | ] }|� tjj� �V qd S r< )�formatr� �implementation� cache_tag)r �base��extr r � <genexpr>I s � �z)CommandLineTestsBase.f.<locals>.<genexpr>)z__init__.{}.{}z bar.{}.{}c s g | ] }|� � �r|�qS r )�endswith)r rw r= r r r M s
�z*CommandLineTestsBase.f.<locals>.<listcomp>) r r! r rZ r r+ r� r �sortedrk �listdirrv )r8 r> �switch�expectedr r= r �fD s � �$zCommandLineTestsBase.fZtest_pep3147_paths_c C sL | � dd| j� | �tj�| j�� tg d��}| �tt� | j��|� d S )Nz-br4 )� __init__.pyz __init__.pyczbar.pyzbar.pyc)
r# r rv r r+ r� r rA rk rB )r8 rD r r r �test_legacy_pathsQ s z&CommandLineTestsBase.test_legacy_pathsc C s| | � d| j� | �tj�| j�� tj�| jd�}| �tj�|�� | � d| j� | �tj�| j�� | �tj�|�� d S )Nr4 r� ) r# r rZ r r+ r� r r, rv )r8 Z
cachecachedirr r r �test_multiple_runs[ s z'CommandLineTestsBase.test_multiple_runsc C s� | � d| j� tj�| j�}t�|t�� d fd � t� |�j
}| � d| j� t� |�j
}| � ||� | � dd| j� t� |�j
}| � ||� d S )Nr4 r. r r/ )
r# r r r r r r rS r0 r rJ rk rm )r8 r1 rO r2 r r r �
test_forceh s z CommandLineTestsBase.test_forcec C s� t j�| jd�}t �|� t�|dd�}t�|dd�}| �dd| j� | �|� | � t j�
t j�|d��� | �d| j� | � |� | � |� d S )N�spamr� r �hamr4 z-lr� ) r r+ r, r r5 r r� r# r) rv r� r' )r8 �
subpackage� subinitfn�hamfnr r r �test_recursion_controlx s
z+CommandLineTestsBase.test_recursion_controlc C sV t j�| jd�}t j�|d�}t j�|d�}|||fD ]}t�|� qt j�|d�}t�|dd�}t�|dd�}t�|dd�}| �dd| j� | �|� | � t j�
t j�|d ��� | �dd
| j� | � |� | � |� | �|� | �dd | j� | � |� | � |� | � |� | �|� | �dd | j� | � |� | � |� | � |� | � |� d S )
NrJ rK ZeggsrF r Zeggr4 z-r 0r� z-r 1z-r 2z-r 5) r r+ r, r r �make_pkgr� r# r) rv r� r' ) r8 rL Z subpackage2Z subpackage3�pkgrM rN ZspamfnZeggfnr r r �test_recursion_limit� s8
�
z)CommandLineTestsBase.test_recursion_limitc C st t j�| jd�}t�|� t �dt j�|d�� t �dt j�|d�� | �d| j� | �t j�| jdddd�� d S )NrJ �.ZevilZevil2r4 rF ) r r+ r, r r rP r� r# r' )r8 rQ r r r �test_symlink_loop� s
�z&CommandLineTestsBase.test_symlink_loopc C s6 | � | j�}| � d| j�}| �d|� | �d|� d S )Nr4 r )r# r rm rk )r8 r5 re r r r �
test_quiet� s z CommandLineTestsBase.test_quietc C sT t �| jdd� | �d| j�\}}}| �d| j�\}}}| �d|� | �d|� d S )N� crunchyfrog�
bad(syntaxr4 z-qqr )r r� r r$ rm rk )r8 r� re Zsilentr r r � test_silent� s
z CommandLineTestsBase.test_silentc C s. | � ddd| j� | �| j� | �| j� d S )Nr4 z-xz
ba[^\\/]*$)r# r r) r r' r r? r r r � test_regexp� s z CommandLineTestsBase.test_regexpc C sv t j�| jd�}t �|� t�|dd�}t�|dd�}| �d| j|� | � | j
� | � | j � | � |� | � |� d S )NZfoo2r� r �bar2r4 ) r r+ r, r* r5 r r� r# r r' r r )r8 Zpkgdir2Zinit2fn�bar2fnr r r �test_multiple_dirs� s
z'CommandLineTestsBase.test_multiple_dirsc C s8 t �| jdd� | �ddd| j�\}}}| �|d� d S )NrV rW r4 �-d�dinsdale� File "dinsdale)r r� r r$ r| �r8 r� r� r� r r r �test_d_compile_error� s z)CommandLineTestsBase.test_d_compile_errorc C s� t �| jdd�}| �ddd| j� t �| jdd�}tj�|�}t�|tj �
| jd�� t� |� t j |d d
�\}}}| �
|d � d S ) Nr* zraise Exceptionr4 r] r^ �bingz
import bazzbaz.pycF)�
__isolatedr_ )r r� r r# r r r r �renamer+ r, �remover� r| )r8 r+ rw r7 r� r� r� r r r �test_d_runtime_error� s
z)CommandLineTestsBase.test_d_runtime_errorc C sZ | � dtj�| jd�| j�\}}}| �|d� | �|d� | �tj� t
j � | j
��� d S )N�-i�
nosuchfiles rror.*nosuchfiles Traceback)r$ r r+ r, r* r r| ZassertNotRegexrv r� r r r r r` r r r �test_include_bad_file� s
� �z*CommandLineTestsBase.test_include_bad_filec C s� t �| jdd�}t �| jdd�}t �| jdd�}t �| jdd�}ttj�| jd�ddd ��$}|�tj�| jd
�tj � |�tj�| jd �tj � W d � n1 sTw Y | �
d tj�| jd�|� | � |� | � |� | � |� | � |� d S )
N�f1r �f2�f3�f4�l1r$ r% r& �f1.py�f2.pyrg )
r r� r r/ r r+ r, r* r0 �linesepr# r' r) �r8 rj rk rl rm rn r r r �test_include_file_with_arg� s �
z/CommandLineTestsBase.test_include_file_with_argc C s� t �| jdd�}t �| jdd�}t �| jdd�}t �| jdd�}ttj�| jd�ddd ��}|�tj�| jd
�tj � W d � n1 sFw Y | �
d tj�| jd�� | � |� | � |� | � |� | � |� d S ) Nrj r rk rl rm rn r$ r% r& rp rg )
r r� r r/ r r+ r, r* r0 rq r# r) r' rr r r r �test_include_file_no_arg� s �
z-CommandLineTestsBase.test_include_file_no_argc C s� t �| jdd�}t �| jdd�}t �| jdd�}t �| jdd�}t j| �d�ddg � }|j�|tj � d �� t �
|� | � |� | � |� | � |� | � |� d S )
Nrj r rk rl rm r rg �-r )
r r� r Z spawn_pythonr �stdinr0 r rq �encodeZ kill_pythonr) r' )r8 rj rk rl rm �pr r r �test_include_on_stdin s
z*CommandLineTestsBase.test_include_on_stdinc C s\ t �| jdd�}| �d| j|| j�\}}}| �|d� | �|� | �| j� | �| j� d S )Nrb z syntax(errorrh s rror) r r� r r$ r r r| r) r' )r8 Zbingfnr� r� r� r r r �!test_compiles_as_much_as_possible s
�
z6CommandLineTestsBase.test_compiles_as_much_as_possiblec C s | � d�}| �|d� d S )NZ badfilenames Can't list 'badfilename')r# r| )r8 r� r r r �!test_invalid_arg_produces_message s
z6CommandLineTestsBase.test_invalid_arg_produces_messagec C s� t �| jdd� tj�tj�| jd��}| � d| j� t
|d�� }|� � }W d � n1 s/w Y | � t
�|dd� d�d � | � d
| j� t
|d�� }|� � }W d � n1 s]w Y | � t
�|dd� d�d � d S ) Nrj r ro z --invalidation-mode=checked-hashrD � � �littler� z"--invalidation-mode=unchecked-hashr )r r� r r r r r r+ r, r# r/ rH rk rI �
from_bytes)r8 r7 �fprN r r r �test_pyc_invalidation_mode s �
�
� z/CommandLineTestsBase.test_pyc_invalidation_moder� c C s� t �| jdd�}g }td�D ]#}tj�| jd�|��}t�|� t �|dd�}|� t �|dd�� q| �
| jdd� | � |� |D ]}| � |� qAd S )NrZ r r� zfoo{}r� z-j�0) r r� r* r� r r+ r, r9 r5 r� r# r' )r8 r[ �files�suffixr rw r9 r r r � test_workers* s
�z!CommandLineTestsBase.test_workerszcompileall.compile_dirc C sh t jdtj| jdgd�� t�� | �|j� | � |j
d d d� W d � d S 1 s-w Y d S )Nzsys.argvz-j0r� r� r� r ) r r r� �
executabler* r[ �mainrZ r� rk r� )r8 rn r r r � test_workers_available_cores9 s � "�z1CommandLineTestsBase.test_workers_available_coresc C s� g d�}t jj| jg|�R � }t �|� t�|dd�}tj� |�}t jj| jg|d d� �R � }d}| �
d|d||� t� |�\}}} t jj|g|dd � �R � }
| � |
t
| t�� d�� | �|t
| t�� d�� d S ) Nr� r� r� r r� z-sz-pr& )r r+ r, r* r� r r� r r r r# r� r� r� r� r� r� r� r r r r� A s$
��z+CommandLineTestsBase.test_strip_and_prependc
C s� t j�| jd�}t �|� t�|dd�}g }dD ]
}|�tj j
||d�� qddgddgddgg d �g}|D ]4}| j |gd
d � |D ��R � |D ]!}| � t j�
|t|� �� z t �|| � W qH tyi Y qHw q6d S ) NZ
optimizationsr� r� r� r r� �1�2)r� r� r� c s s � | ] }d t |� V qdS )z-oN)r� )r �nr r r r? d s � zICommandLineTestsBase.test_multiple_optimization_levels.<locals>.<genexpr>)r r+ r, r* r� r r� r� r r r r# rZ ru rI rt r� )r8 r+ r ro r� r� r� r r r r� U s6
�
�� ���z6CommandLineTestsBase.test_multiple_optimization_levelsc
C s� t j�| jddd�}t j�| jddd�}t j�| jddd�}t �|� t �|� t �|� t�|dd�}t�|dd�}t j�|d �}t j�|d
�}t �||� t �||� tj �
|�}tj �
|�} | � |d |� | � t j�
|�� | �t j�
| �� d S ) Nr� r� r� r� r� r� r� r� r� r� z-e)r r+ r, r* r� r r� r� r r r r# rZ ru rv r r r r r l s"
z4CommandLineTestsBase.test_ignore_symlink_destinationc C s | � | jdd� d S )N�-o 1�--hardlink-dupes)r$ r* r? r r r �test_hardlink_bad_args� s z+CommandLineTestsBase.test_hardlink_bad_argsc
C s� dD ]r}t �� �d}| j|d��N t�|dd�}t|�}g d�}|r&|�d� | j|g|�R � | �t |d |d �|� | �t |d |d �|� | �t |d |d �|� W d � n1 s`w Y W d � n1 sow Y qd S )
N)TF��dedupr r� )r4 z-o 0r� z-o 2r� r r r )
r( �TemporaryDirectory�subTestr r� r r� r# rk r! )r8 r� r+ r �pycsr r r r �
test_hardlink� s"
�� ���z"CommandLineTestsBase.test_hardlinkN)6r r r r r: rV �contextmanagerr r r# r$ r' r) r- r r3 r6 �namer> rC rE �localsrG rH rI rO rR r r rT rU rX rY r\ ra rf ri rs rt ry rz r{ r� r r r� r r r� r� r� r r� r� r r r r r � sf
�
"
r c @ r )� CommandLineTestsWithSourceEpochNr
r r r r r� � r r� c @ r )� CommandLineTestsNoSourceEpochNr
r r r r r� � r r� c @ s� e Zd Zdd� Zejdd� �Zd#dd�Zdd d
d �d d
�Zdd� Z d$dd�Z
dd� Z dd� Z d$dd�Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"S )%�HardlinkDedupTestsBasec C s
d | _ d S r< r� r? r r r r: � s
z HardlinkDedupTestsBase.setUpc c sB � t �� �}|| _|V d | _W d � d S 1 sw Y d S r< )r( r� r+ )r8 r+ r r r �temporary_directory� s �
"�z*HardlinkDedupTestsBase.temporary_directoryr c C s t �| j||�S r< )r r� r+ )r8 �coder� r r r r� � rA z"HardlinkDedupTestsBase.make_scriptTr F)r� r� rd c C s t j| jd|||d� d S )NT)re r� �hardlink_dupesrd )r[ rn r+ )r8 r� r� rd r r r rn � s
�z"HardlinkDedupTestsBase.compile_dirc C s� | � � �V | �d� | �t�� tj| jdddd� W d � n1 s$w Y | �t�� tj| jdddgdd� W d � n1 sEw Y W d � d S W d � d S 1 s]w Y d S )N�passTr )re r� r� )r� r� r� r� r[ rn r+ r? r r r �
test_bad_args� s
� � � ��"�z$HardlinkDedupTestsBase.test_bad_argsc C s4 g }|r |� d� |� d� |r|� d� d�|�S )Nz'module docstring'zx = 1z
assert x == 1�
)r� r, )r8 � docstring� assertion�linesr r r � create_code� s
z"HardlinkDedupTestsBase.create_codec c s4 � dD ]}dD ]}| j ||d�}|||fV qqd S )N)FT�r� r� )r� )r8 r� r� r� r r r �
iter_codes� s ���z!HardlinkDedupTestsBase.iter_codesc
C s� | � � D ]c\}}}| j||d��O | �� �; | �|�}t|�}| jdd� | �t|d |d �� | �t|d |d �� | �t|d |d �� W d � n1 sSw Y W d � n1 sbw Y qd S )Nr� Fr� r r r )r� r� r� r� r rn rv r! )r8 r� r� r� r r� r r r �
test_disabled� s
�� ���z$HardlinkDedupTestsBase.test_disabledc C sf t |�}| �t|d |d �| � | �t|d |d �| o!| � | �t|d |d �| � d S )Nr r r )r rk r! )r8 r r� r� r� r r r �check_hardlinks� s �
��z&HardlinkDedupTestsBase.check_hardlinksc
C s� | � � D ]@\}}}| j||d��, | �� � | �|�}| �� | �|||� W d � n1 s0w Y W d � n1 s?w Y qd S )Nr� )r� r� r� r� rn r� )r8 r� r� r� r r r r r� � s
�� ���z$HardlinkDedupTestsBase.test_hardlinkc
C s� dD ]O}| j |d��? | �� �+ | �| �� �}| j|d� t||d �}t||d �}| �t||�� W d � n1 s=w Y W d � n1 sLw Y qd S )N))r r )r r )r r )�opts�r� r r )r� r� r� r� rn r rZ r! )r8 r� r �pyc1�pyc2r r r �test_only_two_levels� s
�� ���z+HardlinkDedupTestsBase.test_only_two_levelsc C sp | � � �* | �| �� �}| jg d�d� t|d�}t|d�}| �t||�� W d � d S 1 s1w Y d S )N)r r r r r� r r )r� r� r� rn r rZ r! )r8 r r� r� r r r �test_duplicated_levels s
"�z-HardlinkDedupTestsBase.test_duplicated_levelsc C s� | � � �d | �d�}| �� | �|� t|�}t�|d �j}| �d�}| jddgdd� | �|t�|d �j� | � t
|d |d �� | � |t�|d �j� | � t
j|d |d dd�� W d � d S 1 skw Y d S ) Nr� r �print(0)r T)r� rd r ��shallow)r� r� rn r� r r r r rk rZ r! rm rv �filecmp�cmp�r8 r r� �inoder r r �test_recompilation s
"�z)HardlinkDedupTestsBase.test_recompilationc C s� | � � �l | j| �� dd�}| �� | �|� t|�}t�|d �j}| jddd�}t j
dddd| j d � | � |t�|d �j� | � |t�|d
�j� | �
t|d |d
�� | �
tj|d |d
d d
�� W d � d S 1 ssw Y d S )N�module)r� r r� r8 z-cz
import moduleF)rc r r r Tr� )r� r� r� rn r� r r r r r r! r+ rk rv r! r� r� r� r r r � test_import0 s
� "�z"HardlinkDedupTestsBase.test_importNr )FF)r r r r: rV r� r� r� rn r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� c @ r )�!HardlinkDedupTestsWithSourceEpochNr
r r r r r� L r r� c @ r )� HardlinkDedupTestsNoSourceEpochNr
r r r r r� S r r� �__main__)0r[ rV r� �importlib.utilr rX r rz ri r3 rK r� r( Ztest.test_importlib.utilr� r0 rh r r �concurrent.futuresr �concurrent.futures.processr r �NotImplementedErrorr Z test.supportr r Ztest.test_py_compiler r
r r r! r" ZTestCaser r
r r r� r� r� r� r� r r� r r r r �<module> s� � � � >� � � �
�