B v9aF@sddlZddlZddlZddlZddlZddlZddlmZmZmZddl m Z ddl Z ddl Z ddl Z ddlmZmZddlmZddlZddlZddlZddlZddlmZddlmZeZde_de_d ejkrd e_d ejkrd e_eadt_ dt_!dt_"dt_#d t_$d d Z%e%ddZ&d@ddZ'dAddZ(dBddZ)dZ*ej+dkre,-Z.e.e,krte,/Z.e j 0e.sdZ*e&d dZ1e2ej3e2e1j3dZ3ddZ4dCddZ5ej6d d!Z7ej6dDdd"d#d$Z8ej6d%d&Z9Gd'd(d(Z:Gd)d*d*e:Z;Gd+d,d,e:Zej6d1d2Z?d3d4Z@Gd5d6d6ZAdEd8d9ZBGd:d;d;ejCZDGdd?d?eEZFdS)FN) machineryutilinvalidate_caches)ResourceReader)PathPurePath)support)data01) zipdata01errno importlibZ _testcapicCsdx^tjD]T}xNtjD]D}tj|}tj||}tj|r|t_|t_ |t_ |t_ dSqWqWdS)N) syspathrEXTENSION_SUFFIXES EXTENSIONSnameosjoinexistsextfilename file_path)rrrrr/usr/lib/python3.7/util.py_extension_details&s    rcCs4d|kr dnd}t|}tj||dd}||dS)zAImport a module from importlib both w/ and w/o _frozen_importlib..)r r)_frozen_importlib_frozen_importlib_external)freshZblocked)FrozenSource)rZimport_fresh_module)Z module_namerfrozensourcerrrimport_importlib6s  r$c Ks|dkrtj}nt|ts"||}d||j}||f}t||}|j|_|j|_ ||_ x(| D]\}}||} t ||| qdW|S)Nz{}_{}) unittestZTestCase isinstancetypeformat__name__types new_class __module__Z_NAMEZ_KINDitemssetattr) clskindbasekwargsrbasesZ specializedattrvaluesvaluerrrspecialize_class?s  r7cKs(t|d|f|}t|d|f|}||fS)Nr r!)r7)r/r1r2r"r#rrr split_frozenRsr8cKst||f|S)N)r8)Z test_classr1r2rrr test_bothXsr9T)win32cygwinFr!)r r!cCstt d|S)zRClass decorator that nullifies tests requiring a case-insensitive file system.z&requires a case-insensitive filesystem)r%skipIfCASE_INSENSITIVE_FS)testrrrcase_insensitive_testsksr?c CsBtj||d}t|d}||WdQRXd|||fS)Nz.pywz{}.{})rrropenwriter()parentrZpkg_dircontentrZsubfilerrr submodulers rFcgsxD|D]<}|dkr td|y tj|=Wqtk r@YqXqWz dVWdx.|D]&}y tj|=WqXtk r|YqXXqXWXdS)zUncache a module from sys.modules. A basic sanity check is performed to prevent uncaching modules that either cannot/shouldn't be uncached. )rmarshalZimpzcannot uncache {0}N) ValueErrorr(rmodulesKeyError)namesrrrruncacheys       rL)pkgc #sfddtjD}td}tf|t|zttj |}|rntj |d}t n|d}|dkrd}|dk rt |d}| |WdQRX|VWdQRXWdQRXWdQRXdS)Ncs"g|]}|ddkr|qS)rr) partition).0n)rrr sztemp_module..z __init__.pyz.pyr@rA) rrIrZtemp_cwdrLZ DirsOnSysPathrrrrmkdirrBrC)rrErM conflictscwdlocationZmodpathZmodfiler)rr temp_modules     rVc ksi}zx^dgfdgfdgfdiffD]>\}}tt|||<||krR||}||=n|}tt||q$Wt|rtd|dVWdx |D]\}}tt||qWXdS)a,Context manager to manage the various importers and stored state in the sys module. The 'modules' attribute is not supported as the interpreter state stores a pointer to the dict that the interpreter uses internally; reassigning to sys.modules does not have the desired effect. meta_pathr path_hookspath_importer_cachezunrecognized arguments: {0}N)getattrrr.lenrHr(keysr-)r2Z originalsr4defaultZ new_valuer6rrr import_states"  r^c@s6eZdZdZidddZddZddZd d Zd S) _ImporterMockz0Base class to help with creating importer mocks.) module_codecGsi|_i|_x|D]}|ds&|}n|dtd }d|krFd}n||kr`|ddd}n|}t|}||_d|_||_ ||_ ||krdg|_ ||j|<||kr|||j|<qWdS)Nz .__init__rr rzz) rIr`endswithr[rsplitr* ModuleType __loader____file__ __package__r4__path__)selfr`rKr import_namepackagemodulerrr__init__s*    z_ImporterMock.__init__cCs |j|S)N)rI)rhrrrr __getitem__sz_ImporterMock.__getitem__cCst|j|_|j|S)N)rLrIr\_uncache __enter__)rhrrrros z_ImporterMock.__enter__cGs|jddddS)N)rn__exit__)rhexc_inforrrrpsz_ImporterMock.__exit__N)r)r, __qualname____doc__rlrmrorprrrrr_s r_c@s"eZdZdZdddZddZdS) mock_modulesz!Importer mock using PEP 302 APIs.NcCs||jkrdS|SdS)N)rI)rhfullnamerrrr find_modules zmock_modules.find_modulecCsh||jkrtnT|j|tj|<||jkrZy|j|Wntk rXtj|=YnX|j|SdS)N)rI ImportErrorrr` Exception)rhrurrr load_modules  zmock_modules.load_module)N)r)r,rrrsrvryrrrrrts rtc@s*eZdZdZd ddZddZddZdS) mock_specz!Importer mock using PEP 451 APIs.NcCsDy|j|}Wntk r"dSXtj||j|t|ddd}|S)Nrg)loadersubmodule_search_locations)rIrJrspec_from_file_locationrerZ)rhrurrDrkspecrrr find_specszmock_spec.find_speccCs|j|jkrt|j|jS)N)rrIrw)rhr~rrr create_modules zmock_spec.create_modulecCs.y|j|jjWntk r(YnXdS)N)r`__spec__rrJ)rhrkrrr exec_moduleszmock_spec.exec_module)NN)r)r,rrrsrrrrrrrrz s rzcs(tjrddStfdd}|S)ztDecorator to protect sys.dont_write_bytecode from mutation and to skip tests that require it to be set to False.c_sdS)Nr)argsr2rrr)z'writes_bytecode_files..cs(tj}dt_z||}Wd|t_X|S)NF)rdont_write_bytecode)rr2ZoriginalZ to_return)fxnrrwrapper*s z&writes_bytecode_files..wrapper)rr functoolswraps)rrr)rrwrites_bytecode_files%s rc CsLyttj|Wn0tk rF}z|jtjkr6Wdd}~XYnXdS)zEnsure that the __pycache__ directory for PEP 3147 pyc file exists. :param bytecode_path: File system path to PEP 3147 pyc file. N)rrRrdirnameOSErrorr ZEEXIST) bytecode_patherrorrrrensure_bytecode_path6s  rc gsd}g}i}d}d}z0t}||d<t}x|D]}|dsH|} n|dtd } || | tjkrvtj| =|d} |} xB| ddD]2} t j | | } t j | st | || qWt j | | dd} t| d } | ||WdQRX|| | ||<q4Wt|}|t|gd }||VWd|dk r`|ddd|dk rx|dddt|XdS) aRTemporarily create each named module with an attribute (named 'attr') that contains the name passed into the context manager that caused the creation of the module. All files are created in a temporary directory returned by tempfile.mkdtemp(). This directory is inserted at the beginning of sys.path. When the context manager exits all created files (source and bytecode) are explicitly deleted. No magic is performed when creating packages! This means that if you create a module within a package you must also create the package's __init__ as well. z attr = {0!r}Nz.rootrlz .__init__rz.pyrA)r)tempfileZmkdtempsetrar[addrrIsplitrrrrrRappendrBrCr(rLror^rprZrmtree)rKr#Z created_pathsmappingZ state_managerZuncache_managerZtemp_dirZ import_namesrriZ name_partsrZ directoryfilerrrcreate_modulesBsL              rcsfdd}|S)zA mock sys.path_hooks entry.cs|kr tS)N)rw)entry)entriesimporterrrhook~szmock_path_hook..hookr)rrrr)rrrmock_path_hook|src@seZdZddZdS)CASEOKTestBasecs,d}tfdd|D|kr(ddS)N)s PYTHONCASEOK PYTHONCASEOKc3s|]}|jjjjkVqdS)N)r _bootstrap_external_osenviron)rOx)rhrr sz4CASEOKTestBase.caseok_env_changed..z/os.environ changes not reflected in _os.environ)anyZskipTest)rhZ should_existZ possibilitiesr)rhrcaseok_env_changeds z!CASEOKTestBase.caseok_env_changedN)r)r,rrrrrrrrsrrc sNGfdddt}d}t|}|}tj||d|d}||_||_|S)NcsHeZdZddZfddZfddZfddZfd d Zd S) zcreate_package..ReadercSs|S)Nr)rhrjrrrget_resource_readersz2create_package..Reader.get_resource_readercs||_ttrnSdS)N)_pathr&rx)rhr)rrr open_resources z,create_package..Reader.open_resourcecs||_ttrnSdS)N)rr&rx)rhpath_)rrr resource_paths z,create_package..Reader.resource_pathcsL||_ttrx2D]*}|d}t|dkr|d|krdSqWdS)N/r rTF)rr&rxrr[)rhrrparts)contentsrrr is_resources   z*create_package..Reader.is_resourcec3s&ttrxD] }|VqWdS)N)r&rx)rhr)rrrrrs  z'create_package..Reader.contentsN)r)r,rrrrrrrr)rrrrrReaders    rZtestingpackagezdoes-not-exist)origin is_package)rr*rcr ModuleSpecrrd) rrrrrrrkr{r~r)rrrrcreate_packages# rc@seZdZejddZddZddZddZe e j d kd d d Z d dZddZddZddZddZe e j dkdddZe e j dkdddZddZdS) CommonResourceTestscCstdS)N)NotImplementedError)rhrjrrrrexecuteszCommonResourceTests.executecCs|tjddS)Nz utf-8.file)rr r))rhrrrtest_package_namesz%CommonResourceTests.test_package_namecCs|tddS)Nz utf-8.file)rr )rhrrrtest_package_objectsz'CommonResourceTests.test_package_objectcCsd}|t|dS)Nz utf-8.file)rr )rhrrrrtest_string_pathsz$CommonResourceTests.test_string_path)zrequires os.PathLike supportcCstd}|t|dS)Nz utf-8.file)rrr )rhrrrrtest_pathlib_pathsz%CommonResourceTests.test_pathlib_pathc Cs8tt}|jd}|t|t|WdQRXdS)Nz utf-8.file)rrerD assertRaisesrHrr )rhr full_pathrrrtest_absolute_paths  z&CommonResourceTests.test_absolute_pathc Cs&|t|tdWdQRXdS)Nz../data01/utf-8.file)rrHrr )rhrrrtest_relative_paths z&CommonResourceTests.test_relative_pathcCstjtj=|tjddS)Nz utf-8.file)rrIr r)r)rhrrr$test_importing_module_as_side_effects z8CommonResourceTests.test_importing_module_as_side_effectc Cs&|t|tdWdQRXdS)Nz utf-8.file)r TypeErrorrr))rhrrrtest_non_package_by_names z,CommonResourceTests.test_non_package_by_namec Cs0|ttjd}||dWdQRXdS)Nztest.test_importlib.utilz utf-8.file)rrrrIr)rhrkrrrtest_non_package_by_packages  z/CommonResourceTests.test_non_package_by_package)rzNo ResourceReader in Python 2cCs8td}t|td}||d||jjddS)Ns Hello, world!)rrz utf-8.file)ioBytesIOrFileNotFoundErrorr assertEqualrdr)rh bytes_datarjrrrtest_resource_openers  z(CommonResourceTests.test_resource_openercCs:td}t}t||d}||d||jjddS)Ns Hello, world!)rrz utf-8.file)rrrerrrrdr)rhrrrjrrrtest_resource_paths    z&CommonResourceTests.test_resource_pathc Cs6tttd}|t||dWdQRXdS)N)rrz utf-8.file)rrrr)rhrjrrrtest_useless_loaders  z'CommonResourceTests.test_useless_loaderN)r)r,rrabcabstractmethodrrrrr%r<r version_inforrrrrrrrrrrrrrsrc@s0eZdZdZeddZeddZddZdS) ZipSetupBaseNcCs>t|jj}|j}t|d|_tj|jt d|_ dS)Nzziptestdata.zipZ ziptestdata) r ZIP_MODULErerDstr _zip_pathrrrr import_moduledata)r/Z data_pathZdata_dirrrr setUpClass s  zZipSetupBase.setUpClasscCs~ytj|jWntk r&YnXytj|j=tj|jj=Wnt k rVYnXy |`|`Wnt k rxYnXdS)N) rrremoverrHrYrIrr)rJAttributeError)r/rrr tearDownClasss zZipSetupBase.tearDownClasscCst}|jtjf|dS)N)rZ modules_setupZ addCleanupZmodules_cleanup)rhrIrrrsetUp(szZipSetupBase.setUp)r)r,rrr classmethodrrrrrrrr s  rc@seZdZeZdS)ZipSetupN)r)r,rrr rrrrrr-sr)N)N)N)r@)r@)Tr)Grbuiltins contextlibr rr rrrZ importlib.abcrrrZos.pathZpathlibrrr>rr%rrr*r@r r SimpleNamespaceZBUILTINSZ good_nameZbad_namebuiltin_module_namesrrrrrrrr$r7r8r9r=platformreupperZ changed_namelowerrZsource_importlib staticmethod __import__r?rFcontextmanagerrLrVr^r_rtrzrrrrrrABCrrrrrrrs              ) :  2J#