Your IP : 216.73.216.213
��Yf�{ � @ s d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l m Z m
Z
m Z m Z m Z m Z m Z d d l m Z m Z m Z m Z d d l Z d d l m Z m Z m Z m Z d Z d d d d d d d d d
d d d i Z d e d
e d d d g d e j d d d d d d e e d d d d d d �i Z! e! j" e � Gd d � d � Z# Gd d � d e j$ e# � Z% Gd d � d e j$ e# � Z& Gd d! � d! e j$ e# � Z' e( d" k re j) � d S)#� 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__�sys�runpy�resultzTop level assignmentzLower level reference� run_argv0�run_name_in_sys_modulesF�module_in_sys_modules�nested�x� z<run>�
__loader__c @ s@ e Z d Z d d d d d d g Z d d � Z d d
� Z d S) �CodeExecutionMixin�name�parent�origin�cached� has_location�submodule_search_locationsc C s� | j � } | j � } xl t | � D]^ } | j d � r% | j d � r% | | k rb | j | � | | d k r% | d j | � q% W| j d � } | j d � } | d k r� | j | � n� | j d k r� | j t | j � t | j � � xU | j D]J } d | } | t
| | � f } | t
| | � f } | j | | � q� W| j t | � t | � � x> | D]6 } | | | f } | | | f } | j | | � qmWd S)zbCheck two namespaces match.
Ignores any unspecified interpreter created names
�__r r Nz __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�attrZactualZexpected� r9 �4/opt/alt/python35/lib64/python3.5/test/test_runpy.py�assertNamespaceMatchesM s2
z)CodeExecutionMixin.assertNamespaceMatchesc
C s t � } | j � } | d } t j d } t j j | | � } | d � } | j | | � | j t j d | � | j t j j | | � | � d | i } | | d <| | � } | j | | � | j | | � | j t j d | � | j t j j | | � | � d S)z�Check that an interface runs the example code correctly
First argument is a callable accepting the initial globals and
using them to create the actual namespace
Second argument is the expected result
r r N�sentinel) �objectr) r �argv�modules�getr; ZassertIsZ assertIsNot)
r4 Zcreate_namespaceZexpected_namespacer< r6 �run_nameZ saved_argv0Z saved_modr5 Z
initial_nsr9 r9 r: �check_code_executionu s
z'CodeExecutionMixin.check_code_executionN)r �
__module__� __qualname__r1 r; rB r9 r9 r9 r: r! A s (r! c @ s. e Z d Z d Z d d � Z d d � Z d S)�ExecutionLayerTestCasez9Unit tests for runpy._run_code and runpy._run_module_codec C s? t j � } | j d d i � d d � } | j | | � d S)Nr c S s t t i | � S)N)r
�example_source)� init_globalsr9 r9 r: � create_ns� s z7ExecutionLayerTestCase.test_run_code.<locals>.create_ns)�example_namespacer) �updaterB )r4 r6 rH r9 r9 r: �
test_run_code� s
z$ExecutionLayerTestCase.test_run_codec s� d � d } d } d } t j j � d | d | �� t j � } | j d � d | d | d
| d � d | d
d d d i � � � f d d � } | j | | � d S)Nz
<Nonsense>zSome other nonsensezNow you're just being silly� r$ r. r r r r r r r Tr c s t t | � � � S)N)r rF )rG )�mod_name�mod_specr9 r: rH � s z>ExecutionLayerTestCase.test_run_module_code.<locals>.create_ns)� importlib� machinery�
ModuleSpecrI r) rJ rB )r4 � mod_fnameZ
mod_loaderZ mod_packager6 rH r9 )rM rN r: �test_run_module_code� s&
z+ExecutionLayerTestCase.test_run_module_codeN)r rC rD r rK rS r9 r9 r9 r: rE � s rE c @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d d
� Z d d d d
d d d �Z d d � Z d d � Z
d d d d
d d d �Z d d d d
d d d �Z d d � Z
d d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d S);�RunModuleTestCasez Unit tests for runpy.run_modulec C s8 y t | � Wn t k
r" Yn X| j d | � d S)NzExpected import error for )r � ImportError�fail)r4 rM r9 r9 r: �expect_import_error� s
z%RunModuleTestCase.expect_import_errorc C sy | j d � | j d � | j d � | j d � | j d � | j d � | j d � | j d � | j d � d S)
Nr z sys.imp.ericz os.path.halfza.beez.howardz..eatenz .test_runpyz .unittestZmultiprocessing)rW )r4 r9 r9 r: �test_invalid_names� s
z$RunModuleTestCase.test_invalid_namesc C s | j t d � d d � d S)Nr r )r/ r )r4 r9 r9 r: �test_library_module� s z%RunModuleTestCase.test_library_moduleFc C s: t j | � | r d St j j | d � } t | � | S)Nz __init__.py)�os�mkdir�path�joinr )r4 �pkg_dir� namespace� pkg_fnamer9 r9 r: � _add_pkg_dir� s
z RunModuleTestCase._add_pkg_dirZ
runpy_testr_ �parent_namespacesc C s� | s | r | r t d � � d } | t j d } t j j t j � � } } t d k rk t d | � t j j
d | � t d k r� t d t j d � | r%| g | }
| |
d <xg |
D]_ } t j j | | � } | j | | � } t d k rt d | � t d k r� t d | � q� Wt j j | | � }
t
|
d
� } | j | � | j � t d k ryt d |
� | d | | } t j j | |
� } | |
| | f S)
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�.���)� RuntimeErrorrZ �extsepr\ �realpath�tempfileZmkdtempr �printr �insertr] ra �open�write�closerO �util�spec_from_file_location)r4 �source�depthZmod_baser_ rb �pkg_nameZ
test_fnamer^ Zsub_dirZnamespace_flagsZnamespace_flagr` rR �mod_filerM rN r9 r9 r: � _make_pkg� s<
zRunModuleTestCase._make_pkgc C s� x0 t t j � D] } | j d � r t j | =q Wt d k rI t d � t j d =t d k ri t d � x
t j | d d �D]� \ } } } xl | D]d } y t j t j j
| | � � Wq� t k
r� } z t d k r� t | � WYd d } ~ Xq� Xq� Wxr | D]j } t j j
| | � } y t j | � Wqt k
rm} z t d k r[t | � WYd d } ~ XqXqWq Wy' t j | � t d k r�t d � Wn; t k
r�} z t d k r�t | � WYd d } ~ Xn Xd S) Nrc r z Removed sys.modules entriesr z Removed sys.path entry�topdownFz Removed package tree)
r* r r? r+ r rl r\ rZ �walk�remover] �OSError�rmdir) r4 �top�entry�root�dirs�filesr" Zex�fullnamer9 r9 r: �_del_pkg� s:
%
!
%
zRunModuleTestCase._del_pkgc C sj d } | d | 7<| d | d <| d } t j j | j | d � } | | d <| rf | d | 7<d S)N�cr r r r )rO rq rr r" )r4 �ns� alter_sysZ char_to_add�specZnew_specr9 r9 r: �_fix_ns_for_legacy_pyc s
z(RunModuleTestCase._fix_ns_for_legacy_pycc
s� | j t | d | d | �\ } } � } t � � t j � } | j d � d | d | j d � j d � d d | i � � r� | j d
| d d d
d i � � � f d d � } z� t d k r� t d � � | j
| | � t j � t
� � t j | � t j sbt | � t � � t j � t d k rBt d � � | j | � � | j
| | � Wd | j | � Xt d k r�t d � d S)Nr_ rb r r r r rf r r r r Tr c s t � | d � �S)Nr� )r )rG )r� rM r9 r: rH 4 s z2RunModuleTestCase._check_module.<locals>.create_nsr zRunning from source:zRunning from compiled:z Module executed successfully)rw rF r rI r) rJ r% �
rpartitionr rl rB rO �invalidate_caches�
__import__rZ rz r �dont_write_bytecoder r r� r� )
r4 rt r� r_ rb r^ rR rN r6 rH r9 )r� rM r: �
_check_module sF
z RunModuleTestCase._check_modulec s� | j t | d d | d | �\ } } } } | j d � d � t | � t j � } | j d | d | d t j j | � d � d
| i � � r� | j d | d d
d d
i � � � f d d � }
z� t
d k r� t d � � | j |
| � t j
� t | � t j | � t j stt | � t | � t
d k rJt d � � t j
� | j | � � | j |
| � Wd | j | � Xt
d k r�t d � d S)N�__main__r_ rb rf r r r r r r r r Tr c s t � | d � �S)Nr� )r )rG )r� ru r9 r: rH ] s z3RunModuleTestCase._check_package.<locals>.create_nsr zRunning from source:zRunning from compiled:z Package executed successfully)rw rF r� r rI r) rJ rO rq �cache_from_sourcer rl rB r� r� rZ rz r r� r r r� r� ) r4 rt r� r_ rb r^ rR rM rN r6 rH r9 )r� ru r: �_check_packageG sH
z RunModuleTestCase._check_packagec C s@ | d k r t d � � d } | } x/ t | � D]! } | } t j j | | � } q1 Wt j j | d � } t | � t d k r� t d | � t j j | d � } | j | � t d k r� t d | � t j j | d � }
| j |
� t d k rt d |
� t j j |
d
� } t | � t d k r<t d | � d S) Nr z$Relative module test needs depth > 1rc z
sibling.pyz Added sibling module:�unclez Added uncle package:Zcousinz Added cousin package:z nephew.pyz Added nephew module:) �
ValueError�rangerZ r\ r] r r rl ra ) r4 Zbase_dirrs rt ru Z
module_dir�iZ
parent_dirZ
sibling_fnameZ uncle_dirZ
cousin_dirZ nephew_fnamer9 r9 r: �_add_relative_modulesp s.
z'RunModuleTestCase._add_relative_modulesNc C s� d } | j | | � \ } } } } | d k r9 | } n | } z^| j | | | � | j d � d } t d k r� t d | � t | d | �}
| j |
d | � | j |
d | � | j d |
� | j d
|
� ~
t j � t
| � t j | � t
j s�t | � t | � t d k r5t d | � t j � t | d | �} | j | d | � | j | d | � | j d | � | j d
| � ~ Wd | j | � Xt d k r�t d � d S)
Nza\
from __future__ import absolute_import
from . import sibling
from ..uncle.cousin import nephew
rf r r zRunning from source:rA r r ZsiblingZnephewzRunning from compiled:z Module executed successfully)rw r� r� r rl r r/ �assertInrO r� r� rZ rz r r� r r r� ) r4 rt rA �contentsr^ rR rM rN �
expected_nameru Zd1Zd2r9 r9 r: �_check_relative_imports� sD
z)RunModuleTestCase._check_relative_importsc C sA x: t d � D], } t d k r, t d | � | j | � q
Wd S)N� r zTesting package depth:)r� r rl r� )r4 rt r9 r9 r: �test_run_module� s
z!RunModuleTestCase.test_run_modulec C sP xI t d d � D]8 } t d k r/ t d | � | j | d d d d �q Wd S)Nr r� zTesting package depth:r_ Trb )r� r rl r� )r4 rt r9 r9 r: �$test_run_module_in_namespace_package� s
z6RunModuleTestCase.test_run_module_in_namespace_packagec C sD x= t d d � D], } t d k r/ t d | � | j | � q Wd S)Nr r� zTesting package depth:)r� r rl r� )r4 rt r9 r9 r: �test_run_package� s
z"RunModuleTestCase.test_run_packagec - C s� | j d d d � } | \ } } } } | j d d � } | j | j | � t j j | d d � } t t t t
f } x1| D])} | j } | j | � �d j
| � } t | d d d
�� }
|
j | � Wd QRXy t | � Wn; | k
r"} z | j d t
| � � WYd d } ~ Xn X| j d j
| � � y t | d
� Wn; | k
r�} z | j d t
| � � WYd d } ~ Xn X| j d j
| � � Wd QRXq} Wd S)NrL r r� z .__main__rc z __init__.pyz raise {0}('{0} in __init__.py.')Zwt�encoding�asciiz finding speczNothing raised; expected {}z
.submodule)rw �replace�
addCleanupr� rZ r\ r] rU �AttributeError� TypeErrorr� r ZsubTest�formatrn ro r Z assertNotInrV ) r4 r r^ �_rM Zinit�
exceptionsZ exceptionr" rs rv �errr9 r9 r: � test_run_package_init_exceptions� s,
))z2RunModuleTestCase.test_run_package_init_exceptionsc C se | j d d � \ } } } } z2 t | � | j t d � � t | � Wd QRXWd | j | � Xd S)NrL r zfound in sys\.modules)rw r� �assertWarnsRegex�RuntimeWarningr r� )r4 r^ r� rM r9 r9 r: � test_submodule_imported_warning� s
z1RunModuleTestCase.test_submodule_imported_warningc C s� | j d d d � \ } } } } | j | j | � | j d d � } t | � | j | t j � t j � � t j
d t � t | � Wd QRXt | � | j
t d � � t | � Wd QRXd S)NrL r r� z .__main__�errorzfound in sys\.modules)rw r� r� r� r� r� r r? �warnings�catch_warnings� simplefilterr� r r� )r4 r^ r� rM �packager9 r9 r: � test_package_imported_no_warning� s !
z2RunModuleTestCase.test_package_imported_no_warningc C sJ xC t d d � D]2 } t d k r/ t d | � | j | d d �q Wd S)Nr r� zTesting package depth:rb T)r� r rl r� )r4 rt r9 r9 r: �%test_run_package_in_namespace_package� s
z7RunModuleTestCase.test_run_package_in_namespace_packagec C sJ xC t d d � D]2 } t d k r/ t d | � | j | d d �q Wd S)Nr r� zTesting package depth:r_ T)r� r rl r� )r4 rt r9 r9 r: �test_run_namespace_package� s
z,RunModuleTestCase.test_run_namespace_packagec C sP xI t d d � D]8 } t d k r/ t d | � | j | d d d d �q Wd S)Nr r� zTesting package depth:r_ Trb )r� r rl r� )r4 rt r9 r9 r: �/test_run_namespace_package_in_namespace_package� s
zARunModuleTestCase.test_run_namespace_package_in_namespace_packagec C sG x@ t d � D]2 } t d k r, t d | � | j | d d �q
Wd S)Nr� r zTesting package depth:r� T)r� r rl r� )r4 rt r9 r9 r: �test_run_module_alter_sys s
z+RunModuleTestCase.test_run_module_alter_sysc C sJ xC t d d � D]2 } t d k r/ t d | � | j | d d �q Wd S)Nr r� zTesting package depth:r� T)r� r rl r� )r4 rt r9 r9 r: �test_run_package_alter_sys s
z,RunModuleTestCase.test_run_package_alter_sysc C sD x= t d d � D], } t d k r/ t d | � | j | � q Wd S)N� � r z"Testing relative imports at depth:)r� r rl r� )r4 rt r9 r9 r: � test_explicit_relative_import s
z/RunModuleTestCase.test_explicit_relative_importc C sG x@ t d d � D]/ } t d k r/ t d | � | j | d � q Wd S)Nr� r� r z'Testing main relative imports at depth:r� )r� r rl r� )r4 rt r9 r9 r: �test_main_relative_import s
z+RunModuleTestCase.test_main_relative_importc s� d } d � | j t | � \ } } � } t � � t j � } | j d � d | d t j j | � d � j d � d d | i � � � f d
d � } z | j
| | � Wd | j | � Xd S) Nr z*And now for something completely differentr r r r rf r r c s t � | � � S)N)r )rG )rM rA r9 r: rH $ s z2RunModuleTestCase.test_run_name.<locals>.create_ns) rw rF r rI r) rJ rO rq r� r� rB r� )r4 rt r^ rR rN r6 rH r9 )rM rA r: �
test_run_name s
z RunModuleTestCase.test_run_namec C s� d d l } d } d } d d g } d | d g } t � } t � } x� t d | � D]u } d j | g | � } | j | � x# | D] }
| j | d |
� q� Wx# | D] }
| j | d |
� q� WqV Wd j | g | � } | j | � | j | d
� | j d | � \ } } }
} | j | j | � x. t d | d � D] } | j | d | � qHWx\ | j | g � D]H \ } }
} | j
| t j j
� | r�| j |
� qx| j |
� qxW| j t | � d | � | j t | � d | � d S)
Nr r� rc r� z uncle.cousinzuncle.cousin.nephewz.siblingr rf z .runpy_testrL r� )�pkgutilr3 r� r] �addrw r� r� r� Z
walk_packagesZassertIsInstancerO rP �
FileFinderrz r/ �len)r4 r� Z max_depthZ base_nameZpackage_suffixesZmodule_suffixesZexpected_packagesZexpected_modulesrt ru r" r^ rR rM rN �finderZispkgr9 r9 r: �test_pkgutil_walk_packages+ s:
"
z,RunModuleTestCase.test_pkgutil_walk_packages) r rC rD r rW rX rY ra rw r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r9 r9 r9 r: rT � s< '( (
rT c @ s� e Z d Z d Z d d d d � Z d d d d d � Z d d
� Z d d � Z d
d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z e d d � � Z d d � Z d S)!�RunPathTestCasez Unit tests for runpy.run_pathNFc C s% | d k r t } t | | | | � S)N)rF r
)r4 �
script_dirZscript_basenamers � omit_suffixr9 r9 r: �_make_test_scriptQ s z!RunPathTestCase._make_test_scriptTc
sV � f d d � } t j � } | d k r3 | }
n | }
| ro t j j |
| � } | j } | sy d | _ n
d } } | j d | d | d | d d d | d | d
d d d i � | j | | � d
� � � f d d � } | r!| d k r!t j j � | � } | sd | _ | | d <� | d <� j d � d | d <| j | | � d S)Nc s
t � | � S)N)r )rG )� script_namer9 r: rH \ s z0RunPathTestCase._check_script.<locals>.create_nsr r r r rL r r r Tr zprove.issue15230.is.fixedc s t � | � � S)N)r )rG )rA r� r9 r: rH y s rf r )
rI r) rO rq rr r% r. rJ rB r� )
r4 r� r� Z
expected_fileZexpected_argv0rM � expect_spec� check_loaderrH r6 Z spec_namerN Z
mod_cachedr9 )rA r� r: �
_check_scriptX sB
z RunPathTestCase._check_scriptc C s) t j | � } | j t | t | � d S)N)�re�escape�assertRaisesRegexrU r )r4 r� �msgr9 r9 r: �_check_import_error� s z#RunPathTestCase._check_import_errorc C sK t � �; } d } | j | | � } | j | d | | d d �Wd QRXd S)N�scriptz
<run_path>r� F)r r� r� )r4 r� rM r� r9 r9 r: �test_basic_script� s
z!RunPathTestCase.test_basic_scriptc C sQ t � �A } d } | j | | d d �} | j | d | | d d �Wd QRXd S)Nr� r� Tz
<run_path>r� F)r r� r� )r4 r� rM r� r9 r9 r: �test_basic_script_no_suffix� s z+RunPathTestCase.test_basic_script_no_suffixc C sm t � �] } d } | j | | � } t j | d d �} t j | � | j | d | | d d �Wd QRXd S)Nr� �doraiseTz
<run_path>r� F)r r� �
py_compile�compilerZ rz r� )r4 r� rM r� �
compiled_namer9 r9 r: �test_script_compiled� s
z$RunPathTestCase.test_script_compiledc C sK t � �; } d } | j | | � } | j | d | | d | �Wd QRXd S)Nr� z
<run_path>rM )r r� r� )r4 r� rM r� r9 r9 r: �test_directory� s
z RunPathTestCase.test_directoryc C s� t � �r } d } | j | | � } t j | d d �} t j | � t j sw t | � } | j | d | | d | �Wd QRXd S)Nr� r� Tz
<run_path>rM )
r r� r� r� rZ rz r r� r r� )r4 r� rM r� r� Z
legacy_pycr9 r9 r: �test_directory_compiled� s
z'RunPathTestCase.test_directory_compiledc
C sI t � �9 } d } | j | | � } d | } | j | | � Wd QRXd S)N�not_mainz"can't find '__main__' module in %r)r r� r� )r4 r� rM r� r� r9 r9 r: �test_directory_error� s
z$RunPathTestCase.test_directory_errorc C si t � �Y } d } | j | | � } t | d | � \ } } | j | d | | d | d d �Wd QRXd S)Nr� �test_zipz
<run_path>rM r� F)r r� r r� )r4 r� rM r� �zip_name�fnamer9 r9 r: � test_zipfile� s z RunPathTestCase.test_zipfilec C s~ t � �n } d } | j | | � } t j | d d �} t | d | � \ } } | j | d | | d | d d �Wd QRXd S) Nr� r� Tr� z
<run_path>rM r� F)r r� r� r� r r� )r4 r� rM r� r� r� r� r9 r9 r: �test_zipfile_compiled� s z%RunPathTestCase.test_zipfile_compiledc C sa t � �Q } d } | j | | � } t | d | � \ } } d | } | j | | � Wd QRXd S)Nr� r� z"can't find '__main__' module in %r)r r� r r� )r4 r� rM r� r� r� r� r9 r9 r: �test_zipfile_error� s
z"RunPathTestCase.test_zipfile_errorc C s� t � �s } t � �` } d } d | } | j | | | � } t | d | � \ } } d } | j t | t | � Wd QRXWd QRXd S)Nr� z import runpy
runpy.run_path(%r)
r� zrecursion depth exceeded)r r� r r� �RecursionErrorr ) r4 r� Z dummy_dirrM rs r� r� r� r� r9 r9 r: �test_main_recursion_error� s z)RunPathTestCase.test_main_recursion_errorc C sx t � �h } t j j | d � } t | d d d �� } | j d � Wd QRXt | � } | j | d d � Wd QRXd S)Nz script.pyre r� �latin1u%
#coding:latin1
s = "non-ASCII: hé"
�su non-ASCII: hé)r rZ r\ r] rn ro r r/ )r4 r� �filename�fr r9 r9 r: �
test_encoding� s z RunPathTestCase.test_encoding)r rC rD r r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r9 r9 r9 r: r� N s +
r� r� )*ZunittestrZ Zos.pathr r� rk rO �importlib.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 rF Zimplicit_namespacer> �dictrI rJ r! ZTestCaserE rT r� r �mainr9 r9 r9 r: �<module> sF $ 4" "
O ' � � �