Your IP : 216.73.216.213
�
f f�r � @ 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
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 m Z d d l Z d d l m Z m Z m Z m Z d Z i d d 6d d 6d d 6d d 6d d
6d d 6Z i e d 6e d
6d d g d 6e j d d 6d d 6d d 6e e d d d d d d �d 6Z! 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 r!e j) � n d S)#� N)�forget�make_legacy_pyc� run_unittest�unload�verbose�
no_tracing�create_empty_file)�make_pkg� make_script� make_zip_pkg�make_zip_script�temp_dir)� _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�runpyzTop level assignmentzLower level reference�result� run_argv0F�run_name_in_sys_modules�module_in_sys_modules�x� z<run>�
__loader__�nestedc @ 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 � } xu t | � D]g } | j d � r% | j d � r% | | k re | j | � n | | d k r� | d j | � q� q% q% W| j d � } | j d � } | d k r� | j | � n� | j d k r� | j t | j � t | j � � n xU | j D]J } d | } | t
| | � f } | t
| | � f } | j | | � qW| j t | � t | � � x> | D]6 } | | | f } | | | f } | j | | � qyWd 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� r: �4/opt/alt/python34/lib64/python3.4/test/test_runpy.py�assertNamespaceMatchesL 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 | | � | � i | d 6} | | 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)
r5 Zcreate_namespaceZexpected_namespacer= r7 �run_nameZ saved_argv0Z saved_modr6 Z
initial_nsr: r: r; �check_code_executiont s
z'CodeExecutionMixin.check_code_executionN)r �
__module__� __qualname__r2 r< rC r: r: r: r; r" @ 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 i d d 6� 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 )r5 r7 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 i � d 6| d 6| d 6| d
6� d 6| d 6d
d 6d
d 6� � � f d d � } | j | | � d S)Nz
<Nonsense>zSome other nonsensezNow you're just being silly� r% r/ r r r r r r Tr 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 )r5 � mod_fnameZ
mod_loaderZ mod_packager7 rI r: )rN rO r; �test_run_module_code� s& z+ExecutionLayerTestCase.test_run_module_codeN)r rD rE r rL rT r: r: r: r; rF � s rF 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 d S)5�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 � ImportErrorZfail)r5 rN r: r: r; �expect_import_error� s
z%RunModuleTestCase.expect_import_errorc C s_ | 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..eatenZmultiprocessing)rW )r5 r: r: r; �test_invalid_names� s
z$RunModuleTestCase.test_invalid_namesc C s | j t d � d d � d S)Nr r )r0 r )r5 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 )r5 �pkg_dir� namespace� pkg_fnamer: r: r; � _add_pkg_dir� s
z RunModuleTestCase._add_pkg_dirZ
runpy_testr_ �parent_namespacesc C s� | s | r" | r" t d � � n d } | t j d } t j j t j � � } } t d k rq t d | � n t j j
d | � t d k r� t d t j d � n | r7| g | }
| |
d <xp |
D]e } t j j | | � } | j | | � } t d k rt d | � n t d k r� t d | � q� q� Wn t j j | | � }
t
|
d
� } | j | � | j � t d k r�t d |
� n | 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�closerP �util�spec_from_file_location)r5 �source�depthZmod_baser_ rb �pkg_nameZ
test_fnamer^ Zsub_dirZnamespace_flagsZnamespace_flagr` rS Zmod_filerN rO r: r: r; � _make_pkg� s<
zRunModuleTestCase._make_pkgc C s� x3 t t j � D]" } | j d � r t j | =q q Wt d k rO t d � n t j d =t d k rr t d � n xt j | d d �D]� \ } } } xo | D]g } y t j t j j
| | � � Wq� t k
r} z t d k r� t | � n WYd d } ~ Xq� Xq� Wxu | D]m } t j j
| | � }
y t j |
� Wqt k
r|} z t d k rjt | � n WYd d } ~ XqXqWq� Wy* t j | � t d k r�t d � n Wn> t k
r�} z t d k r�t | � n 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) r5 �toprt rN �entry�root�dirs�filesr# Zex�fullnamer: r: r; �_del_pkg� s:
%
$
(
zRunModuleTestCase._del_pkgc C sy t r d n d } | d | 7<| d | d <| d } t j j | j | d � } | | d <| ru | d | 7<n d S)N�c�or r r r )� __debug__rP rq rr r# )r5 �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 i � d 6| d 6| j d 6� j d � d d 6| d 6� � r� | j i | d
6d d 6d d
6� n � � f d d � } z� t d k r� t d � � n | j
| | � t j � t
� � t j | � t j svt | � t � � t j � t d k rSt d � � n | j | � � | j
| | � n Wd | j | | � � Xt d k r�t d � n d S)Nr_ rb r r r rf r r r r Tr r c s t � | d � �S)Nr� )r )rH )r� rN r: r; rI 0 s z2RunModuleTestCase._check_module.<locals>.create_nsr zRunning from source:zRunning from compiled:z Module executed successfully)rv rG r rJ r* rK r& �
rpartitionr rl rC rP �invalidate_caches�
__import__rZ ry r �dont_write_bytecoder r r� r� )
r5 rt r� r_ rb r^ rS rO r7 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 i | d 6| d 6t j j | � d 6� d 6| d
6� � r� | j i | d 6d d
6d d 6� n � � f d d � }
z� t
d k r� t d � � n | j |
| � t j
� t | � t j | � t j s�t | � t | � t
d k r[t d � � n t j
� | j | � � | j |
| � n Wd | j | | � � Xt
d k r�t d � n d S)N�__main__r_ rb rf r r r r r r r Tr r c s t � | d � �S)Nr� )r )rH )r� ru r: r; rI Y s z3RunModuleTestCase._check_package.<locals>.create_nsr zRunning from source:zRunning from compiled:z Package executed successfully)rv rG r� r rJ r* rK rP rq �cache_from_sourcer rl rC r� r� rZ ry r r� r r r� r� ) r5 rt r� r_ rb r^ rS rN rO r7 rI r: )r� ru r; �_check_packageC sH
z RunModuleTestCase._check_packagec C sO | d k r t d � � n d } | } x/ t | � D]! } | } t j j | | � } q4 Wt j j | d � } t | � t d k r� t d | � n t j j | d � } | j | � t d k r� t d | � n t j j | d � }
| j |
� t d k rt d |
� n t j j |
d
� } t | � t d k rKt d | � n 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 ) r5 Zbase_dirrs rt ru Z
module_dir�iZ
parent_dirZ
sibling_fnameZ uncle_dirZ
cousin_dirZ nephew_fnamer: r: r; �_add_relative_modulesl s.
z'RunModuleTestCase._add_relative_modulesNc
C s� d } | j | | � \ } } } } | d k r9 | } n | } zg| j | | | � | j d � d } t d k r� t d | � n t | d | �}
| j |
d | � | j |
d | � | j d |
� | j d
|
� ~
t j � t
| � t j | � t
j s�t | � t | � t d k r;t d | � n t j � t | d | �} | j | d | � | j | d | � | j d | � | j d
| � ~ n Wd | j | | | � Xt d k r�t d � n d S)
Nza\
from __future__ import absolute_import
from . import sibling
from ..uncle.cousin import nephew
rf r r zRunning from source:rB r r ZsiblingZnephewzRunning from compiled:z Module executed successfully)rv r� r� r rl r r0 ZassertInrP r� r� rZ ry r r� r r r� ) r5 rt rB �contentsr^ rS rN rO �
expected_nameru Zd1Zd2r: r: r; �_check_relative_imports� sD
z)RunModuleTestCase._check_relative_importsc C sD x= t d � D]/ } t d k r/ t d | � n | j | � q
Wd S)N� r zTesting package depth:)r� r rl r� )r5 rt r: r: r; �test_run_module� s z!RunModuleTestCase.test_run_modulec C sS xL t d d � D]; } t d k r2 t d | � n | j | d d d d �q Wd S)Nr r� zTesting package depth:r_ Trb )r� r rl r� )r5 rt r: r: r; �$test_run_module_in_namespace_package� s z6RunModuleTestCase.test_run_module_in_namespace_packagec C sG x@ t d d � D]/ } t d k r2 t d | � n | j | � q Wd S)Nr r� zTesting package depth:)r� r rl r� )r5 rt r: r: r; �test_run_package� s z"RunModuleTestCase.test_run_packagec C sM xF t d d � D]5 } t d k r2 t d | � n | j | d d �q Wd S)Nr r� zTesting package depth:rb T)r� r rl r� )r5 rt r: r: r; �%test_run_package_in_namespace_package� s z7RunModuleTestCase.test_run_package_in_namespace_packagec C sM xF t d d � D]5 } t d k r2 t d | � n | j | d d �q Wd S)Nr r� zTesting package depth:r_ T)r� r rl r� )r5 rt r: r: r; �test_run_namespace_package� s z,RunModuleTestCase.test_run_namespace_packagec C sS xL t d d � D]; } t d k r2 t d | � n | j | d d d d �q Wd S)Nr r� zTesting package depth:r_ Trb )r� r rl r� )r5 rt r: r: r; �/test_run_namespace_package_in_namespace_package� s zARunModuleTestCase.test_run_namespace_package_in_namespace_packagec C sJ xC t d � D]5 } t d k r/ t d | � n | j | d d �q
Wd S)Nr� r zTesting package depth:r� T)r� r rl r� )r5 rt r: r: r; �test_run_module_alter_sys� s z+RunModuleTestCase.test_run_module_alter_sysc C sM xF t d d � D]5 } t d k r2 t d | � n | j | d d �q Wd S)Nr r� zTesting package depth:r� T)r� r rl r� )r5 rt r: r: r; �test_run_package_alter_sys� s z,RunModuleTestCase.test_run_package_alter_sysc C sG x@ t d d � D]/ } t d k r2 t d | � n | j | � q Wd S)N� � r z"Testing relative imports at depth:)r� r rl r� )r5 rt r: r: r; � test_explicit_relative_import� s z/RunModuleTestCase.test_explicit_relative_importc C sJ xC t d d � D]2 } t d k r2 t d | � n | j | d � q Wd S)Nr� r� r z'Testing main relative imports at depth:r� )r� r rl r� )r5 rt r: r: r; �test_main_relative_import� s z+RunModuleTestCase.test_main_relative_importc
s� d } d � | j t | � \ } } � } t � � t j � } | j i � d 6| d 6t j j | � d 6� j d � d d 6| d 6� � � f d
d � } z | j
| | � Wd | j | | � � Xd S) Nr z*And now for something completely differentr r r rf r r r c s t � | � � S)N)r )rH )rN rB r: r; rI � s z2RunModuleTestCase.test_run_name.<locals>.create_ns) rv rG r rJ r* rK rP rq r� r� rC r� )r5 rt r^ rS rO r7 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 | � qNWx\ | j | g � D]H \ } }
} | j
| t j j
� | r�| j |
� q~| j |
� q~W| j t | � d | � | j t | � d | � d S)
Nr r� rc r� z uncle.cousinzuncle.cousin.nephewz.siblingr rf z .runpy_testrM r� )�pkgutilr4 r� r] �addrv Z
addCleanupr� r� Z
walk_packagesZassertIsInstancerP rQ �
FileFinderry r0 �len)r5 r� Z max_depthZ base_nameZpackage_suffixesZmodule_suffixesZexpected_packagesZexpected_modulesrt ru r# r^ rS rN rO �finderZispkgr: r: r; �test_pkgutil_walk_packages� s:
"
z,RunModuleTestCase.test_pkgutil_walk_packages)r rD rE r rW rX rY ra rv r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r: r: r: r; rU � s6 '( ( rU 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 } n t | | | | � S)N)rG r
)r5 �
script_dirZscript_basenamers � omit_suffixr: r: r; �_make_test_script s z!RunPathTestCase._make_test_scriptTc
sg � f d d � } t j � } | d k r3 | }
n | }
| rr t j j |
| � } | j } | s| d | _ q| n
d } } | j i | d 6| d 6| d 6d d 6| d 6| d 6d
d 6d
d 6� | j | | � d
� � � f d d � } | r2| d k r2t j j � | � } | s%d | _ n | | d <n � | 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 rM r r r Tr r zprove.issue15230.is.fixedc s t � | � � S)N)r )rH )rB r� r: r; rI @ s rf r )
rJ r* rP rq rr r& r/ rK rC r� )
r5 r� r� Z
expected_fileZexpected_argv0rN � expect_spec� check_loaderrI r7 Z spec_namerO Z
mod_cachedr: )rB r� r; �
_check_script sB
z RunPathTestCase._check_scriptc C s) t j | � } | j t | t | � d S)N)�re�escape�assertRaisesRegexrV r )r5 r� �msgr: r: r; �_check_import_errorL s z#RunPathTestCase._check_import_errorc C sJ t � �; } d } | j | | � } | j | d | | d d �Wd QXd S)N�scriptz
<run_path>r� F)r
r� r� )r5 r� rN r� r: r: r; �test_basic_scriptP s
z!RunPathTestCase.test_basic_scriptc C sP t � �A } d } | j | | d d �} | j | d | | d d �Wd QXd S)Nr� r� Tz
<run_path>r� F)r
r� r� )r5 r� rN r� r: r: r; �test_basic_script_no_suffixW s z+RunPathTestCase.test_basic_script_no_suffixc C sl t � �] } d } | j | | � } t j | d d �} t j | � | j | d | | d d �Wd QXd S)Nr� �doraiseTz
<run_path>r� F)r
r� �
py_compile�compilerZ ry r� )r5 r� rN r� �
compiled_namer: r: r; �test_script_compiled_ s
z$RunPathTestCase.test_script_compiledc C sJ t � �; } d } | j | | � } | j | d | | d | �Wd QXd S)Nr� z
<run_path>rN )r
r� r� )r5 r� rN r� r: r: r; �test_directoryh s
z RunPathTestCase.test_directoryc C s� t � �u } d } | j | | � } t j | d d �} t j | � t j sz t | � } | j | d | | d | �n Wd QXd S)Nr� r� Tz
<run_path>rN )
r
r� r� r� rZ ry r r� r r� )r5 r� rN r� r� Z
legacy_pycr: r: r; �test_directory_compiledo s
z'RunPathTestCase.test_directory_compiledc
C sH t � �9 } d } | j | | � } d | } | j | | � Wd QXd S)N�not_mainz"can't find '__main__' module in %r)r
r� r� )r5 r� rN r� r� r: r: r; �test_directory_errorz s
z$RunPathTestCase.test_directory_errorc C sh t � �Y } d } | j | | � } t | d | � \ } } | j | d | | d | d d �Wd QXd S)Nr� �test_zipz
<run_path>rN r� F)r
r� r r� )r5 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 QXd S) Nr� r� Tr� z
<run_path>rN r� F)r
r� r� r� r r� )r5 r� rN r� r� r� r� r: r: r; �test_zipfile_compiled� s z%RunPathTestCase.test_zipfile_compiledc C s` t � �Q } d } | j | | � } t | d | � \ } } d | } | j | | � Wd QXd S)Nr� r� z"can't find '__main__' module in %r)r
r� r r� )r5 r� rN r� r� r� r� r: r: r; �test_zipfile_error� s
z"RunPathTestCase.test_zipfile_errorc C s� t � �r } t � �` } d } d | } | j | | | � } t | d | � \ } } d } | j t | t | � Wd QXWd QXd S)Nr� z import runpy
runpy.run_path(%r)
r� zrecursion depth exceeded)r
r� r r� rh r ) r5 r� Z dummy_dirrN rs r� r� r� r� r: r: r; �test_main_recursion_error� s z)RunPathTestCase.test_main_recursion_errorc C sv t � �g } t j j | d � } t | d d d �� } | j d � Wd QXt | � } | j | d d � Wd QXd S)Nz script.pyre �encoding�latin1u%
#coding:latin1
s = "non-ASCII: hé"
�su non-ASCII: hé)r
rZ r\ r] rn ro r r0 )r5 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; r� s +
r� r� )*ZunittestrZ Zos.pathr r� rk rP �importlib.machinery�importlib.utilr� Z test.supportr r r r r r r Ztest.script_helperr r
r r r
r r r r r rG Zimplicit_namespacer? �dictrJ rK r" ZTestCaserF rU r� r �mainr: r: r: r; �<module> sH $ 4( "
O ' � ` �