File: //usr/local/lib/python3.7/test/__pycache__/test_runpy.cpython-37.opt-2.pyc
B
��g0| � @ s^ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl m Z m
Z
mZmZmZmZmZ d dlmZmZmZmZ d dlZd dlmZmZmZm Z dZ ddddddd�Z eeddgej d d d e e d
d dd �d
�Z!e!�"e � G dd� d�Z#G dd� de j$e#�Z%G dd� de j$e#�Z&G dd� de j$e#�Z'e(dk�rZe �)� dS )� N)�forget�make_legacy_pyc�unload�verbose�
no_tracing�create_empty_file�temp_dir)�make_pkg� make_script� make_zip_pkg�make_zip_script)� _run_code�_run_module_code�
run_module�run_patha� # Check basic code execution
result = ['Top level assignment']
def f():
result.append('Lower level reference')
f()
del f
# Check the sys module
import sys
run_argv0 = sys.argv[0]
run_name_in_sys_modules = __name__ in sys.modules
module_in_sys_modules = (run_name_in_sys_modules and
globals() is sys.modules[__name__].__dict__)
# Check nested operation
import runpy
nested = runpy._run_module_code('x=1\n', mod_name='<run>')
)�__name__�__file__�
__cached__� __package__�__doc__�__spec__zTop level assignmentzLower level referenceF� z<run>)�xr �
__loader__)�sys�runpy�result� run_argv0�run_name_in_sys_modules�module_in_sys_modules�nestedc @ s, e Zd ZddddddgZdd� Zd d
� Zd S ) �CodeExecutionMixin�name�parent�origin�cached� has_location�submodule_search_locationsc C s8 |� � }|� � }xPt|�D ]D}|�d�r|�d�r||krD|�|� ||d kr|d �|� qW |�d�}|�d�}|d kr�| �|� nb|jd k r�| �t|j�t|j�� x>| j D ]4}d| }|t
||�f}|t
||�f}| �||� q�W | �t |�t |�� x2|D ]*}||| f}||| f}| �||� �qW d S )N�__r r z __spec__.) �copy�list�
startswith�endswith�popZ assertIsNone�loader� assertEqual�type�CHECKED_SPEC_ATTRIBUTES�getattr�set) �self� result_ns� expected_ns�kZ result_specZ
expected_spec�attr�actualZexpected� r: �+/usr/local/lib/python3.7/test/test_runpy.py�assertNamespaceMatchesM s2
z)CodeExecutionMixin.assertNamespaceMatchesc
C s� t � }|�� }|d }tjd }tj�||�}|d �}| �||� | �tjd |� | �tj�||�|� d|i} ||d< || �}| �|| � | �||� | �tjd |� | �tj�||�|� d S )Nr r �sentinel) �objectr) r �argv�modules�getr< ZassertIsZ assertIsNot)
r4 Zcreate_namespaceZexpected_namespacer= r6 �run_nameZ saved_argv0Z saved_modr5 Z
initial_nsr: r: r; �check_code_executionu s
z'CodeExecutionMixin.check_code_executionN)r �
__module__� __qualname__r1 r< rC r: r: r: r; r! A s (r! c @ s e Zd Zdd� Zdd� ZdS )�ExecutionLayerTestCasec C s. t �� }|�dd i� dd� }| �||� d S )Nr c S s t ti | �S )N)r
�example_source)� init_globalsr: r: r; � create_ns� s z7ExecutionLayerTestCase.test_run_code.<locals>.create_ns)�example_namespacer) �updaterC )r4 r6 rI r: r: r; �
test_run_code� s
z$ExecutionLayerTestCase.test_run_codec sd d� d}d}d}t jj� ||d��t�� }|�� |||�|ddd�� � �fdd �}| �||� d S )
Nz
<Nonsense>zSome other nonsensezNow you're just being silly� )r$ r. T)r r r r r r r r c s t t| � ��S )N)r rG )rH )�mod_name�mod_specr: r; rI � s z>ExecutionLayerTestCase.test_run_module_code.<locals>.create_ns)� importlib� machinery�
ModuleSpecrJ r) rK rC )r4 � mod_fnameZ
mod_loaderZ mod_packager6 rI r: )rN rO r; �test_run_module_code� s&
z+ExecutionLayerTestCase.test_run_module_codeN)r rD rE rL rT r: r: r: r; rF � s rF c @ s� e Zd Zdd� Zdd� Zdd� Zd9dd �Zd:ddd �d d
�Zdd� Zdd� Z d;ddd �dd�Z
d<ddd �dd�Z dd� Z d=dd�Z
dd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Zd7d8� Z dS )>�RunModuleTestCasec C s4 y t |� W n tk
r Y nX | �d| � d S )NzExpected import error for )r � ImportError�fail)r4 rN r: r: r; �expect_import_error� s
z%RunModuleTestCase.expect_import_errorc C s^ | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d� | � d � d S )
Nr z sys.imp.ericz os.path.halfza.beez.howardz..eatenz .test_runpyz .unittestZmultiprocessing)rX )r4 r: r: r; �test_invalid_names� s
z$RunModuleTestCase.test_invalid_namesc C s | � td�d d� d S )Nr r )r/ r )r4 r: r: r; �test_library_module� s z%RunModuleTestCase.test_library_moduleFc C s, t �|� |rd S t j�|d�}t|� |S )Nz __init__.py)�os�mkdir�path�joinr )r4 �pkg_dir� namespace� pkg_fnamer: r: r; � _add_pkg_dir� s
z RunModuleTestCase._add_pkg_dir�
runpy_test)r` �parent_namespacesc C s< |s|r|st d��d}|tj d }tj�t�� � }} tdkrLtd| � t j�
d|� tdkrrtdt jd � |r�|g| }
||
d< xJ|
D ]B} tj� | |�} | � | | �} tdkr�td | � tdkr�td
| � q�W tj� | |�}
t
|
d �}|�|� |�� tdk�rtd
|
� |d | | }tj�||
�}||
||fS )
Nz2Can't mark top level module as a namespace package�
__runpy_pkg__�pyr z Package tree in:r z Updated sys.path:���z Next level in:z
Created:�w�.)� RuntimeErrorr[ �extsepr] �realpath�tempfileZmkdtempr �printr �insertr^ rb �open�write�closerP �util�spec_from_file_location)r4 �source�depthZmod_baser` rd �pkg_nameZ
test_fnamer_ Zsub_dirZnamespace_flagsZnamespace_flagra rS �mod_filerN rO r: r: r; � _make_pkg� s<
zRunModuleTestCase._make_pkgc C s� x$t tj�D ]}|�d�r tj|= q W tdkr6td� tjd= tdkrNtd� x�tj|dd�D ]�\}}}xX|D ]P}yt� tj�
||�� W qn t k
r� } ztdkr�t|� W d d }~X Y qnX qnW x`|D ]X}tj�
||�}yt� |� W q� t k
�r } ztdk�rt|� W d d }~X Y q�X q�W q^W y t� |� tdk�rFtd� W n6 t k
�r~ } ztdk�rnt|� W d d }~X Y nX d S ) Nre r z Removed sys.modules entriesr z Removed sys.path entryF)�topdownz Removed package tree)
r* r r@ r+ r rn r] r[ �walk�remover^ �OSError�rmdir) r4 �top�entry�root�dirs�filesr" Zex�fullnamer: r: r; �_del_pkg� s:
"
zRunModuleTestCase._del_pkgc C s\ d}|d |7 < |d |d< |d }t j�|j|d �}||d< |rX|d |7 < d S )N�cr r r r )rP rs rt r" )r4 �ns� alter_sysZ char_to_add�specZnew_specr: r: r; �_fix_ns_for_legacy_pyc s
z(RunModuleTestCase._fix_ns_for_legacy_pycc
s | j t|||d�\}}�}t�� t�� }|��||j��d�d |d�� � rb|�|ddd�� � �fdd�} z�td kr�t d
�� | �
| |� t � � t
�� t�|� tjs�t|� t�� t � � td kr�t d �� | �|� � | �
| |� W d | �|� X td k�rt d � d S )
N)r` rd ri r )r r r r r T)r r r c s t �| � d�S )N)r� )r )rH )r� rN r: r; rI 4 s z2RunModuleTestCase._check_module.<locals>.create_nsr zRunning from source:zRunning from compiled:z Module executed successfully)ry rG r rJ r) rK r% �
rpartitionr rn rC rP �invalidate_caches�
__import__r[ r| r �dont_write_bytecoder r r� r� )
r4 rv r� r` rd r_ rS rO r6 rI r: )r� rN r; �
_check_module sF
z RunModuleTestCase._check_modulec s$ | j t|d||d�\}}}}|�d�d �t|� t�� } | �||tj� |��|d�� � rn| �|ddd�� � �fdd �}
z�t
d
kr�t d �� | � |
| � t�
� t|� t�|� tjs�t|� t|� t
d
kr�t d �� t�
� | �| � � | � |
| � W d | �|� X t
d
k�r t d
� d S )N�__main__)r` rd ri r )r r r r r T)r r r c s t �| � d�S )N)r� )r )rH )r� rw r: r; rI ] s z3RunModuleTestCase._check_package.<locals>.create_nsr zRunning from source:zRunning from compiled:z Package executed successfully)ry rG r� r rJ r) rK rP rs �cache_from_sourcer rn rC r� r� r[ r| r r� r r r� r� ) r4 rv r� r` rd r_ rS rN rO r6 rI r: )r� rw r; �_check_packageG sH
z RunModuleTestCase._check_packagec C s� |dkrt d��d}|}x"t|�D ]}|}tj�||�}q"W tj�|d�}t|� tdkrdtd|� tj�|d�} | �| � tdkr�td| � tj�| d�}
| �|
� tdkr�td |
� tj�|
d
�} t| � tdkr�td | � d S ) Nr z$Relative module test needs depth > 1re z
sibling.pyz Added sibling module:�unclez Added uncle package:Zcousinz Added cousin package:z nephew.pyz Added nephew module:) �
ValueError�ranger[ r] r^ r r rn rb ) r4 Zbase_dirru rv rw Z
module_dir�iZ
parent_dirZ
sibling_fnameZ uncle_dirZ
cousin_dirZ nephew_fnamer: r: r; �_add_relative_modulesp s.
z'RunModuleTestCase._add_relative_modulesNc C sZ d}| � ||�\}}}}|d kr&|}n|}�z
| �|||� |�d�d } tdkr\td|� t||d�}
| �|
d |� | �|
d | � | �d |
� | �d
|
� ~
t� � t
|� t � |� t
j�s4t|� t|� tdkr�td |� t� � t||d�} | �| d |� | �| d | � | �d | � | �d
| � ~ W d | �|� X tdk�rVtd � d S )
Nza\
from __future__ import absolute_import
from . import sibling
from ..uncle.cousin import nephew
ri r r zRunning from source:)rB r r ZsiblingZnephewzRunning from compiled:z Module executed successfully)ry r� r� r rn r r/ �assertInrP r� r� r[ r| r r� r r r� ) r4 rv rB �contentsr_ rS rN rO �
expected_namerw Zd1Zd2r: r: r; �_check_relative_imports� sD
z)RunModuleTestCase._check_relative_importsc C s2 x,t d�D ] }tdkr td|� | �|� q
W d S )N� r zTesting package depth:)r� r rn r� )r4 rv r: r: r; �test_run_module� s
z!RunModuleTestCase.test_run_modulec C s: x4t dd�D ]&}tdkr"td|� | j|ddd� q W d S )Nr r� zTesting package depth:T)r` rd )r� r rn r� )r4 rv r: r: r; �$test_run_module_in_namespace_package� s
z6RunModuleTestCase.test_run_module_in_namespace_packagec C s4 x.t dd�D ] }tdkr"td|� | �|� q W d S )Nr r� zTesting package depth:)r� r rn r� )r4 rv r: r: r; �test_run_package� s
z"RunModuleTestCase.test_run_packagec C sR | � ddd�}|\}}}}|�dd�}| �| j|� tj�|dd�}ttt t
f}x�|D ]�}|j }| � |��� d�
|�} t|dd d
��}
|
�| � W d Q R X y t|� W n2 |k
r� } z| �d t
| �� W d d } ~ X Y nX | �d �
|�� yt|d
� W n4 |k
�r. } z| �d t
| �� W d d } ~ X Y nX | �d �
|�� W d Q R X qVW d S )NrM r r� z .__main__re z __init__.pyz raise {0}('{0} in __init__.py.')Zwt�ascii)�encodingz finding speczNothing raised; expected {}z
.submodule)ry �replace�
addCleanupr� r[ r] r^ rV �AttributeError� TypeErrorr� r ZsubTest�formatrp rq r Z assertNotInrW ) r4 r r_ �_rN Zinit�
exceptionsZ exceptionr" ru rx �errr: r: r; � test_run_package_init_exceptions� s,
""z2RunModuleTestCase.test_run_package_init_exceptionsc C sR | � dd�\}}}}z,t|� | �td�� t|� W d Q R X W d | �|� X d S )NrM r zfound in sys\.modules)ry r� �assertWarnsRegex�RuntimeWarningr r� )r4 r_ r� rN r: r: r; � test_submodule_imported_warning� s z1RunModuleTestCase.test_submodule_imported_warningc C s� | � ddd�\}}}}| �| j|� |�dd�}t|� | �|tj� t� � � t�
dt � t |� W d Q R X t|� | �
t d�� t |� W d Q R X d S )NrM r r� z .__main__�errorzfound in sys\.modules)ry r� r� r� r� r� r r@ �warnings�catch_warnings� simplefilterr� r r� )r4 r_ r� rN �packager: r: r; � test_package_imported_no_warning� s
z2RunModuleTestCase.test_package_imported_no_warningc C s8 x2t dd�D ]$}tdkr"td|� | j|dd� q W d S )Nr r� zTesting package depth:T)rd )r� r rn r� )r4 rv r: r: r; �%test_run_package_in_namespace_package� s
z7RunModuleTestCase.test_run_package_in_namespace_packagec C s8 x2t dd�D ]$}tdkr"td|� | j|dd� q W d S )Nr r� zTesting package depth:T)r` )r� r rn r� )r4 rv r: r: r; �test_run_namespace_package� s
z,RunModuleTestCase.test_run_namespace_packagec C s: x4t dd�D ]&}tdkr"td|� | j|ddd� q W d S )Nr r� zTesting package depth:T)r` rd )r� r rn r� )r4 rv r: r: r; �/test_run_namespace_package_in_namespace_package� s
zARunModuleTestCase.test_run_namespace_package_in_namespace_packagec C s6 x0t d�D ]$}tdkr td|� | j|dd� q
W d S )Nr� r zTesting package depth:T)r� )r� r rn r� )r4 rv r: r: r; �test_run_module_alter_sys s
z+RunModuleTestCase.test_run_module_alter_sysc C s8 x2t dd�D ]$}tdkr"td|� | j|dd� q W d S )Nr r� zTesting package depth:T)r� )r� r rn r� )r4 rv r: r: r; �test_run_package_alter_sys s
z,RunModuleTestCase.test_run_package_alter_sysc C s4 x.t dd�D ] }tdkr"td|� | �|� q W d S )N� � r z"Testing relative imports at depth:)r� r rn r� )r4 rv r: r: r; � test_explicit_relative_import s
z/RunModuleTestCase.test_explicit_relative_importc C s6 x0t dd�D ]"}tdkr"td|� | �|d� q W d S )Nr� r� r z'Testing main relative imports at depth:r� )r� r rn r� )r4 rv r: r: r; �test_main_relative_import s
z+RunModuleTestCase.test_main_relative_importc s� d}d�| � t|�\}}� }t� � t�� }|��|tj�|�� � d�d |d�� � �fdd�}z| �
||� W d | � |� X d S )Nr z*And now for something completely differentri r )r r r r r c s t � | ��S )N)r )rH )rN rB r: r; rI $ s z2RunModuleTestCase.test_run_name.<locals>.create_ns) ry rG r rJ r) rK rP rs r� r� rC r� )r4 rv r_ rS rO r6 rI r: )rN rB r; �
test_run_name s
z RunModuleTestCase.test_run_namec C s� dd l }d}d}ddg}d|d g}t� }t� }xltd|�D ]^}d �|g| �} |�| � x |D ]}
|�| d |
� q`W x |D ]}
|�| d |
� q�W q<W d �|g| �} |�| � |�| d
� | �d |�\} } }
}| �| j| � x$td |d �D ]}| �| d |� q�W xX|� | g�D ]H}| �
||j � | �
|j t
jj� |j�rV|�|j� n |�|j� �q W | �t|�d|� | �t|�d|� d S )
Nr r� re r� z uncle.cousinzuncle.cousin.nephewz.siblingr ri z .runpy_testrM r� )�pkgutilr3 r� r^ �addry r� r� r� Z
walk_packagesZassertIsInstanceZ
ModuleInfoZ
module_finderrP rQ �
FileFinderZispkgr| r" r/ �len)r4 r� Z max_depthZ base_nameZpackage_suffixesZmodule_suffixesZexpected_packagesZexpected_modulesrv rw r" r_ rS rN rO Z
moduleinfor: r: r; �test_pkgutil_walk_packages+ s<
z,RunModuleTestCase.test_pkgutil_walk_packages)F)rc )F)F)N) r rD rE rX rY rZ rb ry r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r: r: r: r; rU � s:
'(
(
rU c @ s� e Zd Zd dd�Zd!dd�Zdd � Zd
d � Zd d
� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zed d � �Zd d � ZdS )"�RunPathTestCaseNFc C s |d kr t }t||||�S )N)rG r
)r4 �
script_dirZscript_basenameru � omit_suffixr: r: r; �_make_test_scriptR s z!RunPathTestCase._make_test_scriptTc
s� �fdd�}t �� } |d kr"|}
n|}
|rJtj�|
|�} | j} |sRd | _nd } } | �||| d| |ddd�� | �|| � d� � �fdd�}|r�|d kr�tj�� |�} |s�d | _| | d< � | d < � � d
�d | d < | �|| � d S )
Nc s
t � | �S )N)r )rH )� script_namer: r; rI ] s z0RunPathTestCase._check_script.<locals>.create_nsrM T)r r r r r r r r zprove.issue15230.is.fixedc s t �| � �S )N)r )rH )rB r� r: r; rI z s r r ri r r )
rJ r) rP rs rt r% r. rK rC r� )
r4 r� r� Z
expected_fileZexpected_argv0rN � expect_spec� check_loaderrI r6 Z spec_namerO Z
mod_cachedr: )rB r� r; �
_check_scriptY sB
z RunPathTestCase._check_scriptc C s t �|�}| �t|t|� d S )N)�re�escape�assertRaisesRegexrV r )r4 r� �msgr: r: r; �_check_import_error� s
z#RunPathTestCase._check_import_errorc C s: t � �*}d}| �||�}| j|d||dd� W d Q R X d S )N�scriptz
<run_path>F)r� )r r� r� )r4 r� rN r� r: r: r; �test_basic_script� s
z!RunPathTestCase.test_basic_scriptc C s> t � �.}d}| j||dd�}| j|d||dd� W d Q R X d S )Nr� T)r� z
<run_path>F)r� )r r� r� )r4 r� rN r� r: r: r; �test_basic_script_no_suffix� s
z+RunPathTestCase.test_basic_script_no_suffixc C sR t � �B}d}| �||�}tj|dd�}t�|� | j|d||dd� W d Q R X d S )Nr� T)�doraisez
<run_path>F)r� )r r� �
py_compile�compiler[ r| r� )r4 r� rN r� �
compiled_namer: r: r; �test_script_compiled� s
z$RunPathTestCase.test_script_compiledc C s: t � �*}d}| �||�}| j|d|||d� W d Q R X d S )Nr� z
<run_path>)rN )r r� r� )r4 r� rN r� r: r: r; �test_directory� s
z RunPathTestCase.test_directoryc C s` t � �P}d}| �||�}tj|dd�}t�|� tjsRt|�}| j |d|||d� W d Q R X d S )Nr� T)r� z
<run_path>)rN )
r r� r� r� r[ r| r r� r r� )r4 r� rN r� r� Z
legacy_pycr: r: r; �test_directory_compiled� s
z'RunPathTestCase.test_directory_compiledc C s: t � �*}d}| �||�}d| }| �||� W d Q R X d S )N�not_mainz"can't find '__main__' module in %r)r r� r� )r4 r� rN r� r� r: r: r; �test_directory_error� s
z$RunPathTestCase.test_directory_errorc C sL t � �<}d}| �||�}t|d|�\}}| j|d|||dd� W d Q R X d S )Nr� �test_zipz
<run_path>F)rN r� )r r� r r� )r4 r� rN r� �zip_name�fnamer: r: r; � test_zipfile� s z RunPathTestCase.test_zipfilec C sZ t � �J}d}| �||�}tj|dd�}t|d|�\}}| j|d|||dd� W d Q R X d S )Nr� T)r� r� z
<run_path>F)rN r� )r r� r� r� r r� )r4 r� rN r� r� r� r� r: r: r; �test_zipfile_compiled� s
z%RunPathTestCase.test_zipfile_compiledc C sJ t � �:}d}| �||�}t|d|�\}}d| }| �||� W d Q R X d S )Nr� r� z"can't find '__main__' module in %r)r r� r r� )r4 r� rN r� r� r� r� r: r: r; �test_zipfile_error� s z"RunPathTestCase.test_zipfile_errorc
C sf t � �V}t � �D}d}d| }| �|||�}t|d|�\}}d}| �t|t|� W d Q R X W d Q R X d S )Nr� z import runpy
runpy.run_path(%r)
r� zrecursion depth exceeded)r r� r r� �RecursionErrorr ) r4 r� Z dummy_dirrN ru r� r� r� r� r: r: r; �test_main_recursion_error� s z)RunPathTestCase.test_main_recursion_errorc
C s` t � �P}tj�|d�}t|ddd��}|�d� W d Q R X t|�}| �|d d� W d Q R X d S )Nz script.pyrh �latin1)r� u%
#coding:latin1
s = "non-ASCII: hé"
�su non-ASCII: hé)r r[ r] r^ rp rq r r/ )r4 r� �filename�fr r: r: r; �
test_encoding� s z RunPathTestCase.test_encoding)NF)NTT)r rD rE r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r: r: r: r; r� O s
+
r� r� )*Zunittestr[ Zos.pathr r� rm rP Zimportlib.machinery�importlib.utilr� r� Z test.supportr r r r r r r Ztest.support.script_helperr r
r r r r
r r r rG Zimplicit_namespacer? �dictrJ rK r! ZTestCaserF rU r� r �mainr: r: r: r; �<module> sL $
O'