a zel @sddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlmZmZmZmZmZmZmZddlmZmZddlZddlmZmZmZmZdZ dddddddZ!e eddge j"dd d e#e!d d dd d Z$e$%e!GdddZ&Gdddej'e&Z(Gdddej'e&Z)Gdddej'e&Z*Gdddej'Z+e,dkr|e-dS)N)forgetmake_legacy_pycunloadverbose no_tracingcreate_empty_filetemp_dir) make_scriptmake_zip_script) _run_code_run_module_code run_modulerun_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='') )__name____file__ __cached__ __package____doc____spec__zTop level assignmentzLower level referenceFz)xr __loader__)sysrunpyresult run_argv0run_name_in_sys_modulesmodule_in_sys_modulesnestedc@s$eZdZgdZddZddZdS)CodeExecutionMixin)nameparentorigincached has_locationsubmodule_search_locationsc Cs*|}|}t|D]D}|dr|dr||vrB||||dvr|d|q|d}|d}|dur||n^|jdur|t|jt|j|j D]4}d|}|t ||f}|t ||f}|||q|t |t ||D](}|||f}|||f}|||qdS)zbCheck two namespaces match. Ignores any unspecified interpreter created names __rrNz __spec__.) copylist startswithendswithpopZ assertIsNoneloader assertEqualtypeCHECKED_SPEC_ATTRIBUTESgetattrset) self result_ns expected_nskZ result_specZ expected_specattractualZexpectedr8C/opt/bitninja-python-dojo/embedded/lib/python3.9/test/test_runpy.pyassertNamespaceMatchesPs4           z)CodeExecutionMixin.assertNamespaceMatchesc Cst}|}|d}tjd}tj||}|d}||||tjd||tj|||d|i} ||d<|| }||| ||||tjd||tj|||dS)zCheck 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 rrNsentinel) objectr'rargvmodulesgetr:ZassertIsZ assertIsNot) r2Zcreate_namespaceZexpected_namespacer;r4run_nameZ saved_argv0Z saved_modr3Z initial_nsr8r8r9check_code_executionxs     z'CodeExecutionMixin.check_code_executionN)r __module__ __qualname__r/r:rAr8r8r8r9rDs (rc@s eZdZdZddZddZdS)ExecutionLayerTestCasez9Unit tests for runpy._run_code and runpy._run_module_codecCs.t}|ddidd}|||dS)NrcSs tti|SN)r example_sourceZ init_globalsr8r8r9 create_nssz7ExecutionLayerTestCase.test_run_code..create_ns)example_namespacer'updaterA)r2r4rHr8r8r9 test_run_codes z$ExecutionLayerTestCase.test_run_codec sddd}d}d}tjj||dt}|||||dddfdd }|||dS) Nz zSome other nonsensezNow you're just being silly)r"r,T)rrrrrrrrcstt|SrE)r rFrGmod_namemod_specr8r9rHs z>ExecutionLayerTestCase.test_run_module_code..create_ns) importlib machinery ModuleSpecrIr'rJrA)r2 mod_fnameZ mod_loaderZ mod_packager4rHr8rMr9test_run_module_codes* z+ExecutionLayerTestCase.test_run_module_codeN)rrBrCrrKrTr8r8r8r9rDs rDc@seZdZdZddZddZddZd:d d Zd;ddd d dZddZ ddZ dddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8d9ZdS)?RunModuleTestCasezUnit tests for runpy.run_modulecCs2z t|WntyYn0|d|dS)NzExpected import error for )r ImportErrorfail)r2rNr8r8r9expect_import_errors   z%RunModuleTestCase.expect_import_errorcCs^|d|d|d|d|d|d|d|d|d dS) Nrz sys.imp.ericz os.path.halfza.beez.howardz..eatenz .test_runpyz .unittestmultiprocessing)rXr2r8r8r9test_invalid_namess        z$RunModuleTestCase.test_invalid_namescCs|tddddS)Nrr)r-r rZr8r8r9test_library_modulesz%RunModuleTestCase.test_library_moduleFcCs,t||rdStj|d}t||S)N __init__.py)osmkdirpathjoinr)r2pkg_dir namespace pkg_fnamer8r8r9 _add_pkg_dirs  zRunModuleTestCase._add_pkg_dir runpy_testrcparent_namespacescCsR|s|r|stdd}|tjd}tjt}} tdkrLtd| t j d|tdkrrtdt jd|r|g|} || d<| D]B} tj | |} | | | } tdkrtd | tdkrtd | qtj | |} t | d }||Wdn1s 0Ytdkr(td | |d ||}tj|| }|| ||fS) Nz2Can't mark top level module as a namespace package __runpy_pkg__pyrz Package tree in:rz Updated sys.path:z Next level in:z Created:w.) RuntimeErrorr^extsepr`realpathtempfilemkdtemprprintrinsertrareopenwriterPutilspec_from_file_location)r2sourcedepthZmod_basercrhpkg_nameZ test_fnamerbZsub_dirZnamespace_flagsZnamespace_flagrdrSmod_filerNrOr8r8r9 _make_pkgs2    *zRunModuleTestCase._make_pkgc CszttjD]}|dr tj|=q tdkr2tdtjd=tdkrJtdtj|ddD]\}}}|D]R}zt tj ||Wqft y}ztdkrt|WYd}~qfd}~00qf|D]Z}tj ||}zt |Wqt y}ztdkrt|WYd}~qd}~00qqXz t |tdkr:tdWn8t yt}ztdkr`t|WYd}~n d}~00dS) Nrirz Removed sys.modules entriesrz Removed sys.path entryF)topdownz Removed package tree) r(rr>r)rrsr`r^walkremoveraOSErrorrmdir) r2topentryrootdirsfilesr exfullnamer8r8r9_del_pkgs.  (, zRunModuleTestCase._del_pkgcCs\d}|d|7<|d|d<|d}tj|j|d}||d<|rX|d|7<dS)Ncrrrr)rPrwrxr )r2ns alter_sysZ char_to_addspecZnew_specr8r8r9_fix_ns_for_legacy_pycs  z(RunModuleTestCase._fix_ns_for_legacy_pycc s"|jt|||d\}}}tt}|||jdd|drb||dddfdd} ztd krt d | | |t t t|tjst|tt td krt d ||| | |W||n ||0td krt d dS) NrgrmrrrrrrTrrrcst|dSNrr rGrrNr8r9rH6sz2RunModuleTestCase._check_module..create_nsrRunning from source:Running from compiled:Module executed successfully)r}rFrrIr'rJr# rpartitionrrsrArPinvalidate_caches __import__r^rrdont_write_bytecoderrrr) r2rzrrcrhrbrSrOr4rHr8rr9 _check_module!sH     zRunModuleTestCase._check_modulec s.|jt|d||d\}}}}|ddt|t} | ||tj ||drn| |dddfdd } zt d krt d | | | t t|t|tjst|t|t d krt d t || | | | W||n ||0t d kr*t d dS)N__main__rgrmrrTrcst|dSrrrGrr{r8r9rH_sz3RunModuleTestCase._check_package..create_nsrrrzPackage executed successfully)r}rFrrrIr'rJrPrwcache_from_sourcerrsrArrr^rrrrrrr) r2rzrrcrhrbrSrNrOr4rHr8rr9_check_packageIsJ      z RunModuleTestCase._check_packagec Cs|dkrtdd}|}t|D]}|}tj||}q tj|d}t|tdkr`td|tj|d} || tdkrtd| tj| d} || tdkrtd | tj| d } t| tdkrtd | dS) Nrz$Relative module test needs depth > 1riz sibling.pyz Added sibling module:unclez Added uncle package:Zcousinz Added cousin package:z nephew.pyz Added nephew module:) ValueErrorranger^r`rarrrsre) r2base_dirryrzr{Z module_diriZ parent_dirZ sibling_fnameZ uncle_dirZ cousin_dirZ nephew_fnamer8r8r9_add_relative_modulesrs&   z'RunModuleTestCase._add_relative_modulesNc Csdd}|||\}}}}|dur&|}n|}z|||||dd} tdkr\td|t||d} || d||| d| |d | |d | ~ t t |t |t js4t|t|tdkrtd |t t||d} || d||| d| |d | |d | ~ W||n ||0tdkr`td dS) Nza\ from __future__ import absolute_import from . import sibling from ..uncle.cousin import nephew rmrrr)r@rrZsiblingZnephewrr)r}rrrrsr r-assertInrPrrr^rrrrrr) r2rzr@contentsrbrSrNrO expected_namer{Zd1Zd2r8r8r9_check_relative_importss@        z)RunModuleTestCase._check_relative_importscCs.tdD] }tdkrtd|||qdS)NrTesting package depth:rrrsrr2rzr8r8r9test_run_modules z!RunModuleTestCase.test_run_modulecCs6tddD]&}tdkr td||j|dddq dSNrrrTrgrrr8r8r9$test_run_module_in_namespace_packagesz6RunModuleTestCase.test_run_module_in_namespace_packagecCs0tddD] }tdkr td|||q dS)Nrrrrrrsrrr8r8r9test_run_packagesz"RunModuleTestCase.test_run_packagec Cs~|ddd}|\}}}}|dd}||j|tj|dd}ttt t f}|D]"}|j }| |d |} t|dd d } | | Wdn1s0Yz t|Wn4|y} z|d t | WYd} ~ nd} ~ 00|d |zt|d Wn6|yF} z|d t | WYd} ~ nd} ~ 00|d |WdqT1sn0YqTdS)NrLrr .__main__rir]z raise {0}('{0} in __init__.py.')Zwtasciiencodingz finding speczNothing raised; expected {}z .submodule)r}replace addCleanuprr^r`rarVAttributeError TypeErrorrrZsubTestformatrurvr Z assertNotInrW) r2rrb_rNinit exceptions exceptionr ryr|errr8r8r9 test_run_package_init_exceptionss,      ( &&z2RunModuleTestCase.test_run_package_init_exceptionsc Csp|dd\}}}}zJt||tdt|Wdn1sH0YW||n ||0dS)NrLrfound in sys\.modules)r}rassertWarnsRegexRuntimeWarningr r)r2rbrrNr8r8r9test_submodule_imported_warnings(z1RunModuleTestCase.test_submodule_imported_warningcCs|ddd\}}}}||j||dd}t|||tjt $t dt t |Wdn1sx0Yt|| t dt |Wdn1s0YdS)NrLrrrerrorr)r}rrrrrrr>warningscatch_warnings simplefilterrr r)r2rbrrNpackager8r8r9 test_package_imported_no_warnings   &z2RunModuleTestCase.test_package_imported_no_warningcCs4tddD]$}tdkr td||j|ddq dS)NrrrT)rhrrr8r8r9%test_run_package_in_namespace_packagesz7RunModuleTestCase.test_run_package_in_namespace_packagecCs4tddD]$}tdkr td||j|ddq dS)NrrrT)rcrrr8r8r9test_run_namespace_packagesz,RunModuleTestCase.test_run_namespace_packagecCs6tddD]&}tdkr td||j|dddq dSrrrr8r8r9/test_run_namespace_package_in_namespace_packageszARunModuleTestCase.test_run_namespace_package_in_namespace_packagecCs2tdD]$}tdkrtd||j|ddqdS)NrrrTrrrr8r8r9test_run_module_alter_syss z+RunModuleTestCase.test_run_module_alter_syscCs4tddD]$}tdkr td||j|ddq dS)NrrrTrrrr8r8r9test_run_package_alter_sys sz,RunModuleTestCase.test_run_package_alter_syscCs0tddD] }tdkr td|||q dS)Nrz"Testing relative imports at depth:rrrsrrr8r8r9test_explicit_relative_importsz/RunModuleTestCase.test_explicit_relative_importcCs2tddD]"}tdkr td|||dq dS)Nrrrz'Testing main relative imports at depth:rrrr8r8r9test_main_relative_importsz+RunModuleTestCase.test_main_relative_importc sd}d|t|\}}}tt}||tj| dd|dfdd}z| ||W| |n | |0dS)Nrz*And now for something completely differentrmrrcs t|SrErrGrNr@r8r9rH&sz2RunModuleTestCase.test_run_name..create_ns) r}rFrrIr'rJrPrwrrrAr)r2rzrbrSrOr4rHr8rr9 test_run_names"   zRunModuleTestCase.test_run_namecCs|ddl}d}d}ddg}d|dg}t}t}td|D]V}d |g|} || |D]} || d | q\|D]} || d | qxq:d |g|} || || d |d |\} } } }||j| td |dD]}|| d |q| | gD]H}| ||j | |j t jj|jrD||jn ||jq |t|d||t|d|dS) Nrrrirz uncle.cousinzuncle.cousin.nephewz.siblingrrmz .runpy_testrLr)pkgutilr1rraaddr}rrrZ walk_packagesZassertIsInstanceZ ModuleInfoZ module_finderrPrQ FileFinderZispkgrr r-len)r2rZ max_depth base_nameZpackage_suffixesZmodule_suffixesZexpected_packagesZexpected_modulesrzr{r rbrSrNrOZ moduleinfor8r8r9test_pkgutil_walk_packages-s@    z,RunModuleTestCase.test_pkgutil_walk_packages)F)rf)F)F)N)rrBrCrrXr[r\rer}rrrrrrrrrrrrrrrrrrrrrr8r8r8r9rUsB    ( ) ( rUc@seZdZdZd#ddZd$ddZd d Zd d Zd dZddZ ddZ ddZ ddZ ddZ ddZddZddZedd Zd!d"ZdS)%RunPathTestCasezUnit tests for runpy.run_pathNFcCs|dur t}t||||SrE)rFr )r2 script_dirZscript_basenamery omit_suffixr8r8r9_make_test_scriptTs z!RunPathTestCase._make_test_scriptTc sfdd}t} |dur"|} n|} |rJtj| |} | j} |sRd| _nd} } | ||| d| |ddd||| dfdd}|r|durtj|} |sd| _| | d<| d < d d | d <||| dS) Ncs t|SrErrG) script_namer8r9rH_sz0RunPathTestCase._check_script..create_nsrLT)rrrrrrrrzprove.issue15230.is.fixedcs t|SrErrGr@rr8r9rH|srrrmrr) rIr'rPrwrxr#r,rJrAr) r2rrZ expected_fileZexpected_argv0rN expect_spec check_loaderrHr4Z spec_namerOZ mod_cachedr8rr9 _check_script[sH   zRunPathTestCase._check_scriptcCst|}|t|t|dSrE)reescapeassertRaisesRegexrVr)r2rmsgr8r8r9_check_import_errors z#RunPathTestCase._check_import_errorcCsNt4}d}|||}|j|d||ddWdn1s@0YdSNscript Frrrrr2rrNrr8r8r9test_basic_scripts   z!RunPathTestCase.test_basic_scriptcCsTt:}d}t|||}|j|d||ddWdn1sF0YdSr)rpathlibPathrrrr8r8r9"test_basic_script_with_path_objects  z2RunPathTestCase.test_basic_script_with_path_objectcCsRt8}d}|j||dd}|j|d||ddWdn1sD0YdS)NrT)rrFrrrr8r8r9test_basic_script_no_suffixs z+RunPathTestCase.test_basic_script_no_suffixcCsftL}d}|||}tj|dd}t||j|d||ddWdn1sX0YdS)NrTdoraiserFr)rr py_compilecompiler^rr)r2rrNr compiled_namer8r8r9test_script_compileds   z$RunPathTestCase.test_script_compiledcCsNt4}d}|||}|j|d|||dWdn1s@0YdS)NrrrNrrr8r8r9test_directorys   zRunPathTestCase.test_directorycCsttZ}d}|||}tj|dd}t|tjsRt|}|j |d|||dWdn1sf0YdS)NrTrrr) rrrrr^rrrrr)r2rrNrrZ legacy_pycr8r8r9test_directory_compileds   z'RunPathTestCase.test_directory_compiledcCsNt4}d}|||}d|}|||Wdn1s@0YdS)Nnot_main"can't find '__main__' module in %r)rrr)r2rrNrrr8r8r9test_directory_errors  z$RunPathTestCase.test_directory_errorc Cs`tF}d}|||}t|d|\}}|j|d|||ddWdn1sR0YdS)Nrtest_ziprFrNr)rrr r)r2rrNrzip_namefnamer8r8r9 test_zipfiles  zRunPathTestCase.test_zipfilec CsntT}d}|||}tj|dd}t|d|\}}|j|d|||ddWdn1s`0YdS)NrTrrrFr)rrrrr r)r2rrNrrrr r8r8r9test_zipfile_compileds  z%RunPathTestCase.test_zipfile_compiledcCs^tD}d}|||}t|d|\}}d|}|||Wdn1sP0YdS)Nrrr)rrr r)r2rrNrrr rr8r8r9test_zipfile_errors  z"RunPathTestCase.test_zipfile_errorc Cstt}tN}d}d|}||||}t|d|\}}d}|t|t|Wdn1sb0YWdn1s0YdS)Nrz import runpy runpy.run_path(%r) rzrecursion depth exceeded)rrr rRecursionErrorr) r2rZ dummy_dirrNryrrr rr8r8r9test_main_recursion_errorsz)RunPathTestCase.test_main_recursion_errorc Cstn}tj|d}t|ddd}|dWdn1sD0Yt|}||ddWdn1sz0YdS)Nz script.pyrllatin1ru% #coding:latin1 s = "non-ASCII: hé" sunon-ASCII: hé)rr^r`rarurvrr-)r2rfilenamefrr8r8r9 test_encodings (zRunPathTestCase.test_encoding)NF)NTT)rrBrCrrrrrrrrrrrr r r rrrr8r8r8r9rQs(  -    rcseZdZdZejdkrenej Ze e j ddZ fddZ ddZd d Zd d Zd dZddZddZddZddZZS)TestExitl:win32cos6t}t|VWdn1s(0YdSrE)rrr)argskwargsZtmp_fnr8r8r9tmp_pathszTestExit.tmp_pathcs\|@}|d|_}|tdtj|i|Wdn1sN0YdS)Nzham.pyz@ raise KeyboardInterrupt )rham write_texttextwrapdedentsuperrun)r2rrtmpr __class__r8r9rs z TestExit.runcOsBtj|i|dtjd}||jd||j|jdS)NT)textstderrz KeyboardInterrupt ) subprocessrPIPEZ assertTruer#r*r- returncode EXPECTED_CODE)r2rrprocr8r8r9 assertSigIntszTestExit.assertSigIntcCs|tj|jgdSrE)r)r executablerrZr8r8r9test_pymain_run_fileszTestExit.test_pymain_run_filecCs8|jj}|d}|td|jtj|g|ddS)Nz run_module.pyzU import runpy runpy.run_module("ham") cwdrr!rrrr)rr*)r2rr r8r8r9%test_pymain_run_file_runpy_run_modulesz.TestExit.test_pymain_run_file_runpy_run_modulecCs8|jj}|d}|td|jtj|g|ddS)Nzrun_module_as_main.pyz^ import runpy runpy._run_module_as_main("ham") r,r.)r2rZrun_module_as_mainr8r8r9-test_pymain_run_file_runpy_run_module_as_main%sz6TestExit.test_pymain_run_file_runpy_run_module_as_maincCs|jtjddg|jjddS)N-cz%import runpy; runpy.run_module('ham')r,r)rr*rr!rZr8r8r9"test_pymain_run_command_run_module2s z+TestExit.test_pymain_run_command_run_modulecCs|jtjddg|jjddS)Nr1 import hamr,r2rZr8r8r9test_pymain_run_command8sz TestExit.test_pymain_run_commandcCs|jtjgd|jjddS)Nr4)inputr-r2rZr8r8r9test_pymain_run_stdin;szTestExit.test_pymain_run_stdincCs$|j}|jtjd|jg|jddS)Nz-mr,)rr)rr*stemr!)r2rr8r8r9test_pymain_run_module>szTestExit.test_pymain_run_module)rrBrCZSTATUS_CONTROL_C_EXITrplatformsignalSIGINTr' staticmethod contextlibcontextmanagerrrr)r+r/r0r3r5r7r9 __classcell__r8r8r r9rs"    rr).r>Zimportlib.machineryrPimportlib.utilZos.pathr^rrrr;r$rrqrZunittestrZ test.supportrrrrrrrZtest.support.script_helperr r rr r r rrFZimplicit_namespacer=dictrIrJrZTestCaserDrUrrrmainr8r8r8r9s\$  O'&M