Your IP : 216.73.216.213
�
g f�9 � @ sM d d l m Z m Z m Z m Z d d l Z d d l Z d d l Z d d l m 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 Gd d � d e j � Z Gd d � d e j � Z Gd d � d e j � Z Gd
d � d e j � Z Gd d
� d
e j � Z d d � Z e d k rIe � n d S)� )� run_unittest�unload�check_warnings� CleanImportN)�spec_from_file_locationc @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) � PkgutilTestsc C s? t j � | _ | j t j | j � t j j d | j � d S)Nr ) �tempfile�mkdtemp�dirname�
addCleanup�shutil�rmtree�sys�path�insert)�self� r �6/opt/alt/python34/lib64/python3.4/test/test_pkgutil.py�setUp s zPkgutilTests.setUpc C s t j d =d S)Nr )r r )r r r r �tearDown s zPkgutilTests.tearDownc C s4 d } d } t j j | j | � } t j | � t t j j | d � d � } | j � t t j j | d � d � } | j | � | j � t j t j j | d � � t t j j | d d � d � } | j | � | j � t j | d � } | j
| | � t j | d � } | j
| | � t j | =d S)N�test_getdata_filesyss% Hello, world!
Second line
Third linez __init__.py�wbzres.txt�subz sub/res.txt)
�osr �joinr
�mkdir�open�close�write�pkgutil�get_data� assertEqualr �modules)r �pkg�
RESOURCE_DATAZ package_dir�f�res1�res2r r r r s$
!
z!PkgutilTests.test_getdata_filesysc C s= d } d } d } t j j | j | � } t j | d � } | j | d d � | j | d | � | j | d | � | j � t j j d | � t
j | d
� } | j | | � t
j | d � } | j | | � g } x0 t
j
| g � D] \ } }
} | j |
� q� W| j | d g � t j d =t j | =d S) Nztest_getdata_zipfile.zip�test_getdata_zipfiles% Hello, world!
Second line
Third line�wz /__init__.py� z/res.txtz /sub/res.txtr zres.txtz sub/res.txt)r r r r
�zipfileZZipFileZwritestrr r r r r r! Z iter_modules�appendr" ) r �zipr# r$ Zzip_file�zr& r' �names�loader�nameZispkgr r r r( ; s(
"
z!PkgutilTests.test_getdata_zipfilec C su d } t j j | j | � } t j | d � | j t j | � x- t j d | j g � D] } | j d � qZ Wd S)NZunreadable_packager r zunexpected package found)
r r r r
r r �rmdirr Z
walk_packagesZfail)r Z package_name�d�tr r r � test_unreadable_dir_on_syspath] s z+PkgutilTests.test_unreadable_dir_on_syspathN)�__name__�
__module__� __qualname__r r r r( r5 r r r r r s
"r c @ sl e Z d Z Gd d � d e � Z Gd d � d e � Z d d � Z d d � Z d d
� Z d d � Z d
S)�PkgutilPEP302Testsc @ s( e Z d Z d d � Z d d � Z d S)z PkgutilPEP302Tests.MyTestLoaderc C s$ | j j d d � d | j d <d S)N�loadsr � )�__dict__�get)r �modr r r � exec_modulek s z+PkgutilPEP302Tests.MyTestLoader.exec_modulec C s d S)Nz
Hello, world!r )r r r r r r o s z(PkgutilPEP302Tests.MyTestLoader.get_dataN)r6 r7 r8 r? r r r r r � MyTestLoaderj s r@ c @ s" e Z d Z d d d d � Z d S)z!PkgutilPEP302Tests.MyTestImporterNc C s/ t j � } t | d | j j d | d g �S)Nz<%s>r0 �submodule_search_locations)r9 r@ r � __class__r6 )r �fullnamer �targetr0 r r r � find_specs s
z+PkgutilPEP302Tests.MyTestImporter.find_spec)r6 r7 r8 rE r r r r �MyTestImporterr s rF c C s t j j d | j � � d S)Nr )r � meta_pathr rF )r r r r r z s zPkgutilPEP302Tests.setUpc C s t j d =d S)Nr )r rG )r r r r r } s zPkgutilPEP302Tests.tearDownc C s* | j t j d d � d � t j d =d S)N�foo�dummyz
Hello, world!)r! r r r r" )r r r r �test_getdata_pep302� s z&PkgutilPEP302Tests.test_getdata_pep302c C s\ d d l } | j | j d � | j t j d d � d � | j | j d � t j d =d S)Nr r; rH rI z
Hello, world!)rH r! r: r r r r" )r rH r r r �test_alreadyloaded� s
z%PkgutilPEP302Tests.test_alreadyloadedN)
r6 r7 r8 �objectr@ rF r r rJ rK r r r r r9 h s r9 c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �ExtendPathTestsc C sy t j � } t j j d | � t j j | | � } t j | � t t j j | d � d � � } | j d � Wd QX| S)Nr z __init__.pyr) zKfrom pkgutil import extend_path
__path__ = extend_path(__path__, __name__)
)
r r r r r r r r r r )r �pkgnamer
Zpkgdir�flr r r � create_init� s
!zExtendPathTests.create_initc C sQ t j j | | | d � } t | d � � } t d j | � d | �Wd QXd S)Nz.pyr) zvalue={}�file)r r r r �print�format)r r
rN Zsubmodule_name�valueZ module_namerO r r r �create_submodule� s z ExtendPathTests.create_submodulec C s' d } | j | � } | j | � } | j | | d d � | j | | d d � d d l } d d l } | j | j j d � | j | j j d � | j t | j � t t
j j | | � t
j j | | � g � � t
j | � t
j | � t j d =t j d =t j d =t j d =t j d =d S)NrH �barr �bazr; zfoo.barzfoo.baz)rP rU �foo.bar�foo.bazr! rV rT rW �sorted�__path__r r r r r
r r" )r rN � dirname_0� dirname_1rH r r r � test_simple� s&
zExtendPathTests.test_simplec
C s� t j } t j } d } d } | j | � } t j j | | � } d j | | � } t j j
| d � t j j
| d � z4| j | | | d � t j
| � t | | � � } | | � } x� | D]� }
t j |
| � } | j } y
| j } Wn t k
rYn X| j |
t j j � | j |
| � | j | t j j � | j t j |
| � � q� W| j t � � t | d � � Wd QX| j t � � t | d � � Wd QXWd t j | � t j d =y t j d =t j d =Wn t k
r�Yn XXd S)N�spamZeggsz{}.{}r zinvalid.modulez.spamz spam.eggs) r �iter_importers� get_importerrP r r r rS r r" �poprU � importlib�
import_module�list� _get_specr0 �AttributeErrorZassertIsInstance� machinery�
FileFinderr! �SourceFileLoader� assertIsNoneZ assertRaises� ImportErrorr r
�KeyError)
r r` ra rN �modnamer
ZpathitemrC Z importersZexpected_importer�finder�specr0 r r r �test_iter_importers� sJ
z#ExtendPathTests.test_iter_importersc C sF d } | j | � } | j | � } | j | | d d � t j t j j | | d � � | j | | d d � d d l } d d l } | j | j j
d � | j | j j
d � | j t | j
� t t j j | | � t j j | | � g � � t j | � t j | � t j d =t j d =t j d =t j d =t j d =d S) NrH rV r z __init__.pyrW r; zfoo.barzfoo.baz)rP rU r �unlinkr r rX rY r! rV rT rW rZ r[ r r
r r" )r rN r\ r] rH r r r �test_mixed_namespace� s(
z$ExtendPathTests.test_mixed_namespaceN)r6 r7 r8 rP rU r^ rq rs r r r r rM � s
0rM c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
�NestedNamespacePackageTestc C s) t j � | _ t j d d � | _ d S)N)r r �basedirr r �old_path)r r r r r s z NestedNamespacePackageTest.setUpc C s* | j t j d d � <t j | j � d S)N)rv r r r r
ru )r r r r r s z#NestedNamespacePackageTest.tearDownc C s� | j d d � \ } } t j j | j | j d t j j � � } t j | d d �t t j j | | d � d � � } | j | � Wd QXd S)N�.r; �exist_okTz.pyr) )
�rsplitr r r ru �replace�sep�makedirsr r )r r1 �contents�base�final� base_pathr% r r r �
create_module s
*%z(NestedNamespacePackageTest.create_modulec C sa d } | j d | � | j d | � | j d | � | j d | � | j d d � | j d d � t j j d
t j j | j d � � t j j d
t j j | j d � � d
d l } | j t d
� | j
t | j � d � d
d l
} | j t d � | j
t | j j � d � d
d l m } d
d l m } | j
| d � | j
| d � d S)NzBimport pkgutil; __path__ = pkgutil.extend_path(__path__, __name__)za.pkg.__init__zb.pkg.__init__za.pkg.subpkg.__init__zb.pkg.subpkg.__init__za.pkg.subpkg.czc = 1zb.pkg.subpkg.dzd = 2r �a�br# � z
pkg.subpkg)�c)r3 r; )r� r r r r r ru r# r r r! �lenr[ Z
pkg.subpkgZsubpkgZ pkg.subpkg.cr� Z pkg.subpkg.dr3 )r Zpkgutil_boilerplater# r� r3 r r r � test_nested s& %% z&NestedNamespacePackageTest.test_nestedN)r6 r7 r8 r r r� r� r r r r rt s rt c @ s� e Z 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 d d � Z
d d � Z d S)�ImportlibMigrationTestsc C s t d t f � S)Nz5This emulation is deprecated, use 'importlib' instead)r �DeprecationWarning)r r r r �check_deprecated= s z(ImportlibMigrationTests.check_deprecatedc C s& | j � � t j d � } Wd QXd S)Nr* )r� r Z ImpImporter)r �xr r r �test_importer_deprecatedB s
z0ImportlibMigrationTests.test_importer_deprecatedc C s/ | j � � t j d d d d � } Wd QXd S)Nr* )r� r Z ImpLoader)r r� r r r �test_loader_deprecatedF s
z.ImportlibMigrationTests.test_loader_deprecatedc
C sq t � �b } | j t j d � � | j t j d � � | j t j d � � | j t | j � d � Wd QXd S)Nr r z test.supportr )r �assertIsNotNoner �
get_loaderr! r� �warnings)r r) r r r � test_get_loader_avoids_emulationJ s
z8ImportlibMigrationTests.test_get_loader_avoids_emulationc C s\ t } b zE t � �6 } | j t j t � � | j t | j � d � Wd QXWd | a Xd S)Nr ) �
__loader__r r� r r� r6 r! r� r� )r Z this_loaderr) r r r �0test_get_loader_handles_missing_loader_attributeQ s #zHImportlibMigrationTests.test_get_loader_handles_missing_loader_attributec
C s^ d } t t � | � } | ` t | � �! | t j | <t j | � } Wd QX| j | � d S)Nr_ )�typer �__spec__r r" r r� rk )r r1 r> r0 r r r �.test_get_loader_handles_missing_spec_attribute\ s
zFImportlibMigrationTests.test_get_loader_handles_missing_spec_attributec
C sa d } t t � | � } d | _ t | � �! | t j | <t j | � } Wd QX| j | � d S)Nr_ )r� r r� r r" r r� rk )r r1 r> r0 r r r �+test_get_loader_handles_spec_attribute_nonee s
zCImportlibMigrationTests.test_get_loader_handles_spec_attribute_nonec C sE d } d t j | <z t j | � } Wd t j | =X| j | � d S)Nz
totally bogus)r r" r r� rk )r r1 r0 r r r �#test_get_loader_None_in_sys_modulesn s
z;ImportlibMigrationTests.test_get_loader_None_in_sys_modulesc C s& d } t j | � } | j | � d S)Nz
totally bogus)r � find_loaderrk )r r1 r0 r r r � test_find_loader_missing_modulew s z7ImportlibMigrationTests.test_find_loader_missing_modulec
C sq t � �b } | j t j d � � | j t j d � � | j t j d � � | j t | j � d � Wd QXd S)Nr r z test.supportr )r r� r r� r! r� r� )r r) r r r �!test_find_loader_avoids_emulation| s
z9ImportlibMigrationTests.test_find_loader_avoids_emulationc
C sE t � �6 } | j t j d � � | j t | j � d � Wd QXd S)Nz*??r )r rk r ra r! r� r� )r r) r r r �"test_get_importer_avoids_emulation� s z:ImportlibMigrationTests.test_get_importer_avoids_emulationc
C sF t � �7 } x t j � D] } q W| j t | j � d � Wd QXd S)Nr )r r r` r! r� r� )r r) Zimporterr r r �$test_iter_importers_avoids_emulation� s z<ImportlibMigrationTests.test_iter_importers_avoids_emulationN)r6 r7 r8 r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� 8 s r� c C sI t t t t t t � d d l } d d l } | j j � | j
� d S)Nr ) r r r9 rM rt r� � zipimportrc �_zip_directory_cache�clear�invalidate_caches)r� rc r r r � test_main� s
r� �__main__)Z test.supportr r r r Zunittestr rc �importlib.utilr r r Zos.pathr �typesr r+ ZTestCaser r9 rM rt r� r� r6 r r r r �<module> s& " T*})W