Your IP : 216.73.216.85
���]�{ � @ 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)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: �//opt/alt/python35/lib64/python3.5/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)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 Z d Z d d � Z d d � Z d S)�ExecutionLayerTestCasec 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_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 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 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 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. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z d8 d9 � Z d S):�RunModuleTestCasec C s8 y t | � Wn t k
r" Yn X| j 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 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 .unittest�multiprocessing)rX )r4 r: r: r; �test_invalid_names� s
z$RunModuleTestCase.test_invalid_namesc C s | j t d � d d � d S)Nr r )r/ r )r4 r: r: 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_fnamer: r: r; � _add_pkg_dir� s
z RunModuleTestCase._add_pkg_dirZ
runpy_testra �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�.���)� RuntimeErrorr\ �extsepr^ �realpath�tempfile�mkdtempr �printr �insertr_ rc �open�write�closerP �util�spec_from_file_location)r4 �source�depthZmod_basera rd �pkg_nameZ
test_fnamer` Zsub_dirZnamespace_flagsZnamespace_flagrb rS �mod_filerN rO r: r: 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) Nre r z Removed sys.modules entriesr z Removed sys.path entry�topdownFz Removed package tree)
r* r r@ r+ r ro r^ r\ �walk�remover_ �OSError�rmdir) r4 �top�entry�root�dirs�filesr" �ex�fullnamer: r: 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 )rP rt ru 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 | 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)Nra rd r r r r rh r r r r Tr c s t � | d � �S)Nr� )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)rz rG r rJ r) rK r% �
rpartitionr ro rC rP �invalidate_caches�
__import__r\ r} r �dont_write_bytecoder r r� r� )
r4 rw r� ra rd r` rS rO r6 rI r: )r� rN 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__ra rd rh r r r r r r r r Tr c s t � | d � �S)Nr� )r )rH )r� rx r: r; rI ] s z3RunModuleTestCase._check_package.<locals>.create_nsr zRunning from source:zRunning from compiled:z Package executed successfully)rz rG r� r rJ r) rK rP rt �cache_from_sourcer ro rC r� r� r\ r} r r� r r r� r� ) r4 rw r� ra rd r` rS rN rO r6 rI r: )r� rx 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 > 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 ro rc ) r4 �base_dirrv rw rx 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 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
rh r r zRunning from source:rB r r ZsiblingZnephewzRunning from compiled:z Module executed successfully)rz r� r� r ro r r/ �assertInrP r� r� r\ r} r r� r r r� ) r4 rw rB �contentsr` rS rN rO �
expected_namerx Zd1Zd2r: r: 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 ro r� )r4 rw r: r: 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:ra Trd )r� r ro r� )r4 rw r: r: 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 ro r� )r4 rw r: r: 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)NrM r r� z .__main__re z __init__.pyz raise {0}('{0} in __init__.py.')Zwt�encoding�asciiz finding speczNothing raised; expected {}z
.submodule)rz �replace�
addCleanupr� r\ r^ r_ rV �AttributeError� TypeErrorr� r ZsubTest�formatrq rr r Z assertNotInrW ) r4 r r` �_rN Zinit�
exceptions� exceptionr" rv ry �errr: r: 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)NrM r zfound in sys\.modules)rz r� �assertWarnsRegex�RuntimeWarningr r� )r4 r` r� rN r: r: 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)NrM r r� z .__main__�errorzfound in sys\.modules)rz 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 sJ xC t d d � D]2 } t d k r/ t d | � | j | d d �q Wd S)Nr r� zTesting package depth:rd T)r� r ro r� )r4 rw r: r: 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:ra T)r� r ro r� )r4 rw r: r: 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:ra Trd )r� r ro r� )r4 rw r: r: 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 ro r� )r4 rw r: r: 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 ro r� )r4 rw r: r: 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 ro r� )r4 rw r: r: 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 ro r� )r4 rw r: r: 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 rh r r c s t � | � � S)N)r )rH )rN rB r: r; rI $ s z2RunModuleTestCase.test_run_name.<locals>.create_ns) rz rG r rJ r) rK rP rt r� r� rC r� )r4 rw r` rS rO r6 rI r: )rN rB 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� re r� z uncle.cousinzuncle.cousin.nephewz.siblingr rh z .runpy_testrM r� )�pkgutilr3 r� r_ �addrz r� r� r� Z
walk_packagesZassertIsInstancerP rQ �
FileFinderr} r/ �len)r4 r� Z max_depth� base_nameZpackage_suffixesZmodule_suffixesZexpected_packagesZexpected_modulesrw rx r" r` rS rN rO �finderZispkgr: r: r; �test_pkgutil_walk_packages+ s:
"
z,RunModuleTestCase.test_pkgutil_walk_packages) r rD rE rX rZ r[ rc rz 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 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) �RunPathTestCaseNFc C s% | d k r t } t | | | | � S)N)rG r
)r4 �
script_dirZscript_basenamerv � omit_suffixr: r: 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 )rH )� script_namer: r; rI \ s z0RunPathTestCase._check_script.<locals>.create_nsr r r r rM r r r Tr zprove.issue15230.is.fixedc s t � | � � S)N)r )rH )rB r� r: r; rI y s rh r )
rJ r) rP rt ru 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_scriptX sB
z RunPathTestCase._check_scriptc C s) t j | � } | j t | t | � d S)N)�re�escape�assertRaisesRegexrV r )r4 r� �msgr: r: 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� rN r� r: r: 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� rN r� r: r: 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�compiler\ r} r� )r4 r� rN r� �
compiled_namer: r: 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>rN )r r� r� )r4 r� rN r� r: r: 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>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 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� rN r� r� r: r: 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>rN r� F)r r� r r� )r4 r� rN r� �zip_name�fnamer: r: 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>rN r� F)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 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� rN r� r� r� r� r: r: 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_dirrN rv r� r� r� r� r: r: 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.pyrg r� �latin1u%
#coding:latin1
s = "non-ASCII: hé"
�su non-ASCII: hé)r r\ r^ r_ rq rr r r/ )r4 r� �filename�fr r: r: r; �
test_encoding� s z RunPathTestCase.test_encoding)r rD rE r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r: r: r: r; r� N 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> sF $ 4" "
O ' � � �