Your IP : 216.73.216.196
B
f� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z dd� Z dd� Z
G dd� dee j��ZG d d
� d
�ZG d d � d ee jed
d�ZG dd� dee jedd�Zedkr�e �� dS )� N)�supportc s t �� �� fdd��}|S )Nc s( t �� �}|�d� � | |�S Q R X d S )N�SOURCE_DATE_EPOCH)r �EnvironmentVarGuardZunset)�args�kwargs�env)�fxn� �9/opt/alt/python37/lib64/python3.7/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 s& t �� �}d|d< � | |�S Q R X d S )NZ 123456789r )r r )r r r )r r r
r s
z'with_source_date_epoch.<locals>.wrapper)r r
)r r r )r r
�with_source_date_epoch s r c s e Zd Z� fdd�Z� ZS )�SourceDateEpochTestMetac s^ t � �| |||�}xFt|�D ]:}|�d�r t||�}|rBt|�}nt|�}t|||� q W |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� Zdd� Ze�ee d�o�e �� dkd�e�e jdkd�dd� ��Zdd� Zd d � Ze�ejjdkd �d d � �Zd!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 Q R X d S )N)r z_test.py�cr �wzx = 123
)�tempfileZmkdtemp�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
�setUp5 s
zPyCompileTestsBase.setUpc C s" t �| j� | jr t�| j� d S )N)�shutilZrmtreer( r2 r& r3 )r6 r r r
�tearDownE s zPyCompileTestsBase.tearDownc C s< t �| j| j� | �tj�| j�� | �tj�| j �� d S )N)
�
py_compile�compiler+ r, �
assertTruer& r) �exists� assertFalser0 )r6 r r r
�test_absolute_pathJ s z%PyCompileTestsBase.test_absolute_pathc C sf yt �| jd | j� W n" ttfk
r: | �d� Y n(X | �t�� t� | j
| j� W d Q R X d S )Nz.actualz.need to be able to create a symlink for a file) r& �symlinkr, �NotImplementedError�OSErrorZskipTest� assertRaises�FileExistsErrorr; r< r+ )r6 r r r
� test_do_not_overwrite_symlinksO 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 Q R X d S )N)rD rE r; r< r+ r& �devnull)r6 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 )N)r; r< r+ r= r& r) r> r0 )r6 r r r
�test_cache_pathc s z"PyCompileTestsBase.test_cache_pathc C sd t �| j��& t�tj�| j�tj�| j �� W d Q R X | �
tj� | j �� | � tj� | j
�� d S )N)r Z
change_cwdr( r; r< r& r) �basenamer+ r, r= r> r? r0 )r6 r r r
�test_cwdg s
zPyCompileTestsBase.test_cwdc C s. t j�t j�t�d�}| �tj|dd�� d S )Nzbad_coding2.pyF)�doraise)r& r) r* �dirname�__file__� assertIsNoner; r< )r6 �
bad_codingr r r
�test_bad_codingn s z"PyCompileTestsBase.test_bad_codingc C sL t �tj�| j�tj�| j�� | �tj�| j�� | � tj�| j
�� d S )N) r; r< r& r) �relpathr+ r, r= r> r? r0 )r6 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 d Q R X W d t �| j|j � X d S )N) r& �statr( �chmod�S_IREADrD �IOErrorr; r< r+ r, �st_mode)r6 �moder r r
�test_exceptions_propagatex s z,PyCompileTestsBase.test_exceptions_propagatec C s\ t j�t j�t�d�}t�� � | �tj |dd�� W d Q R X | �
t j� t j
�|��� d S )Nzbad_coding2.pyF)rL )r& r) r* rM rN r Zcaptured_stderrrO r; r< r? r> r- r. r/ )r6 rP r r r
rQ � s
c C s� t �| j| j� | �tj�| j�� | �tj�| j �� t
| jd��}t j �
|�� di �}W d Q R X tj�d�rvd}nd}| �||� d S )N�rb�testr � r )r; r< r+ r, r= r& r) r> r? r0 r4 r- �_bootstrap_external�
_classify_pyc�read�environ�get� assertEqual)r6 �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 Q R X t�|� | �t j�|�� | �t j�|�� d S )Nz
foo.bar.pyr# � __pycache__zfoo.bar.{}.pycr$ zx = 123
)r& r) r* r( r- r. r/ �splitrJ re �format�sys�implementation� cache_tagr4 r5 r; r< r= r> r? )r6 Z
weird_pathr0 r, �head�tailZpenultimate_tailr7 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)ZassertInr; r< r+ )r6 r r r
�test_optimization_path� s z)PyCompileTestsBase.test_optimization_pathc C s� t j| jt jjd� t| jd��}tj� |�
� di �}W d Q R X | � |d� t j| jt jj d� t| jd��}tj� |�
� di �}W d Q R X | � |d� d S )N)�invalidation_moder] r^ r_ � )
r; r< r+ �PycInvalidationMode� CHECKED_HASHr4 r0 r- r` ra rb re �UNCHECKED_HASH)r6 rf rg r r r
�test_invalidation_mode� s z)PyCompileTestsBase.test_invalidation_modeN) r r r r8 r: r@ rF �unittestZskipIfr& r) r> rG �isfilerH rI rK rQ rS �hasattrrT r r\ rh rl rg rs rq rt rz r r r r
r" 3 s&
r" c @ s e Zd ZdS )� PyCompileTestsWithSourceEpochN)r r r r r r r
r~ � s r~ T)� metaclassr c @ s e Zd ZdS )� PyCompileTestsWithoutSourceEpochN)r r r r r r r
r� � s r� F�__main__)r �importlib.utilr- r& r; r9 rV rl r% r{ r^ r r r �typeZTestCaser r" r~ r� r �mainr r r r
�<module> s0