Your IP : 216.73.216.213
�
g f� � @ su d d l m Z d d l m Z m Z d d l Z d d l m Z d d l Z d d l
Z
d d l Z d d � Z d d � Z
d Z e
j d k r� e j � Z e e k r� e j � Z n e j j e � s� d Z q� n d
d � Z d d d � Z e d d � � Z e d d d d d �� Z e d d � � Z Gd d � d � Z Gd d � d e � Z Gd d � d e � Z d S) � )�contextmanager)�util�invalidate_cachesN)�supportc C sL d | k r d n f } t j | � } t j | d | d d �} | | f S)zAImport a module from importlib both w/ and w/o _frozen_importlib.�.� importlib�freshZblocked�_frozen_importlib)z importlib)z_frozen_importlib)r Zimport_fresh_module)Z module_namer �frozen�source� r �=/opt/alt/python34/lib64/python3.4/test/test_importlib/util.py�import_importlib
s
r c K s� t j d | j | t j f � } t j d | j | t j f � } | j | _ | _ x@ | j � D]2 \ } \ } } t | | | � t | | | � qd W| | f S)NZFrozen_ZSource_)�types� new_class�__name__�unittestZTestCase�
__module__�items�setattr)Z
test_class�kwargsZ frozen_testsZ source_tests�attrZ frozen_valueZ source_valuer r r
� test_both s r T�win32�cygwinFc C s t j t d � | � S)zRClass decorator that nullifies tests requiring a case-insensitive
file system.z&requires a case-insensitive filesystem)r ZskipIf�CASE_INSENSITIVE_FS)�testr r r
�case_insensitive_tests* s
r � c C sT t j j | | d � } t | d � � } | j | � Wd QXd j | | � | f S)Nz.py�wz{}.{})�os�path�join�open�write�format)�parent�nameZpkg_dir�contentr! Zsubfiler r r
� submodule1 s r) c g s� xU | D]M } | d k r1 t d j | � � � n y t j | =Wq t k
rS Yq Xq Wz d VWd x1 | D]) } y t j | =Wqk t k
r� Yqk Xqk WXd S)z�Uncache a module from sys.modules.
A basic sanity check is performed to prevent uncaching modules that either
cannot/shouldn't be uncached.
�sys�marshal�impzcannot uncache {0}N)zsyszmarshalr, )�
ValueErrorr% r* �modules�KeyError)�namesr' r r r
�uncache8 s
r1 �pkgc # s � f d d � t j D� } t j d � �� } t � | � �� t j | � �� t � t j j | � � } | r� t j j | d � } t j
� � n | d } | d k r� d } n | d k r� t | d � � } | j | � Wd QXn | VWd QXWd QXWd QXd S)Nc s/ g | ]% } | j d � d � k r | � q S)r r )� partition)�.0�n)r' r r
�
<listcomp>T s z temp_module.<locals>.<listcomp>z __init__.pyz.pyr r )
r* r. r Ztemp_cwdr1 Z
DirsOnSysPathr r r! r" �mkdirr# r$ )r' r( r2 Z conflicts�cwd�locationZmodpathZmodfiler )r' r
� temp_moduleR s
r: c k s� i } z� x� d g f d g f d g f d i f f D]U \ } } t t | � | | <| | k rs | | } | | =n | } t t | | � q4 Wt | � r� t d j | j � � � � n d VWd x* | j � D] \ } } t t | | � q� WXd S)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_hooks�path_importer_cachezunrecognized arguments: {0}N)�getattrr* r �lenr- r% �keysr )r Z originalsr �defaultZ new_value�valuer r r
� import_statej s"
rC c @ sL e Z d Z d Z d i d d � Z d d � Z d d � Z d d
� Z d S) �
_ImporterMockz0Base class to help with creating importer mocks.� module_codec G s i | _ i | _ x� | D]� } | j d � s7 | } n | d t d � � } d | k rc d } n+ | | k r� | j d d � d } n | } t j | � } | | _ d | _ | | _ | | _
| | k r� d g | _ n | | j | <| | k r | | | j | <q q Wd S)Nz .__init__r � r z<mock __file__>z<mock __path__>) r. rE �endswithr? �rsplitr �
ModuleType�
__loader__�__file__� __package__r �__path__)�selfrE r0 r' Z import_name�package�moduler r r
�__init__� s*
z_ImporterMock.__init__c C s | j | S)N)r. )rN r' r r r
� __getitem__� s z_ImporterMock.__getitem__c C s) t | j j � � | _ | j j � | S)N)r1 r. r@ �_uncache� __enter__)rN r r r
rT � s
z_ImporterMock.__enter__c G s | j j d d d � d S)N)rS �__exit__)rN �exc_infor r r
rU � s z_ImporterMock.__exit__N)r r � __qualname__�__doc__rQ rR rT rU r r r r
rD � s
rD c @ s1 e Z d Z d Z d d d � Z d d � Z d S)� mock_modulesz!Importer mock using PEP 302 APIs.Nc C s | | j k r d S| Sd S)N)r. )rN �fullnamer! r r r
� find_module� s zmock_modules.find_modulec C s� | | j k r t � ne | j | t j | <| | j k rr y | j | � Wqr t k
rn t j | =� Yqr Xn | j | Sd S)N)r. � ImportErrorr* rE � Exception)rN rZ r r r
� load_module� s
zmock_modules.load_module)r r rW rX r[ r^ r r r r
rY � s rY c @ s@ e Z d Z d Z d d d d � Z d d � Z d d � Z d S) � mock_specz!Importer mock using PEP 451 APIs.Nc C sj y | j | } Wn t k
r) d SYn Xt | d � } t j | | j d | d t | d d � �} | S)NrM �loader�submodule_search_locations)r. r/ �hasattrr �spec_from_file_locationrK r> )rN rZ r! r& rP �
is_package�specr r r
� find_spec� s
zmock_spec.find_specc C s) | j | j k r t � n | j | j S)N)r' r. r\ )rN re r r r
�
create_module� s zmock_spec.create_modulec C s1 y | j | j j � Wn t k
r, Yn Xd S)N)rE �__spec__r' r/ )rN rP r r r
� exec_module� s
zmock_spec.exec_module)r r rW rX rf rg ri r r r r
r_ � s r_ )zwin32zcygwin) �
contextlibr r r r Zos.pathr r r r r* r r r r �platformrK �upperZ changed_name�lowerr! �existsr r) r1 r: rC rD rY r_ r r r r
�<module> s0 )