Your IP : 216.73.216.85
a
�Shk+ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z d dl
mZ dd� Zdd� ZG dd � d ee
j��ZG d
d � d �ZG d d
� d
ee
jedd�ZG dd� dee
jedd�ZG dd� de
j�Zedkr�e
�� dS )� N)�support)�
script_helperc s t �� �� fdd��}|S )Nc sN t �� �2}|�d� |�d� � | i |��W d � S 1 s@0 Y d S )N�SOURCE_DATE_EPOCH�RPM_BUILD_ROOT�r ZEnvironmentVarGuardZunset��args�kwargs�env��fxn� �9/opt/alt/python39/lib64/python3.9/test/test_py_compile.py�wrapper s
z*without_source_date_epoch.<locals>.wrapper�� functools�wraps�r r r
r r �without_source_date_epoch s r c s t �� �� fdd��}|S )Nc sL t �� �0}d|d< |�d� � | i |��W d � S 1 s>0 Y d S )NZ 123456789r r r r r r
r r s
z'with_source_date_epoch.<locals>.wrapperr r r
r r �with_source_date_epoch s r c s e Zd Z� fdd�Z� ZS )�SourceDateEpochTestMetac sZ t � �| |||�}t|�D ]:}|�d�rt||�}|r@t|�}nt|�}t|||� q|S )NZtest_)�super�__new__�dir�
startswith�getattrr r �setattr) �mcls�name�basesZdct�source_date_epoch�cls�attrZmethr �� __class__r
r r ( s
z SourceDateEpochTestMeta.__new__)�__name__�
__module__� __qualname__r �
__classcell__r
r
r# r r ' s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Ze�e j
� e j � pHe j
�
e j �d �d
d � �Zd d
� Zdd� Zdd� Ze�ee d�o�e �� dkd�e�e jdkd�dd� ��Zdd� Zdd � Ze�ejjdkd �d d � �Zd d!� Z d"d#� Z d$d%� Z d&S )'�PyCompileTestsBasec C s� t jt�� d�| _tj�| jd�| _| jd | _t j
� | j�| _ tj�
t�� �d | _tj�
| j�d }|rvt�|� t| jd��}|�d� W d � n1 s�0 Y d S )N)r �_test.py�cr �w�x = 123
)�tempfile�mkdtemp�os�getcwd� directory�path�join� source_path�pyc_path� importlib�util�cache_from_source�
cache_path�
splitdrive� cwd_drive�chdir�open�write)�selfZdrive�filer
r
r �setUp9 s
zPyCompileTestsBase.setUpc C s" t �| j� | jr t�| j� d S �N)�shutil�rmtreer2 r< r0 r= �r@ r
r
r �tearDownI s zPyCompileTestsBase.tearDownc C s< t �| j| j� | �tj�| j�� | �tj�| j �� d S rC )
�
py_compile�compiler5 r6 �
assertTruer0 r3 �exists� assertFalser: rF r
r
r �test_absolute_pathN s z%PyCompileTestsBase.test_absolute_pathc C sx zt �| jd | j� W n ttfy8 | �d� Y n<0 | �t�� t� | j
| j� W d � n1 sj0 Y d S )Nz.actualz.need to be able to create a symlink for a file) r0 �symlinkr6 �NotImplementedError�OSErrorZskipTest� assertRaises�FileExistsErrorrH rI r5 rF r
r
r � test_do_not_overwrite_symlinksS s z1PyCompileTestsBase.test_do_not_overwrite_symlinksz7requires os.devnull and for it to be a non-regular filec C s> | � t�� t�| jtj� W d � n1 s00 Y d S rC )rQ rR rH rI r5 r0 �devnullrF r
r
r �&test_do_not_overwrite_nonregular_files_ s z9PyCompileTestsBase.test_do_not_overwrite_nonregular_filesc C s$ t �| j� | �tj�| j�� d S rC )rH rI r5 rJ r0 r3 rK r: rF r
r
r �test_cache_pathg s z"PyCompileTestsBase.test_cache_pathc C sx t �| j��0 t�tj�| j�tj�| j �� W d � n1 sB0 Y | �
tj� | j �� | � tj� | j
�� d S rC )r Z
change_cwdr2 rH rI r0 r3 �basenamer5 r6 rJ rK rL r: rF r
r
r �test_cwdk s �"zPyCompileTestsBase.test_cwdc C sL t �tj�| j�tj�| j�� | �tj�| j�� | � tj�| j
�� d S rC ) rH rI r0 r3 �relpathr5 r6 rJ rK rL r: rF r
r
r �test_relative_pathr s
�z%PyCompileTestsBase.test_relative_path�geteuidr znon-root user required�ntz/cannot control directory permissions on Windowsc
C s� t �| j�}t �| jtj� zN| �t�� t�| j | j
� W d � n1 sN0 Y W t �| j|j � nt �| j|j � 0 d S rC ) r0 �statr2 �chmod�S_IREADrQ �IOErrorrH rI r5 r6 �st_mode)r@ �moder
r
r �test_exceptions_propagatex s 0z,PyCompileTestsBase.test_exceptions_propagatec C sp t j�t j�t�d�}t�� �$ | �tj |dd�� W d � n1 sH0 Y | �
t j� t j
�|��� d S )N�bad_coding2.pyF)�doraise)r0 r3 r4 �dirname�__file__r �captured_stderr� assertIsNonerH rI rL rK r7 r8 r9 )r@ �
bad_codingr
r
r �test_bad_coding� s
2
�z"PyCompileTestsBase.test_bad_codingc C s� t �| j| j� | �tj�| j�� | �tj�| j �� t
| jd��$}t j �
|�� di �}W d � n1 sn0 Y tj�d�r�d}nd}| �||� d S )N�rb�testr � r )rH rI r5 r6 rJ r0 r3 rK rL r: r> r7 �_bootstrap_external�
_classify_pyc�read�environ�get� assertEqual)r@ �fp�flagsZexpected_flagsr
r
r �test_source_date_epoch� s
�" z)PyCompileTestsBase.test_source_date_epochztest does not work with -Oc C s� t j�| jd�}tj�|�}|d }t j�|�\}}t j�|�}| � t j�||�t j�dd�
t j j
��� t|d��}|�d� W d � n1 s�0 Y t�|� | �t j�|�� | �t j�|�� d S )Nz
foo.bar.pyr+ � __pycache__zfoo.bar.{}.pycr, r- )r0 r3 r4 r2 r7 r8 r9 �splitrW rt �format�sys�implementation� cache_tagr> r? rH rI rJ rK rL )r@ Z
weird_pathr: r6 �head�tailZpenultimate_tailrA r
r
r �test_double_dot_no_clobber� s" �� (
z-PyCompileTestsBase.test_double_dot_no_clobberc C s | � dtj| jdd�� d S )Nzopt-2� )�optimize)�assertInrH rI r5 rF r
r
r �test_optimization_path� s z)PyCompileTestsBase.test_optimization_pathc C s� t j| jt jjd� t| jd��$}tj� |�
� di �}W d � n1 sJ0 Y | � |d� t j| jt jj d� t| jd��$}tj� |�
� di �}W d � n1 s�0 Y | � |d� d S )N)�invalidation_moderl rm rn � )
rH rI r5 �PycInvalidationMode� CHECKED_HASHr> r: r7 ro rp rq rt �UNCHECKED_HASH)r@ ru rv r
r
r �test_invalidation_mode� s$ �
�" �
�"z)PyCompileTestsBase.test_invalidation_modec C s� t j�t j�t�d�}t�� ��}| �tj |ddd�� | �tj |ddd�� | �
|� � d� | � tj
�� tj |ddd� W d � n1 s�0 Y W d � n1 s�0 Y d S )Nrd Fr� )re �quietT� r� )r0 r3 r4 rf rg r rh ri rH rI rt �getvaluerQ �PyCompileError)r@ rj �stderrr
r
r �
test_quiet� s
z PyCompileTestsBase.test_quietN) r% r& r' rB rG rM rS �unittestZskipIfr0 r3 rK rT �isfilerU rV rX rZ �hasattrr[ r rc rk rw r{ rv r� r� r� r� r� r
r
r
r r) 7 s2 �
� �
r) c @ s e Zd ZdS )� PyCompileTestsWithSourceEpochN�r% r& r' r
r
r
r r� � s r� T)� metaclassr c @ s e Zd ZdS )� PyCompileTestsWithoutSourceEpochNr� r
r
r
r r� � s r� Fc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�PyCompileCLITestCasec C sf t �� | _tj�| jd�| _tj� | j�| _
t | jd��}|� d� W d � n1 sX0 Y d S )Nr* r, r- )
r. r/ r2 r0 r3 r4 r5 r7 r8 r9 r: r> r? )r@ rA r
r
r rB � s
zPyCompileCLITestCase.setUpc C s t �| j� d S rC )r rE r2 rF r
r
r rG � s z PyCompileCLITestCase.tearDownc O sT |r:|d dkr:d|v r:t jtjdddg|d �� dd�S tjddg|�R i |��S )Nr �-�input�-mrH T)r� Zcapture_output)�
subprocess�runr{ �
executable�encoder Zassert_python_ok)r@ r r r
r
r � pycompilecmd� s
�z!PyCompileCLITestCase.pycompilecmdc G s t jddg|�R � S )Nr� rH )r Zassert_python_failure)r@ r r
r
r �pycompilecmd_failure� s z)PyCompileCLITestCase.pycompilecmd_failurec C sR | j d| jd�}| �|jd� | �|jd� | �|jd� | �tj� | j
�� d S )Nr� )r� r � ) r� r5 rt �
returncode�stdoutr� rJ r0 r3 rK r: )r@ �resultr
r
r �
test_stdin� s
z PyCompileCLITestCase.test_stdinc C sR | � | j| j�\}}}| �|d� | �|d� | �|d� | �tj�| j�� d S )Nr r� )r� r5 rt rJ r0 r3 rK r: )r@ �rcr� r� r
r
r �test_with_files� s
z$PyCompileCLITestCase.test_with_filesc C sN t j�t j�t�d�}| �|�\}}}| �|d� | �|d� | �d|� d S )Nzbadsyntax_3131.pyr� r� s SyntaxError)r0 r3 r4 rf rg r� rt r� )r@ Z
bad_syntaxr� r� r� r
r
r �test_bad_syntax s
z$PyCompileCLITestCase.test_bad_syntaxc C sR t j�t j�t�d�}| �| j|�\}}}| �|d� | �|d� | �d|� d S )Nzshould_not_exists.pyr� r� s No such file or directory) r0 r3 r4 rf rg r� r5 rt r� )r@ Zshould_not_existsr� r� r� r
r
r �test_file_not_exists
s
z)PyCompileCLITestCase.test_file_not_existsN) r% r& r' rB rG r� r� r� r� r� r� r
r
r
r r� � s
r� �__main__)r �importlib.utilr7 r0 rH rD r] r� r{ r. r� rm r Z test.supportr r r �typeZTestCaser r) r� r� r� r% �mainr
r
r
r �<module> s: � �6