File: //usr/local/lib/python3.7/test/__pycache__/test_py_compile.cpython-37.pyc
B
��g� � @ 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 )z+Runs function with SOURCE_DATE_EPOCH unset.c s( t �� �}|�d� � | |�S Q R X d S )N�SOURCE_DATE_EPOCH)r �EnvironmentVarGuardZunset)�args�kwargs�env)�fxn� �0/usr/local/lib/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 )z)Runs function with SOURCE_DATE_EPOCH set.c 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� Ze�ee d�o�e �� dkd�e�e jdkd�dd� ��Zdd� Zdd � 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 sx yt �| jd | j� W n" ttfk
r: | �d� Y n:X t j�| j�sNt�| � 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�OSErrorZskipTestr) �islink�AssertionError� 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)rF rG 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 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_pathn 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_IREADrF �IOErrorr; r< r+ r, �st_mode)r6 �moder r r
�test_exceptions_propagatet 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)�doraise)r& r) r* �dirname�__file__r Zcaptured_stderrZ assertIsNoner; r< r? r> r- r. r/ )r6 Z
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 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/ �splitrL 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@ rH �unittestZskipIfr& r) r> rI �isfilerJ rK rM rO �hasattrrP r rX 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 rR rl r% r{ r^ r r r �typeZTestCaser r" r~ r� r �mainr r r r
�<module> s0