Your IP : 216.73.217.129
U
�M�fk+ � @ 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 )z+Runs function with SOURCE_DATE_EPOCH unset.c
s> t �� �,}|�d� |�d� � | |�W 5 Q R � S Q R X d S )N�SOURCE_DATE_EPOCH�RPM_BUILD_ROOT�r ZEnvironmentVarGuardZunset��args�kwargs�env��fxn� �9/opt/alt/python38/lib64/python3.8/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 )z)Runs function with SOURCE_DATE_EPOCH set.c
s< t �� �*}d|d< |�d� � | |�W 5 Q R � S Q R X 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 5 Q R X 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" ttfk
r: | �d� Y n:X t j�| j�sNt�| � t
�� t � | j
| j� W 5 Q R X d S )Nz.actualz.need to be able to create a symlink for a file)r0 �symlinkr6 �NotImplementedError�OSErrorZskipTestr3 �islink�AssertionError� 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 5 Q R X d S rC )rS rT 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 sd t �| j��& t�tj�| j�tj�| j �� W 5 Q R X | �
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� z*| �t�� t� | j
| j � W 5 Q R X W 5 t �| j|j� X d S rC ) r0 �statr2 �chmod�S_IREAD�st_moderS �IOErrorrH rI r5 r6 )r@ �moder
r
r �test_exceptions_propagatex s z,PyCompileTestsBase.test_exceptions_propagatec C s\ t j�t j�t�d�}t�� � | �tj |dd�� W 5 Q R X | �
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
�z"PyCompileTestsBase.test_bad_codingc C s� t �| j| j� | �tj�| j�� | �tj�| j �� t
| jd��}t j �
|�� di �}W 5 Q R X tj�d�rvd}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 5 Q R X t�|� | �t j�|�� | �t j�|�� d S )Nz
foo.bar.pyr+ � __pycache__zfoo.bar.{}.pycr, r- )r0 r3 r4 r2 r7 r8 r9 �splitrY rv �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 5 Q R X | � |d� t j| jt jj d� t| jd��}tj� |�
� di �}W 5 Q R X | � |d� d S )N)�invalidation_modern ro rp � )
rH rI r5 �PycInvalidationMode� CHECKED_HASHr> r: r7 rq rr rs rv �UNCHECKED_HASH)r@ rw rx r
r
r �test_invalidation_mode� s, � � � �z)PyCompileTestsBase.test_invalidation_modec
C s� t j�t j�t�d�}t�� �j}| �tj |ddd�� | �tj |ddd�� | �
|� � d� | � tj
�� tj |ddd� W 5 Q R X W 5 Q R X d S )Nrf Fr� )rg �quietT� r� )r0 r3 r4 rh ri r rj rk rH rI rv �getvaluerS �PyCompileError)r@ rl �stderrr
r
r �
test_quiet� s
z PyCompileTestsBase.test_quietN) r% r& r' rB rG rM rU �unittestZskipIfr0 r3 rK rV �isfilerW rX rZ r\ �hasattrr] r re rm ry r} rx 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 sR t �� | _tj�| jd�| _tj� | j�| _
t | jd��}|� d� W 5 Q R X 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 sJ |r:|d dkr:d|kr:t jtjdddg|d �� dd�S tjd|�|�S ) Nr �-�input�-mrH T)r� Zcapture_output)r� rH )�
subprocess�runr} �
executable�encoder Zassert_python_ok)r@ r r r
r
r � pycompilecmd� s
�z!PyCompileCLITestCase.pycompilecmdc G s t jd|�� S )Nr� rH )r� 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 rv �
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 rv 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 rh ri r� rv 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 rh ri r� r5 rv 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� ro r Z test.supportr r r �typeZTestCaser r) r� r� r� r% �mainr
r
r
r �<module> s: � �6